-
Notifications
You must be signed in to change notification settings - Fork 0
/
codefile.go
57 lines (49 loc) · 1.02 KB
/
codefile.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
package generator
import (
"io/ioutil"
"os"
"path/filepath"
)
type CodeFile struct {
Path string
Content string
}
func exists(path string) bool {
if _, err := os.Stat(path); err != nil {
if os.IsNotExist(err) {
return false
}
}
return true
}
func WriteFile(file *CodeFile, overwrite bool) (bool, error) {
if overwrite || !exists(file.Path) {
data := []byte(file.Content)
dir := filepath.Dir(file.Path)
_ = os.MkdirAll(dir, os.ModePerm)
err := ioutil.WriteFile(file.Path, data, 0644)
if err != nil {
return false, err
}
return true, nil
} else {
return false, nil
}
}
type ProcessedFileCallback func(wrote bool, fullpath string)
func WriteFiles(files []CodeFile, overwrite bool, processedFile ProcessedFileCallback) error {
for _, file := range files {
fullpath, err := filepath.Abs(file.Path)
if err != nil {
return err
}
wrote, err := WriteFile(&file, overwrite)
if err != nil {
return err
}
if processedFile != nil {
processedFile(wrote, fullpath)
}
}
return nil
}