Upstream/harbor
1
0
Fork 0
mirror of https://github.com/goharbor/harbor.git synced 2024-12-24 17:47:46 +01:00
Code Issues Projects Releases Wiki Activity

refactor config API

Browse Source
This commit is contained in:
Wenkai Yin 2017-05-25 14:44:05 +08:00
parent 565110d9f1
commit 8ae5e46d63
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/ui/api/config.go
View File

@ -20,7 +20,6 @@ import (
"strconv" "strconv"
"github.com/vmware/harbor/src/common" "github.com/vmware/harbor/src/common"
"github.com/vmware/harbor/src/common/api"
"github.com/vmware/harbor/src/common/dao" "github.com/vmware/harbor/src/common/dao"
"github.com/vmware/harbor/src/common/utils/log" "github.com/vmware/harbor/src/common/utils/log"
"github.com/vmware/harbor/src/ui/config" "github.com/vmware/harbor/src/ui/config"
@ -91,20 +90,19 @@ var (
// ConfigAPI ... // ConfigAPI ...
type ConfigAPI struct { type ConfigAPI struct {
api.BaseAPI BaseController
} }
// Prepare validates the user // Prepare validates the user
func (c *ConfigAPI) Prepare() { func (c *ConfigAPI) Prepare() {
userID := c.ValidateUser() c.BaseController.Prepare()
isSysAdmin, err := dao.IsAdminRole(userID) if !c.SecurityCtx.IsAuthenticated() {
if err != nil { c.HandleUnauthorized()
log.Errorf("failed to check the role of user: %v", err) return
c.CustomAbort(http.StatusInternalServerError, http.StatusText(http.StatusInternalServerError))
} }
if !c.SecurityCtx.IsSysAdmin() {
if !isSysAdmin { c.HandleForbidden(c.SecurityCtx.GetUsername())
c.CustomAbort(http.StatusForbidden, http.StatusText(http.StatusForbidden)) return
} }
} }