package config

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

// URLConfig 存储所有URL配置
type URLConfig struct {
	URLs []string `json:"urls"`
}

// SaveURLsToFile 将URLs保存到文件
func SaveURLsToFile(urls []string, filename string) error {
	config := URLConfig{
		URLs: urls,
	}

	data, err := json.MarshalIndent(config, "", "  ")
	if err != nil {
		return err
	}

	return os.WriteFile(filename, data, 0644)
}

// GetProtectedURLs 从文件中读取需要保护的URLs
func GetProtectedURLs() ([]string, error) {
	configPath := filepath.Join("config", "urls.json")
	data, err := os.ReadFile(configPath)
	if err != nil {
		return nil, err
	}

	var config URLConfig
	if err := json.Unmarshal(data, &config); err != nil {
		return nil, err
	}

	return config.URLs, nil
}