main.go

package main

import (
	"log"
	"os"
	"os/exec"
	"time"

	ui "github.com/gizak/termui/v3"
	flag "github.com/spf13/pflag"

	_ "github.com/go-sql-driver/mysql"
)

const version = "0.1"

type Conf struct {
	Host             string
	DBPwd            string
	DBUser           string
	Port             int
	NumProcessToShow int
}

var (
	host  = flag.StringP("host", "h", "127.0.0.1", "host")
	pwd   = flag.StringP("password", "p", "", "pwd")
	user  = flag.StringP("user", "u", "root", "user")
	port  = flag.IntP("port", "P", 3306, "port")
	count = flag.IntP("count", "n", 50, "Number of process to show")

	cfg *Conf
)

func Config() *Conf {
	return cfg
}

func InitConfig() {
	flag.Parse()
	cfg = &Conf{}

	cfg.DBUser = *user
	cfg.DBPwd = *pwd
	cfg.Host = *host
	cfg.Port = *port
	cfg.NumProcessToShow = *count
}

func main() {
	InitConfig()

	if err := ui.Init(); err != nil {
		log.Print(err)
		os.Exit(-1)
	}
	if err := InitDB(); err != nil {
		cleanExit(err)
	}
	defer func() {
		ui.Close()
		DB().Close()
	}()

	go refreshWorker()
	refreshUI()
}

func cleanExit(err error) {
	ui.Close()
	exec.Command("clear").Run()
	if err != nil {
		log.Print(err)
	}
	os.Exit(0)
}

// refreshUI periodically refreshes the screen.
func refreshUI() {
	controllers := []UIController{
		newProcessListController(),
		newOverviewController(),
		newIOStatController(),
	}
	redraw := make(chan struct{})
	go func() {
		for {
			for _, c := range controllers {
				c.UpdateData()
			}

			redraw <- struct{}{}
			// update every 2 seconds
			time.Sleep(2 * time.Second)
		}
	}()

	evt := ui.PollEvents()
	for {
		select {
		case e := <-evt:
			if e.Type == ui.KeyboardEvent && (e.ID == "q" || e.ID == "<C-c>") {
				cleanExit(nil)
			}
			if e.ID == "<Resize>" {
				for _, c := range controllers {
					c.OnResize(e.Payload.(ui.Resize))
				}
			}

		case <-redraw:
			for _, c := range controllers {
				c.Render()
			}
		}
	}
}