package config

import (
	"agent-wdd/log"
	"encoding/json"
	"io"
	"net/http"
	"time"
)

// 能够联网,就大于这个数字  外网9 国内7 不能联网1 未知0
const InternetBaseLine = 1

// 定义响应数据结构体
type IPInfo struct {
	IP       string `json:"ip"`
	City     string `json:"city"`
	Region   string `json:"region"`
	Country  string `json:"country"`
	Loc      string `json:"loc"`
	Org      string `json:"org"`
	Postal   string `json:"postal"`
	Timezone string `json:"timezone"`
}

// Gather 获取网络相关的信息
func (network *Network) Gather() {

	log.Info("Gathering INFO => NETWORK !")
	// 能够联网
	network.Internet = CanConnectInternet()

	// 获取公网的相关信息
	pub := PublicInfo{}
	pub.GetPublicInfo()

	network.Public = pub

	//获取本机网卡相关的内容
}

// CanConnectInternet 判定主机能够上网 外网为9 国内为7 不能上网为1
func CanConnectInternet() int {

	// 读取 config 文件，判定能否上网
	internetCode := ConfigCache.Agent.Network.Internet

	if internetCode == 0 {
		// 没有相关的信息,需要重新判定
		internetCode = judgeCanConnectInternet()

		// 持久化保存
		ConfigCache.Agent.Network.Internet = internetCode
		SaveConfig()
	}

	return internetCode
}

// judgeCanConnectInternet 请求网址判定主机能否上网 请求 www.google.com 如果请求正常 返回9 如果超时三秒 请求baidu.com，如果没有错误返回7 如果错误返回1
func judgeCanConnectInternet() int {
	client := http.Client{
		Timeout: 3 * time.Second,
	}

	_, err := client.Get("https://www.google.com")
	if err == nil {
		return 9
	}

	_, err = client.Get("https://www.baidu.com")
	if err == nil {
		return 7
	}

	return 1
}

// GetPublicInfo 获取服务器的公网信息
func (p PublicInfo) GetPublicInfo() {

	if CanConnectInternet() == InternetBaseLine {
		// 无法联网, 假信息

		fakePublicInfo := PublicInfo{
			IPv4:    "1.1.1.1",
			IPv6:    "2400::1",
			Country: "CN",
			City:    "Shanghai",
			ASN:     "Wdd Inc",
		}

		ConfigCache.Agent.Network.Public = fakePublicInfo

		SaveConfig()
		return
	}

	// 可以联网
	// 创建带有超时的HTTP客户端
	client := &http.Client{
		Timeout: 3 * time.Second,
	}

	// 创建新的请求对象
	req, err := http.NewRequest("GET", "https://ipinfo.io", nil)
	if err != nil {
		log.Error("创建请求失败: %v", err)
	}

	// 设置请求头
	req.Header.Add("Authorization", "Bearer 6ecb0db9bd8f19")
	req.Header.Add("Accept", "application/json")

	// 发送请求
	resp, err := client.Do(req)
	if err != nil {
		log.Error("请求PublicInfo失败: %v", err)
	}
	defer resp.Body.Close()

	// 检查响应状态码
	if resp.StatusCode != http.StatusOK {
		log.Error("非200状态码: %d", resp.StatusCode)
	}

	// 读取响应体
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		log.Error("读取响应失败: %v", err)
	}

	// 解析JSON数据
	var info IPInfo
	if err := json.Unmarshal(body, &info); err != nil {
		log.Error("JSON解析失败: %v", err)
	}

	// 打印解析结果
	log.Info("IP信息:\n%+v\n", info)

	// 保存信息
	p.IPv4 = info.IP
	p.ASN = info.Org
	p.Country = info.Country
	p.City = info.City

	ConfigCache.Agent.Network.Public = p

	SaveConfig()

}