package osinfo

import (
	"ahenk-go/pkg/utils"
	"syscall"

	"github.com/shirou/gopsutil/disk"
)

const KB = uint64(1024)

// return kernel information as string map with calling syscall (like uname command)
func GetKernelInfo() map[string]string {
	var uname syscall.Utsname
	if err := syscall.Uname(&uname); err != nil {
		panic(err)
	}
	return map[string]string{
		"Sysname":    utils.Byte2String(uname.Sysname[:]),
		"Nodename":   utils.Byte2String(uname.Nodename[:]),
		"Release":    utils.Byte2String(uname.Release[:]),
		"Version":    utils.Byte2String(uname.Version[:]),
		"Machine":    utils.Byte2String(uname.Machine[:]),
		"Domainname": utils.Byte2String(uname.Domainname[:]),
	}
}

// return all physical disks, USB and CD-ROM devices
//
// An example output:
// device: /dev/example, mountpoint: /, fstype: ext4, opts:rw,noatime
func GetDisks() []disk.PartitionStat {
	parts, err := disk.Partitions(false)
	utils.Check(err)
	return parts
}

// return disk usage as MiB
// TODO different function for all disks / a specified disk?
func GetDiskUsage() map[string]float64 {
	var totalSize, freeSize, usedSize uint64
	for _, part := range GetDisks() {
		u, err := disk.Usage(part.Mountpoint)
		utils.Check(err)
		totalSize += u.Total
		freeSize += u.Free
		usedSize += u.Used
	}
	return map[string]float64{
		"total": utils.Byte2GiB(totalSize),
		"free":  utils.Byte2GiB(freeSize),
		"used":  utils.Byte2GiB(usedSize),
	}
}