oh-my-posh/segment_path.go

package main

import (
	"bytes"
	"fmt"
	"path/filepath"
	"strings"
)

type path struct {
	props *properties
	env   environmentInfo
}

const (
	//FolderSeparatorIcon the path which is split will be separated by this icon
	FolderSeparatorIcon Property = "folder_separator_icon"
	//HomeIcon indicates the $HOME location
	HomeIcon Property = "home_icon"
	//FolderIcon identifies one folder
	FolderIcon Property = "folder_icon"
	//WindowsRegistryIcon indicates the registry location on Windows
	WindowsRegistryIcon Property = "windows_registry_icon"
	//Agnoster displays a short path with separator icon, this the default style
	Agnoster string = "agnoster"
	//Short displays a shorter path
	Short string = "short"
	//Full displays the full path
	Full string = "full"
	//Folder displays the current folder
	Folder string = "folder"
)

func (pt *path) enabled() bool {
	return true
}

func (pt *path) string() string {
	switch style := pt.props.getString(Style, Agnoster); style {
	case Agnoster:
		return pt.getAgnosterPath()
	case Short:
		return pt.getShortPath()
	case Full:
		return pt.env.getcwd()
	case Folder:
		return base(pt.env.getcwd(), pt.env)
	default:
		return fmt.Sprintf("Path style: %s is not available", style)
	}
}

func (pt *path) init(props *properties, env environmentInfo) {
	pt.props = props
	pt.env = env
}

func (pt *path) getShortPath() string {
	pwd := pt.env.getcwd()
	mappedLocations := map[string]string{
		"HKCU:": pt.props.getString(WindowsRegistryIcon, "REGISTRY"),
		"HKLM:": pt.props.getString(WindowsRegistryIcon, "REGISTRY"),
		"Microsoft.PowerShell.Core\\FileSystem::": "",
		pt.env.homeDir(): pt.props.getString(HomeIcon, "~"),
	}
	for location, value := range mappedLocations {
		if strings.HasPrefix(pwd, location) {
			return strings.Replace(pwd, location, value, 1)
		}
	}
	return pwd
}

func (pt *path) getAgnosterPath() string {
	pwd := pt.env.getcwd()
	buffer := new(bytes.Buffer)
	buffer.WriteString(pt.rootLocation(pwd))
	pathDepth := pt.pathDepth(pwd)
	for i := 1; i < pathDepth; i++ {
		buffer.WriteString(fmt.Sprintf("%s%s", pt.props.getString(FolderSeparatorIcon, pt.env.getPathSeperator()), pt.props.getString(FolderIcon, "..")))
	}
	if pathDepth > 0 {
		buffer.WriteString(fmt.Sprintf("%s%s", pt.props.getString(FolderSeparatorIcon, pt.env.getPathSeperator()), base(pwd, pt.env)))
	}
	return buffer.String()
}

func (pt *path) inHomeDir(pwd string) bool {
	return strings.HasPrefix(pwd, pt.env.homeDir())
}

func (pt *path) rootLocation(pwd string) string {
	//See https://community.idera.com/database-tools/powershell/powertips/b/tips/posts/correcting-powershell-paths
	if strings.HasPrefix(pwd, "Microsoft.PowerShell.Core\\FileSystem::") {
		pwd = strings.Replace(pwd, "Microsoft.PowerShell.Core\\FileSystem::", "", 1)
	}
	if pt.inHomeDir(pwd) {
		return pt.props.getString(HomeIcon, "~")
	}
	pwd = strings.TrimPrefix(pwd, pt.env.getPathSeperator())
	splitted := strings.Split(pwd, pt.env.getPathSeperator())
	rootLocation := splitted[0]
	mappedLocations := map[string]string{
		"HKCU:": pt.props.getString(WindowsRegistryIcon, "HK:"),
	}
	if val, ok := mappedLocations[rootLocation]; ok {
		return val
	}
	return rootLocation
}

func (pt *path) pathDepth(pwd string) int {
	if pt.inHomeDir(pwd) {
		pwd = strings.Replace(pwd, pt.env.homeDir(), "root", 1)
	}
	splitted := strings.Split(pwd, pt.env.getPathSeperator())
	var validParts []string
	for _, part := range splitted {
		if part != "" {
			validParts = append(validParts, part)
		}
	}
	depth := len(validParts)
	return depth - 1
}

// Base returns the last element of path.
// Trailing path separators are removed before extracting the last element.
// If the path is empty, Base returns ".".
// If the path consists entirely of separators, Base returns a single separator.
func base(path string, env environmentInfo) string {
	if path == "" {
		return "."
	}
	// Strip trailing slashes.
	for len(path) > 0 && string(path[len(path)-1]) == env.getPathSeperator() {
		path = path[0 : len(path)-1]
	}
	// Throw away volume name
	path = path[len(filepath.VolumeName(path)):]
	// Find the last element
	i := len(path) - 1
	for i >= 0 && string(path[i]) != env.getPathSeperator() {
		i--
	}
	if i >= 0 {
		path = path[i+1:]
	}
	// If empty now, it had only slashes.
	if path == "" {
		return string(env.getPathSeperator())
	}
	return path
}