-
Notifications
You must be signed in to change notification settings - Fork 18
/
assets.go
110 lines (95 loc) · 3.03 KB
/
assets.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
// Package static has runtime resources for various commands.
package static
import (
"bytes"
"compress/gzip"
"embed"
"io"
"os"
"path"
"path/filepath"
"strings"
)
//go:embed configuration.toml default_cover.jpeg dictionaries profiles resources sentences
var content embed.FS
// -------------------------------------------------------------------------------------------------------------------------
// Following functions provide minimal emulation of go-bindata generated code to avoid changing the rest of the program.
// Individual files could be gzip compressed - this slows down access but otherwise transparent to the outside code.
// NOTE: this is not a generic implementation - it relies on details of the usage!
// -------------------------------------------------------------------------------------------------------------------------
// Asset loads and returns the asset for the given name.
// It returns an error if the asset could not be found or could not be loaded.
func Asset(name string) ([]byte, error) {
data, err := content.ReadFile(name)
if err != nil {
if !os.IsNotExist(err) {
return nil, err
}
data, err = content.ReadFile(name + ".gz")
if err != nil {
return nil, err
}
gzr, err := gzip.NewReader(bytes.NewBuffer(data))
if err != nil {
return nil, err
}
buf := new(bytes.Buffer)
if _, err := io.Copy(buf, gzr); err != nil {
return nil, err
}
return buf.Bytes(), nil
}
return data, nil
}
// AssetDir returns the file names below a certain directory embedded in the file by go-bindata.
// For example if you run go-bindata on data/... and data contains the
// following hierarchy:
// data/
// foo.txt
// img/
// a.png
// b.png
// then AssetDir("data") would return []string{"foo.txt", "img"}
// AssetDir("data/img") would return []string{"a.png", "b.png"}
// AssetDir("foo.txt") and AssetDir("notexist") would return an error
// AssetDir("") will return []string{"data"}.
func AssetDir(name string) ([]string, error) {
name = path.Clean(filepath.ToSlash(name))
dirEntries, err := content.ReadDir(name)
if err != nil {
return nil, err
}
var entries []string
for _, de := range dirEntries {
entries = append(entries, strings.TrimSuffix(de.Name(), ".gz"))
}
return entries, nil
}
func restoreFile(dir, name string) error {
data, err := Asset(name)
if err != nil {
return err
}
if err := os.MkdirAll(filepath.Join(dir, filepath.Dir(name)), os.FileMode(0755)); err != nil {
return err
}
if err := os.WriteFile(filepath.Join(dir, name), data, os.FileMode(0644)); err != nil {
return err
}
return nil
}
// RestoreAssets restores an asset under the given directory recursively.
func RestoreAssets(dir, name string) error {
dir, name = path.Clean(filepath.ToSlash(dir)), path.Clean(filepath.ToSlash(name))
dirEntries, err := content.ReadDir(name)
if err != nil {
return restoreFile(dir, name)
}
for _, de := range dirEntries {
err := RestoreAssets(dir, path.Join(name, strings.TrimSuffix(de.Name(), ".gz")))
if err != nil {
return err
}
}
return nil
}