TRKJ/EdgeAdmin
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAdmin.git synced 2025-11-18 14:50:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
刘祥超
2020-11-20 21:59:12 +08:00
parent 8d8460c3fe
commit 3ec89a432c
25 changed files with 609 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
internal/web/helpers/user_must_auth.go
View File

@@ -35,6 +35,12 @@ func (this *UserMustAuth) BeforeAction(actionPtr actions.ActionWrapper, paramNam
}
action.AddHeader("Content-Security-Policy", "default-src 'self' data:; script-src 'self' 'unsafe-inline' 'unsafe-eval'; style-src 'self' 'unsafe-inline'")
// 检查IP
if !checkIP(securityConfig, action.RequestRemoteIP()) {
action.ResponseWriter.WriteHeader(http.StatusForbidden)
return false
}
// 检查系统是否已经配置过
if !setup.IsConfigured() {
action.RedirectURL("/setup")
@@ -82,8 +88,48 @@ func (this *UserMustAuth) BeforeAction(actionPtr actions.ActionWrapper, paramNam
return action.Data["teaTitle"].(string)
})
// 初始化变量
modules := []maps.Map{
action.Data["teaTitle"] = teaconst.ProductNameZH
action.Data["teaName"] = teaconst.ProductNameZH
resp, err := rpc.AdminRPC().FindAdminFullname(rpc.Context(0), &pb.FindAdminFullnameRequest{AdminId: int64(this.AdminId)})
if err != nil {
utils.PrintError(err)
action.Data["teaUsername"] = ""
} else {
action.Data["teaUsername"] = resp.Fullname
}
action.Data["teaUserAvatar"] = ""
action.Data["teaMenu"] = ""
action.Data["teaModules"] = this.modules()
action.Data["teaSubMenus"] = []map[string]interface{}{}
action.Data["teaTabbar"] = []map[string]interface{}{}
action.Data["teaVersion"] = teaconst.Version
action.Data["teaIsSuper"] = false
action.Data["teaDemoEnabled"] = teaconst.IsDemo
if !action.Data.Has("teaSubMenu") {
action.Data["teaSubMenu"] = ""
}
// 菜单
action.Data["firstMenuItem"] = ""
// 未读消息数
action.Data["teaBadge"] = 0
// 调用Init
initMethod := reflect.ValueOf(actionPtr).MethodByName("Init")
if initMethod.IsValid() {
initMethod.Call([]reflect.Value{})
}
return true
}
// 菜单配置
func (this *UserMustAuth) modules() []maps.Map {
return []maps.Map{
{
"code": "servers",
"name": "网站服务",
@@ -136,46 +182,9 @@ func (this *UserMustAuth) BeforeAction(actionPtr actions.ActionWrapper, paramNam
"icon": "history",
},
}
action.Data["teaTitle"] = teaconst.ProductNameZH
action.Data["teaName"] = teaconst.ProductNameZH
resp, err := rpc.AdminRPC().FindAdminFullname(rpc.Context(0), &pb.FindAdminFullnameRequest{AdminId: int64(this.AdminId)})
if err != nil {
utils.PrintError(err)
action.Data["teaUsername"] = ""
} else {
action.Data["teaUsername"] = resp.Fullname
}
action.Data["teaUserAvatar"] = ""
action.Data["teaMenu"] = ""
action.Data["teaModules"] = modules
action.Data["teaSubMenus"] = []map[string]interface{}{}
action.Data["teaTabbar"] = []map[string]interface{}{}
action.Data["teaVersion"] = teaconst.Version
action.Data["teaIsSuper"] = false
action.Data["teaDemoEnabled"] = teaconst.IsDemo
if !action.Data.Has("teaSubMenu") {
action.Data["teaSubMenu"] = ""
}
// 菜单
action.Data["firstMenuItem"] = ""
// 未读消息数
action.Data["teaBadge"] = 0
// 调用Init
initMethod := reflect.ValueOf(actionPtr).MethodByName("Init")
if initMethod.IsValid() {
initMethod.Call([]reflect.Value{})
}
return true
}
// 跳转到登录页
func (this *UserMustAuth) login(action *actions.ActionObject) {
action.RedirectURL("/")
}