import os
from dotenv import load_dotenv

from config import rest_config
from custom_logger.main_log import main_logger as LOG


class ApiKeyManager:
    
    def __init__(self):
        self.api_key = None
        
    def set_api_key(self, env_path=rest_config.api_path):
        
        if os.path.exists(env_path):
            
            load_dotenv(dotenv_path=env_path)
            key = os.getenv("GEMINI_API_KEY")
            
            if key is None:
                LOG.error(f"api key 파일에 GEMINI_API_KEY라는 변수가 없습니다")
            elif key == "":
                LOG.error(f"api key 파일에 내부 변수 GEMINI_API_KEY 값이 빈값입니다 값을 설정해주세요")
            else:
                self.api_key = key
            
        else:
            LOG.error(f"api key 파일이 없습니다 : {os.path.abspath(env_path)}")
        
    def get_api_key(self):
        if self.api_key is not None:
            return self.api_key