package widgets

import (
	"time"

	ui "github.com/cjbassi/gotop/termui"
	ps "github.com/shirou/gopsutil/mem"
)

type Mem struct {
	*ui.LineGraph
	interval time.Duration
}

func NewMem() *Mem {
	m := &Mem{ui.NewLineGraph(), time.Second}
	m.Label = "Memory Usage"
	m.Data["Main"] = []float64{0} // Sets initial data to 0
	m.Data["Swap"] = []float64{0}
	m.LineColor["Main"] = ui.ColorMagenta
	m.LineColor["Swap"] = ui.ColorYellow

	go m.update()
	ticker := time.NewTicker(m.interval)
	go func() {
		for range ticker.C {
			m.update()
		}
	}()

	return m
}

func (m *Mem) update() {
	main, _ := ps.VirtualMemory()
	swap, _ := ps.SwapMemory()
	m.Data["Main"] = append(m.Data["Main"], main.UsedPercent)
	m.Data["Swap"] = append(m.Data["Swap"], swap.UsedPercent)
}