// Demo code for the bar chart primitive.
package main

import (
	"math/rand"
	"time"

	"github.com/gdamore/tcell/v2"
	"github.com/navidys/tgraph"
	"github.com/rivo/tview"
)

func main() {
	app := tview.NewApplication()

	barGraph := tgraph.NewBarChart()
	barGraph.SetBorder(true)
	barGraph.SetTitle("System Resource Usage")
	// display system metric usage
	barGraph.AddBar("cpu", 20, tcell.ColorBlue)
	barGraph.AddBar("mem", 60, tcell.ColorRed)
	barGraph.AddBar("swap", 80, tcell.ColorGreen)
	barGraph.AddBar("disk", 100, tcell.ColorOrange)
	barGraph.SetMaxValue(100)

	flex := tview.NewFlex().SetDirection(tview.FlexColumn)
	flex.AddItem(barGraph, 40, 0, false)
	flex.AddItem(tview.NewBox().SetBorder(true), 40, 0, false)

	flex.SetRect(0, 0, 100, 15)

	update := func() {
		rand.Seed(time.Now().UnixNano())
		tick := time.NewTicker(1000 * time.Millisecond)
		for {
			select {
			case <-tick.C:
				rangeLower := 0
				rangeUpper := 100
				randomNum := rangeLower + rand.Intn(rangeUpper-rangeLower+1)
				barGraph.SetBarValue("cpu", randomNum)
				randomNum = rangeLower + rand.Intn(rangeUpper-rangeLower+1)
				barGraph.SetBarValue("memory", randomNum)
				randomNum = rangeLower + rand.Intn(rangeUpper-rangeLower+1)
				barGraph.SetBarValue("swap", randomNum)
				randomNum = rangeLower + rand.Intn(rangeUpper-rangeLower+1)
				barGraph.SetBarValue("disk", randomNum)
				app.Draw()
			}
		}
	}
	go update()

	if err := app.SetRoot(flex, false).EnableMouse(true).Run(); err != nil {
		panic(err)
	}
}