parent
7d17032fc2
commit
c6cbcd2e66
|
@ -10,6 +10,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
YAUTH_ADMIN_SECRET = ""
|
||||||
ENV = ""
|
ENV = ""
|
||||||
DB_TYPE = ""
|
DB_TYPE = ""
|
||||||
DB_URL = ""
|
DB_URL = ""
|
||||||
|
@ -38,7 +39,7 @@ func init() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Error loading .env file")
|
log.Println("Error loading .env file")
|
||||||
}
|
}
|
||||||
|
YAUTH_ADMIN_SECRET = os.Getenv("YAUTH_ADMIN_SECRET")
|
||||||
ENV = os.Getenv("ENV")
|
ENV = os.Getenv("ENV")
|
||||||
DB_TYPE = os.Getenv("DB_TYPE")
|
DB_TYPE = os.Getenv("DB_TYPE")
|
||||||
DB_URL = os.Getenv("DB_URL")
|
DB_URL = os.Getenv("DB_URL")
|
||||||
|
@ -60,6 +61,10 @@ func init() {
|
||||||
// FACEBOOK_CLIENT_ID = os.Getenv("FACEBOOK_CLIENT_ID")
|
// FACEBOOK_CLIENT_ID = os.Getenv("FACEBOOK_CLIENT_ID")
|
||||||
// FACEBOOK_CLIENT_SECRET = os.Getenv("FACEBOOK_CLIENT_SECRET")
|
// FACEBOOK_CLIENT_SECRET = os.Getenv("FACEBOOK_CLIENT_SECRET")
|
||||||
|
|
||||||
|
if YAUTH_ADMIN_SECRET == "" {
|
||||||
|
panic("Yauth admin secret is required")
|
||||||
|
}
|
||||||
|
|
||||||
if ENV == "" {
|
if ENV == "" {
|
||||||
ENV = "production"
|
ENV = "production"
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,10 +6,20 @@ import (
|
||||||
|
|
||||||
"github.com/yauthdev/yauth/server/db"
|
"github.com/yauthdev/yauth/server/db"
|
||||||
"github.com/yauthdev/yauth/server/graph/model"
|
"github.com/yauthdev/yauth/server/graph/model"
|
||||||
|
"github.com/yauthdev/yauth/server/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
func Users(ctx context.Context) ([]*model.User, error) {
|
func Users(ctx context.Context) ([]*model.User, error) {
|
||||||
|
gc, err := utils.GinContextFromContext(ctx)
|
||||||
var res []*model.User
|
var res []*model.User
|
||||||
|
if err != nil {
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !utils.IsSuperAdmin(gc) {
|
||||||
|
return res, fmt.Errorf("unauthorized")
|
||||||
|
}
|
||||||
|
|
||||||
users, err := db.Mgr.GetUsers()
|
users, err := db.Mgr.GetUsers()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return res, err
|
return res, err
|
||||||
|
|
15
server/utils/validateSuperAdmin.go
Normal file
15
server/utils/validateSuperAdmin.go
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/yauthdev/yauth/server/constants"
|
||||||
|
)
|
||||||
|
|
||||||
|
func IsSuperAdmin(gc *gin.Context) bool {
|
||||||
|
secret := gc.Request.Header.Get("x-yauth-admin-secret")
|
||||||
|
if secret == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return secret == constants.YAUTH_ADMIN_SECRET
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user