package config

import (
	"os"
	"time"
)

type Config struct {
	Server    ServerConfig
	Database  DatabaseConfig
	Redis     RedisConfig
	JWT       JWTConfig
	Dify      DifyConfig
	LLM       LLMConfig
	Embedding EmbeddingConfig
	Gateway   GatewayConfig
	MinIO     MinIOConfig
	PPTWorker PPTWorkerConfig
}

type PPTWorkerConfig struct {
	URL string // PPT Worker 微服务地址
}

type LLMConfig struct {
	Provider         string // "openai" or "anthropic"
	OpenAIKey        string
	OpenAIBaseURL    string
	OpenAIModel      string
	AnthropicKey     string
	AnthropicBaseURL string
	AnthropicModel   string
}

type EmbeddingConfig struct {
	APIKey     string // Embedding API 密钥
	BaseURL    string // Embedding API 基础 URL（OpenAI 兼容格式）
	Model      string // 向量模型名称
	Dimensions int    // 向量维度
}

type ServerConfig struct {
	Host string
	Port string
}

type DatabaseConfig struct {
	URL string
}

type RedisConfig struct {
	URL string
}

type JWTConfig struct {
	Secret        string
	AccessExpiry  time.Duration
	RefreshExpiry time.Duration
}

type DifyConfig struct {
	APIURL string
	APIKey string
}

type GatewayConfig struct {
	URL string
}

type MinIOConfig struct {
	Endpoint  string
	AccessKey string
	SecretKey string
	Bucket    string
	UseSSL    bool
}

func Load() *Config {
	return &Config{
		Server: ServerConfig{
			Host: getEnv("SERVER_HOST", "0.0.0.0"),
			Port: getEnv("SERVER_PORT", "8080"),
		},
		Database: DatabaseConfig{
			URL: getEnv("DATABASE_URL", "postgres://localhost:5432/govai_portal?sslmode=disable"),
		},
		Redis: RedisConfig{
			URL: getEnv("REDIS_URL", "redis://localhost:6379"),
		},
		JWT: JWTConfig{
			Secret:        getEnv("JWT_SECRET", "dev-secret-change-in-production"),
			AccessExpiry:  24 * time.Hour,
			RefreshExpiry: 7 * 24 * time.Hour,
		},
		Dify: DifyConfig{
			APIURL: getEnv("DIFY_API_URL", "http://localhost:5001/v1"),
			APIKey: getEnv("DIFY_API_KEY", ""),
		},
		LLM: LLMConfig{
			Provider:         getEnv("LLM_PROVIDER", "openai"),
			OpenAIKey:        getEnv("OPENAI_API_KEY", ""),
			OpenAIBaseURL:    getEnv("OPENAI_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
			OpenAIModel:      getEnv("OPENAI_MODEL", "qwen-plus"),
			AnthropicKey:     getEnv("ANTHROPIC_API_KEY", ""),
			AnthropicBaseURL: getEnv("ANTHROPIC_BASE_URL", "https://api.anthropic.com"),
			AnthropicModel:   getEnv("ANTHROPIC_MODEL", "claude-sonnet-4-20250514"),
		},
		Embedding: EmbeddingConfig{
			APIKey:     getEnv("EMBEDDING_API_KEY", ""),
			BaseURL:    getEnv("EMBEDDING_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
			Model:      getEnv("EMBEDDING_MODEL", "text-embedding-v3"),
			Dimensions: 1024,
		},
		Gateway: GatewayConfig{
			URL: getEnv("MODEL_GATEWAY_URL", "http://localhost:8081"),
		},
		MinIO: MinIOConfig{
			Endpoint:  getEnv("MINIO_ENDPOINT", "localhost:9000"),
			AccessKey: getEnv("MINIO_ACCESS_KEY", "minioadmin"),
			SecretKey: getEnv("MINIO_SECRET_KEY", "minioadmin"),
			Bucket:    getEnv("MINIO_BUCKET", "aily-files"),
			UseSSL:    false,
		},
		PPTWorker: PPTWorkerConfig{
			URL: getEnv("PPT_WORKER_URL", "http://localhost:8090"),
		},
	}
}

func getEnv(key, fallback string) string {
	if v := os.Getenv(key); v != "" {
		return v
	}
	return fallback
}