-
Notifications
You must be signed in to change notification settings - Fork 5
/
public.go
113 lines (93 loc) · 2.42 KB
/
public.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package gen
import (
"io/fs"
"os"
"path/filepath"
"slices"
"github.com/livefir/fir/internal/logger"
gitignore "github.com/sabhiram/go-gitignore"
)
type publicOpt struct {
inDir string
outDir string
extensions []string
}
// PublicDirOption is a function that can be used to configure generation of public directory using GeneratePublic.
type PublicDirOption func(*publicOpt)
// InDir sets the input directory for the public directory.
func InDir(path string) PublicDirOption {
return func(o *publicOpt) {
o.inDir = path
}
}
// OutputDir sets the output directory for the public directory.
func OutDir(path string) PublicDirOption {
return func(o *publicOpt) {
o.outDir = path
}
}
// Extension adds an extension to the list of extensions that will be copied over.
func PublicFileExtensions(extensions []string) PublicDirOption {
return func(o *publicOpt) {
for _, ext := range extensions {
if !slices.Contains(o.extensions, ext) {
o.extensions = append(o.extensions, ext)
}
}
}
}
// GeneratePublicDir generates the public directory which can be then embedded into the binary.
func GeneratePublicDir(options ...PublicDirOption) error {
opt := &publicOpt{
inDir: ".",
outDir: "./public",
extensions: []string{".html"},
}
for _, option := range options {
option(opt)
}
if err := os.MkdirAll(opt.outDir, os.ModePerm); err != nil {
return err
}
ignore, err := gitignore.CompileIgnoreFile(filepath.Join(opt.inDir, ".gitignore"))
if err != nil {
logger.Warnf("failed to compile .gitignore: %v\n", err)
}
err = filepath.WalkDir(opt.inDir, func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if d.IsDir() {
if d.Name() == filepath.Base(opt.outDir) {
return filepath.SkipDir
}
if d.Name() == ".git" {
return filepath.SkipDir
}
if ignore != nil && ignore.MatchesPath(d.Name()) {
return filepath.SkipDir
}
return nil
}
if ignore != nil && ignore.MatchesPath(path) {
return nil
}
if !slices.Contains(opt.extensions, filepath.Ext(path)) {
return nil
}
relpath, err := filepath.Rel(opt.inDir, path)
if err != nil {
return err
}
outPath := filepath.Join(opt.outDir, relpath)
if err := os.MkdirAll(filepath.Dir(outPath), os.ModePerm); err != nil {
return err
}
data, err := os.ReadFile(path)
if err != nil {
return err
}
return os.WriteFile(outPath, data, os.ModePerm)
})
return err
}