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

4
internal/bootstrap/data/data.go
View File

@ -1,11 +1,11 @@
package data
import "github.com/alist-org/alist/v3/cmd/args"
import "github.com/alist-org/alist/v3/cmd/flags"
func InitData() {
initUser()
initSettings()
if args.Dev {
if flags.Dev {
initDevData()
initDevDo()
}

4
internal/bootstrap/data/dev.go
View File

@ -3,7 +3,7 @@ package data
import (
"context"
"github.com/alist-org/alist/v3/cmd/args"
"github.com/alist-org/alist/v3/cmd/flags"
"github.com/alist-org/alist/v3/internal/db"
"github.com/alist-org/alist/v3/internal/message"
"github.com/alist-org/alist/v3/internal/model"
@ -35,7 +35,7 @@ func initDevData() {
}
func initDevDo() {
if args.Dev {
if flags.Dev {
go func() {
err := message.GetMessenger().WaitSend(map[string]string{
"type": "dev",

6
internal/bootstrap/data/setting.go
View File

@ -1,7 +1,7 @@
package data
import (
"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/internal/db"
"github.com/alist-org/alist/v3/internal/model"
@ -60,7 +60,7 @@ func isActive(key string) bool {
func initialSettings() {
var token string
if args.Dev {
if flags.Dev {
token = "dev_token"
} else {
token = random.Token()
@ -96,7 +96,7 @@ func initialSettings() {
// single settings
{Key: conf.Token, Value: token, Type: conf.TypeString, Group: model.SINGLE, Flag: model.PRIVATE},
}
if args.Dev {
if flags.Dev {
initialSettingItems = append(initialSettingItems, model.SettingItem{Key: "test_deprecated", Value: "test_value", Type: conf.TypeString, Flag: model.DEPRECATED})
}
}

7
internal/bootstrap/data/user.go
View File

@ -1,7 +1,7 @@
package data
import (
"github.com/alist-org/alist/v3/cmd/args"
"github.com/alist-org/alist/v3/cmd/flags"
"github.com/alist-org/alist/v3/internal/db"
"github.com/alist-org/alist/v3/internal/model"
"github.com/alist-org/alist/v3/pkg/utils/random"
@ -13,7 +13,7 @@ import (
func initUser() {
admin, err := db.GetAdmin()
adminPassword := random.String(8)
if args.Dev {
if flags.Dev {
adminPassword = "admin"
}
if err != nil {
@ -26,6 +26,8 @@ func initUser() {
}
if err := db.CreateUser(admin); err != nil {
panic(err)
} else {
log.Infof("Successfully created the administrator user and the initial password is: %s", admin.Password)
}
} else {
panic(err)
@ -48,5 +50,4 @@ func initUser() {
panic(err)
}
}
log.Infof("admin password: %+v", admin.Password)
}