forked from lukesampson/figlet
/
fonts_loader.go
80 lines (66 loc) · 1.58 KB
/
fonts_loader.go
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package figletlib
import (
"errors"
"go/build"
"os"
"path/filepath"
"strings"
)
const (
pkgName = "github.com/lukesampson/figlet"
)
func GuessFontsDirectory() string {
bin := os.Args[0]
if !filepath.IsAbs(bin) {
maybeBin, err := filepath.Abs(bin)
if err == nil {
bin = maybeBin
}
}
// try <bindir>
bindir := filepath.Dir(bin)
dirsToTry := []string{
filepath.Join(bindir, "figletlib", "fonts"),
filepath.Join(bindir, "fonts"),
}
// try src directory
ctx := build.Default
if p, err := ctx.Import(pkgName, "", build.FindOnly); err == nil {
dirsToTry = append(dirsToTry, filepath.Join(p.Dir, "figletlib", "fonts"))
dirsToTry = append(dirsToTry, filepath.Join(p.Dir, "fonts"))
}
for _, fontsDir := range dirsToTry {
fontsGlob := filepath.Join(fontsDir, "*.flf")
matches, err := filepath.Glob(fontsGlob)
if err == nil && len(matches) > 0 {
return fontsDir
}
}
return ""
}
func FontNamesInDir(dir string) ([]string, error) {
glob := filepath.Join(dir, "*.flf")
matches, err := filepath.Glob(glob)
if err != nil {
return nil, err
}
fontNames := make([]string, 0)
for _, filename := range matches {
base := filepath.Base(filename)
fontNames = append(fontNames, strings.TrimSuffix(base, ".flf"))
}
return fontNames, nil
}
func GetFontByName(dirname, name string) (*Font, error) {
if dirname == "" {
dirname := GuessFontsDirectory()
if dirname == "" {
return nil, errors.New("Could not find fonts directory!")
}
}
if !strings.HasSuffix(name, ".flf") {
name += ".flf"
}
fontpath := filepath.Join(dirname, name)
return ReadFont(fontpath)
}