oh-my-posh/src/segment_shell.go

package main

import "strings"

type shell struct {
	props Properties
	env   Environment

	Name string
}

const (
	// MappedShellNames allows for custom text in place of shell names
	MappedShellNames Property = "mapped_shell_names"
)

func (s *shell) enabled() bool {
	mappedNames := s.props.getKeyValueMap(MappedShellNames, make(map[string]string))
	s.Name = s.env.getShellName()
	for key, val := range mappedNames {
		if strings.EqualFold(s.Name, key) {
			s.Name = val
			break
		}
	}
	return true
}

func (s *shell) string() string {
	segmentTemplate := s.props.getString(SegmentTemplate, "{{.Name}}")
	template := &textTemplate{
		Template: segmentTemplate,
		Context:  s,
		Env:      s.env,
	}
	text, err := template.render()
	if err != nil {
		return err.Error()
	}
	return text
}

func (s *shell) init(props Properties, env Environment) {
	s.props = props
	s.env = env
}