"""Storage factory for creating different storage implementations."""
from typing import Dict, Type

from application.storage.base import BaseStorage
from application.storage.local import LocalStorage
from application.storage.s3 import S3Storage


class StorageCreator:
    storages: Dict[str, Type[BaseStorage]] = {
        "local": LocalStorage,
        "s3": S3Storage,
    }
    
    @classmethod
    def create_storage(cls, type_name: str, *args, **kwargs) -> BaseStorage:
        storage_class = cls.storages.get(type_name.lower())
        if not storage_class:
            raise ValueError(f"No storage implementation found for type {type_name}")
        
        return storage_class(*args, **kwargs)