package system

import (
	"runtime"
	"strconv"

	"github.com/oneclickvirt/basics/model"
	"github.com/oneclickvirt/basics/system/utils"
)

var (
	expectDiskFsTypes = []string{
		"apfs", "ext4", "ext3", "ext2", "f2fs", "reiserfs", "jfs", "btrfs",
		"fuseblk", "zfs", "simfs", "ntfs", "fat32", "exfat", "xfs", "fuse.rclone",
	}
	cpuType string
)

// GetSystemInfo 获取主机硬件信息
func GetSystemInfo() *model.SystemInfo {
	var ret = &model.SystemInfo{}
	// 系统信息查询
	cpuType, ret.Uptime, ret.Platform, ret.Kernel, ret.Arch, ret.VmType, ret.NatType, ret.TimeZone, _ = getHostInfo()
	// CPU信息查询
	ret, _ = getCpuInfo(ret, cpuType)
	// 硬盘信息查询
	ret.DiskTotal, ret.DiskUsage, ret.Percentage, ret.BootPath, _ = getDiskInfo()
	// 内存信息查询
	ret.MemoryTotal, ret.MemoryUsage, ret.SwapTotal, ret.SwapUsage, ret.VirtioBalloon, ret.KSM = getMemoryInfo()
	// 获取负载信息
	load1, load5, load15, err := getSystemLoad()
	if err != nil {
		load1, load5, load15 = 0, 0, 0
	}
	ret.Load = strconv.FormatFloat(load1, 'f', 2, 64) + " / " +
		strconv.FormatFloat(load5, 'f', 2, 64) + " / " +
		strconv.FormatFloat(load15, 'f', 2, 64)
	// 获取TCP控制算法
	ret.TcpAccelerationMethod = utils.GetTCPAccelerateStatus()
	return ret
}

func CheckSystemInfo(language string) string {
	ret := GetSystemInfo()
	var res string
	if language == "en" {
		res += " CPU Model           : " + ret.CpuModel + "\n"
		res += " CPU Cores           : " + ret.CpuCores + "\n"
		if ret.CpuCache != "" {
			res += " CPU Cache           : " + ret.CpuCache + "\n"
		}
		if runtime.GOOS != "windows" && runtime.GOOS != "macos" {
			res += " AES-NI              : " + ret.CpuAesNi + "\n"
		}
		res += " VM-x/AMD-V/Hyper-V  : " + ret.CpuVAH + "\n"
		res += " RAM                 : " + ret.MemoryUsage + " / " + ret.MemoryTotal + "\n"
		if ret.VirtioBalloon != "" {
			res += " Virtio Balloon      : " + ret.VirtioBalloon + "\n"
		}
		if ret.KSM != "" {
			res += " KSM                 : " + ret.KSM + "\n"
		}
		if ret.SwapTotal == "" && ret.SwapUsage == "" {
			res += " Swap                : [ no swap partition or swap file detected ]" + "\n"
		} else if ret.SwapTotal != "" && ret.SwapUsage != "" {
			res += " Swap                : " + ret.SwapUsage + " / " + ret.SwapTotal + "\n"
		}
		res += " Disk                : " + ret.DiskUsage + " / " + ret.DiskTotal
		if ret.Percentage != "" {
			res += " [" + ret.Percentage + "] " + "\n"
		} else {
			res += "\n"
		}
		res += " Boot Path           : " + ret.BootPath + "\n"
		res += " OS Release          : " + ret.Platform + " [" + ret.Arch + "] " + "\n"
		if ret.Kernel != "" {
			res += " Kernel              : " + ret.Kernel + "\n"
		}
		res += " Uptime              : " + ret.Uptime + "\n"
		res += " Current Time Zone   : " + ret.TimeZone + "\n"
		res += " Load                : " + ret.Load + "\n"
		res += " VM Type             : " + ret.VmType + "\n"
		res += " NAT Type            : " + ret.NatType + "\n"
		if ret.TcpAccelerationMethod != "" {
			res += " Tcp Accelerate      : " + ret.TcpAccelerationMethod + "\n"
		}
	} else if language == "zh" {
		res += " CPU 型号            : " + ret.CpuModel + "\n"
		res += " CPU 数量            : " + ret.CpuCores + "\n"
		if ret.CpuCache != "" {
			res += " CPU 缓存            : " + ret.CpuCache + "\n"
		}
		if runtime.GOOS != "windows" && runtime.GOOS != "macos" {
			res += " AES-NI              : " + ret.CpuAesNi + "\n"
		}
		res += " VM-x/AMD-V/Hyper-V  : " + ret.CpuVAH + "\n"
		res += " 内存                : " + ret.MemoryUsage + " / " + ret.MemoryTotal + "\n"
		if ret.VirtioBalloon != "" {
			res += " Virtio Balloon      : " + ret.VirtioBalloon + "\n"
		}
		if ret.KSM != "" {
			res += " KSM                 : " + ret.KSM + "\n"
		}
		if ret.SwapTotal == "" && ret.SwapUsage == "" {
			res += " Swap                : [ no swap partition or swap file detected ]" + "\n"
		} else if ret.SwapTotal != "" && ret.SwapUsage != "" {
			res += " Swap                : " + ret.SwapUsage + " / " + ret.SwapTotal + "\n"
		}
		res += " 硬盘空间            : " + ret.DiskUsage + " / " + ret.DiskTotal
		if ret.Percentage != "" {
			res += " [" + ret.Percentage + "] " + "\n"
		} else {
			res += "\n"
		}
		res += " 启动盘路径          : " + ret.BootPath + "\n"
		res += " 系统                : " + ret.Platform + " [" + ret.Arch + "] " + "\n"
		if ret.Kernel != "" {
			res += " 内核                : " + ret.Kernel + "\n"
		}
		res += " 系统在线时间        : " + ret.Uptime + "\n"
		res += " 时区                : " + ret.TimeZone + "\n"
		res += " 负载                : " + ret.Load + "\n"
		res += " 虚拟化架构          : " + ret.VmType + "\n"
		res += " NAT类型             : " + ret.NatType + "\n"
		if ret.TcpAccelerationMethod != "" {
			res += " TCP加速方式         : " + ret.TcpAccelerationMethod + "\n"
		}
	}
	return res
}