import os
import shutil
import logging

logger = logging.getLogger(__name__)

BASE_DIRECTORY = './embedding_data'

class FileSystemStorage:
    def __init__(self, base_directory=BASE_DIRECTORY):
        self.base_dir = base_directory

    def create_group(self, auth_token, group_id):
        path = os.path.join(self.base_dir, group_id)
        if os.path.exists(path):
            raise FileExistsError(f"{group_id}")
        else:
            try:
                os.makedirs(path)
            except OSError as e:
                logger.error(f"Error creating directory {path}: {e}")
                raise

    def get(self, auth_token, group_id, item_id):
        try:
            path = os.path.join(self.base_dir, group_id, item_id)
            with open(path, 'rb') as f:
                return f.read()
        except Exception as e:
            logger.error(f"Error reading from {path}: {e}")
            raise

    def save(self, auth_token, group_id, item_id, content):
        try:
            path = os.path.join(self.base_dir, group_id, item_id)
            with open(path, 'wb') as f:
                f.write(content)
        except Exception as e:
            logger.error(f"Error saving to {path}: {e}")
            raise
            
    def remove(slef, auth_token, group_id, item_id):
        try:
            path = os.path.join(self.base_dir, group_id, item_id)
            os.remove(path)
        except Exception as e:
            logger.error(f"Error remove item {path}: {e}")
            raise
            
    def remove_group(self, auth_token, group_id):
        try:
            path = os.path.join(self.base_dir, group_id)
            shutil.rmtree(path)
        except Exception as e:
            logger.error(f"Error remove group {path}: {e}")
            raise