from __future__ import annotations

from storage_service.depends.depend_queue import dependency_queue
from storage_service.depends.depend_s3_service import (
    dependency_storage_service,
)
from storage_service.service.storage.storage_service import StorageService
from storage_service.utils.enums.file_type import FileType
from storage_service.utils.file_name_hash import file_name_hash
from storage_service.worker.storage_file_worker import storage_file_worker

from fastapi import Body, Depends
from fastapi_utils.cbv import cbv
from fastapi_utils.inferring_router import InferringRouter
from rq import Queue

from typing import Annotated

s3_router = InferringRouter()


@cbv(s3_router)
class StorageController:
    queue: Queue = Depends(dependency_queue, use_cache=True)
    storage_service: StorageService = Depends(
        dependency_storage_service, use_cache=True
    )

    @s3_router.post("/file/", status_code=200)
    def new_file_url(
        self,
        username: Annotated[str, Body(embed=True)],
        file_postfix: Annotated[str, Body(embed=True)],
        file_type: Annotated[FileType, Body(embed=True)],
    ) -> dict[str, str]:
        return self.storage_service.get_temp_upload_link(
            file_name_hash(username, file_postfix), file_type
        )

    @s3_router.get("/file/", status_code=200)
    def file_url(self, username: str, file_postfix: str) -> dict[str, str | None]:
        return self.storage_service.get_temp_read_link(
            file_name_hash(username, file_postfix)
        )

    @s3_router.delete("/file/", status_code=204)
    def delete_file(self, username: str, file_postfix: str):
        return self.storage_service.delete_file(
            file_name_hash(username, file_postfix)
        )

    @s3_router.post("/file/process", status_code=200)
    def process_file(
        self,
        username: Annotated[str, Body(embed=True)],
        file_postfix: Annotated[str, Body(embed=True)],
    ):
        self.queue.enqueue(storage_file_worker, username, file_postfix)