12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121 |
- package controllers
- import (
- "context"
- "encoding/json"
- "errors"
- "fmt"
- "net"
- "net/http"
- "net/url"
- "strings"
- "github.com/gorilla/mux"
- "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"
- proAuth "github.com/gravitl/netmaker/pro/auth"
- "github.com/gravitl/netmaker/pro/email"
- proLogic "github.com/gravitl/netmaker/pro/logic"
- "github.com/gravitl/netmaker/servercfg"
- "golang.org/x/exp/slog"
- )
- func UserHandlers(r *mux.Router) {
- r.HandleFunc("/api/oauth/login", proAuth.HandleAuthLogin).Methods(http.MethodGet)
- r.HandleFunc("/api/oauth/callback", proAuth.HandleAuthCallback).Methods(http.MethodGet)
- r.HandleFunc("/api/oauth/headless", proAuth.HandleHeadlessSSO)
- r.HandleFunc("/api/oauth/register/{regKey}", proAuth.RegisterHostSSO).Methods(http.MethodGet)
- // User Role Handlers
- r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(getRole))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(createRole))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(updateRole))).Methods(http.MethodPut)
- r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(deleteRole))).Methods(http.MethodDelete)
- // User Group Handlers
- r.HandleFunc("/api/v1/users/groups", logic.SecurityCheck(true, http.HandlerFunc(listUserGroups))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(getUserGroup))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(createUserGroup))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(updateUserGroup))).Methods(http.MethodPut)
- r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(deleteUserGroup))).Methods(http.MethodDelete)
- // User Invite Handlers
- r.HandleFunc("/api/v1/users/invite", userInviteVerify).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/users/invite-signup", userInviteSignUp).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(inviteUsers))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(listUserInvites))).Methods(http.MethodGet)
- r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(deleteUserInvite))).Methods(http.MethodDelete)
- r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(deleteAllUserInvites))).Methods(http.MethodDelete)
- 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)
- r.HandleFunc("/api/users/{username}/remote_access_gw/{remote_access_gateway_id}", logic.SecurityCheck(true, http.HandlerFunc(attachUserToRemoteAccessGw))).Methods(http.MethodPost)
- r.HandleFunc("/api/users/{username}/remote_access_gw/{remote_access_gateway_id}", logic.SecurityCheck(true, http.HandlerFunc(removeUserFromRemoteAccessGW))).Methods(http.MethodDelete)
- r.HandleFunc("/api/users/{username}/remote_access_gw", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUserRemoteAccessGwsV1)))).Methods(http.MethodGet)
- r.HandleFunc("/api/users/ingress/{ingress_id}", logic.SecurityCheck(true, http.HandlerFunc(ingressGatewayUsers))).Methods(http.MethodGet)
- }
- // swagger:route POST /api/v1/users/invite-signup user userInviteSignUp
- //
- // user signup via invite.
- //
- // Schemes: https
- //
- // Responses:
- // 200: ReturnSuccessResponse
- func userInviteSignUp(w http.ResponseWriter, r *http.Request) {
- email, _ := url.QueryUnescape(r.URL.Query().Get("email"))
- code, _ := url.QueryUnescape(r.URL.Query().Get("invite_code"))
- in, err := logic.GetUserInvite(email)
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if code != in.InviteCode {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid invite code"), "badrequest"))
- return
- }
- // check if user already exists
- _, err = logic.GetUser(email)
- if err == nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user already exists"), "badrequest"))
- 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 user.UserName != email {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username not matching with invite"), "badrequest"))
- return
- }
- if user.Password == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("password cannot be empty"), "badrequest"))
- return
- }
- user.UserGroups = in.UserGroups
- user.PlatformRoleID = models.UserRoleID(in.PlatformRoleID)
- if user.PlatformRoleID == "" {
- user.PlatformRoleID = models.ServiceUser
- }
- user.NetworkRoles = in.NetworkRoles
- err = logic.CreateUser(&user)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- // delete invite
- logic.DeleteUserInvite(email)
- logic.DeletePendingUser(email)
- w.Header().Set("Access-Control-Allow-Origin", "*")
- logic.ReturnSuccessResponse(w, r, "created user successfully "+email)
- }
- // swagger:route GET /api/v1/users/invite user userInviteVerify
- //
- // verfies user invite.
- //
- // Schemes: https
- //
- // Responses:
- // 200: ReturnSuccessResponse
- func userInviteVerify(w http.ResponseWriter, r *http.Request) {
- email, _ := url.QueryUnescape(r.URL.Query().Get("email"))
- code, _ := url.QueryUnescape(r.URL.Query().Get("invite_code"))
- err := logic.ValidateAndApproveUserInvite(email, code)
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponse(w, r, "invite is valid")
- }
- // swagger:route POST /api/v1/users/invite user inviteUsers
- //
- // invite users.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: userBodyResponse
- func inviteUsers(w http.ResponseWriter, r *http.Request) {
- var inviteReq models.InviteUsersReq
- err := json.NewDecoder(r.Body).Decode(&inviteReq)
- if err != nil {
- slog.Error("error decoding request body", "error",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- callerUserName := r.Header.Get("user")
- if r.Header.Get("ismaster") != "yes" {
- caller, err := logic.GetUser(callerUserName)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
- return
- }
- if inviteReq.PlatformRoleID == models.SuperAdminRole.String() {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("super admin cannot be invited"), "badrequest"))
- return
- }
- if inviteReq.PlatformRoleID == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("platform role id cannot be empty"), "badrequest"))
- return
- }
- if (inviteReq.PlatformRoleID == models.AdminRole.String() ||
- inviteReq.PlatformRoleID == models.SuperAdminRole.String()) && caller.PlatformRoleID != models.SuperAdminRole {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can invite admin users"), "forbidden"))
- return
- }
- }
- //validate Req
- err = proLogic.IsGroupsValid(inviteReq.UserGroups)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.IsNetworkRolesValid(inviteReq.NetworkRoles)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- // check platform role
- _, err = logic.GetRole(models.UserRoleID(inviteReq.PlatformRoleID))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- for _, inviteeEmail := range inviteReq.UserEmails {
- // check if user with email exists, then ignore
- if !email.IsValid(inviteeEmail) {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid email "+inviteeEmail), "badrequest"))
- return
- }
- _, err := logic.GetUser(inviteeEmail)
- if err == nil {
- // user exists already, so ignore
- continue
- }
- invite := models.UserInvite{
- Email: inviteeEmail,
- PlatformRoleID: inviteReq.PlatformRoleID,
- UserGroups: inviteReq.UserGroups,
- NetworkRoles: inviteReq.NetworkRoles,
- InviteCode: logic.RandomString(8),
- }
- frontendURL := strings.TrimSuffix(servercfg.GetFrontendURL(), "/")
- if frontendURL == "" {
- frontendURL = fmt.Sprintf("https://dashboard.%s", servercfg.GetNmBaseDomain())
- }
- u, err := url.Parse(fmt.Sprintf("%s/invite?email=%s&invite_code=%s",
- frontendURL, url.QueryEscape(invite.Email), url.QueryEscape(invite.InviteCode)))
- if err != nil {
- slog.Error("failed to parse to invite url", "error", err)
- return
- }
- if servercfg.DeployedByOperator() {
- u, err = url.Parse(fmt.Sprintf("%s/invite?tenant_id=%s&email=%s&invite_code=%s",
- proLogic.GetAccountsUIHost(), url.QueryEscape(servercfg.GetNetmakerTenantID()), url.QueryEscape(invite.Email), url.QueryEscape(invite.InviteCode)))
- if err != nil {
- slog.Error("failed to parse to invite url", "error", err)
- return
- }
- }
- invite.InviteURL = u.String()
- err = logic.InsertUserInvite(invite)
- if err != nil {
- slog.Error("failed to insert invite for user", "email", invite.Email, "error", err)
- }
- // notify user with magic link
- go func(invite models.UserInvite) {
- // Set E-Mail body. You can set plain text or html with text/html
- e := email.UserInvitedMail{
- BodyBuilder: &email.EmailBodyBuilderWithH1HeadlineAndImage{},
- InviteURL: invite.InviteURL,
- }
- n := email.Notification{
- RecipientMail: invite.Email,
- }
- err = email.GetClient().SendEmail(context.Background(), n, e)
- if err != nil {
- slog.Error("failed to send email invite", "user", invite.Email, "error", err)
- }
- }(invite)
- }
- }
- // swagger:route GET /api/v1/users/invites user listUserInvites
- //
- // lists all pending invited users.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: ReturnSuccessResponseWithJson
- func listUserInvites(w http.ResponseWriter, r *http.Request) {
- usersInvites, err := logic.ListUserInvites()
- if err != nil {
- logger.Log(0, "failed to fetch users: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, usersInvites, "fetched pending user invites")
- }
- // swagger:route DELETE /api/v1/users/invite user deleteUserInvite
- //
- // delete pending invite.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: ReturnSuccessResponse
- func deleteUserInvite(w http.ResponseWriter, r *http.Request) {
- email, _ := url.QueryUnescape(r.URL.Query().Get("invitee_email"))
- err := logic.DeleteUserInvite(email)
- if err != nil {
- logger.Log(0, "failed to delete user invite: ", email, err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponse(w, r, "deleted user invite")
- }
- // swagger:route DELETE /api/v1/users/invites user deleteAllUserInvites
- //
- // deletes all pending invites.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: ReturnSuccessResponse
- func deleteAllUserInvites(w http.ResponseWriter, r *http.Request) {
- err := database.DeleteAllRecords(database.USER_INVITES_TABLE_NAME)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending user invites "+err.Error()), "internal"))
- return
- }
- logic.ReturnSuccessResponse(w, r, "cleared all pending user invites")
- }
- // 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 := proLogic.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) {
- gid, _ := url.QueryUnescape(r.URL.Query().Get("group_id"))
- if gid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
- return
- }
- group, err := proLogic.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 userGroupReq models.CreateGroupReq
- err := json.NewDecoder(r.Body).Decode(&userGroupReq)
- if err != nil {
- slog.Error("error decoding request body", "error",
- err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.ValidateCreateGroupReq(userGroupReq.Group)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.CreateUserGroup(userGroupReq.Group)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- for _, userID := range userGroupReq.Members {
- user, err := logic.GetUser(userID)
- if err != nil {
- continue
- }
- if len(user.UserGroups) == 0 {
- user.UserGroups = make(map[models.UserGroupID]struct{})
- }
- user.UserGroups[userGroupReq.Group.ID] = struct{}{}
- logic.UpsertUser(*user)
- }
- logic.ReturnSuccessResponseWithJson(w, r, userGroupReq.Group, "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
- }
- // fetch curr group
- currUserG, err := proLogic.GetUserGroup(userGroup.ID)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.ValidateUpdateGroupReq(userGroup)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.UpdateUserGroup(userGroup)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- // reset configs for service user
- go proLogic.UpdatesUserGwAccessOnGrpUpdates(currUserG.NetworkRoles, userGroup.NetworkRoles)
- 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
- //
- // @Summary Delete user group.
- // @Router /api/v1/user/group [delete]
- // @Tags Users
- // @Param group_id param string true "group id required to delete the role"
- // @Success 200 {string} string
- // @Failure 500 {object} models.ErrorResponse
- func deleteUserGroup(w http.ResponseWriter, r *http.Request) {
- gid, _ := url.QueryUnescape(r.URL.Query().Get("group_id"))
- if gid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- userG, err := proLogic.GetUserGroup(models.UserGroupID(gid))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- err = proLogic.DeleteUserGroup(models.UserGroupID(gid))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- go proLogic.UpdatesUserGwAccessOnGrpUpdates(userG.NetworkRoles, make(map[models.NetworkID]map[models.UserRoleID]struct{}))
- logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user group")
- }
- // @Summary lists all user roles.
- // @Router /api/v1/user/roles [get]
- // @Tags Users
- // @Param role_id param string true "roleid required to get the role details"
- // @Success 200 {object} []models.UserRolePermissionTemplate
- // @Failure 500 {object} models.ErrorResponse
- func ListRoles(w http.ResponseWriter, r *http.Request) {
- platform, _ := url.QueryUnescape(r.URL.Query().Get("platform"))
- var roles []models.UserRolePermissionTemplate
- var err error
- if platform == "true" {
- roles, err = logic.ListPlatformRoles()
- } else {
- roles, err = proLogic.ListNetworkRoles()
- }
- 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")
- }
- // @Summary Get user role permission template.
- // @Router /api/v1/user/role [get]
- // @Tags Users
- // @Param role_id param string true "roleid required to get the role details"
- // @Success 200 {object} models.UserRolePermissionTemplate
- // @Failure 500 {object} models.ErrorResponse
- func getRole(w http.ResponseWriter, r *http.Request) {
- rid, _ := url.QueryUnescape(r.URL.Query().Get("role_id"))
- if rid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- role, err := logic.GetRole(models.UserRoleID(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")
- }
- // @Summary Create user role permission template.
- // @Router /api/v1/user/role [post]
- // @Tags Users
- // @Param body models.UserRolePermissionTemplate true "user role template"
- // @Success 200 {object} models.UserRolePermissionTemplate
- // @Failure 500 {object} models.ErrorResponse
- 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 = proLogic.ValidateCreateRoleReq(&userRole)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- userRole.Default = false
- userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
- err = proLogic.CreateRole(userRole)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, userRole, "created user role")
- }
- // @Summary Update user role permission template.
- // @Router /api/v1/user/role [put]
- // @Tags Users
- // @Param body models.UserRolePermissionTemplate true "user role template"
- // @Success 200 {object} userBodyResponse
- // @Failure 500 {object} models.ErrorResponse
- 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
- }
- currRole, err := logic.GetRole(userRole.ID)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.ValidateUpdateRoleReq(&userRole)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
- err = proLogic.UpdateRole(userRole)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- // reset configs for service user
- go proLogic.UpdatesUserGwAccessOnRoleUpdates(currRole.NetworkLevelAccess, userRole.NetworkLevelAccess, string(userRole.NetworkID))
- logic.ReturnSuccessResponseWithJson(w, r, userRole, "updated user role")
- }
- // @Summary Delete user role permission template.
- // @Router /api/v1/user/role [delete]
- // @Tags Users
- // @Param role_id param string true "roleid required to delete the role"
- // @Success 200 {string} string
- // @Failure 500 {object} models.ErrorResponse
- func deleteRole(w http.ResponseWriter, r *http.Request) {
- rid, _ := url.QueryUnescape(r.URL.Query().Get("role_id"))
- if rid == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- role, err := logic.GetRole(models.UserRoleID(rid))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
- return
- }
- err = proLogic.DeleteRole(models.UserRoleID(rid), false)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- go proLogic.UpdatesUserGwAccessOnRoleUpdates(role.NetworkLevelAccess, make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope), role.NetworkID.String())
- logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user role")
- }
- // @Summary Attach user to a remote access gateway
- // @Router /api/users/{username}/remote_access_gw/{remote_access_gateway_id} [post]
- // @Tags PRO
- // @Accept json
- // @Produce json
- // @Param username path string true "Username"
- // @Param remote_access_gateway_id path string true "Remote Access Gateway ID"
- // @Success 200 {object} models.ReturnUser
- // @Failure 400 {object} models.ErrorResponse
- // @Failure 500 {object} models.ErrorResponse
- func attachUserToRemoteAccessGw(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- username := params["username"]
- remoteGwID := params["remote_access_gateway_id"]
- if username == "" || remoteGwID == "" {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- errors.New("required params `username` and `remote_access_gateway_id`"),
- "badrequest",
- ),
- )
- return
- }
- user, err := logic.GetUser(username)
- if err != nil {
- slog.Error("failed to fetch user: ", "username", username, "error", err.Error())
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- fmt.Errorf("failed to fetch user %s, error: %v", username, err),
- "badrequest",
- ),
- )
- return
- }
- if user.PlatformRoleID == models.AdminRole || user.PlatformRoleID == models.SuperAdminRole {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("superadmins/admins have access to all gateways"), "badrequest"))
- return
- }
- node, err := logic.GetNodeByID(remoteGwID)
- if err != nil {
- slog.Error("failed to fetch gateway node", "nodeID", remoteGwID, "error", err)
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- fmt.Errorf("failed to fetch remote access gateway node, error: %v", err),
- "badrequest",
- ),
- )
- return
- }
- if !node.IsIngressGateway {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(fmt.Errorf("node is not a remote access gateway"), "badrequest"),
- )
- return
- }
- if user.RemoteGwIDs == nil {
- user.RemoteGwIDs = make(map[string]struct{})
- }
- user.RemoteGwIDs[node.ID.String()] = struct{}{}
- err = logic.UpsertUser(*user)
- if err != nil {
- slog.Error("failed to update user's gateways", "user", username, "error", err)
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- fmt.Errorf("failed to fetch remote access gateway node,error: %v", err),
- "badrequest",
- ),
- )
- return
- }
- json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
- }
- // @Summary Remove user from a remote access gateway
- // @Router /api/users/{username}/remote_access_gw/{remote_access_gateway_id} [delete]
- // @Tags PRO
- // @Accept json
- // @Produce json
- // @Param username path string true "Username"
- // @Param remote_access_gateway_id path string true "Remote Access Gateway ID"
- // @Success 200 {object} models.ReturnUser
- // @Failure 400 {object} models.ErrorResponse
- // @Failure 500 {object} models.ErrorResponse
- func removeUserFromRemoteAccessGW(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- username := params["username"]
- remoteGwID := params["remote_access_gateway_id"]
- if username == "" || remoteGwID == "" {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- errors.New("required params `username` and `remote_access_gateway_id`"),
- "badrequest",
- ),
- )
- return
- }
- user, err := logic.GetUser(username)
- if err != nil {
- logger.Log(0, username, "failed to fetch user: ", err.Error())
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- fmt.Errorf("failed to fetch user %s, error: %v", username, err),
- "badrequest",
- ),
- )
- return
- }
- delete(user.RemoteGwIDs, remoteGwID)
- go func(user models.User, remoteGwID string) {
- extclients, err := logic.GetAllExtClients()
- if err != nil {
- slog.Error("failed to fetch extclients", "error", err)
- return
- }
- for _, extclient := range extclients {
- if extclient.OwnerID == user.UserName && remoteGwID == extclient.IngressGatewayID {
- err = logic.DeleteExtClientAndCleanup(extclient)
- if err != nil {
- slog.Error("failed to delete extclient",
- "id", extclient.ClientID, "owner", user.UserName, "error", err)
- } else {
- if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
- slog.Error("error setting ext peers: " + err.Error())
- }
- }
- }
- }
- if servercfg.IsDNSMode() {
- logic.SetDNS()
- }
- }(*user, remoteGwID)
- err = logic.UpsertUser(*user)
- if err != nil {
- slog.Error("failed to update user gateways", "user", username, "error", err)
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- errors.New("failed to fetch remote access gaetway node "+err.Error()),
- "badrequest",
- ),
- )
- return
- }
- json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
- }
- // @Summary Get Users Remote Access Gw.
- // @Router /api/users/{username}/remote_access_gw [get]
- // @Tags Users
- // @Param username path string true "Username to fetch all the gateways with access"
- // @Success 200 {object} map[string][]models.UserRemoteGws
- // @Failure 500 {object} models.ErrorResponse
- func getUserRemoteAccessGwsV1(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- username := params["username"]
- if username == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params username"), "badrequest"))
- return
- }
- user, err := logic.GetUser(username)
- if err != nil {
- logger.Log(0, username, "failed to fetch user: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
- return
- }
- remoteAccessClientID := r.URL.Query().Get("remote_access_clientid")
- var req models.UserRemoteGwsReq
- if remoteAccessClientID == "" {
- err := json.NewDecoder(r.Body).Decode(&req)
- if err != nil {
- slog.Error("error decoding request body: ", "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- }
- reqFromMobile := r.URL.Query().Get("from_mobile") == "true"
- if req.RemoteAccessClientID == "" && remoteAccessClientID == "" {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("remote access client id cannot be empty"), "badrequest"))
- return
- }
- if req.RemoteAccessClientID == "" {
- req.RemoteAccessClientID = remoteAccessClientID
- }
- userGws := make(map[string][]models.UserRemoteGws)
- allextClients, err := logic.GetAllExtClients()
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- userGwNodes := proLogic.GetUserRAGNodes(*user)
- for _, extClient := range allextClients {
- node, ok := userGwNodes[extClient.IngressGatewayID]
- if !ok {
- continue
- }
- if extClient.RemoteAccessClientID == req.RemoteAccessClientID && extClient.OwnerID == username {
- host, err := logic.GetHost(node.HostID.String())
- if err != nil {
- continue
- }
- network, err := logic.GetNetwork(node.Network)
- if err != nil {
- slog.Error("failed to get node network", "error", err)
- }
- gws := userGws[node.Network]
- if net.ParseIP(extClient.IngressGatewayEndpoint).To16() != nil {
- extClient.IngressGatewayEndpoint = fmt.Sprintf("[%s]", extClient.IngressGatewayEndpoint)
- }
- extClient.AllowedIPs = logic.GetExtclientAllowedIPs(extClient)
- gws = append(gws, models.UserRemoteGws{
- GwID: node.ID.String(),
- GWName: host.Name,
- Network: node.Network,
- GwClient: extClient,
- Connected: true,
- IsInternetGateway: node.IsInternetGateway,
- GwPeerPublicKey: host.PublicKey.String(),
- GwListenPort: logic.GetPeerListenPort(host),
- Metadata: node.Metadata,
- AllowedEndpoints: getAllowedRagEndpoints(&node, host),
- NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
- })
- userGws[node.Network] = gws
- delete(userGwNodes, node.ID.String())
- }
- }
- // add remaining gw nodes to resp
- for gwID := range userGwNodes {
- node, err := logic.GetNodeByID(gwID)
- if err != nil {
- continue
- }
- if !node.IsIngressGateway {
- continue
- }
- if node.PendingDelete {
- continue
- }
- host, err := logic.GetHost(node.HostID.String())
- if err != nil {
- continue
- }
- network, err := logic.GetNetwork(node.Network)
- if err != nil {
- slog.Error("failed to get node network", "error", err)
- }
- gws := userGws[node.Network]
- gws = append(gws, models.UserRemoteGws{
- GwID: node.ID.String(),
- GWName: host.Name,
- Network: node.Network,
- IsInternetGateway: node.IsInternetGateway,
- GwPeerPublicKey: host.PublicKey.String(),
- GwListenPort: logic.GetPeerListenPort(host),
- Metadata: node.Metadata,
- AllowedEndpoints: getAllowedRagEndpoints(&node, host),
- NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
- })
- userGws[node.Network] = gws
- }
- if reqFromMobile {
- // send resp in array format
- userGwsArr := []models.UserRemoteGws{}
- for _, userGwI := range userGws {
- userGwsArr = append(userGwsArr, userGwI...)
- }
- logic.ReturnSuccessResponseWithJson(w, r, userGwsArr, "fetched gateways for user"+username)
- return
- }
- slog.Debug("returned user gws", "user", username, "gws", userGws)
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(userGws)
- }
- // @Summary List users attached to an remote access gateway
- // @Router /api/nodes/{network}/{nodeid}/ingress/users [get]
- // @Tags PRO
- // @Accept json
- // @Produce json
- // @Param ingress_id path string true "Ingress Gateway ID"
- // @Success 200 {array} models.IngressGwUsers
- // @Failure 400 {object} models.ErrorResponse
- // @Failure 500 {object} models.ErrorResponse
- func ingressGatewayUsers(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- ingressID := params["ingress_id"]
- node, err := logic.GetNodeByID(ingressID)
- if err != nil {
- slog.Error("failed to get ingress node", "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- gwUsers, err := logic.GetIngressGwUsers(node)
- if err != nil {
- slog.Error(
- "failed to get users on ingress gateway",
- "nodeid",
- ingressID,
- "network",
- node.Network,
- "user",
- r.Header.Get("user"),
- "error",
- err,
- )
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(gwUsers)
- }
- func getAllowedRagEndpoints(ragNode *models.Node, ragHost *models.Host) []string {
- endpoints := []string{}
- if len(ragHost.EndpointIP) > 0 {
- endpoints = append(endpoints, ragHost.EndpointIP.String())
- }
- if len(ragHost.EndpointIPv6) > 0 {
- endpoints = append(endpoints, fmt.Sprintf("[%s]", ragHost.EndpointIPv6.String()))
- }
- if servercfg.IsPro {
- for _, ip := range ragNode.AdditionalRagIps {
- if ip.To16() != nil {
- endpoints = append(endpoints, fmt.Sprintf("[%s]", ip.String()))
- } else {
- endpoints = append(endpoints, ip.String())
- }
- }
- }
- return endpoints
- }
- // @Summary Get all pending users
- // @Router /api/users_pending [get]
- // @Tags Users
- // @Success 200 {array} models.User
- // @Failure 500 {object} models.ErrorResponse
- 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)
- }
- // @Summary Approve a pending user
- // @Router /api/users_pending/user/{username} [post]
- // @Tags Users
- // @Param username path string true "Username of the pending user to approve"
- // @Success 200 {string} string
- // @Failure 500 {object} models.ErrorResponse
- 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 = logic.FetchPassValue("")
- if fetchErr != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
- return
- }
- if err = logic.CreateUser(&models.User{
- UserName: user.UserName,
- Password: newPass,
- PlatformRoleID: models.ServiceUser,
- }); 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)
- }
- // @Summary Delete a pending user
- // @Router /api/users_pending/user/{username} [delete]
- // @Tags Users
- // @Param username path string true "Username of the pending user to delete"
- // @Success 200 {string} string
- // @Failure 500 {object} models.ErrorResponse
- 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)
- }
- // @Summary Delete all pending users
- // @Router /api/users_pending [delete]
- // @Tags Users
- // @Success 200 {string} string
- // @Failure 500 {object} models.ErrorResponse
- func deleteAllPendingUsers(w http.ResponseWriter, r *http.Request) {
- // set header.
- 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")
- }
|