254 lines
10 KiB
Go
254 lines
10 KiB
Go
package dao
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"gorm.io/gorm/schema"
|
|
"hy2xs-admin/model/constant"
|
|
"hy2xs-admin/model/entity"
|
|
"hy2xs-admin/util"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var sqlInitStr = "CREATE TABLE IF NOT EXISTS account\n(\n id INTEGER PRIMARY KEY AUTOINCREMENT,\n username TEXT NOT NULL UNIQUE DEFAULT '',\n pass TEXT NOT NULL DEFAULT '',\n con_pass TEXT NOT NULL DEFAULT '',\n quota INTEGER NOT NULL DEFAULT 0,\n download INTEGER NOT NULL DEFAULT 0,\n upload INTEGER NOT NULL DEFAULT 0,\n expire_time INTEGER NOT NULL DEFAULT 0,\n kick_util_time INTEGER NOT NULL DEFAULT 0,\n device_no INTEGER NOT NULL DEFAULT 3,\n role TEXT NOT NULL DEFAULT 'user',\n deleted INTEGER NOT NULL DEFAULT 0,\n create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,\n update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP\n);\nALTER TABLE account\n ADD COLUMN login_at INTEGER NOT NULL DEFAULT 0;\nALTER TABLE account\n ADD COLUMN con_at INTEGER NOT NULL DEFAULT 0;\nALTER TABLE account\n ADD COLUMN remark INTEGER NOT NULL DEFAULT '';\nALTER TABLE account\n ADD COLUMN force_password_change INTEGER NOT NULL DEFAULT 0;\nCREATE INDEX IF NOT EXISTS account_deleted_index ON account (deleted);\nCREATE INDEX IF NOT EXISTS account_username_index ON account (username);\nCREATE INDEX IF NOT EXISTS account_con_pass_index ON account (con_pass);\nCREATE INDEX IF NOT EXISTS account_pass_index ON account (pass);\nCREATE TABLE IF NOT EXISTS config\n(\n id INTEGER PRIMARY KEY AUTOINCREMENT,\n key TEXT NOT NULL UNIQUE DEFAULT '',\n value TEXT NOT NULL DEFAULT '',\n remark TEXT NOT NULL DEFAULT '',\n create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,\n update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP\n);\nCREATE INDEX IF NOT EXISTS config_key_index ON config (key);\nINSERT INTO config (key, value, remark)\nSELECT 'H_UI_WEB_PORT', '8081', 'HY2XS admin Web Port'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'H_UI_WEB_PORT');\nINSERT INTO config (key, value, remark)\nSELECT 'H_UI_WEB_CONTEXT', '/', 'HY2XS admin Web Context'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'H_UI_WEB_CONTEXT');\nINSERT INTO config (key, value, remark)\nSELECT 'H_UI_CRT_PATH', '', 'HY2XS admin CRT File Path'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'H_UI_CRT_PATH');\nINSERT INTO config (key, value, remark)\nSELECT 'H_UI_KEY_PATH', '', 'HY2XS admin KEY File Path'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'H_UI_KEY_PATH');\nINSERT INTO config (key, value, remark)\nSELECT 'JWT_SECRET', hex(randomblob(10)), 'JWT Secret'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'JWT_SECRET');\nINSERT INTO config (key, value, remark)\nSELECT 'HYSTERIA2_ENABLE', '0', 'Hysteria2 Switch'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'HYSTERIA2_ENABLE');\nINSERT INTO config (key, value, remark)\nSELECT 'HYSTERIA2_CONFIG', '', 'Hysteria2 Config'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'HYSTERIA2_CONFIG');\nINSERT INTO config (key, value, remark)\nSELECT 'HYSTERIA2_TRAFFIC_TIME', '1', 'Hysteria2 Traffic Time'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'HYSTERIA2_TRAFFIC_TIME');\nINSERT INTO config (key, value, remark)\nSELECT 'HYSTERIA2_CONFIG_REMARK', '', 'Hysteria2 Config Remark'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'HYSTERIA2_CONFIG_REMARK');\nINSERT INTO config (key, value, remark)\nSELECT 'RESET_TRAFFIC_CRON', '', 'Reset Traffic Cron'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'RESET_TRAFFIC_CRON');\nINSERT INTO config (key, value, remark)\nSELECT 'CLASH_EXTENSION', '', 'Clash Subscription Extension'\n WHERE NOT EXISTS (SELECT 1 FROM config WHERE key = 'CLASH_EXTENSION');"
|
|
|
|
var sqliteDB *gorm.DB
|
|
|
|
func InitSqliteDB() error {
|
|
var err error
|
|
sqliteDB, err = gorm.Open(sqlite.Open(constant.SqliteDBPath), &gorm.Config{
|
|
TranslateError: true,
|
|
Logger: logger.New(
|
|
log.New(os.Stdout, "\r\n", log.LstdFlags),
|
|
logger.Config{
|
|
SlowThreshold: time.Second,
|
|
LogLevel: logger.Silent,
|
|
IgnoreRecordNotFoundError: true,
|
|
ParameterizedQueries: true,
|
|
Colorful: false,
|
|
},
|
|
),
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
logrus.Errorf("sqlite open err: %v", err)
|
|
return errors.New("sqlite open err")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func InitSql(port string) error {
|
|
if err := InitSqliteDB(); err != nil {
|
|
return err
|
|
}
|
|
if err := sqliteInit(sqlInitStr); err != nil {
|
|
return err
|
|
}
|
|
if port != "" {
|
|
var result string
|
|
db, err := sqliteDB.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := db.QueryRow("SELECT value from config where key = 'H_UI_WEB_PORT' limit 1").Scan(&result); err != nil {
|
|
logrus.Errorf("sqlite exec err: %v", err)
|
|
return errors.New("sqlite exec err")
|
|
}
|
|
|
|
if result == "8081" {
|
|
if tx := sqliteDB.Exec("UPDATE config set value = ? where key = 'H_UI_WEB_PORT'", port); tx.Error != nil {
|
|
logrus.Errorf("sqlite exec err: %v", tx.Error)
|
|
return errors.New("sqlite exec err")
|
|
}
|
|
}
|
|
}
|
|
|
|
if err := ensureAccountSchema(); err != nil {
|
|
return err
|
|
}
|
|
if err := ensureSecureBootstrapAdmin(); err != nil {
|
|
return err
|
|
}
|
|
if err := ensureTrafficStatsSecret(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func envInt(name string, fallback int) int {
|
|
if raw := strings.TrimSpace(os.Getenv(name)); raw != "" {
|
|
if value, err := strconv.Atoi(raw); err == nil {
|
|
return value
|
|
}
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func envBoolAsInt(name string, fallback int) int {
|
|
raw := strings.TrimSpace(strings.ToLower(os.Getenv(name)))
|
|
if raw == "" {
|
|
return fallback
|
|
}
|
|
if raw == "true" || raw == "1" {
|
|
return 1
|
|
}
|
|
if raw == "false" || raw == "0" {
|
|
return 0
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func ensureAccountSchema() error {
|
|
if tx := sqliteDB.Exec("ALTER TABLE account ADD COLUMN force_password_change INTEGER NOT NULL DEFAULT 0"); tx.Error != nil && !strings.Contains(tx.Error.Error(), "duplicate column name") {
|
|
logrus.Errorf("sqlite exec err: %v", tx.Error)
|
|
return errors.New("sqlite exec err")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ensureSecureBootstrapAdmin() error {
|
|
adminUser := strings.TrimSpace(os.Getenv("HY2XS_ADMIN_USER"))
|
|
if adminUser == "" {
|
|
adminUser = "hy2xsadmin"
|
|
}
|
|
adminPassword := strings.TrimSpace(os.Getenv("HY2XS_ADMIN_INITIAL_PASSWORD"))
|
|
if adminPassword == "" {
|
|
password, pwdErr := util.RandomString(18)
|
|
if pwdErr != nil {
|
|
return pwdErr
|
|
}
|
|
adminPassword = password
|
|
}
|
|
forcePasswordChange := envBoolAsInt("HY2XS_FORCE_PASSWORD_CHANGE", 1)
|
|
quota := int64(-1)
|
|
expireTime := int64(253370736000000)
|
|
deviceNo := int64(envInt("HY2XS_ADMIN_DEVICE_NO", 6))
|
|
role := "admin"
|
|
deleted := int64(0)
|
|
conPass := strings.TrimSpace(os.Getenv("HY2XS_ADMIN_CON_PASS"))
|
|
if conPass == "" {
|
|
generated, genErr := util.RandomString(28)
|
|
if genErr != nil {
|
|
return genErr
|
|
}
|
|
conPass = generated
|
|
}
|
|
hash, hashErr := util.HashPassword(adminPassword)
|
|
if hashErr != nil {
|
|
return hashErr
|
|
}
|
|
|
|
admin, err := GetAccount("role = 'admin' and deleted = 0")
|
|
if err != nil {
|
|
username := adminUser
|
|
account := entity.Account{
|
|
Username: &username,
|
|
Pass: &hash,
|
|
ConPass: &conPass,
|
|
Quota: "a,
|
|
ExpireTime: &expireTime,
|
|
DeviceNo: &deviceNo,
|
|
Role: &role,
|
|
Deleted: &deleted,
|
|
ForcePasswordChange: func() *int64 { v := int64(forcePasswordChange); return &v }(),
|
|
}
|
|
if _, saveErr := SaveAccount(account); saveErr != nil {
|
|
return saveErr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if admin.Pass == nil {
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ensureTrafficStatsSecret() error {
|
|
envSecret := strings.TrimSpace(os.Getenv("HY2XS_HYSTERIA_TRAFFIC_STATS_SECRET"))
|
|
if envSecret != "" {
|
|
if existing, err := GetConfig("key = ?", constant.Hysteria2TrafficStatsSecret); err == nil {
|
|
if existing.Value != nil && *existing.Value != envSecret {
|
|
return UpdateConfig([]string{constant.Hysteria2TrafficStatsSecret}, map[string]interface{}{"value": envSecret})
|
|
}
|
|
return nil
|
|
}
|
|
key := constant.Hysteria2TrafficStatsSecret
|
|
remark := "Hysteria2 trafficStats API secret"
|
|
if _, saveErr := SaveConfig(entity.Config{Key: &key, Value: &envSecret, Remark: &remark}); saveErr != nil {
|
|
return saveErr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if _, err := GetConfig("key = ?", constant.Hysteria2TrafficStatsSecret); err == nil {
|
|
return nil
|
|
}
|
|
secret, secErr := util.RandomString(32)
|
|
if secErr != nil {
|
|
return secErr
|
|
}
|
|
key := constant.Hysteria2TrafficStatsSecret
|
|
remark := "Hysteria2 trafficStats API secret"
|
|
if _, saveErr := SaveConfig(entity.Config{Key: &key, Value: &secret, Remark: &remark}); saveErr != nil {
|
|
return saveErr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func sqliteInit(sqlStr string) error {
|
|
if sqliteDB != nil {
|
|
sqls := strings.Split(strings.Replace(sqlStr, "\r\n", "\n", -1), ";\n")
|
|
for _, s := range sqls {
|
|
s = strings.TrimSpace(s)
|
|
if s != "" {
|
|
tx := sqliteDB.Exec(s)
|
|
if tx.Error != nil && !strings.HasPrefix(tx.Error.Error(), "SQL logic error: duplicate column name") {
|
|
logrus.Errorf("sqlite exec err: %v", tx.Error)
|
|
return errors.New("sqlite exec err")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func CloseSqliteDB() error {
|
|
if sqliteDB != nil {
|
|
db, err := sqliteDB.DB()
|
|
if err != nil {
|
|
logrus.Errorf("sqlite err: %v", err)
|
|
return errors.New("sqlite err")
|
|
}
|
|
if err = db.Close(); err != nil {
|
|
logrus.Errorf("sqlite close err: %v", err)
|
|
return errors.New("sqlite close err")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Paginate(pageNum *int64, pageSize *int64) func(db *gorm.DB) *gorm.DB {
|
|
var num int64 = 1
|
|
var size int64 = 10
|
|
if pageNum != nil && *pageNum > 0 {
|
|
num = *pageNum
|
|
}
|
|
if pageSize != nil && *pageSize > 0 {
|
|
size = *pageSize
|
|
}
|
|
return func(db *gorm.DB) *gorm.DB {
|
|
return db.Offset(int((num - 1) * size)).Limit(int(size))
|
|
}
|
|
}
|