package config

import (
	"encoding/json"
	"os"
	"path/filepath"
)

type Config struct {
	App     AppConfig     `json:"app"`
	Capture CaptureConfig `json:"capture"`
	Parser  ParserConfig  `json:"parser"`
	Log     LogConfig     `json:"log"`
}

type AppConfig struct {
	Name    string `json:"name"`
	Version string `json:"version"`
	Debug   bool   `json:"debug"`
}

type CaptureConfig struct {
	DefaultFilter  string `json:"default_filter"`
	DefaultTimeout int    `json:"default_timeout"`
	BufferSize     int    `json:"buffer_size"`
	MaxPacketSize  int    `json:"max_packet_size"`
}

type ParserConfig struct {
	MaxDataSize      int  `json:"max_data_size"`
	EnableValidation bool `json:"enable_validation"`
}

type LogConfig struct {
	Level      string `json:"level"`
	File       string `json:"file"`
	MaxSize    int    `json:"max_size"`
	MaxBackups int    `json:"max_backups"`
	MaxAge     int    `json:"max_age"`
	Compress   bool   `json:"compress"`
}

// Load 加载配置文件
func Load() (*Config, error) {
	configPath := getConfigPath()

	// 如果配置文件不存在，创建默认配置
	if _, err := os.Stat(configPath); os.IsNotExist(err) {
		if err := createDefaultConfig(configPath); err != nil {
			return nil, err
		}
	}

	file, err := os.Open(configPath)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	var config Config
	if err := json.NewDecoder(file).Decode(&config); err != nil {
		return nil, err
	}

	return &config, nil
}

func getConfigPath() string {
	homeDir, err := os.UserHomeDir()
	if err != nil {
		return "config.json"
	}
	return filepath.Join(homeDir, ".equipment-analyzer", "config.json")
}

func createDefaultConfig(path string) error {
	config := &Config{
		App: AppConfig{
			Name:    "equipment-analyzer",
			Version: "1.0.0",
			Debug:   false,
		},
		Capture: CaptureConfig{
			DefaultFilter:  "tcp and ( port 5222 or port 3333 )",
			DefaultTimeout: 3000,
			BufferSize:     1600,
			MaxPacketSize:  65535,
		},
		Parser: ParserConfig{
			MaxDataSize:      1024 * 1024, // 1MB
			EnableValidation: true,
		},
		Log: LogConfig{
			Level:      "info",
			File:       "equipment-analyzer.log",
			MaxSize:    100, // MB
			MaxBackups: 3,
			MaxAge:     28, // days
			Compress:   true,
		},
	}

	// 创建目录
	dir := filepath.Dir(path)
	if err := os.MkdirAll(dir, 0755); err != nil {
		return err
	}

	// 写入配置文件
	file, err := os.Create(path)
	if err != nil {
		return err
	}
	defer file.Close()

	encoder := json.NewEncoder(file)
	encoder.SetIndent("", "  ")
	return encoder.Encode(config)
}