from pydantic import BaseSettings
from typing import Optional
from pathlib import Path


PROJECT_ROOT = Path(__file__).resolve().parents[2]
ENV_FILE = PROJECT_ROOT / ".env"


class Settings(BaseSettings):
    class Config:
        env_file = str(ENV_FILE)
        env_file_encoding = "utf-8"
        extra = "ignore"

    APP_NAME: str = "TradeMate"

    SECRET_KEY: str
    JWT_ALGORITHM: str = "HS256"
    ACCESS_TOKEN_EXPIRE_MINUTES: int = 60
    REFRESH_TOKEN_EXPIRE_DAYS: int = 30

    DATABASE_URL: str
    DB_ECHO: bool = False

    REDIS_URL: str = "redis://localhost:6379/0"

    CELERY_BROKER_URL: str = "redis://localhost:6379/1"
    CELERY_RESULT_BACKEND: str = "redis://localhost:6379/2"

    OPENAI_API_KEY: Optional[str] = None
    ANTHROPIC_API_KEY: Optional[str] = None
    DEEPL_API_KEY: Optional[str] = None

    SENSENOVA_API_KEY: Optional[str] = None
    SENSENOVA_BASE_URL: str = "https://token.sensenova.cn/v1"
    SENSENOVA_MODEL: str = "sensenova-6.7-flash-lite"

    IFLYTEK_API_KEY: Optional[str] = None
    IFLYTEK_API_BASE: str = "https://maas-api.cn-huabei-1.xf-yun.com/v2"
    IFLYTEK_MODEL: str = "astron-code-latest"

    LOCAL_MODEL_ENABLED: bool = False
    LOCAL_MODEL_URL: str = "http://localhost:8001"

    OPENCODE_GO_API_KEY: Optional[str] = None
    OPENCODE_GO_BASE_URL: str = "https://opencode.ai/zen/go/v1"
    OPENCODE_GO_MODEL: str = "deepseek-v4-flash"

    WHATSAPP_API_TOKEN: Optional[str] = None
    WHATSAPP_PHONE_NUMBER_ID: Optional[str] = None
    WHATSAPP_WEBHOOK_VERIFY_TOKEN: Optional[str] = None

    WECHAT_APP_ID: Optional[str] = None
    WECHAT_APP_SECRET: Optional[str] = None
    WECHAT_PUSH_TEMPLATE_ID: Optional[str] = None

    EXCHANGE_RATE_API_KEY: Optional[str] = None

    UPLOAD_DIR: str = "./uploads"
    MAX_UPLOAD_SIZE: int = 10 * 1024 * 1024

    FRONTEND_URL: str = "http://localhost:3000"
    BACKEND_URL: str = "http://localhost:8000"

    SENTRY_DSN: Optional[str] = None
    DEBUG: bool = True

    AI_ROUTING: dict = {
        "translate": {"primary": "opencode_go", "fallback": ["sensenova", "openai", "local"]},
        "reply": {"primary": "opencode_go", "fallback": ["sensenova", "anthropic", "local"]},
        "marketing": {"primary": "opencode_go", "fallback": ["sensenova", "openai", "local"]},
        "extract": {"primary": "opencode_go", "fallback": ["sensenova", "openai"]},
        "quotation": {"primary": "opencode_go", "fallback": ["sensenova", "openai"]},
    }

    FREE_DAILY_TRANSLATE_CHARS: int = 5000
    FREE_DAILY_REPLIES: int = 20
    FREE_DAILY_MARKETING: int = 5
    FREE_MAX_CUSTOMERS: int = 5
    FREE_MAX_PRODUCTS: int = 1
    FREE_DAILY_QUOTATIONS: int = 3

    PRO_DAILY_TRANSLATE_CHARS: int = 50000
    PRO_DAILY_REPLIES: int = 200
    PRO_DAILY_MARKETING: int = 50
    PRO_MAX_CUSTOMERS: int = 100
    PRO_MAX_PRODUCTS: int = 20
    PRO_DAILY_QUOTATIONS: int = 30


settings = Settings()