mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-07 02:20:24 +08:00
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package services
|
||
|
||
import (
|
||
"context"
|
||
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
||
"github.com/TeaOSLab/EdgeAPI/internal/errors"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
|
||
"github.com/iwind/TeaGo/lists"
|
||
)
|
||
|
||
type SysSettingService struct {
|
||
BaseService
|
||
}
|
||
|
||
// UpdateSysSetting 更改配置
|
||
func (this *SysSettingService) UpdateSysSetting(ctx context.Context, req *pb.UpdateSysSettingRequest) (*pb.RPCSuccess, error) {
|
||
// 校验请求
|
||
// 不要允许用户修改
|
||
_, err := this.ValidateAdmin(ctx)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
var tx = this.NullTx()
|
||
|
||
err = models.SharedSysSettingDAO.UpdateSetting(tx, req.Code, req.ValueJSON)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return this.Success()
|
||
}
|
||
|
||
// ReadSysSetting 读取配置
|
||
func (this *SysSettingService) ReadSysSetting(ctx context.Context, req *pb.ReadSysSettingRequest) (*pb.ReadSysSettingResponse, error) {
|
||
// 校验请求
|
||
_, userId, err := this.ValidateAdminAndUser(ctx)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
var tx = this.NullTx()
|
||
|
||
// 检查权限
|
||
if userId > 0 {
|
||
// TODO 限制用户只能为专有用户，比如1_000_000_000
|
||
if !lists.ContainsString([]string{
|
||
systemconfigs.SettingCodeUserRegisterConfig,
|
||
systemconfigs.SettingCodeUserServerConfig,
|
||
systemconfigs.SettingCodeUserUIConfig,
|
||
systemconfigs.SettingCodeNSUserConfig,
|
||
systemconfigs.SettingCodeUserOrderConfig,
|
||
}, req.Code) {
|
||
return nil, errors.New("can not read setting code '" + req.Code + "'")
|
||
}
|
||
}
|
||
|
||
valueJSON, err := models.SharedSysSettingDAO.ReadSetting(tx, req.Code)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &pb.ReadSysSettingResponse{ValueJSON: valueJSON}, nil
|
||
}
|