theme.go

package theme

import (
	"encoding/json"
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/charmbracelet/glamour"
	"github.com/charmbracelet/glamour/ansi"
	"github.com/charmbracelet/lipgloss"
)

// Default is the default colorscheme
var Default = Colors{
	BgDark:        "#161622",
	BgDarker:      "#11111a",
	Text:          "#FFFFFF",
	TextDark:      "#676985",
	Color1:        "#c29fec",
	Color2:        "#ddbec0",
	Color3:        "#89b4fa",
	Color4:        "#e06c75",
	Color5:        "#98c379",
	Color6:        "#fab387",
	Color7:        "#f1c1e4",
	MarkdownStyle: glamour.DraculaStyleConfig,
}

// Colors is a struct that contains all the colors for the application
type Colors struct {
	MarkdownStyle ansi.StyleConfig `json:"-"` // Just generate this at runtime
	Color2        lipgloss.Color   `json:"color2"`
	BgDarker      lipgloss.Color   `json:"bg_darker"`
	Text          lipgloss.Color   `json:"text"`
	TextDark      lipgloss.Color   `json:"text_dark"`
	Color1        lipgloss.Color   `json:"color1"`
	FilePath      string           `json:"-"`
	Color3        lipgloss.Color   `json:"color3"`
	Color4        lipgloss.Color   `json:"color4"`
	Color5        lipgloss.Color   `json:"color5"`
	Color6        lipgloss.Color   `json:"color6"`
	Color7        lipgloss.Color   `json:"color7"`
	BgDark        lipgloss.Color   `json:"bg_dark"`
}

// New will create a new colorscheme and try to load it
func New(path string) (*Colors, error) {
	log.Println("Initializing colorscheme")
	if path == "" {
		defaultPath, err := getDefaultPath()
		if err != nil {
			return nil, fmt.Errorf("theme.New: %w", err)
		}

		path = defaultPath
	}

	colors := Default
	colors.FilePath = path
	colors.genMarkdownStyle()
	return &colors, nil
}

// Load will load the colorscheme from a JSON file
func (c *Colors) Load() error {
	fileContent, err := os.ReadFile(c.FilePath)
	if err != nil {
		return fmt.Errorf("theme.Load: %w", err)
	}

	if err = json.Unmarshal(fileContent, c); err != nil {
		return fmt.Errorf("theme.Load: %w", err)
	}

	c.genMarkdownStyle()
	return nil
}

// Save saves the colorscheme to a JSON file
func (c Colors) Save() error {
	jsonData, err := json.MarshalIndent(c, "", "  ")
	if err != nil {
		return fmt.Errorf("theme.Save: %w", err)
	}

	if err = os.WriteFile(c.FilePath, jsonData, 0600); err != nil {
		if err = os.MkdirAll(filepath.Dir(c.FilePath), 0755); err != nil {
			return fmt.Errorf("theme.Save: %w", err)
		}

		if err = os.WriteFile(c.FilePath, jsonData, 0600); err != nil {
			return fmt.Errorf("theme.Save: %w", err)
		}
	}

	return nil
}

// Convert takes the information from a pywal file and converts it to a colorscheme
func (c *Colors) Convert(pywalFilePath string) error {
	if pywalFilePath == "" {
		cacheDir, err := os.UserCacheDir()
		if err != nil {
			return fmt.Errorf("theme.Convert: %w", err)
		}

		pywalFilePath = filepath.Join(cacheDir, "wal", "colors.json")
	}

	fileContent, err := os.ReadFile(pywalFilePath)
	if err != nil {
		return fmt.Errorf("theme.Convert: %w", err)
	}

	var walColorscheme map[string]interface{}
	if err = json.Unmarshal(fileContent, &walColorscheme); err != nil {
		return fmt.Errorf("theme.Convert: %w", err)
	}

	// Set the colors
	c.BgDark = lipgloss.Color(walColorscheme["special"].(map[string]interface{})["background"].(string))
	c.BgDarker = lipgloss.Color(walColorscheme["special"].(map[string]interface{})["background"].(string))
	c.Text = lipgloss.Color(walColorscheme["special"].(map[string]interface{})["foreground"].(string))
	c.TextDark = lipgloss.Color(walColorscheme["special"].(map[string]interface{})["foreground"].(string))
	c.Color1 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color1"].(string))
	c.Color2 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color2"].(string))
	c.Color3 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color3"].(string))
	c.Color4 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color4"].(string))
	c.Color5 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color5"].(string))
	c.Color6 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color6"].(string))
	c.Color7 = lipgloss.Color(walColorscheme["colors"].(map[string]interface{})["color7"].(string))

	return nil
}

