GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-05-10 17:52:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: (WIP) migrate storage configuration to user-specific models and remove deprecated storage loading

Browse Source
This commit is contained in:
krau
2025-02-18 19:45:06 +08:00
parent 968547b005
commit dfde65c28e
8 changed files with 193 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
config/viper.go
View File

@@ -3,9 +3,11 @@ package config
import (
"fmt"
"os"
"strconv"
"strings"
"github.com/spf13/viper"
"gorm.io/datatypes"
)
type Config struct {
@@ -53,6 +55,7 @@ type proxyConfig struct {
/*
在配置文件中定义的存储将会为telegram.admins中的每个用户创建一个存储模型
*/
// these config will be removed in the future.
type storageConfig struct {
Alist AlistConfig `toml:"alist" mapstructure:"alist"`
Local LocalConfig `toml:"local" mapstructure:"local"`
@@ -65,8 +68,13 @@ type AlistConfig struct {
Username string `toml:"username" mapstructure:"username"`
Password string `toml:"password" mapstructure:"password"`
Token string `toml:"token" mapstructure:"token"`
BasePath string `toml:"base_path" mapstructure:"base_path"`
TokenExp int64 `toml:"token_exp" mapstructure:"token_exp"`
BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
TokenExp int64 `toml:"token_exp" mapstructure:"token_exp" json:"token_exp"`
}
func (a *AlistConfig) ToJSON() datatypes.JSON {
tokenExp := strconv.FormatInt(a.TokenExp, 10)
return datatypes.JSON([]byte(`{"url":"` + a.URL + `","username":"` + a.Username + `","password":"` + a.Password + `","token":"` + a.Token + `","base_path":"` + a.BasePath + `","token_exp":` + tokenExp + `}`))
}
type LocalConfig struct {
@@ -74,6 +82,10 @@ type LocalConfig struct {
BasePath string `toml:"base_path" mapstructure:"base_path"`
}
func (l *LocalConfig) ToJSON() datatypes.JSON {
return datatypes.JSON([]byte(`{"base_path":"` + l.BasePath + `"}`))
}
type WebdavConfig struct {
Enable bool `toml:"enable" mapstructure:"enable"`
URL string `toml:"url" mapstructure:"url"`
@@ -82,6 +94,10 @@ type WebdavConfig struct {
BasePath string `toml:"base_path" mapstructure:"base_path"`
}
func (w *WebdavConfig) ToJSON() datatypes.JSON {
return datatypes.JSON([]byte(`{"url":"` + w.URL + `","username":"` + w.Username + `","password":"` + w.Password + `","base_path":"` + w.BasePath + `"}`))
}
var Cfg *Config
func Init() {
@@ -109,9 +125,6 @@ func Init() {
viper.SetDefault("db.path", "data/saveany.db")
viper.SetDefault("storage.alist.base_path", "/")
viper.SetDefault("storage.alist.token_exp", 3600)
viper.SafeWriteConfigAs("config.toml")
if err := viper.ReadInConfig(); err != nil {
@@ -124,6 +137,9 @@ func Init() {
fmt.Println("Error unmarshalling config file, ", err)
os.Exit(1)
}
if Cfg.Storage != (storageConfig{}) {
fmt.Println("警告: 存储配置已经废弃, 未来版本将会移除.\n请直接使用 Bot 命令添加存储.")
}
if Cfg.Workers < 1 || Cfg.Retry < 1 {
fmt.Println("Invalid workers or retry value")
os.Exit(1)