from logger import get_logger
from models.settings import get_supabase_client
from modules.knowledge.repository.storage_interface import StorageInterface

logger = get_logger(__name__)

class Storage(StorageInterface):
    def __init__(self):
        supabase_client = get_supabase_client()
        self.db = supabase_client

    def upload_file(self, file_name: str):
        """
        Upload file to storage
        """
        self.db.storage.from_("quivr").download(file_name)

    def remove_file(self, file_name: str):
        """
        Remove file from storage
        """
        try:
            response = self.db.storage.from_("quivr").remove([file_name])
            return response
        except Exception as e:
            logger.error(e)
            # raise e