ProjectHentai/alist
6
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: use cobra and add some command

Browse Source
This commit is contained in:
Noah Hsu
2022-08-07 13:09:59 +08:00
parent 2b5da3ef34
commit 0df3473337
20 changed files with 279 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
internal/bootstrap/config.go
View File

@ -5,7 +5,7 @@ import (
"os"
"path/filepath"
"github.com/alist-org/alist/v3/cmd/args"
"github.com/alist-org/alist/v3/cmd/flags"
"github.com/alist-org/alist/v3/internal/conf"
"github.com/alist-org/alist/v3/pkg/utils"
"github.com/caarlos0/env/v6"
@ -13,36 +13,35 @@ import (
)
func InitConfig() {
log.Infof("reading config file: %s", args.Config)
if !utils.Exists(args.Config) {
log.Infof("reading config file: %s", flags.Config)
if !utils.Exists(flags.Config) {
log.Infof("config file not exists, creating default config file")
_, err := utils.CreateNestedFile(args.Config)
_, err := utils.CreateNestedFile(flags.Config)
if err != nil {
log.Fatalf("failed to create config file: %+v", err)
}
conf.Conf = conf.DefaultConfig()
if !utils.WriteToJson(args.Config, conf.Conf) {
if !utils.WriteToJson(flags.Config, conf.Conf) {
log.Fatalf("failed to create default config file")
}
} else {
configBytes, err := ioutil.ReadFile(args.Config)
configBytes, err := ioutil.ReadFile(flags.Config)
if err != nil {
log.Fatalf("reading config file error:%s", err.Error())
log.Fatalf("reading config file error: %+v", err)
}
conf.Conf = conf.DefaultConfig()
err = utils.Json.Unmarshal(configBytes, conf.Conf)
if err != nil {
log.Fatalf("load config error: %s", err.Error())
log.Fatalf("load config error: %+v", err)
}
log.Debugf("config:%+v", conf.Conf)
// update config.json struct
confBody, err := utils.Json.MarshalIndent(conf.Conf, "", " ")
if err != nil {
log.Fatalf("marshal config error:%s", err.Error())
log.Fatalf("marshal config error: %+v", err)
}
err = ioutil.WriteFile(args.Config, confBody, 0777)
err = ioutil.WriteFile(flags.Config, confBody, 0777)
if err != nil {
log.Fatalf("update config struct error: %s", err.Error())
log.Fatalf("update config struct error: %+v", err)
}
}
if !conf.Conf.Force {
@ -54,7 +53,7 @@ func InitConfig() {
if !filepath.IsAbs(conf.Conf.TempDir) {
absPath, err = filepath.Abs(conf.Conf.TempDir)
if err != nil {
log.Fatalf("get abs path error: %s", err.Error())
log.Fatalf("get abs path error: %+v", err)
}
}
conf.Conf.TempDir = absPath
@ -64,20 +63,20 @@ func InitConfig() {
}
err = os.MkdirAll(conf.Conf.TempDir, 0700)
if err != nil {
log.Fatalf("create temp dir error: %s", err.Error())
log.Fatalf("create temp dir error: %+v", err)
}
log.Debugf("config: %+v", conf.Conf)
}
func confFromEnv() {
prefix := "ALIST_"
if args.NoPrefix {
if flags.NoPrefix {
prefix = ""
}
log.Infof("load config from env with prefix: %s", prefix)
if err := env.Parse(conf.Conf, env.Options{
Prefix: prefix,
}); err != nil {
log.Fatalf("load config from env error: %s", err.Error())
log.Fatalf("load config from env error: %+v", err)
}
}