// PrettyPrint displays the colorscheme in the terminal
func (c Colors) PrettyPrint() string {
	result := []string{"A table of all the colors:"}

	for _, color := range []lipgloss.Color{c.BgDark, c.BgDarker, c.Text, c.TextDark, c.Color1, c.Color2, c.Color3, c.Color4, c.Color5, c.Color6, c.Color7} {
		foreground := lipgloss.NewStyle().Foreground(color)
		background := lipgloss.NewStyle().Background(color)
		result = append(result, fmt.Sprintf(
			"%s %s %s",
			foreground.Render("foreground"),
			background.Render("background"),
			color,
		))
	}

	return strings.Join(result, "\n")
}

// getDefaultPath returns the default path for the colorscheme file
func getDefaultPath() (string, error) {
	// Get the default config path
	configDir, err := os.UserConfigDir()
	if err != nil {
		return "", fmt.Errorf("theme.getDefaultPath: %w", err)
	}

	// Create the config path
	return filepath.Join(configDir, "goread", "colorscheme.json"), nil
}

// generateMarkDownStyle generates the markdown style from the colorscheme
func (c *Colors) genMarkdownStyle() {
	c.MarkdownStyle = ansi.StyleConfig{
		Document: ansi.StyleBlock{
			StylePrimitive: ansi.StylePrimitive{
				BlockPrefix: "\n",
				BlockSuffix: "\n",
				Color:       stringPtr(string(c.Text)),
			},
			Margin: uintPtr(2),
		},
		BlockQuote: ansi.StyleBlock{
			StylePrimitive: ansi.StylePrimitive{
				Color:  stringPtr(string(c.Color7)),
				Italic: boolPtr(true),
			},
			Indent:      uintPtr(1),
			IndentToken: stringPtr("│ "),
		},
		List: ansi.StyleList{
			LevelIndent: 2,
			StyleBlock: ansi.StyleBlock{
				StylePrimitive: ansi.StylePrimitive{Color: stringPtr(string(c.Text))},
			},
		},
		Heading: ansi.StyleBlock{
			StylePrimitive: ansi.StylePrimitive{
				BlockSuffix: "\n",
				Color:       stringPtr(string(c.Color3)),
				Bold:        boolPtr(true),
			},
		},
		H1:            ansi.StyleBlock{StylePrimitive: ansi.StylePrimitive{Prefix: "# "}},
		H2:            ansi.StyleBlock{StylePrimitive: ansi.StylePrimitive{Prefix: "## "}},
		H3:            ansi.StyleBlock{StylePrimitive: ansi.StylePrimitive{Prefix: "### "}},
		H4:            ansi.StyleBlock{StylePrimitive: ansi.StylePrimitive{Prefix: "#### "}},
		H5:            ansi.StyleBlock{StylePrimitive: ansi.StylePrimitive{Prefix: "##### "}},
		H6:            ansi.StyleBlock{StylePrimitive: ansi.StylePrimitive{Prefix: "###### "}},
		Strikethrough: ansi.StylePrimitive{CrossedOut: boolPtr(true)},
		Emph: ansi.StylePrimitive{
			Italic: boolPtr(true),
			Color:  stringPtr(string(c.Color1)),
		},
		Strong: ansi.StylePrimitive{
			Bold:  boolPtr(true),
			Color: stringPtr(string(c.Color1)),
		},
		HorizontalRule: ansi.StylePrimitive{
			Color:  stringPtr(string(c.TextDark)),
			Format: "\n--------\n",
		},
		Item: ansi.StylePrimitive{
			BlockPrefix: "• ",
		},
		Enumeration: ansi.StylePrimitive{
			BlockPrefix: ". ",
			Color:       stringPtr("#8be9fd"),
		},
		Task: ansi.StyleTask{
			StylePrimitive: ansi.StylePrimitive{},
			Ticked:         "[✓] ",
			Unticked:       "[ ] ",
		},
		Link: ansi.StylePrimitive{
			Color:     stringPtr(string(c.Color6)),
			Underline: boolPtr(true),
		},
		LinkText: ansi.StylePrimitive{
			Color: stringPtr(string(c.Color5)),
			Bold:  boolPtr(true),
		},
		Image: ansi.StylePrimitive{
			Color:     stringPtr(string(c.Color6)),
			Underline: boolPtr(true),
		},
		ImageText: ansi.StylePrimitive{
			Color:  stringPtr(string(c.TextDark)),
			Format: "Image: {{.text}} →",
		},
		Code: ansi.StyleBlock{
			StylePrimitive: ansi.StylePrimitive{
				Prefix:          " ",
				Suffix:          " ",
				Color:           stringPtr(string(c.Text)),
				BackgroundColor: stringPtr(string(c.BgDark)),
			},
		},
		CodeBlock: ansi.StyleCodeBlock{
			StyleBlock: ansi.StyleBlock{
				StylePrimitive: ansi.StylePrimitive{
					Color: stringPtr(string(c.Text)),
				},
				Margin: uintPtr(2),
			},
			Chroma: &ansi.Chroma{
				Text: ansi.StylePrimitive{
					Color: stringPtr(string(c.Text)),
				},
				Error: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color4)),
				},
				Comment: ansi.StylePrimitive{
					Color: stringPtr(string(c.TextDark)),
				},
				CommentPreproc: ansi.StylePrimitive{
					Color: stringPtr(string(c.TextDark)),
				},
				Keyword: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color1)),
				},
				KeywordReserved: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color1)),
				},
				KeywordNamespace: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color1)),
				},
				KeywordType: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color2)),
				},
				Operator: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color3)),
				},
				Punctuation: ansi.StylePrimitive{
					Color: stringPtr(string(c.Text)),
				},
				Name: ansi.StylePrimitive{
					Color: stringPtr(string(c.Text)),
				},
				NameBuiltin: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color6)),
				},
				NameTag: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color6)),
				},
				NameAttribute: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color6)),
				},
				NameClass: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color2)),
				},
				NameConstant: ansi.StylePrimitive{
					Color: stringPtr(string(c.Text)),
				},
				NameDecorator: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color3)),
				},
				NameFunction: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color3)),
				},
				LiteralNumber: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color5)),
				},
				LiteralString: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color5)),
				},
				LiteralStringEscape: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color5)),
				},
				GenericDeleted: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color4)),
				},
				GenericEmph: ansi.StylePrimitive{
					Color:  stringPtr(string(c.Color1)),
					Italic: boolPtr(true),
				},
				GenericInserted: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color1)),
				},
				GenericStrong: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color1)),
					Bold:  boolPtr(true),
				},
				GenericSubheading: ansi.StylePrimitive{
					Color: stringPtr(string(c.Color2)),
				},
				Background: ansi.StylePrimitive{
					BackgroundColor: stringPtr(string(c.BgDark)),
				},
			},
		},
		Table: ansi.StyleTable{
			StyleBlock: ansi.StyleBlock{
				StylePrimitive: ansi.StylePrimitive{},
			},
			CenterSeparator: stringPtr("┼"),
			ColumnSeparator: stringPtr("│"),
			RowSeparator:    stringPtr("─"),
		},
		DefinitionDescription: ansi.StylePrimitive{
			BlockPrefix: "\n🠶 ",
		},
	}
}

func boolPtr(b bool) *bool       { return &b }
func stringPtr(s string) *string { return &s }
func uintPtr(u uint) *uint       { return &u }