oh-my-posh/src/segment_shell.go

package main

import (
	"oh-my-posh/environment"
	"oh-my-posh/properties"
	"strings"
)

type Shell struct {
	props properties.Properties
	env   environment.Environment

	Name string
}

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

func (s *Shell) template() string {
	return "{{ .Name }}"
}

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

func (s *Shell) init(props properties.Properties, env environment.Environment) {
	s.props = props
	s.env = env
}