TRKJ/EdgeAdmin
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAdmin.git synced 2026-03-15 08:55:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

安全设置增加允许访问的国家地区、省份、是否局域网访问

Browse Source
This commit is contained in:
GoEdgeLab
2020-11-20 21:59:12 +08:00
parent b764f470d9
commit ea76242ef1
25 changed files with 609 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/securitymanager/security_config.go
View File

@@ -22,7 +22,10 @@ const (
var sharedSecurityConfig *SecurityConfig = nil
type SecurityConfig struct {
Frame string `json:"frame"`
Frame string `json:"frame"`
AllowCountryIds []int64 `json:"allowCountryIds"`
AllowProvinceIds []int64 `json:"allowProvinceIds"`
AllowLocal bool `json:"allowLocal"`
}
func LoadSecurityConfig() (*SecurityConfig, error) {
@@ -50,7 +53,7 @@ func UpdateSecurityConfig(securityConfig *SecurityConfig) error {
if err != nil {
return err
}
_, err = rpcClient.SysSettingRPC().UpdateSysSetting(rpcClient.Context(1), &pb.UpdateSysSettingRequest{
_, err = rpcClient.SysSettingRPC().UpdateSysSetting(rpcClient.Context(0), &pb.UpdateSysSettingRequest{
Code: SecuritySettingName,
ValueJSON: valueJSON,
})
@@ -69,7 +72,7 @@ func loadSecurityConfig() (*SecurityConfig, error) {
if err != nil {
return nil, err
}
resp, err := rpcClient.SysSettingRPC().ReadSysSetting(rpcClient.Context(1), &pb.ReadSysSettingRequest{
resp, err := rpcClient.SysSettingRPC().ReadSysSetting(rpcClient.Context(0), &pb.ReadSysSettingRequest{
Code: SecuritySettingName,
})
if err != nil {
@@ -93,6 +96,7 @@ func loadSecurityConfig() (*SecurityConfig, error) {
func defaultSecurityConfig() *SecurityConfig {
return &SecurityConfig{
Frame: FrameSameOrigin,
Frame: FrameSameOrigin,
AllowLocal: true,
}
}