-
Notifications
You must be signed in to change notification settings - Fork 0
/
bases_repo.go
70 lines (57 loc) · 1.2 KB
/
bases_repo.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
package basesrepo
import (
"embed"
"io/fs"
"path"
"path/filepath"
)
const (
DEFAULT_JSON_FILENAME string = "recommended.json"
baseDir string = "bases/bases"
)
type BasesRepo struct {
fs embed.FS
}
func NewBasesRepo(fs embed.FS) *BasesRepo {
return &BasesRepo{
fs: fs,
}
}
func (b *BasesRepo) TemplateFiles() ([]TemplateFile, error) {
files := []TemplateFile{}
d, err := fs.ReadDir(b.fs, baseDir)
if err != nil {
return files, err
}
for _, v := range d {
if v.IsDir() {
continue
}
if filepath.Ext(v.Name()) != ".json" {
continue
}
files = append(files, TemplateFile{
Name: v.Name(),
Path: path.Join(baseDir, v.Name()),
})
}
return files, nil
}
// ReadFile reads a file from the file system.
func (b *BasesRepo) ReadFile(path string) ([]byte, error) {
return b.fs.ReadFile(path)
}
type TemplateFile struct {
Name string
Path string
}
// GetFileMap returns a map of file names to paths and a slice of file names.
func GetFileMap(files []TemplateFile) (map[string]string, []string) {
fileMap := make(map[string]string)
var names []string
for _, f := range files {
fileMap[f.Name] = f.Path
names = append(names, f.Name)
}
return fileMap, names
}