gatus/config/ui/ui.go

package ui

import (
	"bytes"
	"html/template"
)

const (
	defaultTitle  = "Health Dashboard | Gatus"
	defaultHeader = "Health Status"
	defaultLogo   = ""
	defaultLink   = ""
)

var (
	// StaticFolder is the path to the location of the static folder from the root path of the project
	// The only reason this is exposed is to allow running tests from a different path than the root path of the project
	StaticFolder = "./web/static"
)

// Config is the configuration for the UI of Gatus
type Config struct {
	Title  string `yaml:"title,omitempty"`  // Title of the page
	Header string `yaml:"header,omitempty"` // Header is the text at the top of the page
	Logo   string `yaml:"logo,omitempty"`   // Logo to display on the page
	Link   string `yaml:"link,omitempty"`   // Link to open when clicking on the logo
}

// GetDefaultConfig returns a Config struct with the default values
func GetDefaultConfig() *Config {
	return &Config{
		Title:  defaultTitle,
		Header: defaultHeader,
		Logo:   defaultLogo,
		Link:   defaultLink,
	}
}

// ValidateAndSetDefaults validates the UI configuration and sets the default values if necessary.
func (cfg *Config) ValidateAndSetDefaults() error {
	if len(cfg.Title) == 0 {
		cfg.Title = defaultTitle
	}
	if len(cfg.Header) == 0 {
		cfg.Header = defaultHeader
	}
	if len(cfg.Header) == 0 {
		cfg.Header = defaultLink
	}
	t, err := template.ParseFiles(StaticFolder + "/index.html")
	if err != nil {
		return err
	}
	var buffer bytes.Buffer
	err = t.Execute(&buffer, cfg)
	if err != nil {
		return err
	}
	return nil
}