forked from trustwallet/assets
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
61 lines (48 loc) · 1.56 KB
/
main.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 main
import (
"flag"
log "github.com/sirupsen/logrus"
"github.com/trustwallet/assets/internal/config"
"github.com/trustwallet/assets/internal/file"
"github.com/trustwallet/assets/internal/processor"
"github.com/trustwallet/assets/internal/service"
)
var (
configPath, root, script string
)
func main() {
setup()
paths, err := file.ReadLocalFileStructure(root, config.Default.ValidatorsSettings.RootFolder.SkipFiles)
if err != nil {
log.WithError(err).Fatal("Failed to load file structure.")
}
fileStorage := file.NewService(paths...)
validatorsService := processor.NewService(fileStorage)
assetfsProcessor := service.NewService(fileStorage, validatorsService)
switch script {
case "checker":
assetfsProcessor.RunJob(paths, assetfsProcessor.Check)
case "fixer":
assetfsProcessor.RunJob(paths, assetfsProcessor.Fix)
case "updater-auto":
assetfsProcessor.RunUpdateAuto()
case "updater-manual":
assetfsProcessor.RunUpdateManual()
default:
log.Info("Nothing to launch. Use --script flag to choose a script to run.")
}
}
func setup() {
flag.StringVar(&configPath, "config", "./.github/assets.config.yaml", "path to config file")
flag.StringVar(&root, "root", "./", "path to the root of the dir")
flag.StringVar(&script, "script", "", "script type to run")
flag.Parse()
if err := config.SetConfig(configPath); err != nil {
log.WithError(err).Fatal("Failed to set config.")
}
logLevel, err := log.ParseLevel(config.Default.App.LogLevel)
if err != nil {
log.WithError(err).Fatal("Failed to parse log level.")
}
log.SetLevel(logLevel)
}