package clocks

import (
	"fmt"
	"sort"
	"strings"
	"time"

	"github.com/olebedev/config"
	"github.com/senorprogrammer/wtf/wtf"
)

const TimeFormat = "15:04 MST"
const DateFormat = "Jan 2"

// Config is a pointer to the global config object
var Config *config.Config

type Widget struct {
	wtf.TextWidget
}

func NewWidget() *Widget {
	widget := Widget{
		TextWidget: wtf.NewTextWidget(" 🕗 World Clocks ", "clocks"),
	}

	return &widget
}

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

func (widget *Widget) Refresh() {
	if widget.Disabled() {
		return
	}

	widget.View.Clear()
	widget.display()
	widget.RefreshedAt = time.Now()
}

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

func (widget *Widget) colorFor(idx int) string {
	rowColor := Config.UString("wtf.mods.clocks.rowcolors.even", "lightblue")

	if idx%2 == 0 {
		rowColor = Config.UString("wtf.mods.clocks.rowcolors.odd", "white")
	}

	return rowColor
}

func (widget *Widget) display() {
	locs := widget.locations(Config.UMap("wtf.mods.clocks.locations"))

	if len(locs) == 0 {
		fmt.Fprintf(widget.View, "\n%s", " no timezone data available")
		return
	}

	labels := widget.sortedLabels(locs)

	tzs := []string{}
	for idx, label := range labels {
		zoneStr := fmt.Sprintf(
			" [%s]%-12s %-10s %7s[white]",
			widget.colorFor(idx),
			label,
			locs[label].Format(TimeFormat),
			locs[label].Format(DateFormat),
		)

		tzs = append(tzs, zoneStr)
	}

	fmt.Fprintf(widget.View, "\n%s", strings.Join(tzs, "\n"))
}

func (widget *Widget) locations(locs map[string]interface{}) map[string]time.Time {
	times := make(map[string]time.Time)

	for label, loc := range locs {
		tzloc, err := time.LoadLocation(loc.(string))

		if err != nil {
			continue
		}

		times[label] = time.Now().In(tzloc)
	}

	return times
}

func (widget *Widget) sortedLabels(locs map[string]time.Time) []string {
	labels := []string{}

	for label, _ := range locs {
		labels = append(labels, label)
	}

	sort.Strings(labels)

	return labels
}