feat(path): allow pattern matching in maped_locations

resolves #6280

Author: JanDeDobbeleer

Diff for: src/regex/regex.go

@@ -110,21 +110,21 @@ func MatchString(pattern, text string) bool {
 	return re.MatchString(text)
 }
 
-func FindStringMatch(pattern, text string, index int) string {
+func FindStringMatch(pattern, text string, index int) (string, bool) {
 	re, err := GetCompiledRegex(pattern)
 	if err != nil {
-		return text
+		return text, false
 	}
 
 	matches := re.FindStringSubmatch(text)
 	if len(matches) <= index {
-		return text
+		return text, false
 	}
 
 	match := matches[index]
 	if len(match) == 0 {
-		return text
+		return text, false
 	}
 
-	return match
+	return match, true
 }

Diff for: src/regex/regex_test.go

@@ -66,7 +66,7 @@ func TestFindStringMatch(t *testing.T) {
 	}
 
 	for _, tc := range cases {
-		got := FindStringMatch(tc.Pattern, tc.Text, tc.Index)
+		got, _ := FindStringMatch(tc.Pattern, tc.Text, tc.Index)
 		assert.Equal(t, tc.Expected, got, tc.Case)
 	}
 }

Diff for: src/segments/golang.go

@@ -80,7 +80,7 @@ func (g *Golang) parseWorkFile() (string, error) {
 	}
 
 	contents := g.env.FileContent(goWork.Path)
-	version := regex.FindStringMatch(`go (\d(\.\d{1,2})?(\.\d{1,2})?)`, contents, 1)
+	version, _ := regex.FindStringMatch(`go (\d(\.\d{1,2})?(\.\d{1,2})?)`, contents, 1)
 	if len(version) > 0 {
 		return version, nil
 	}

Diff for: src/segments/path.go

@@ -16,6 +16,10 @@ import (
 	"github.com/jandedobbeleer/oh-my-posh/src/template"
 )
 
+const (
+	regexPrefix = "re:"
+)
+
 type Folder struct {
 	Name    string
 	Path    string
@@ -570,12 +574,16 @@ func (pt *Path) setMappedLocations() {
 			continue
 		}
 
+		if !strings.HasPrefix(location, regexPrefix) {
+			location = pt.normalize(location)
+		}
+
 		// When two templates resolve to the same key, the values are compared in ascending order and the latter is taken.
-		if v, exist := mappedLocations[pt.normalize(location)]; exist && value <= v {
+		if v, exist := mappedLocations[location]; exist && value <= v {
 			continue
 		}
 
-		mappedLocations[pt.normalize(location)] = value
+		mappedLocations[location] = value
 	}
 
 	pt.mappedLocations = mappedLocations
@@ -610,7 +618,19 @@ func (pt *Path) replaceMappedLocations(inputPath string) (string, string) {
 	}
 
 	for _, key := range keys {
+		if strings.HasPrefix(key, regexPrefix) {
+			match, OK := regex.FindStringMatch(key[len(regexPrefix):], inputPath, 1)
+			if !OK {
+				continue
+			}
+
+			// Replace the first match with the mapped location.
+			inputPath = strings.Replace(inputPath, match, pt.mappedLocations[key], 1)
+			return pt.parsePath(inputPath)
+		}
+
 		keyRoot, keyRelative := pt.parsePath(key)
+
 		matchSubFolders := strings.HasSuffix(keyRelative, pt.pathSeparator+"*")
 
 		if matchSubFolders {