discordo-plus/internal/config/theme.go

package config

import (
	"errors"

	"github.com/ayn2op/tview"
	"github.com/gdamore/tcell/v2"
)

var (
	errInvalidType = errors.New("invalid type")
)

type BorderSetWrapper struct{ tview.BorderSet }

func (bw *BorderSetWrapper) UnmarshalTOML(val any) error {
	s, ok := val.(string)
	if !ok {
		return errInvalidType
	}

	switch s {
	case "plain":
		bw.BorderSet = tview.BorderSetPlain()
	case "round":
		bw.BorderSet = tview.BorderSetRound()
	case "thick":
		bw.BorderSet = tview.BorderSetThick()
	case "double":
		bw.BorderSet = tview.BorderSetDouble()
	}

	return nil
}

type AlignmentWrapper struct{ tview.Alignment }

func (aw *AlignmentWrapper) UnmarshalTOML(v any) error {
	s, ok := v.(string)
	if !ok {
		return errInvalidType
	}

	switch s {
	case "left":
		aw.Alignment = tview.AlignmentLeft
	case "center":
		aw.Alignment = tview.AlignmentCenter
	case "right":
		aw.Alignment = tview.AlignmentRight
	}

	return nil
}

type StyleWrapper struct{ tcell.Style }

func NewStyleWrapper(style tcell.Style) StyleWrapper {
	return StyleWrapper{Style: style}
}

func (sw *StyleWrapper) UnmarshalTOML(v any) error {
	m, ok := v.(map[string]any)
	if !ok {
		return errInvalidType
	}

	for key, val := range m {
		s, ok := val.(string)
		if !ok {
			continue
		}

		switch key {
		case "foreground":
			color := tcell.GetColor(s)
			sw.Style = sw.Foreground(color)
		case "background":
			color := tcell.GetColor(s)
			sw.Style = sw.Background(color)
		}
	}

	return nil
}

type (
	ThemeStyle struct {
		NormalStyle StyleWrapper `toml:"normal_style"`
		ActiveStyle StyleWrapper `toml:"active_style"`
	}

	TitleTheme struct {
		ThemeStyle
		Alignment AlignmentWrapper `toml:"alignment"`
	}

	BorderTheme struct {
		ThemeStyle
		Enabled bool             `toml:"enabled"`
		Padding [4]int           `toml:"padding"`
		Set     BorderSetWrapper `toml:"set"`
	}

	GuildsTreeTheme struct {
		AutoExpandFolders bool `toml:"auto_expand_folders"`

		Graphics      bool   `toml:"graphics"`
		GraphicsColor string `toml:"graphics_color"`

		PrivateChannelStyle StyleWrapper `toml:"private_channel_style"`
		GuildStyle          StyleWrapper `toml:"guild_style"`
		ChannelStyle        StyleWrapper `toml:"channel_style"`
	}

	MessagesTextTheme struct {
		ShowNicknames      bool `toml:"show_user_nicks"`
		ShowUsernameColors bool `toml:"show_user_colors"`

		ReplyIndicator     string `toml:"reply_indicator"`
		ForwardedIndicator string `toml:"forwarded_indicator"`

		AuthorStyle     StyleWrapper `toml:"author_style"`
		MentionStyle    StyleWrapper `toml:"mention_style"`
		EmojiStyle      StyleWrapper `toml:"emoji_style"`
		URLStyle        StyleWrapper `toml:"url_style"`
		AttachmentStyle StyleWrapper `toml:"attachment_style"`
	}

	AutocompleteTheme struct {
		ShowNicknames      bool `toml:"show_user_nicks"`
		ShowUsernameColors bool `toml:"show_user_colors"`

		MinWidth  uint `toml:"min_width"`
		MaxHeight uint `toml:"max_height"`
	}

	Theme struct {
		BackgroundColor string `toml:"background_color"`

		Title  TitleTheme  `toml:"title"`
		Border BorderTheme `toml:"border"`

		GuildsTree   GuildsTreeTheme   `toml:"guilds_tree"`
		MessagesText MessagesTextTheme `toml:"messages_text"`
		Autocomplete AutocompleteTheme `toml:"autocomplete"`
	}
)