diff --git a/controllers/user.go b/controllers/user.go index fff98603..ecf24484 100644 --- a/controllers/user.go +++ b/controllers/user.go @@ -16,6 +16,7 @@ import ( "github.com/gravitl/netmaker/logic" "github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/mq" + "github.com/gravitl/netmaker/schema" "github.com/gravitl/netmaker/servercfg" "golang.org/x/exp/slog" ) @@ -57,7 +58,7 @@ func createUserAccessToken(w http.ResponseWriter, r *http.Request) { // Auth request consists of Mac Address and Password (from node that is authorizing // in case of Master, auth is ignored and mac is set to "mastermac" - var req models.UserAccessToken + var req schema.UserAccessToken err := json.NewDecoder(r.Body).Decode(&req) if err != nil { @@ -139,7 +140,7 @@ func getUserAccessTokens(w http.ResponseWriter, r *http.Request) { logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username is required"), "badrequest")) return } - logic.ReturnSuccessResponseWithJson(w, r, (&models.UserAccessToken{UserName: username}).ListByUser(), "fetched api access tokens for user "+username) + logic.ReturnSuccessResponseWithJson(w, r, (&schema.UserAccessToken{UserName: username}).ListByUser(), "fetched api access tokens for user "+username) } // @Summary Authenticate a user to retrieve an authorization token @@ -157,7 +158,7 @@ func deleteUserAccessTokens(w http.ResponseWriter, r *http.Request) { logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("id is required"), "badrequest")) return } - a := models.UserAccessToken{ + a := schema.UserAccessToken{ ID: id, } err := a.Get() @@ -187,7 +188,7 @@ func deleteUserAccessTokens(w http.ResponseWriter, r *http.Request) { } } - err = (&models.UserAccessToken{ID: id}).Delete() + err = (&schema.UserAccessToken{ID: id}).Delete() if err != nil { logic.ReturnErrorResponse( w, @@ -753,7 +754,7 @@ func updateUser(w http.ResponseWriter, r *http.Request) { } logic.AddGlobalNetRolesToAdmins(&userchange) if userchange.PlatformRoleID != user.PlatformRoleID || !logic.CompareMaps(user.UserGroups, userchange.UserGroups) { - (&models.UserAccessToken{UserName: user.UserName}).DeleteAllUserTokens() + (&schema.UserAccessToken{UserName: user.UserName}).DeleteAllUserTokens() } user, err = logic.UpdateUser(&userchange, user) if err != nil { diff --git a/logic/auth.go b/logic/auth.go index b0856e4d..e2e5a770 100644 --- a/logic/auth.go +++ b/logic/auth.go @@ -14,6 +14,7 @@ import ( "github.com/gravitl/netmaker/database" "github.com/gravitl/netmaker/logger" "github.com/gravitl/netmaker/models" + "github.com/gravitl/netmaker/schema" ) const ( @@ -360,7 +361,7 @@ func DeleteUser(user string) error { return err } go RemoveUserFromAclPolicy(user) - return (&models.UserAccessToken{UserName: user}).DeleteAllUserTokens() + return (&schema.UserAccessToken{UserName: user}).DeleteAllUserTokens() } func SetAuthSecret(secret string) error { diff --git a/logic/jwts.go b/logic/jwts.go index 7d6f0fb9..aada8bfc 100644 --- a/logic/jwts.go +++ b/logic/jwts.go @@ -10,6 +10,7 @@ import ( "github.com/gravitl/netmaker/logger" "github.com/gravitl/netmaker/models" + "github.com/gravitl/netmaker/schema" "github.com/gravitl/netmaker/servercfg" ) @@ -124,7 +125,7 @@ func GetUserNameFromToken(authtoken string) (username string, err error) { if claims.TokenType == models.AccessTokenType { jti := claims.ID if jti != "" { - a := models.UserAccessToken{ID: jti} + a := schema.UserAccessToken{ID: jti} // check if access token is active err := a.Get() if err != nil { @@ -168,7 +169,7 @@ func VerifyUserToken(tokenString string) (username string, issuperadmin, isadmin if claims.TokenType == models.AccessTokenType { jti := claims.ID if jti != "" { - a := models.UserAccessToken{ID: jti} + a := schema.UserAccessToken{ID: jti} // check if access token is active err := a.Get() if err != nil { diff --git a/models/accessToken.go b/schema/accessToken.go similarity index 99% rename from models/accessToken.go rename to schema/accessToken.go index cfa5c042..d2d5be9e 100644 --- a/models/accessToken.go +++ b/schema/accessToken.go @@ -1,4 +1,4 @@ -package models +package schema import ( "context" diff --git a/schema/models.go b/schema/models.go index 3463a820..5c27a7b2 100644 --- a/schema/models.go +++ b/schema/models.go @@ -1,11 +1,9 @@ package schema -import "github.com/gravitl/netmaker/models" - // ListModels lists all the models in this schema. func ListModels() []interface{} { return []interface{}{ &Job{}, - &models.UserAccessToken{}, + &UserAccessToken{}, } }