aboutsummaryrefslogtreecommitdiff
path: root/model/lookup.go
blob: e8cc8bfe457046f39596729bdefffb29f749eae1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package model

import (
	"os"
	"strings"
)

// WordLookup тип определяющий функцию поиска замены слов при
// стрингификации Values.
type WordLookup func(word Word) string

// chainLookup утилитарная функция для последовательных применений функций
// поиска до первого нахождения подстановки. Если returnOrigin == true,
// то в случае неудачи вернёт имя слова.
func chainLookup(lookups ...WordLookup) WordLookup {
	return func(word Word) string {
		for _, lookup := range lookups {
			if v := lookup(word); v != "" {
				return v
			}
		}
		return string(word)
	}
}

// LookupEnv функция типа WordLookup которая пытается подставить вместо word
// соответствующую ему переменную окружения ОС. При этом он срабатывает только
// если слово начинается со знака `$`.
func LookupEnv(word Word) string {
	if !strings.HasPrefix(string(word), "$") {
		return ""
	}
	varName := strings.TrimPrefix(string(word), "$")
	if result, ok := os.LookupEnv(varName); ok {
		return result
	}
	return ""
}

// LookupSubst функция типа WordLookup которая пытается подставить вместо word
// значение из словаря подстановок по соответствующему ключу.
func LookupSubst(subst map[Word]string) WordLookup {
	return func(word Word) string {
		if result, ok := subst[word]; ok {
			return result
		}
		return ""
	}
}