discordo-plus/config/mod.go

package config

import (
	"os"

	"github.com/BurntSushi/toml"
)

type GeneralConfig struct {
	UserAgent          string `json:"userAgent"`
	FetchMessagesLimit int    `json:"fetchMessagesLimit"`
	Mouse              bool   `json:"mouse"`
	Notifications      bool   `json:"notifications"`
	Timestamps         bool   `json:"timestamps"`
}

type KeybindingsConfig struct {
	ToggleGuildsList         string `toml:"toggle_guilds_list"`
	ToggleChannelsTreeView   string `toml:"toggle_channels_tree_view"`
	ToggleMessagesTextView   string `toml:"toggle_messages_text_view"`
	ToggleMessageInputField  string `toml:"toggle_message_input_field"`
	ToggleMessageActionsList string `toml:"toggle_message_actions_list"`
	ToggleExternalEditor     string `toml:"toggle_external_editor"`

	SelectPreviousMessage string `toml:"select_previous_message"`
	SelectNextMessage     string `toml:"select_next_message"`
	SelectFirstMessage    string `toml:"select_first_message"`
	SelectLastMessage     string `toml:"select_last_message"`
}

type Config struct {
	Keybindings KeybindingsConfig `json:"keybindings"`
	General     GeneralConfig     `json:"general"`
}

func Load() Config {
	configPath, err := os.UserConfigDir()
	if err != nil {
		panic(err)
	}

	configPath += "/discordo.toml"
	c := Config{}
	// If the configuration file does not exist, create and write the default configuration to the file.
	if _, err = os.Stat(configPath); os.IsNotExist(err) {
		f, err := os.Create(configPath)
		if err != nil {
			panic(err)
		}
		defer f.Close()

		c = newDefaultConfig()
		err = toml.NewEncoder(f).Encode(c)
		if err != nil {
			panic(err)
		}
	} else {
		_, err = toml.DecodeFile(configPath, &c)
		if err != nil {
			panic(err)
		}
	}

	return c
}

func newDefaultConfig() Config {
	return Config{
		General: GeneralConfig{
			UserAgent:          "Mozilla/5.0 (X11; Linux x86_64; rv:95.0) Gecko/20100101 Firefox/95.0",
			FetchMessagesLimit: 50,
			Mouse:              true,
			Notifications:      true,
			Timestamps:         false,
		},
		Keybindings: KeybindingsConfig{
			ToggleGuildsList:         "g",
			ToggleChannelsTreeView:   "c",
			ToggleMessagesTextView:   "m",
			ToggleMessageInputField:  "i",
			ToggleMessageActionsList: "a",
			ToggleExternalEditor:     "Ctrl-E",

			SelectPreviousMessage: "Up",
			SelectNextMessage:     "Down",
			SelectFirstMessage:    "Home",
			SelectLastMessage:     "End",
		},
	}
}