Upstream/harbor
1
0
Fork 0
mirror of https://github.com/goharbor/harbor.git synced 2025-02-20 05:52:25 +01:00
Code Issues Projects Releases Wiki Activity

refactor email API

Browse Source
This commit is contained in:
Wenkai Yin 2017-05-24 17:59:21 +08:00
parent a901297859
commit 6ee64ba83a
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/email.go
View File

@ -19,8 +19,6 @@ import (
"net/http" "net/http"
"strconv" "strconv"
"github.com/vmware/harbor/src/common/api"
"github.com/vmware/harbor/src/common/dao"
"github.com/vmware/harbor/src/common/utils/email" "github.com/vmware/harbor/src/common/utils/email"
"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"
@ -32,20 +30,20 @@ const (
// EmailAPI ... // EmailAPI ...
type EmailAPI struct { type EmailAPI struct {
api.BaseAPI BaseController
} }
// Prepare ... // Prepare ...
func (e *EmailAPI) Prepare() { func (e *EmailAPI) Prepare() {
userID := e.ValidateUser() e.BaseController.Prepare()
isSysAdmin, err := dao.IsAdminRole(userID) if !e.SecurityCtx.IsAuthenticated() {
if err != nil { e.HandleUnauthorized()
log.Errorf("failed to check the role of user: %v", err) return
e.CustomAbort(http.StatusInternalServerError, http.StatusText(http.StatusInternalServerError))
} }
if !isSysAdmin { if !e.SecurityCtx.IsSysAdmin() {
e.CustomAbort(http.StatusForbidden, http.StatusText(http.StatusForbidden)) e.HandleForbidden(e.SecurityCtx.GetUsername())
return
} }
} }