#
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
#

from datetime import datetime
from typing import Optional

from pydantic import BaseModel


class RemoteFile(BaseModel):
    """
    A file in a file-based stream.
    """

    uri: str
    last_modified: datetime
    file_type: Optional[str] = None

    def extension_agrees_with_file_type(self) -> bool:
        extensions = self.uri.split(".")[1:]
        if not extensions:
            return True
        if not self.file_type:
            return True
        return any(self.file_type.casefold() in e.casefold() for e in extensions)