wtf/wtf.go

package main

import (
	"flag"
	"fmt"
	"os"
	"time"

	"github.com/gdamore/tcell"
	"github.com/olebedev/config"
	"github.com/rivo/tview"
	"github.com/senorprogrammer/wtf/bamboohr"
	"github.com/senorprogrammer/wtf/clocks"
	"github.com/senorprogrammer/wtf/gcal"
	"github.com/senorprogrammer/wtf/git"
	"github.com/senorprogrammer/wtf/github"
	"github.com/senorprogrammer/wtf/jira"
	"github.com/senorprogrammer/wtf/newrelic"
	"github.com/senorprogrammer/wtf/opsgenie"
	"github.com/senorprogrammer/wtf/security"
	"github.com/senorprogrammer/wtf/status"
	"github.com/senorprogrammer/wtf/system"
	"github.com/senorprogrammer/wtf/textfile"
	"github.com/senorprogrammer/wtf/todo"
	"github.com/senorprogrammer/wtf/weather"
	"github.com/senorprogrammer/wtf/wtf"
)

/* -------------------- Built-in Support Documentation -------------------- */

func displayCommandInfo(args []string) {
	if len(args) == 0 {
		return
	}

	cmd := args[0]

	switch cmd {
	case "help", "--help":
		displayHelpInfo()
	case "version", "--version":
		displayVersionInfo()
	}
}

func displayHelpInfo() {
	fmt.Println("Help is coming...")
	os.Exit(0)
}

func displayVersionInfo() {
	fmt.Printf("Version: %s\n", version)
	os.Exit(0)
}

/* -------------------- Functions -------------------- */

func addToGrid(grid *tview.Grid, widget wtf.TextViewer) {
	if widget.Disabled() {
		return
	}

	grid.AddItem(
		widget.TextView(),
		widget.Top(),
		widget.Left(),
		widget.Height(),
		widget.Width(),
		0,
		0,
		false,
	)
}

// Grid stores all the widgets onscreen (like an HTML table)
func buildGrid(modules []wtf.TextViewer) *tview.Grid {
	grid := tview.NewGrid()
	grid.SetColumns(wtf.ToInts(Config.UList("wtf.grid.columns"))...)
	grid.SetRows(wtf.ToInts(Config.UList("wtf.grid.rows"))...)
	grid.SetBorder(false)

	for _, module := range modules {
		addToGrid(grid, module)
		go wtf.Schedule(module)
	}

	return grid
}

func keyboardIntercept(event *tcell.EventKey) *tcell.EventKey {
	switch event.Key() {
	case tcell.KeyCtrlR:
		refreshAllModules()
	case tcell.KeyTab:
		FocusTracker.Next()
	case tcell.KeyBacktab:
		FocusTracker.Prev()
	case tcell.KeyEsc:
		FocusTracker.None()
	default:
		return event
	}

	return event
}

// redrawApp redraws the rendered views to screen on a defined interval (set in config.yml)
// Use this because each textView widget can have it's own update interval, and I don't want to
// manage drawing co-ordination amongst them all. If you need to have a
// widget redraw on it's own schedule, use the view's SetChangedFunc() and pass it `app`.
func redrawApp(app *tview.Application) {
	tick := time.NewTicker(time.Duration(Config.UInt("wtf.refreshInterval", 2)) * time.Second)
	quit := make(chan struct{})

	for {
		select {
		case <-tick.C:
			app.Draw()
		case <-quit:
			tick.Stop()
			return
		}
	}
}

func refreshAllModules() {
	for _, module := range Widgets {
		go module.Refresh()
	}
}

/* -------------------- Main -------------------- */

var Config *config.Config
var FocusTracker wtf.FocusTracker
var Widgets []wtf.TextViewer

var result = wtf.CreateConfigDir()

var (
	builtat = time.Now().Format(wtf.TimestampFormat)
	version = "dev"
)

func main() {
	flagConf := flag.String("config", "~/.wtf/config.yml", "Path to config file")
	flagHelp := flag.Bool("help", false, "Show help")
	flagVers := flag.Bool("version", false, "Show version info")

	flag.Parse()

	if *flagHelp {
		displayHelpInfo()
	}

	if *flagVers {
		displayVersionInfo()
	}

	displayCommandInfo(flag.Args())

	/* -------------------- end flag parsing and handling -------------------- */

	Config = wtf.LoadConfigFile(*flagConf)

	app := tview.NewApplication()
	pages := tview.NewPages()

	bamboohr.Config = Config
	clocks.Config = Config
	gcal.Config = Config
	git.Config = Config
	github.Config = Config
	jira.Config = Config
	newrelic.Config = Config
	opsgenie.Config = Config
	security.Config = Config
	status.Config = Config
	system.Config = Config
	textfile.Config = Config
	todo.Config = Config
	weather.Config = Config
	wtf.Config = Config

	Widgets = []wtf.TextViewer{
		bamboohr.NewWidget(),
		clocks.NewWidget(),
		gcal.NewWidget(),
		git.NewWidget(),
		github.NewWidget(),
		jira.NewWidget(),
		newrelic.NewWidget(),
		opsgenie.NewWidget(),
		security.NewWidget(),
		status.NewWidget(),
		system.NewWidget(builtat, version),
		textfile.NewWidget(),
		todo.NewWidget(app, pages),
		weather.NewWidget(),
	}

	FocusTracker = wtf.FocusTracker{
		App:     app,
		Idx:     0,
		Widgets: Widgets,
	}

	// Loop in a routine to redraw the screen
	go redrawApp(app)

	grid := buildGrid(Widgets)
	pages.AddPage("grid", grid, true, true)
	app.SetInputCapture(keyboardIntercept)

	if err := app.SetRoot(pages, true).Run(); err != nil {
		os.Exit(1)
	}
}