zeit/z/tui.go

package z

import (
	"fmt"
	"math"

	"github.com/gookit/color"
	"github.com/shopspring/decimal"
)

func GetOutputBoxForNumber(number int, clr func(...interface{}) string) string {
	switch number {
	case 0:
		return clr("  ")
	case 1:
		return clr(" ▄")
	case 2:
		return clr("▄▄")
	case 3:
		return clr("▄█")
	case 4:
		return clr("██")
	}

	return clr("  ")
}

func GetOutputBarForHours(hours decimal.Decimal, stats []Statistic) []string {
	bar := []string{
		color.FgGray.Render("····"),
		color.FgGray.Render("····"),
		color.FgGray.Render("····"),
		color.FgGray.Render("····"),
		color.FgGray.Render("····"),
		color.FgGray.Render("····"),
	}

	hoursInt := int((hours.Round(0)).IntPart())
	rest := ((hours.Round(0)).Mod(decimal.NewFromInt(4))).Round(0)
	restInt := int(rest.IntPart())

	divisible := hoursInt - restInt
	fullparts := divisible / 4

	colorsFull := make(map[int](func(...interface{}) string))
	colorsFullIdx := 0

	colorFraction := color.FgWhite.Render
	colorFractionPrevAmount := 0.0

	for _, stat := range stats {
		statHoursInt, _ := stat.Hours.Float64()
		statRest := (stat.Hours.Round(0)).Mod(decimal.NewFromInt(4))
		statRestFloat, _ := statRest.Float64()

		if statRestFloat > colorFractionPrevAmount {
			colorFractionPrevAmount = statRestFloat
			colorFraction = stat.Color
		}

		fullColoredParts := int(math.Round(statHoursInt) / 4)

		if fullColoredParts == 0 && statHoursInt > colorFractionPrevAmount {
			colorFractionPrevAmount = statHoursInt
			colorFraction = stat.Color
		}

		for i := 0; i < fullColoredParts; i++ {
			colorsFull[colorsFullIdx] = stat.Color
			colorsFullIdx++
		}
	}

	iColor := 0
	for i := (len(bar) - 1); i > (len(bar) - 1 - fullparts); i-- {
		if iColor < colorsFullIdx {
			bar[i] = " " + GetOutputBoxForNumber(4, colorsFull[iColor]) + " "
			iColor++
		} else {
			bar[i] = " " + GetOutputBoxForNumber(4, colorFraction) + " "
		}
	}

	if restInt > 0 {
		bar[(len(bar) - 1 - fullparts)] = " " + GetOutputBoxForNumber(restInt, colorFraction) + " "
	}

	return bar
}

func OutputAppendRight(leftStr string, rightStr string, pad int) string {
	var output string = ""
	var rpos int = 0

	left := []rune(leftStr)
	leftLen := len(left)
	right := []rune(rightStr)
	rightLen := len(right)

	for lpos := 0; lpos < leftLen; lpos++ {
		if left[lpos] == '\n' || lpos == (leftLen-1) {
			output = fmt.Sprintf("%s%*s", output, pad, "")
			for rpos = rpos; rpos < rightLen; rpos++ {
				output = fmt.Sprintf("%s%c", output, right[rpos])
				if right[rpos] == '\n' {
					rpos++
					break
				}
			}
			continue
		}
		output = fmt.Sprintf("%s%c", output, left[lpos])
	}

	return output
}

func GetColorFnFromHex(colorHex string) func(...interface{}) string {
	if colorHex == "" {
		colorHex = "#dddddd"
	}
	return color.NewRGBStyle(color.HEX(colorHex)).Sprint
}