Browse Source

fix roles init

abhishek9686 1 year ago
parent
commit
a8c796b735
5 changed files with 10 additions and 7 deletions
  1. 2 2
      logic/user_mgmt.go
  2. 1 1
      main.go
  3. 1 2
      pro/controllers/users.go
  4. 1 1
      pro/initialize.go
  5. 5 1
      pro/logic/user_mgmt.go

+ 2 - 2
logic/user_mgmt.go

@@ -37,7 +37,7 @@ var IsGroupsValid = func(groups map[models.UserGroupID]struct{}) error {
 }
 }
 var RemoveNetworkRoleFromUsers = func(host models.Host, node models.Node) {}
 var RemoveNetworkRoleFromUsers = func(host models.Host, node models.Node) {}
 
 
-var InitialiseRoles = func() {}
+var InitialiseRoles = userRolesInit
 var DeleteNetworkRoles = func(netID string) {}
 var DeleteNetworkRoles = func(netID string) {}
 var CreateDefaultNetworkRoles = func(netID string) {}
 var CreateDefaultNetworkRoles = func(netID string) {}
 
 
@@ -56,7 +56,7 @@ func GetRole(roleID models.UserRole) (models.UserRolePermissionTemplate, error)
 	return ur, nil
 	return ur, nil
 }
 }
 
 
-func UserRolesInit() {
+func userRolesInit() {
 	d, _ := json.Marshal(SuperAdminPermissionTemplate)
 	d, _ := json.Marshal(SuperAdminPermissionTemplate)
 	database.Insert(SuperAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
 	database.Insert(SuperAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
 	d, _ = json.Marshal(AdminPermissionTemplate)
 	d, _ = json.Marshal(AdminPermissionTemplate)

+ 1 - 1
main.go

@@ -89,7 +89,7 @@ func initialize() { // Client Mode Prereq Check
 	migrate.Run()
 	migrate.Run()
 
 
 	logic.SetJWTSecret()
 	logic.SetJWTSecret()
-	logic.UserRolesInit()
+	logic.InitialiseRoles()
 	err = serverctl.SetDefaults()
 	err = serverctl.SetDefaults()
 	if err != nil {
 	if err != nil {
 		logger.FatalLog("error setting defaults: ", err.Error())
 		logger.FatalLog("error setting defaults: ", err.Error())

+ 1 - 2
pro/controllers/users.go

@@ -9,7 +9,6 @@ import (
 	"net/url"
 	"net/url"
 
 
 	"github.com/gorilla/mux"
 	"github.com/gorilla/mux"
-	"github.com/gravitl/netmaker/auth"
 	"github.com/gravitl/netmaker/database"
 	"github.com/gravitl/netmaker/database"
 	"github.com/gravitl/netmaker/logger"
 	"github.com/gravitl/netmaker/logger"
 	"github.com/gravitl/netmaker/logic"
 	"github.com/gravitl/netmaker/logic"
@@ -1148,7 +1147,7 @@ func approvePendingUser(w http.ResponseWriter, r *http.Request) {
 	}
 	}
 	for _, user := range users {
 	for _, user := range users {
 		if user.UserName == username {
 		if user.UserName == username {
-			var newPass, fetchErr = auth.FetchPassValue("")
+			var newPass, fetchErr = logic.FetchPassValue("")
 			if fetchErr != nil {
 			if fetchErr != nil {
 				logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
 				logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
 				return
 				return

+ 1 - 1
pro/initialize.go

@@ -22,7 +22,6 @@ import (
 func InitPro() {
 func InitPro() {
 	servercfg.IsPro = true
 	servercfg.IsPro = true
 	models.SetLogo(retrieveProLogo())
 	models.SetLogo(retrieveProLogo())
-	proLogic.UserRolesInit()
 	controller.HttpMiddlewares = append(
 	controller.HttpMiddlewares = append(
 		controller.HttpMiddlewares,
 		controller.HttpMiddlewares,
 		proControllers.OnlyServerAPIWhenUnlicensedMiddleware,
 		proControllers.OnlyServerAPIWhenUnlicensedMiddleware,
@@ -129,6 +128,7 @@ func InitPro() {
 	logic.FilterNetworksByRole = proLogic.FilterNetworksByRole
 	logic.FilterNetworksByRole = proLogic.FilterNetworksByRole
 	logic.IsGroupsValid = proLogic.IsGroupsValid
 	logic.IsGroupsValid = proLogic.IsGroupsValid
 	logic.RemoveNetworkRoleFromUsers = proLogic.RemoveNetworkRoleFromUsers
 	logic.RemoveNetworkRoleFromUsers = proLogic.RemoveNetworkRoleFromUsers
+	logic.InitialiseRoles = proLogic.UserRolesInit
 }
 }
 
 
 func retrieveProLogo() string {
 func retrieveProLogo() string {

+ 5 - 1
pro/logic/user_mgmt.go

@@ -26,7 +26,11 @@ var PlatformUserUserPermissionTemplate = models.UserRolePermissionTemplate{
 }
 }
 
 
 func UserRolesInit() {
 func UserRolesInit() {
-	d, _ := json.Marshal(ServiceUserPermissionTemplate)
+	d, _ := json.Marshal(logic.SuperAdminPermissionTemplate)
+	database.Insert(logic.SuperAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
+	d, _ = json.Marshal(logic.AdminPermissionTemplate)
+	database.Insert(logic.AdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
+	d, _ = json.Marshal(ServiceUserPermissionTemplate)
 	database.Insert(ServiceUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
 	database.Insert(ServiceUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
 	d, _ = json.Marshal(PlatformUserUserPermissionTemplate)
 	d, _ = json.Marshal(PlatformUserUserPermissionTemplate)
 	database.Insert(PlatformUserUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
 	database.Insert(PlatformUserUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)