package cmd

import (
	"os"

	"github.com/evil7/hostsync/core"
	"github.com/evil7/hostsync/utils"
	"github.com/spf13/cobra"
)

// addCmd 添加命令
var addCmd = &cobra.Command{
	Use:   "add <blockName> <domain> [ip]",
	Short: "添加域名记录",
	Long: `添加域名记录到指定块。如果不存在该块则会自动创建。
如果不指定IP地址，将自动解析域名。

示例:
  hostsync add github github.com 140.82.114.3  # 指定IP添加
  hostsync add github api.github.com           # 自动解析IP添加`,
	Args: cobra.RangeArgs(2, 3),
	Run:  runAdd,
}

func runAdd(cmd *cobra.Command, args []string) {
	utils.CheckAdmin()

	blockName := args[0]
	domain := args[1]
	var ip string

	if !utils.ValidateBlockName(blockName) {
		utils.LogError("无效的块名称: %s", blockName)
		utils.LogInfo("块名称只能包含字母、数字、下划线和连字符")
		os.Exit(1)
	}

	hm := core.NewHostsManager()
	if err := hm.Load(); err != nil {
		utils.LogError("加载hosts文件失败: %v", err)
		os.Exit(1)
	}

	if len(args) == 3 {
		// 使用指定的IP
		ip = args[2]
	} else {
		// 自动解析IP
		utils.LogInfo("正在解析域名: %s", domain)
		resolver := core.NewDNSResolver()
		var err error
		ip, err = resolver.ResolveDomain(domain, "", "")
		if err != nil {
			utils.LogError("解析域名失败: %v", err)
			os.Exit(1)
		}
		utils.LogSuccess("解析成功: %s -> %s", domain, ip)
	}

	if err := hm.AddEntry(blockName, domain, ip); err != nil {
		utils.LogError("添加记录失败: %v", err)
		os.Exit(1)
	}

	utils.LogSuccess("已添加记录: %s -> %s (块: %s)", domain, ip, blockName)
}