oh-my-posh/settings.go

package main

import (
	"encoding/json"
	"fmt"
	"os"
)

//Settings holds all the theme for rendering the prompt
type Settings struct {
	ConsoleBackgroundColor string   `json:"console_background_color"`
	RightSegmentOffset     int      `json:"right_segment_offset"`
	EndSpaceEnabled        bool     `json:"end_space_enabled"`
	Blocks                 []*Block `json:"blocks"`
}

//BlockType type of block
type BlockType string

//BlockAlignment aligment of a Block
type BlockAlignment string

const (
	//Prompt writes one or more Segments
	Prompt BlockType = "prompt"
	//LineBreak creates a line break in the prompt
	LineBreak BlockType = "newline"
	//Left aligns left
	Left BlockAlignment = "left"
	//Right aligns right
	Right BlockAlignment = "right"
)

//Block defines a part of the prompt with optional segments
type Block struct {
	Type                          BlockType      `json:"type"`
	Alignment                     BlockAlignment `json:"alignment"`
	PowerlineSeparator            string         `json:"powerline_separator"`
	InvertPowerlineSeparatorColor bool           `json:"invert_powerline_separator_color"`
	LineOffset                    int            `json:"line_offset"`
	Segments                      []*Segment     `json:"segments"`
}

//GetSettings returns the default configuration including possible user overrides
func GetSettings(env environmentInfo) *Settings {
	settings, err := loadUserConfiguration(env)
	if err != nil {
		return getDefaultSettings()
	}
	return settings
}

func loadUserConfiguration(env environmentInfo) (*Settings, error) {
	var settings Settings
	settingsFileLocation := fmt.Sprintf("%s/.go_my_posh", env.getenv("HOME"))
	if _, err := os.Stat(*env.getArgs().Config); !os.IsNotExist(err) {
		settingsFileLocation = *env.getArgs().Config
	}
	defaultSettings, err := os.Open(settingsFileLocation)
	defer func() {
		_ = defaultSettings.Close()
	}()
	if err != nil {
		return nil, err
	}
	jsonParser := json.NewDecoder(defaultSettings)
	err = jsonParser.Decode(&settings)
	return &settings, err
}

func getDefaultSettings() *Settings {
	settings := &Settings{
		EndSpaceEnabled:        true,
		ConsoleBackgroundColor: "#193549",
		Blocks: []*Block{
			{
				Type:               Prompt,
				Alignment:          Left,
				PowerlineSeparator: "",
				Segments: []*Segment{
					{
						Type:       Root,
						Style:      Powerline,
						Background: "#ffe9aa",
						Foreground: "#100e23",
					},
					{
						Type:       Session,
						Style:      Powerline,
						Background: "#ffffff",
						Foreground: "#100e23",
					},
					{
						Type:       Path,
						Style:      Powerline,
						Background: "#91ddff",
						Foreground: "#100e23",
					},
					{
						Type:       Git,
						Style:      Powerline,
						Background: "#95ffa4",
						Foreground: "#100e23",
					},
					{
						Type:       Venv,
						Style:      Powerline,
						Background: "#906cff",
						Foreground: "#100e23",
					},
					{
						Type:       Exit,
						Style:      Powerline,
						Background: "#ff8080",
						Foreground: "#ffffff",
					},
				},
			},
		},
	}
	return settings
}