import dataclasses
import os
from typing import Optional

from . import types as upload_types
from ..message_processing import messages
from ..types import AttachmentEntityType


@dataclasses.dataclass
class FileUploadOptions:
    file_path: str
    file_name: str
    file_size: int
    mime_type: Optional[str]
    entity_type: AttachmentEntityType
    entity_id: str
    project_name: str
    encoded_url_override: str
    delete_after_upload: bool
    on_upload_success: Optional[upload_types.OnUploadSuccessCallback] = None
    on_upload_failed: Optional[upload_types.OnUploadFailureCallback] = None


def file_upload_options_from_attachment(
    attachment: messages.CreateAttachmentMessage,
    on_upload_success: Optional[upload_types.OnUploadSuccessCallback],
    on_upload_failed: Optional[upload_types.OnUploadFailureCallback],
) -> FileUploadOptions:
    file_size = os.path.getsize(attachment.file_path)

    return FileUploadOptions(
        file_path=attachment.file_path,
        file_name=attachment.file_name,
        file_size=file_size,
        mime_type=attachment.mime_type,
        entity_type=attachment.entity_type,
        entity_id=attachment.entity_id,
        project_name=attachment.project_name,
        encoded_url_override=attachment.encoded_url_override,
        delete_after_upload=attachment.delete_after_upload,
        on_upload_success=on_upload_success,
        on_upload_failed=on_upload_failed,
    )