| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002 |
- package controller
- import (
- "encoding/json"
- "errors"
- "fmt"
- "net/http"
- "github.com/gorilla/mux"
- "github.com/gorilla/websocket"
- "github.com/gravitl/netmaker/auth"
- "github.com/gravitl/netmaker/database"
- "github.com/gravitl/netmaker/logger"
- "github.com/gravitl/netmaker/logic"
- "github.com/gravitl/netmaker/models"
- "github.com/gravitl/netmaker/mq"
- "github.com/gravitl/netmaker/servercfg"
- "golang.org/x/exp/slog"
- )
- var (
- upgrader = websocket.Upgrader{}
- )
- func userHandlers(r *mux.Router) {
- r.HandleFunc("/api/users/adm/hassuperadmin", hasSuperAdmin).Methods(http.MethodGet)
- r.HandleFunc("/api/users/adm/createsuperadmin", createSuperAdmin).Methods(http.MethodPost)
- r.HandleFunc("/api/users/adm/transfersuperadmin/{username}", logic.SecurityCheck(true, http.HandlerFunc(transferSuperAdmin))).Methods(http.MethodPost)
- r.HandleFunc("/api/users/adm/authenticate", authenticateUser).Methods(http.MethodPost)
- r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, http.HandlerFunc(updateUser))).Methods(http.MethodPut)
- r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, checkFreeTierLimits(limitChoiceUsers, http.HandlerFunc(createUser)))).Methods(http.MethodPost)
- r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, http.HandlerFunc(deleteUser))).Methods(http.MethodDelete)
- r.HandleFunc("/api/users/{username}", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUser)))).Methods(http.MethodGet)
- r.HandleFunc("/api/users", logic.SecurityCheck(true, http.HandlerFunc(getUsers))).Methods(http.MethodGet)
- r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(getPendingUsers))).Methods(http.MethodGet)
- r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(deleteAllPendingUsers))).Methods(http.MethodDelete)
- r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(deletePendingUser))).Methods(http.MethodDelete)
- r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(approvePendingUser))).Methods(http.MethodPost)
- // User Role handlers
- r.HandleFunc("/api/v1/user/roles", logic.SecurityCheck(true, http.HandlerFunc(listRoles))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/user/role", logic.SecurityCheck(true, http.HandlerFunc(getRole))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/user/role", logic.SecurityCheck(true, http.HandlerFunc(createRole))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/user/role", logic.SecurityCheck(true, http.HandlerFunc(updateRole))).Methods(http.MethodPut)
- r.HandleFunc("/api/v1/user/role", logic.SecurityCheck(true, http.HandlerFunc(deleteRole))).Methods(http.MethodDelete)
- // User Group Handlers
- r.HandleFunc("/api/v1/user/groups", logic.SecurityCheck(true, http.HandlerFunc(listUserGroups))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/user/group", logic.SecurityCheck(true, http.HandlerFunc(getUserGroup))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/user/group", logic.SecurityCheck(true, http.HandlerFunc(createUserGroup))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/user/group", logic.SecurityCheck(true, http.HandlerFunc(updateUserGroup))).Methods(http.MethodPut)
- r.HandleFunc("/api/v1/user/group", logic.SecurityCheck(true, http.HandlerFunc(deleteUserGroup))).Methods(http.MethodDelete)
- }
- // swagger:route GET /api/v1/user/groups user listUserGroups
- //
- // Get all user groups.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func listUserGroups(w http.ResponseWriter, r *http.Request) {
- groups, err := logic.ListUserGroups()
- if err != nil {
- logic.ReturnErrorResponse(w, r, models.ErrorResponse{
- Code: http.StatusInternalServerError,
- Message: err.Error(),
- })
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, groups, "successfully fetched user groups")
- }
- // swagger:route GET /api/v1/user/group user getUserGroup
- //
- // Get user group.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func getUserGroup(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- gid := params["group_id"]
- if gid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
- return
- }
- group, err := logic.GetUserGroup(models.UserGroupID(gid))
- if err != nil {
- logic.ReturnErrorResponse(w, r, models.ErrorResponse{
- Code: http.StatusInternalServerError,
- Message: err.Error(),
- })
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, group, "successfully fetched user group")
- }
- // swagger:route POST /api/v1/user/group user createUserGroup
- //
- // Create user groups.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func createUserGroup(w http.ResponseWriter, r *http.Request) {
- var userGroup models.UserGroup
- err := json.NewDecoder(r.Body).Decode(&userGroup)
- if err != nil {
- slog.Error("error decoding request body", "error",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = logic.CreateUserGroup(userGroup)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, userGroup, "created user group")
- }
- // swagger:route PUT /api/v1/user/group user updateUserGroup
- //
- // Update user group.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func updateUserGroup(w http.ResponseWriter, r *http.Request) {
- var userGroup models.UserGroup
- err := json.NewDecoder(r.Body).Decode(&userGroup)
- if err != nil {
- slog.Error("error decoding request body", "error",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = logic.UpdateUserGroup(userGroup)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, userGroup, "updated user group")
- }
- // swagger:route DELETE /api/v1/user/group user deleteUserGroup
- //
- // delete user group.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func deleteUserGroup(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- gid := params["group_id"]
- if gid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- err := logic.DeleteUserGroup(models.UserGroupID(gid))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user group")
- }
- // swagger:route GET /api/v1/user/roles user listRoles
- //
- // lists all user roles.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func listRoles(w http.ResponseWriter, r *http.Request) {
- roles, err := logic.ListRoles()
- if err != nil {
- logic.ReturnErrorResponse(w, r, models.ErrorResponse{
- Code: http.StatusInternalServerError,
- Message: err.Error(),
- })
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, roles, "successfully fetched user roles permission templates")
- }
- // swagger:route GET /api/v1/user/role user getRole
- //
- // Get user role permission templates.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func getRole(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- rid := params["role_id"]
- if rid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- role, err := logic.GetRole(rid)
- if err != nil {
- logic.ReturnErrorResponse(w, r, models.ErrorResponse{
- Code: http.StatusInternalServerError,
- Message: err.Error(),
- })
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, role, "successfully fetched user role permission templates")
- }
- // swagger:route POST /api/v1/user/role user createRole
- //
- // Create user role permission template.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func createRole(w http.ResponseWriter, r *http.Request) {
- var userRole models.UserRolePermissionTemplate
- err := json.NewDecoder(r.Body).Decode(&userRole)
- if err != nil {
- slog.Error("error decoding request body", "error",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = logic.CreateRole(userRole)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, userRole, "created user role")
- }
- // swagger:route PUT /api/v1/user/role user updateRole
- //
- // Update user role permission template.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func updateRole(w http.ResponseWriter, r *http.Request) {
- var userRole models.UserRolePermissionTemplate
- err := json.NewDecoder(r.Body).Decode(&userRole)
- if err != nil {
- slog.Error("error decoding request body", "error",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = logic.UpdateRole(userRole)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, userRole, "updated user role")
- }
- // swagger:route DELETE /api/v1/user/role user deleteRole
- //
- // Delete user role permission template.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func deleteRole(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- rid := params["role_id"]
- if rid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- err := logic.DeleteRole(models.UserRole(rid))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, nil, "created user role")
- }
- // swagger:route POST /api/users/adm/authenticate authenticate authenticateUser
- //
- // User authenticates using its password and retrieves a JWT for authorization.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: successResponse
- func authenticateUser(response http.ResponseWriter, request *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 authRequest models.UserAuthParams
- var errorResponse = models.ErrorResponse{
- Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
- }
- if !servercfg.IsBasicAuthEnabled() {
- logic.ReturnErrorResponse(response, request, logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"))
- return
- }
- decoder := json.NewDecoder(request.Body)
- decoderErr := decoder.Decode(&authRequest)
- defer request.Body.Close()
- if decoderErr != nil {
- logger.Log(0, "error decoding request body: ",
- decoderErr.Error())
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- if val := request.Header.Get("From-Ui"); val == "true" {
- // request came from UI, if normal user block Login
- user, err := logic.GetUser(authRequest.UserName)
- if err != nil {
- logger.Log(0, authRequest.UserName, "user validation failed: ",
- err.Error())
- logic.ReturnErrorResponse(response, request, logic.FormatError(err, "unauthorized"))
- return
- }
- if !(user.IsAdmin || user.IsSuperAdmin) {
- logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("only admins can access dashboard"), "unauthorized"))
- return
- }
- }
- username := authRequest.UserName
- jwt, err := logic.VerifyAuthRequest(authRequest)
- if err != nil {
- logger.Log(0, username, "user validation failed: ",
- err.Error())
- logic.ReturnErrorResponse(response, request, logic.FormatError(err, "badrequest"))
- return
- }
- if jwt == "" {
- // very unlikely that err is !nil and no jwt returned, but handle it anyways.
- logger.Log(0, username, "jwt token is empty")
- logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("no token returned"), "internal"))
- return
- }
- var successResponse = models.SuccessResponse{
- Code: http.StatusOK,
- Message: "W1R3: Device " + username + " Authorized",
- Response: models.SuccessfulUserLoginResponse{
- AuthToken: jwt,
- UserName: username,
- },
- }
- // Send back the JWT
- successJSONResponse, jsonError := json.Marshal(successResponse)
- if jsonError != nil {
- logger.Log(0, username,
- "error marshalling resp: ", jsonError.Error())
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- logger.Log(2, username, "was authenticated")
- response.Header().Set("Content-Type", "application/json")
- response.Write(successJSONResponse)
- go func() {
- if servercfg.IsPro && servercfg.GetRacAutoDisable() {
- // enable all associeated clients for the user
- clients, err := logic.GetAllExtClients()
- if err != nil {
- slog.Error("error getting clients: ", "error", err)
- return
- }
- for _, client := range clients {
- if client.OwnerID == username && !client.Enabled {
- slog.Info(fmt.Sprintf("enabling ext client %s for user %s due to RAC autodisabling feature", client.ClientID, client.OwnerID))
- if newClient, err := logic.ToggleExtClientConnectivity(&client, true); err != nil {
- slog.Error("error enabling ext client in RAC autodisable hook", "error", err)
- continue // dont return but try for other clients
- } else {
- // publish peer update to ingress gateway
- if ingressNode, err := logic.GetNodeByID(newClient.IngressGatewayID); err == nil {
- if err = mq.PublishPeerUpdate(false); err != nil {
- slog.Error("error updating ext clients on", "ingress", ingressNode.ID.String(), "err", err.Error())
- }
- }
- }
- }
- }
- }
- }()
- }
- // swagger:route GET /api/users/adm/hassuperadmin user hasSuperAdmin
- //
- // Checks whether the server has an admin.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: hasAdmin
- func hasSuperAdmin(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- hasSuperAdmin, err := logic.HasSuperAdmin()
- if err != nil {
- logger.Log(0, "failed to check for admin: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- json.NewEncoder(w).Encode(hasSuperAdmin)
- }
- // swagger:route GET /api/users/{username} user getUser
- //
- // Get an individual user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func getUser(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- usernameFetched := params["username"]
- user, err := logic.GetReturnUser(usernameFetched)
- if err != nil {
- logger.Log(0, usernameFetched, "failed to fetch user: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logger.Log(2, r.Header.Get("user"), "fetched user", usernameFetched)
- json.NewEncoder(w).Encode(user)
- }
- // swagger:route GET /api/users user getUsers
- //
- // Get all users.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func getUsers(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- users, err := logic.GetUsers()
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.SortUsers(users[:])
- logger.Log(2, r.Header.Get("user"), "fetched users")
- json.NewEncoder(w).Encode(users)
- }
- // swagger:route POST /api/users/adm/createsuperadmin user createAdmin
- //
- // Make a user an admin.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func createSuperAdmin(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var u models.User
- err := json.NewDecoder(r.Body).Decode(&u)
- if err != nil {
- slog.Error("error decoding request body", "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if !servercfg.IsBasicAuthEnabled() {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"))
- return
- }
- err = logic.CreateSuperAdmin(&u)
- if err != nil {
- slog.Error("failed to create admin", "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- logger.Log(1, u.UserName, "was made a super admin")
- json.NewEncoder(w).Encode(logic.ToReturnUser(u))
- }
- // swagger:route POST /api/users/adm/transfersuperadmin user transferSuperAdmin
- //
- // Transfers superadmin role to an admin user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func transferSuperAdmin(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- caller, err := logic.GetUser(r.Header.Get("user"))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- }
- if !caller.IsSuperAdmin {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can assign the superadmin role to another user"), "forbidden"))
- return
- }
- var params = mux.Vars(r)
- username := params["username"]
- u, err := logic.GetUser(username)
- if err != nil {
- slog.Error("error getting user", "user", u.UserName, "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if !u.IsAdmin {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only admins can be promoted to superadmin role"), "forbidden"))
- return
- }
- if !servercfg.IsBasicAuthEnabled() {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"))
- return
- }
- u.IsSuperAdmin = true
- u.IsAdmin = false
- err = logic.UpsertUser(*u)
- if err != nil {
- slog.Error("error updating user to superadmin: ", "user", u.UserName, "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- caller.IsSuperAdmin = false
- caller.IsAdmin = true
- err = logic.UpsertUser(*caller)
- if err != nil {
- slog.Error("error demoting user to admin: ", "user", caller.UserName, "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- slog.Info("user was made a super admin", "user", u.UserName)
- json.NewEncoder(w).Encode(logic.ToReturnUser(*u))
- }
- // swagger:route POST /api/users/{username} user createUser
- //
- // Create a user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func createUser(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- caller, err := logic.GetUser(r.Header.Get("user"))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- var user models.User
- err = json.NewDecoder(r.Body).Decode(&user)
- if err != nil {
- logger.Log(0, user.UserName, "error decoding request body: ",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if !caller.IsSuperAdmin && user.IsAdmin {
- err = errors.New("only superadmin can create admin users")
- slog.Error("error creating new user: ", "user", user.UserName, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
- return
- }
- if user.IsSuperAdmin {
- err = errors.New("additional superadmins cannot be created")
- slog.Error("error creating new user: ", "user", user.UserName, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
- return
- }
- if !servercfg.IsPro && !user.IsAdmin {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("non-admins users can only be created on Pro version"), "forbidden"))
- return
- }
- err = logic.CreateUser(&user)
- if err != nil {
- slog.Error("error creating new user: ", "user", user.UserName, "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- slog.Info("user was created", "username", user.UserName)
- json.NewEncoder(w).Encode(logic.ToReturnUser(user))
- }
- // swagger:route PUT /api/users/{username} user updateUser
- //
- // Update a user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func updateUser(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- // start here
- var caller *models.User
- var err error
- var ismaster bool
- if r.Header.Get("user") == logic.MasterUser {
- ismaster = true
- } else {
- caller, err = logic.GetUser(r.Header.Get("user"))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- }
- }
- username := params["username"]
- user, err := logic.GetUser(username)
- if err != nil {
- logger.Log(0, username,
- "failed to update user info: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- var userchange models.User
- // we decode our body request params
- err = json.NewDecoder(r.Body).Decode(&userchange)
- if err != nil {
- slog.Error("failed to decode body", "error ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if user.UserName != userchange.UserName {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user in param and request body not matching"), "badrequest"))
- return
- }
- selfUpdate := false
- if !ismaster && caller.UserName == user.UserName {
- selfUpdate = true
- }
- if !ismaster && !selfUpdate {
- if caller.IsAdmin && user.IsSuperAdmin {
- slog.Error("non-superadmin user", "caller", caller.UserName, "attempted to update superadmin user", username)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
- return
- }
- if !caller.IsAdmin && !caller.IsSuperAdmin {
- slog.Error("operation not allowed", "caller", caller.UserName, "attempted to update user", username)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
- return
- }
- if caller.IsAdmin && user.IsAdmin {
- slog.Error("admin user cannot update another admin", "caller", caller.UserName, "attempted to update admin user", username)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("admin user cannot update another admin"), "forbidden"))
- return
- }
- if caller.IsAdmin && userchange.IsAdmin {
- err = errors.New("admin user cannot update role of an another user to admin")
- slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
- return
- }
- }
- if !ismaster && selfUpdate {
- if user.IsAdmin != userchange.IsAdmin || user.IsSuperAdmin != userchange.IsSuperAdmin {
- slog.Error("user cannot change his own role", "caller", caller.UserName, "attempted to update user role", username)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user not allowed to self assign role"), "forbidden"))
- return
- }
- }
- if ismaster {
- if !user.IsSuperAdmin && userchange.IsSuperAdmin {
- slog.Error("operation not allowed", "caller", logic.MasterUser, "attempted to update user role to superadmin", username)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("attempted to update user role to superadmin"), "forbidden"))
- return
- }
- }
- if auth.IsOauthUser(user) == nil && userchange.Password != "" {
- err := fmt.Errorf("cannot update user's password for an oauth user %s", username)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
- return
- }
- user, err = logic.UpdateUser(&userchange, user)
- if err != nil {
- logger.Log(0, username,
- "failed to update user info: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- logger.Log(1, username, "was updated")
- json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
- }
- // swagger:route DELETE /api/users/{username} user deleteUser
- //
- // Delete a user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func deleteUser(w http.ResponseWriter, r *http.Request) {
- // Set header
- w.Header().Set("Content-Type", "application/json")
- // get params
- var params = mux.Vars(r)
- caller, err := logic.GetUser(r.Header.Get("user"))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- }
- username := params["username"]
- user, err := logic.GetUser(username)
- if err != nil {
- logger.Log(0, username,
- "failed to update user info: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- if user.IsSuperAdmin {
- slog.Error(
- "failed to delete user: ", "user", username, "error", "superadmin cannot be deleted")
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("superadmin cannot be deleted"), "internal"))
- return
- }
- if !caller.IsSuperAdmin {
- if caller.IsAdmin && user.IsAdmin {
- slog.Error(
- "failed to delete user: ", "user", username, "error", "admin cannot delete another admin user, including oneself")
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("admin cannot delete another admin user, including oneself"), "internal"))
- return
- }
- }
- success, err := logic.DeleteUser(username)
- if err != nil {
- logger.Log(0, username,
- "failed to delete user: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- } else if !success {
- err := errors.New("delete unsuccessful")
- logger.Log(0, username, err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- // check and delete extclient with this ownerID
- go func() {
- extclients, err := logic.GetAllExtClients()
- if err != nil {
- slog.Error("failed to get extclients", "error", err)
- return
- }
- for _, extclient := range extclients {
- if extclient.OwnerID == user.UserName {
- err = logic.DeleteExtClient(extclient.Network, extclient.ClientID)
- if err != nil {
- slog.Error("failed to delete extclient",
- "id", extclient.ClientID, "owner", user.UserName, "error", err)
- }
- }
- }
- if servercfg.IsDNSMode() {
- logic.SetDNS()
- }
- }()
- logger.Log(1, username, "was deleted")
- json.NewEncoder(w).Encode(params["username"] + " deleted.")
- }
- // Called when vpn client dials in to start the auth flow and first stage is to get register URL itself
- func socketHandler(w http.ResponseWriter, r *http.Request) {
- // Upgrade our raw HTTP connection to a websocket based one
- conn, err := upgrader.Upgrade(w, r, nil)
- if err != nil {
- logger.Log(0, "error during connection upgrade for node sign-in:", err.Error())
- return
- }
- if conn == nil {
- logger.Log(0, "failed to establish web-socket connection during node sign-in")
- return
- }
- // Start handling the session
- go auth.SessionHandler(conn)
- }
- // swagger:route GET /api/users_pending user getPendingUsers
- //
- // Get all pending users.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func getPendingUsers(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- users, err := logic.ListPendingUsers()
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.SortUsers(users[:])
- logger.Log(2, r.Header.Get("user"), "fetched pending users")
- json.NewEncoder(w).Encode(users)
- }
- // swagger:route POST /api/users_pending/user/{username} user approvePendingUser
- //
- // approve pending user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func approvePendingUser(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- username := params["username"]
- users, err := logic.ListPendingUsers()
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- for _, user := range users {
- if user.UserName == username {
- var newPass, fetchErr = auth.FetchPassValue("")
- if fetchErr != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
- return
- }
- if err = logic.CreateUser(&models.User{
- UserName: user.UserName,
- Password: newPass,
- }); err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to create user: %s", err), "internal"))
- return
- }
- err = logic.DeletePendingUser(username)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
- return
- }
- break
- }
- }
- logic.ReturnSuccessResponse(w, r, "approved "+username)
- }
- // swagger:route DELETE /api/users_pending/user/{username} user deletePendingUser
- //
- // delete pending user.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func deletePendingUser(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- username := params["username"]
- users, err := logic.ListPendingUsers()
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- for _, user := range users {
- if user.UserName == username {
- err = logic.DeletePendingUser(username)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
- return
- }
- break
- }
- }
- logic.ReturnSuccessResponse(w, r, "deleted pending "+username)
- }
- // swagger:route DELETE /api/users_pending/{username}/pending user deleteAllPendingUsers
- //
- // delete all pending users.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func deleteAllPendingUsers(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- err := database.DeleteAllRecords(database.PENDING_USERS_TABLE_NAME)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending users "+err.Error()), "internal"))
- return
- }
- logic.ReturnSuccessResponse(w, r, "cleared all pending users")
- }
|