/
fs.go
61 lines (49 loc) · 1.2 KB
/
fs.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
package database
import (
"os"
"path/filepath"
)
func getDatabaseDirPath(dataDir string) string {
return filepath.Join(dataDir, "database")
}
func getGenesisJsonFilePath(dataDir string) string {
return filepath.Join(getDatabaseDirPath(dataDir), "genesis.json")
}
func getBlocksDbFilePath(dataDir string) string {
return filepath.Join(getDatabaseDirPath(dataDir), "block.db")
}
func initDataDirIfNotExists(dataDir string) error {
if fileExist(getGenesisJsonFilePath(dataDir)) {
return nil
}
if err := os.MkdirAll(getDatabaseDirPath(dataDir), os.ModePerm); err != nil {
return err
}
if err := writeGenesisToDisk(getGenesisJsonFilePath(dataDir)); err != nil {
return err
}
if err := writeEmptyBlocksDbToDisk(getBlocksDbFilePath(dataDir)); err != nil {
return err
}
return nil
}
func fileExist(filePath string) bool {
_, err := os.Stat(filePath)
if err != nil && os.IsNotExist(err) {
return false
}
return true
}
func dirExists(path string) (bool, error) {
_, err := os.Stat(path)
if err == nil {
return true, nil
}
if os.IsNotExist(err) {
return false, nil
}
return true, err
}
func writeEmptyBlocksDbToDisk(path string) error {
return os.WriteFile(path, []byte(""), os.ModePerm)
}