config.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"os/user"
)

// Config Struct for Configuration File
type Config struct {
	Format           []string    `json:"format"`           // Format Array for Modules
	ShowASCII        bool        `json:"showAscii"`        // True/False to Display ASCII
	UseDefaultColors bool        `json:"useDefaultColors"` // Use Default Colors
	UseSmallASCII    bool        `json:"useSmallAscii"`    // Use Small ASCII Art
	ASCIIColor       string      `json:"asciiColor"`       // Change ASCII Color
	UseCustomASCII   bool        `json:"useCustomAscii"`   // Use Custom ASCII TRUE/FALSE
	CustomASCIIPath  string      `json:"customAsciiPath"`  // Absolute Path to Custom ASCII
	UserColor        string      `json:"userColor"`        // Color for Username
	SepColor         string      `json:"sepColor"`         // Color for Separator
	TitleColor       string      `json:"titleColor"`       // Color for Title
	InfoColor        string      `json:"infoColor"`        // Color for Information
	Titles           TitleValues `json:"titles"`           // Title Struct
	AutoUpdate       bool        `json:"autoupdate"`       // Should Auto Update?
	Version          int64       `json:"configVersion"`    // Configuration Version
}

// TitleValues Struct for Title Strings
type TitleValues struct {
	Memory         string `json:"memory"`
	CPU            string `json:"cpu"`
	CPUCores       string `json:"cpuCores"`
	CPUThreads     string `json:"cpuThreads"`
	GPUs           string `json:"gpus"`
	DiskSize       string `json:"diskSize"`
	WindowsVersion string `json:"windowsVersion"`
	Bios           string `json:"bios"`
	Baseboard      string `json:"baseboard"`
	ProcessCount   string `json:"processCount"`
	Uptime         string `json:"uptime"`
	Network        string `json:"network"`
}

func updateConfig(config Config) {
	user, _ := user.Current()
	if config.Version == 0 {
		config.AutoUpdate = true
		config.Version = 1
	}
	if config.Version == 1 {
		config.Titles.Network = "Local IP"
		config.Format = append(config.Format, "network")
		config.Version = 2
	}
	file, _ := json.MarshalIndent(config, "", " ")
	_ = ioutil.WriteFile(user.HomeDir+"\\.winfetch.json", file, 0644)
}

func newConfig() Config {
	config := Config{}
	config.Format = []string{"user", "sep", "uptime", "mem", "cpu", "procs", "cpuCores", "cpuThreads", "disk", "wversion", "gpus", "bios", "baseboard", "network"}
	config.ShowASCII = true
	config.UseSmallASCII = false
	config.UseCustomASCII = false
	config.CustomASCIIPath = ""
	config.UseDefaultColors = true
	config.ASCIIColor = "Blue"
	config.UserColor = "Red"
	config.SepColor = "Red"
	config.TitleColor = "Green"
	config.InfoColor = "White"
	config.Titles.Memory = "Memory"
	config.Titles.CPUCores = "CPU Cores"
	config.Titles.CPUThreads = "CPU Threads"
	config.Titles.GPUs = "GPU #"
	config.Titles.DiskSize = "Disk Size"
	config.Titles.WindowsVersion = "Windows Ver."
	config.Titles.Bios = "BIOS"
	config.Titles.Baseboard = "Baseboard"
	config.Titles.Uptime = "Uptime"
	config.Titles.ProcessCount = "Processes"
	config.Titles.CPU = "CPU #"
	config.Titles.Network = "Local IP"
	config.AutoUpdate = true
	config.Version = 1
	return config
}