-
Notifications
You must be signed in to change notification settings - Fork 173
/
embed.go
56 lines (49 loc) · 1.08 KB
/
embed.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
package embed
import (
"embed"
"github.com/chaitin/veinmind-tools/veinmind-malicious/sdk/common"
"io/fs"
"io/ioutil"
"os"
"path"
)
//go:embed scripts/.env
//go:embed template/template.html
//go:embed scripts/docker-compose.yml
var EmbedFile embed.FS
func Open(name string) (fs.File, error) {
if _, err := os.Stat(name); os.IsNotExist(err) {
return EmbedFile.Open(name)
} else {
return os.Open(name)
}
}
func ReadFile(name string) ([]byte, error) {
if _, err := os.Stat(name); os.IsNotExist(err) {
return EmbedFile.ReadFile(name)
} else {
return ioutil.ReadFile(name)
}
}
func ExtractAll() {
extract("scripts/docker-compose.yml")
extract("scripts/.env")
}
// extract
func extract(epath string) {
// extract docker-compose config
composeYamlBytes, err := EmbedFile.ReadFile(epath)
if err != nil {
common.Log.Fatal(err)
}
if _, err := os.Stat(path.Dir(epath)); os.IsNotExist(err) {
err = os.Mkdir(path.Dir(epath), 0755)
if err != nil {
common.Log.Fatal(err)
}
}
err = ioutil.WriteFile(epath, composeYamlBytes, 0755)
if err != nil {
common.Log.Fatal(err)
}
}