slideshow/internal/config/config.go

package config

import (
	"os"
	"strconv"
	"strings"
	"time"

	"git.velvettear.de/velvettear/loggo"
)

var Interval time.Duration
var Directory string
var Resolution string
var PaletteFile string
var PaletteAlgorithm int
var PaletteColors int

// initialize the config
func Initialize() {
	loggo.SetLogLevelByName(os.Getenv("SLIDESHOW_LOGLEVEL"))
	tmpInt, _ := strconv.Atoi(os.Getenv("SLIDESHOW_INTERVAL"))
	if tmpInt <= 0 {
		tmpInt = 60
	}
	Interval = time.Duration(tmpInt) * time.Second
	Directory = os.Getenv("SLIDESHOW_DIRECTORY")
	if len(Directory) == 0 {
		tmp, error := os.UserHomeDir()
		if error != nil {
			loggo.Fatal("encountered an error getting the current user's home directory", error.Error())
		}
		Directory = tmp
	}
	stats, error := os.Stat(Directory)
	if error != nil {
		loggo.Fatal("encountered an error checking the directory '"+Directory+"'", error.Error())
	}
	if !stats.IsDir() {
		loggo.Fatal("configured directory '" + Directory + "' is not a valid directory")
	}
	Resolution = os.Getenv("SLIDESHOW_RESOLUTION")
	if len(Resolution) > 0 {
		width, height, found := strings.Cut(Resolution, "x")
		if !found {
			loggo.Fatal("encountered an error parsing the configured resolution, make sure to specify the format like '1920x1080'")
		}
		_, error = strconv.Atoi(width)
		if error != nil {
			loggo.Fatal("encountered an error parsing the configured width '" + width + "'")
		}
		_, error = strconv.Atoi(height)
		if error != nil {
			loggo.Fatal("encountered an error parsing the configured height '" + height + "'")
		}
	}
	PaletteFile = os.Getenv("SLIDESHOW_PALETTE")
	tmpString := os.Getenv("SLIDESHOW_PALETTE_ALGORITHM")
	if strings.ToLower(tmpString) == "wu" {
		PaletteAlgorithm = 0
	} else {
		PaletteAlgorithm = 1
	}
	tmpInt, _ = strconv.Atoi(os.Getenv("SLIDESHOW_PALETTE_COLORS"))
	if tmpInt <= 0 {
		tmpInt = 16
	}
	PaletteColors = tmpInt
}

// check if a resolution has been specified
func IsResolutionSet() bool {
	return len(Resolution) > 0
}

// check if a color palette should be generated
func IsPaletteSet() bool {
	return len(PaletteFile) > 0
}