package example

import (
	"github.com/shirou/gopsutil/v3/load"
	"github.com/shirou/gopsutil/v3/mem"
	"net"
	"strconv"
	"strings"
)

func GetMemInfo() (free float64) {
	memInfo, err := mem.VirtualMemory()
	if err != nil {
		panic(err)
	}
	free = memInfo.UsedPercent
	return free
}

func GetCpuLoad() (lod int) {
	lo, err := load.Avg()
	if err != nil {
		panic(err)
	}
	lod = int(lo.Load1)
	return lod
}

func GetHostIp() (ips string) {
	addrSlice, _ := net.InterfaceAddrs()
	for _, addr := range addrSlice {
		if ipnet, ok := addr.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
			if nil != ipnet.IP.To4() {
				ip := ipnet.IP.String()
				if !strings.HasPrefix(ip, "169") {
					ips = ip
				}
			}
		}

	}
	return ips
}

// func GetHostAll() (info map[string]int) {
// 	info = map[string]int{
// 		// "hostip": GetHostIp(),
// 		"load": GetCpuLoad(),
// 		"free": int(GetMemInfo()),
// 	}
// 	return info
// }

func GetHostAll() (info map[string]string) {
	info = map[string]string{
		"ip":   GetHostIp(),
		"load": strconv.Itoa(GetCpuLoad()),
		"free": strconv.Itoa(int(GetMemInfo())),
	}
	return info
}