mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 04:22:33 +08:00
376 lines
9.8 KiB
Go
376 lines
9.8 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"log/slog"
|
|
"path/filepath"
|
|
|
|
"github.com/leonelquinteros/gotext"
|
|
"github.com/spf13/cast"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/acepanel/panel/internal/app"
|
|
"github.com/acepanel/panel/internal/biz"
|
|
"github.com/acepanel/panel/internal/http/request"
|
|
"github.com/acepanel/panel/pkg/cert"
|
|
"github.com/acepanel/panel/pkg/config"
|
|
"github.com/acepanel/panel/pkg/firewall"
|
|
"github.com/acepanel/panel/pkg/io"
|
|
"github.com/acepanel/panel/pkg/os"
|
|
"github.com/acepanel/panel/pkg/systemctl"
|
|
)
|
|
|
|
type settingRepo struct {
|
|
t *gotext.Locale
|
|
db *gorm.DB
|
|
log *slog.Logger
|
|
conf *config.Config
|
|
task biz.TaskRepo
|
|
}
|
|
|
|
func NewSettingRepo(t *gotext.Locale, db *gorm.DB, log *slog.Logger, conf *config.Config, task biz.TaskRepo) biz.SettingRepo {
|
|
return &settingRepo{
|
|
t: t,
|
|
db: db,
|
|
log: log,
|
|
conf: conf,
|
|
task: task,
|
|
}
|
|
}
|
|
|
|
func (r *settingRepo) Get(key biz.SettingKey, defaultValue ...string) (string, error) {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
if setting.Value == "" && len(defaultValue) > 0 {
|
|
return defaultValue[0], nil
|
|
}
|
|
|
|
return setting.Value, nil
|
|
}
|
|
|
|
func (r *settingRepo) GetBool(key biz.SettingKey, defaultValue ...bool) (bool, error) {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return false, err
|
|
}
|
|
}
|
|
|
|
if setting.Value == "" && len(defaultValue) > 0 {
|
|
return defaultValue[0], nil
|
|
}
|
|
|
|
return cast.ToBool(setting.Value), nil
|
|
}
|
|
|
|
func (r *settingRepo) GetInt(key biz.SettingKey, defaultValue ...int) (int, error) {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return 0, err
|
|
}
|
|
}
|
|
|
|
if setting.Value == "" && len(defaultValue) > 0 {
|
|
return defaultValue[0], nil
|
|
}
|
|
|
|
return cast.ToInt(setting.Value), nil
|
|
}
|
|
|
|
func (r *settingRepo) GetSlice(key biz.SettingKey, defaultValue ...[]string) ([]string, error) {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// 设置值为空时提前返回
|
|
slice := make([]string, 0)
|
|
if setting.Value == "" {
|
|
if len(defaultValue) > 0 {
|
|
return defaultValue[0], nil
|
|
}
|
|
return slice, nil
|
|
}
|
|
|
|
if err := json.Unmarshal([]byte(setting.Value), &slice); err != nil {
|
|
return nil, err
|
|
}
|
|
if len(slice) == 0 && len(defaultValue) > 0 {
|
|
return defaultValue[0], nil
|
|
}
|
|
|
|
return slice, nil
|
|
}
|
|
|
|
func (r *settingRepo) Set(key biz.SettingKey, value string) error {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return err
|
|
}
|
|
}
|
|
|
|
setting.Key = key
|
|
setting.Value = value
|
|
if err := r.db.Save(setting).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *settingRepo) SetSlice(key biz.SettingKey, value []string) error {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return err
|
|
}
|
|
}
|
|
|
|
setting.Key = key
|
|
if len(value) == 0 {
|
|
setting.Value = "[]"
|
|
} else {
|
|
b, err := json.Marshal(value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
setting.Value = string(b)
|
|
}
|
|
|
|
if err := r.db.Save(setting).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *settingRepo) Delete(key biz.SettingKey) error {
|
|
setting := new(biz.Setting)
|
|
if err := r.db.Where("key = ?", key).Delete(setting).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *settingRepo) GetPanel() (*request.SettingPanel, error) {
|
|
name, err := r.Get(biz.SettingKeyName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
channel, err := r.Get(biz.SettingKeyChannel)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
offlineMode, err := r.GetBool(biz.SettingKeyOfflineMode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
autoUpdate, err := r.GetBool(biz.SettingKeyAutoUpdate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
websitePath, err := r.Get(biz.SettingKeyWebsitePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
backupPath, err := r.Get(biz.SettingKeyBackupPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
projectPath, err := r.Get(biz.SettingKeyProjectPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hiddenMenu, err := r.GetSlice(biz.SettingHiddenMenu)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
customLogo, err := r.Get(biz.SettingKeyCustomLogo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ip, err := r.Get(biz.SettingKeyPublicIPs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
publicIP := make([]string, 0)
|
|
if err = json.Unmarshal([]byte(ip), &publicIP); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
crt, _ := io.Read(filepath.Join(app.Root, "panel/storage/cert.pem"))
|
|
key, _ := io.Read(filepath.Join(app.Root, "panel/storage/cert.key"))
|
|
|
|
return &request.SettingPanel{
|
|
Name: name,
|
|
Channel: channel,
|
|
Locale: r.conf.App.Locale,
|
|
Entrance: r.conf.HTTP.Entrance,
|
|
EntranceError: r.conf.HTTP.EntranceError,
|
|
LoginCaptcha: r.conf.HTTP.LoginCaptcha,
|
|
OfflineMode: offlineMode,
|
|
AutoUpdate: autoUpdate,
|
|
Lifetime: r.conf.Session.Lifetime,
|
|
IPHeader: r.conf.HTTP.IPHeader,
|
|
BindDomain: r.conf.HTTP.BindDomain,
|
|
BindIP: r.conf.HTTP.BindIP,
|
|
BindUA: r.conf.HTTP.BindUA,
|
|
WebsitePath: websitePath,
|
|
BackupPath: backupPath,
|
|
ProjectPath: projectPath,
|
|
HiddenMenu: hiddenMenu,
|
|
CustomLogo: customLogo,
|
|
Port: r.conf.HTTP.Port,
|
|
HTTPS: r.conf.HTTP.TLS,
|
|
ACME: r.conf.HTTP.ACME,
|
|
PublicIP: publicIP,
|
|
Cert: crt,
|
|
Key: key,
|
|
}, nil
|
|
}
|
|
|
|
func (r *settingRepo) UpdatePanel(ctx context.Context, req *request.SettingPanel) (bool, error) {
|
|
if err := r.Set(biz.SettingKeyName, req.Name); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyChannel, req.Channel); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyOfflineMode, cast.ToString(req.OfflineMode)); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyAutoUpdate, cast.ToString(req.AutoUpdate)); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyWebsitePath, req.WebsitePath); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyBackupPath, req.BackupPath); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyProjectPath, req.ProjectPath); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.SetSlice(biz.SettingHiddenMenu, req.HiddenMenu); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.Set(biz.SettingKeyCustomLogo, req.CustomLogo); err != nil {
|
|
return false, err
|
|
}
|
|
if err := r.SetSlice(biz.SettingKeyPublicIPs, req.PublicIP); err != nil {
|
|
return false, err
|
|
}
|
|
|
|
// 下面是需要需要重启的设置
|
|
// 面板HTTPS
|
|
restartFlag := false
|
|
oldCert, _ := io.Read(filepath.Join(app.Root, "panel/storage/cert.pem"))
|
|
oldKey, _ := io.Read(filepath.Join(app.Root, "panel/storage/cert.key"))
|
|
if oldCert != req.Cert || oldKey != req.Key {
|
|
if r.task.HasRunningTask() {
|
|
return false, errors.New(r.t.Get("background task is running, modifying some settings is prohibited, please try again later"))
|
|
}
|
|
restartFlag = true
|
|
}
|
|
if _, err := cert.ParseCert(req.Cert); err != nil {
|
|
return false, errors.New(r.t.Get("failed to parse certificate: %v", err))
|
|
}
|
|
if _, err := cert.ParseKey(req.Key); err != nil {
|
|
return false, errors.New(r.t.Get("failed to parse private key: %v", err))
|
|
}
|
|
if err := io.Write(filepath.Join(app.Root, "panel/storage/cert.pem"), req.Cert, 0600); err != nil {
|
|
return false, err
|
|
}
|
|
if err := io.Write(filepath.Join(app.Root, "panel/storage/cert.key"), req.Key, 0600); err != nil {
|
|
return false, err
|
|
}
|
|
|
|
// 面板主配置
|
|
conf, err := config.Load()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if req.Port != conf.HTTP.Port {
|
|
if os.TCPPortInUse(req.Port) {
|
|
return false, errors.New(r.t.Get("port is already in use"))
|
|
}
|
|
// 放行端口
|
|
if ok, _ := systemctl.IsEnabled("firewalld"); ok {
|
|
fw := firewall.NewFirewall()
|
|
err = fw.Port(firewall.FireInfo{
|
|
Type: firewall.TypeNormal,
|
|
PortStart: req.Port,
|
|
PortEnd: req.Port,
|
|
Direction: firewall.DirectionIn,
|
|
Strategy: firewall.StrategyAccept,
|
|
}, firewall.OperationAdd)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
}
|
|
}
|
|
|
|
conf.App.Locale = req.Locale
|
|
conf.HTTP.Port = req.Port
|
|
conf.HTTP.Entrance = req.Entrance
|
|
conf.HTTP.EntranceError = req.EntranceError
|
|
conf.HTTP.LoginCaptcha = req.LoginCaptcha
|
|
conf.HTTP.TLS = req.HTTPS
|
|
conf.HTTP.ACME = req.ACME
|
|
conf.HTTP.IPHeader = req.IPHeader
|
|
conf.HTTP.BindDomain = req.BindDomain
|
|
conf.HTTP.BindIP = req.BindIP
|
|
conf.HTTP.BindUA = req.BindUA
|
|
conf.Session.Lifetime = req.Lifetime
|
|
|
|
// 检查配置是否有变更
|
|
if same, _ := config.Check(conf); !same {
|
|
if r.task.HasRunningTask() {
|
|
return false, errors.New(r.t.Get("background task is running, modifying some settings is prohibited, please try again later"))
|
|
}
|
|
restartFlag = true
|
|
}
|
|
if err = config.Save(conf); err != nil {
|
|
return false, err
|
|
}
|
|
|
|
// 记录日志
|
|
r.log.Info("panel settings updated", slog.String("type", biz.OperationTypeSetting), slog.Uint64("operator_id", getOperatorID(ctx)))
|
|
|
|
return restartFlag, nil
|
|
}
|
|
|
|
func (r *settingRepo) UpdateCert(req *request.SettingCert) error {
|
|
if r.task.HasRunningTask() {
|
|
return errors.New(r.t.Get("background task is running, modifying some settings is prohibited, please try again later"))
|
|
}
|
|
if _, err := cert.ParseCert(req.Cert); err != nil {
|
|
return errors.New(r.t.Get("failed to parse certificate: %v", err))
|
|
}
|
|
if _, err := cert.ParseKey(req.Key); err != nil {
|
|
return errors.New(r.t.Get("failed to parse private key: %v", err))
|
|
}
|
|
|
|
if err := io.Write(filepath.Join(app.Root, "panel/storage/cert.pem"), req.Cert, 0600); err != nil {
|
|
return err
|
|
}
|
|
if err := io.Write(filepath.Join(app.Root, "panel/storage/cert.key"), req.Key, 0600); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|