sampler-fork/config/default.go

package config

import (
	"github.com/sqshq/sampler/console"
)

const (
	defaultRateMs = 1000
	defaultScale  = 1
	defaultTheme  = console.ThemeDark
)

func (c *Config) setDefaults() {
	c.setDefaultValues()
	c.setDefaultColors()
	c.setDefaultArrangement()
}

func (c *Config) setDefaultValues() {

	if c.Theme == nil {
		t := defaultTheme
		c.Theme = &t
	}

	for i, chart := range c.RunCharts {

		setDefaultTriggersValues(chart.Triggers)
		chart.ComponentConfig.Type = TypeRunChart

		if chart.RateMs == nil {
			r := defaultRateMs
			chart.RateMs = &r
		}
		if chart.Scale == nil {
			p := defaultScale
			chart.Scale = &p
		}
		if chart.Legend == nil {
			chart.Legend = &LegendConfig{true, true}
			c.RunCharts[i] = chart
		}
		c.RunCharts[i] = chart
	}

	for i, line := range c.SparkLines {

		setDefaultTriggersValues(line.Triggers)
		line.ComponentConfig.Type = TypeSparkLine
		line.Item.Label = &line.Title

		if line.RateMs == nil {
			r := defaultRateMs
			line.RateMs = &r
		}
		if line.Scale == nil {
			p := defaultScale
			line.Scale = &p
		}
		c.SparkLines[i] = line
	}

	for i, chart := range c.BarCharts {

		setDefaultTriggersValues(chart.Triggers)
		chart.ComponentConfig.Type = TypeBarChart

		if chart.RateMs == nil {
			r := defaultRateMs
			chart.RateMs = &r
		}
		if chart.Scale == nil {
			p := defaultScale
			chart.Scale = &p
		}
		c.BarCharts[i] = chart
	}

	for i, g := range c.Gauges {

		setDefaultTriggersValues(g.Triggers)
		g.ComponentConfig.Type = TypeGauge

		if g.RateMs == nil {
			r := defaultRateMs
			g.RateMs = &r
		}
		if g.Scale == nil {
			p := defaultScale
			g.Scale = &p
		}

		cur := "cur"
		max := "max"
		min := "min"

		g.Cur.Label = &cur
		g.Max.Label = &max
		g.Min.Label = &min

		c.Gauges[i] = g
	}

	for i, box := range c.AsciiBoxes {

		setDefaultTriggersValues(box.Triggers)
		box.ComponentConfig.Type = TypeAsciiBox

		if box.RateMs == nil {
			r := defaultRateMs
			box.RateMs = &r
		}
		if box.Label == nil {
			label := string(i)
			box.Label = &label
		}
		if box.Font == nil {
			font := console.AsciiFont2D
			box.Font = &font
		}
		if box.Border == nil {
			border := true
			box.Border = &border
		}
		c.AsciiBoxes[i] = box
	}

	for i, box := range c.TextBoxes {

		setDefaultTriggersValues(box.Triggers)
		box.ComponentConfig.Type = TypeTextBox

		if box.RateMs == nil {
			r := defaultRateMs
			box.RateMs = &r
		}
		if box.Label == nil {
			label := string(i)
			box.Label = &label
		}
		if box.Border == nil {
			border := true
			box.Border = &border
		}

		c.TextBoxes[i] = box
	}
}

func setDefaultTriggersValues(triggers []TriggerConfig) {

	defaultTerminalBell := false
	defaultSound := false
	defaultVisual := false

	for i, trigger := range triggers {

		if trigger.Actions == nil {
			trigger.Actions = &ActionsConfig{TerminalBell: &defaultTerminalBell, Sound: &defaultSound, Visual: &defaultVisual, Script: nil}
		} else {
			if trigger.Actions.TerminalBell == nil {
				trigger.Actions.TerminalBell = &defaultTerminalBell
			}
			if trigger.Actions.Sound == nil {
				trigger.Actions.Sound = &defaultSound
			}
			if trigger.Actions.Visual == nil {
				trigger.Actions.Visual = &defaultVisual
			}
		}

		triggers[i] = trigger
	}
}

func (c *Config) setDefaultColors() {

	palette := console.GetPalette(*c.Theme)
	colorsCount := len(palette.ContentColors)

	for _, ch := range c.RunCharts {
		for j, item := range ch.Items {
			if item.Color == nil {
				item.Color = &palette.ContentColors[j%colorsCount]
				ch.Items[j] = item
			}
		}
	}

	for _, b := range c.BarCharts {
		for j, item := range b.Items {
			if item.Color == nil {
				item.Color = &palette.ContentColors[j%colorsCount]
				b.Items[j] = item
			}
		}
	}

	for i, s := range c.SparkLines {
		s.Gradient = &palette.GradientColors[i%(len(palette.GradientColors))]
		c.SparkLines[i] = s
	}

	for i, g := range c.Gauges {
		if g.Color == nil {
			g.Color = &palette.ContentColors[i%colorsCount]
			c.Gauges[i] = g
		}
	}
}