import os
import sys
from dotenv import load_dotenv

# .env 파일 로드
load_dotenv()

class Config:
    # 1. 기본 설정
    BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    
    # 2. NAS 및 AI 서버
    NAS_API_URL = os.getenv("NAS_API_URL", "http://192.168.45.54:18500")
    OLLAMA_API_URL = os.getenv("OLLAMA_API_URL", "http://localhost:11434")
    OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama3.1:8b-instruct-q8_0")
    
    # 3. KIS 한국투자증권
    KIS_ENV_TYPE = os.getenv("KIS_ENV_TYPE", "virtual").lower()
    
    if KIS_ENV_TYPE == "real":
        KIS_APP_KEY = os.getenv("KIS_REAL_APP_KEY")
        KIS_APP_SECRET = os.getenv("KIS_REAL_APP_SECRET")
        KIS_ACCOUNT = os.getenv("KIS_REAL_ACCOUNT")
        KIS_IS_VIRTUAL = False
        KIS_BASE_URL = "https://openapi.koreainvestment.com:9443"
    else:
        KIS_APP_KEY = os.getenv("KIS_VIRTUAL_APP_KEY")
        KIS_APP_SECRET = os.getenv("KIS_VIRTUAL_APP_SECRET")
        KIS_ACCOUNT = os.getenv("KIS_VIRTUAL_ACCOUNT")
        KIS_IS_VIRTUAL = True
        KIS_BASE_URL = "https://openapivts.koreainvestment.com:29443"

    # 4. 텔레그램
    TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
    TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID")
    
    # 5. 매매 설정 (상수)
    MAX_INVESTMENT_PER_STOCK = 3000000  # 종목당 최대 300만원
    
    # 6. 데이터 경로
    DATA_DIR = os.path.join(BASE_DIR, "data")
    if not os.path.exists(DATA_DIR):
        os.makedirs(DATA_DIR, exist_ok=True)
        
    IPC_FILE = os.path.join(DATA_DIR, "bot_ipc.json")
    HISTORY_FILE = os.path.join(DATA_DIR, "daily_trade_history.json")
    WATCHLIST_FILE = os.path.join(DATA_DIR, "watchlist.json")

    # 7. 타임아웃 등
    HTTP_TIMEOUT = 10

    @staticmethod
    def validate():
        """필수 설정 검증"""
        missing = []
        if not Config.KIS_APP_KEY: missing.append("KIS_APP_KEY")
        if not Config.KIS_APP_SECRET: missing.append("KIS_APP_SECRET")
        
        if missing:
            print(f"⚠️ [Config] Missing Env Params: {', '.join(missing)}")
            return False
        return True