users.go 44 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334
  1. package controllers
  2. import (
  3. "context"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "net/http"
  8. "net/url"
  9. "strings"
  10. "github.com/gorilla/mux"
  11. "github.com/gravitl/netmaker/database"
  12. "github.com/gravitl/netmaker/logger"
  13. "github.com/gravitl/netmaker/logic"
  14. "github.com/gravitl/netmaker/models"
  15. "github.com/gravitl/netmaker/mq"
  16. proAuth "github.com/gravitl/netmaker/pro/auth"
  17. "github.com/gravitl/netmaker/pro/email"
  18. proLogic "github.com/gravitl/netmaker/pro/logic"
  19. "github.com/gravitl/netmaker/servercfg"
  20. "golang.org/x/exp/slog"
  21. )
  22. func UserHandlers(r *mux.Router) {
  23. r.HandleFunc("/api/oauth/login", proAuth.HandleAuthLogin).Methods(http.MethodGet)
  24. r.HandleFunc("/api/oauth/callback", proAuth.HandleAuthCallback).Methods(http.MethodGet)
  25. r.HandleFunc("/api/oauth/headless", proAuth.HandleHeadlessSSO)
  26. r.HandleFunc("/api/oauth/register/{regKey}", proAuth.RegisterHostSSO).Methods(http.MethodGet)
  27. // User Role Handlers
  28. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(getRole))).Methods(http.MethodGet)
  29. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(createRole))).Methods(http.MethodPost)
  30. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(updateRole))).Methods(http.MethodPut)
  31. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(deleteRole))).Methods(http.MethodDelete)
  32. // User Group Handlers
  33. r.HandleFunc("/api/v1/users/groups", logic.SecurityCheck(true, http.HandlerFunc(listUserGroups))).Methods(http.MethodGet)
  34. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(getUserGroup))).Methods(http.MethodGet)
  35. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(createUserGroup))).Methods(http.MethodPost)
  36. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(updateUserGroup))).Methods(http.MethodPut)
  37. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(deleteUserGroup))).Methods(http.MethodDelete)
  38. // User Invite Handlers
  39. r.HandleFunc("/api/v1/users/invite", userInviteVerify).Methods(http.MethodGet)
  40. r.HandleFunc("/api/v1/users/invite-signup", userInviteSignUp).Methods(http.MethodPost)
  41. r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(inviteUsers))).Methods(http.MethodPost)
  42. r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(listUserInvites))).Methods(http.MethodGet)
  43. r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(deleteUserInvite))).Methods(http.MethodDelete)
  44. r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(deleteAllUserInvites))).Methods(http.MethodDelete)
  45. r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(getPendingUsers))).Methods(http.MethodGet)
  46. r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(deleteAllPendingUsers))).Methods(http.MethodDelete)
  47. r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(deletePendingUser))).Methods(http.MethodDelete)
  48. r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(approvePendingUser))).Methods(http.MethodPost)
  49. r.HandleFunc("/api/users/{username}/remote_access_gw/{remote_access_gateway_id}", logic.SecurityCheck(true, http.HandlerFunc(attachUserToRemoteAccessGw))).Methods(http.MethodPost)
  50. r.HandleFunc("/api/users/{username}/remote_access_gw/{remote_access_gateway_id}", logic.SecurityCheck(true, http.HandlerFunc(removeUserFromRemoteAccessGW))).Methods(http.MethodDelete)
  51. r.HandleFunc("/api/users/{username}/remote_access_gw", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUserRemoteAccessGwsV1)))).Methods(http.MethodGet)
  52. r.HandleFunc("/api/users/ingress/{ingress_id}", logic.SecurityCheck(true, http.HandlerFunc(ingressGatewayUsers))).Methods(http.MethodGet)
  53. }
  54. // swagger:route POST /api/v1/users/invite-signup user userInviteSignUp
  55. //
  56. // user signup via invite.
  57. //
  58. // Schemes: https
  59. //
  60. // Responses:
  61. // 200: ReturnSuccessResponse
  62. func userInviteSignUp(w http.ResponseWriter, r *http.Request) {
  63. email := r.URL.Query().Get("email")
  64. code := r.URL.Query().Get("invite_code")
  65. in, err := logic.GetUserInvite(email)
  66. if err != nil {
  67. logger.Log(0, "failed to fetch users: ", err.Error())
  68. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  69. return
  70. }
  71. if code != in.InviteCode {
  72. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid invite code"), "badrequest"))
  73. return
  74. }
  75. // check if user already exists
  76. _, err = logic.GetUser(email)
  77. if err == nil {
  78. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user already exists"), "badrequest"))
  79. return
  80. }
  81. var user models.User
  82. err = json.NewDecoder(r.Body).Decode(&user)
  83. if err != nil {
  84. logger.Log(0, user.UserName, "error decoding request body: ",
  85. err.Error())
  86. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  87. return
  88. }
  89. if user.UserName != email {
  90. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username not matching with invite"), "badrequest"))
  91. return
  92. }
  93. if user.Password == "" {
  94. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("password cannot be empty"), "badrequest"))
  95. return
  96. }
  97. user.UserGroups = in.UserGroups
  98. user.PlatformRoleID = models.UserRoleID(in.PlatformRoleID)
  99. if user.PlatformRoleID == "" {
  100. user.PlatformRoleID = models.ServiceUser
  101. }
  102. user.NetworkRoles = in.NetworkRoles
  103. err = logic.CreateUser(&user)
  104. if err != nil {
  105. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  106. return
  107. }
  108. // delete invite
  109. logic.DeleteUserInvite(email)
  110. logic.DeletePendingUser(email)
  111. w.Header().Set("Access-Control-Allow-Origin", "*")
  112. logic.ReturnSuccessResponse(w, r, "created user successfully "+email)
  113. }
  114. // swagger:route GET /api/v1/users/invite user userInviteVerify
  115. //
  116. // verfies user invite.
  117. //
  118. // Schemes: https
  119. //
  120. // Responses:
  121. // 200: ReturnSuccessResponse
  122. func userInviteVerify(w http.ResponseWriter, r *http.Request) {
  123. email := r.URL.Query().Get("email")
  124. code := r.URL.Query().Get("invite_code")
  125. err := logic.ValidateAndApproveUserInvite(email, code)
  126. if err != nil {
  127. logger.Log(0, "failed to fetch users: ", err.Error())
  128. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  129. return
  130. }
  131. logic.ReturnSuccessResponse(w, r, "invite is valid")
  132. }
  133. // swagger:route POST /api/v1/users/invite user inviteUsers
  134. //
  135. // invite users.
  136. //
  137. // Schemes: https
  138. //
  139. // Security:
  140. // oauth
  141. //
  142. // Responses:
  143. // 200: userBodyResponse
  144. func inviteUsers(w http.ResponseWriter, r *http.Request) {
  145. var inviteReq models.InviteUsersReq
  146. err := json.NewDecoder(r.Body).Decode(&inviteReq)
  147. if err != nil {
  148. slog.Error("error decoding request body", "error",
  149. err.Error())
  150. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  151. return
  152. }
  153. callerUserName := r.Header.Get("user")
  154. if r.Header.Get("ismaster") != "yes" {
  155. caller, err := logic.GetUser(callerUserName)
  156. if err != nil {
  157. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
  158. return
  159. }
  160. if inviteReq.PlatformRoleID == models.SuperAdminRole.String() {
  161. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("super admin cannot be invited"), "badrequest"))
  162. return
  163. }
  164. if inviteReq.PlatformRoleID == "" {
  165. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("platform role id cannot be empty"), "badrequest"))
  166. return
  167. }
  168. if (inviteReq.PlatformRoleID == models.AdminRole.String() ||
  169. inviteReq.PlatformRoleID == models.SuperAdminRole.String()) && caller.PlatformRoleID != models.SuperAdminRole {
  170. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can invite admin users"), "forbidden"))
  171. return
  172. }
  173. }
  174. //validate Req
  175. err = proLogic.IsGroupsValid(inviteReq.UserGroups)
  176. if err != nil {
  177. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  178. return
  179. }
  180. err = proLogic.IsNetworkRolesValid(inviteReq.NetworkRoles)
  181. if err != nil {
  182. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  183. return
  184. }
  185. // check platform role
  186. _, err = logic.GetRole(models.UserRoleID(inviteReq.PlatformRoleID))
  187. if err != nil {
  188. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  189. return
  190. }
  191. for _, inviteeEmail := range inviteReq.UserEmails {
  192. // check if user with email exists, then ignore
  193. if !email.IsValid(inviteeEmail) {
  194. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid email "+inviteeEmail), "badrequest"))
  195. return
  196. }
  197. _, err := logic.GetUser(inviteeEmail)
  198. if err == nil {
  199. // user exists already, so ignore
  200. continue
  201. }
  202. invite := models.UserInvite{
  203. Email: inviteeEmail,
  204. PlatformRoleID: inviteReq.PlatformRoleID,
  205. UserGroups: inviteReq.UserGroups,
  206. NetworkRoles: inviteReq.NetworkRoles,
  207. InviteCode: logic.RandomString(8),
  208. }
  209. frontendURL := strings.TrimSuffix(servercfg.GetFrontendURL(), "/")
  210. if frontendURL == "" {
  211. frontendURL = fmt.Sprintf("https://dashboard.%s", servercfg.GetNmBaseDomain())
  212. }
  213. u, err := url.Parse(fmt.Sprintf("%s/invite?email=%s&invite_code=%s",
  214. frontendURL, url.QueryEscape(invite.Email), url.QueryEscape(invite.InviteCode)))
  215. if err != nil {
  216. slog.Error("failed to parse to invite url", "error", err)
  217. return
  218. }
  219. if servercfg.DeployedByOperator() {
  220. u, err = url.Parse(fmt.Sprintf("%s/invite?tenant_id=%s&email=%s&invite_code=%s",
  221. proLogic.GetAccountsUIHost(), url.QueryEscape(servercfg.GetNetmakerTenantID()), url.QueryEscape(invite.Email), url.QueryEscape(invite.InviteCode)))
  222. if err != nil {
  223. slog.Error("failed to parse to invite url", "error", err)
  224. return
  225. }
  226. }
  227. invite.InviteURL = u.String()
  228. err = logic.InsertUserInvite(invite)
  229. if err != nil {
  230. slog.Error("failed to insert invite for user", "email", invite.Email, "error", err)
  231. }
  232. // notify user with magic link
  233. go func(invite models.UserInvite) {
  234. // Set E-Mail body. You can set plain text or html with text/html
  235. e := email.UserInvitedMail{
  236. BodyBuilder: &email.EmailBodyBuilderWithH1HeadlineAndImage{},
  237. InviteURL: invite.InviteURL,
  238. PlatformRoleID: invite.PlatformRoleID,
  239. }
  240. n := email.Notification{
  241. RecipientMail: invite.Email,
  242. }
  243. err = email.GetClient().SendEmail(context.Background(), n, e)
  244. if err != nil {
  245. slog.Error("failed to send email invite", "user", invite.Email, "error", err)
  246. }
  247. }(invite)
  248. }
  249. logic.ReturnSuccessResponse(w, r, "triggered user invites")
  250. }
  251. // swagger:route GET /api/v1/users/invites user listUserInvites
  252. //
  253. // lists all pending invited users.
  254. //
  255. // Schemes: https
  256. //
  257. // Security:
  258. // oauth
  259. //
  260. // Responses:
  261. // 200: ReturnSuccessResponseWithJson
  262. func listUserInvites(w http.ResponseWriter, r *http.Request) {
  263. usersInvites, err := logic.ListUserInvites()
  264. if err != nil {
  265. logger.Log(0, "failed to fetch users: ", err.Error())
  266. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  267. return
  268. }
  269. logic.ReturnSuccessResponseWithJson(w, r, usersInvites, "fetched pending user invites")
  270. }
  271. // swagger:route DELETE /api/v1/users/invite user deleteUserInvite
  272. //
  273. // delete pending invite.
  274. //
  275. // Schemes: https
  276. //
  277. // Security:
  278. // oauth
  279. //
  280. // Responses:
  281. // 200: ReturnSuccessResponse
  282. func deleteUserInvite(w http.ResponseWriter, r *http.Request) {
  283. email := r.URL.Query().Get("invitee_email")
  284. err := logic.DeleteUserInvite(email)
  285. if err != nil {
  286. logger.Log(0, "failed to delete user invite: ", email, err.Error())
  287. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  288. return
  289. }
  290. logic.ReturnSuccessResponse(w, r, "deleted user invite")
  291. }
  292. // swagger:route DELETE /api/v1/users/invites user deleteAllUserInvites
  293. //
  294. // deletes all pending invites.
  295. //
  296. // Schemes: https
  297. //
  298. // Security:
  299. // oauth
  300. //
  301. // Responses:
  302. // 200: ReturnSuccessResponse
  303. func deleteAllUserInvites(w http.ResponseWriter, r *http.Request) {
  304. err := database.DeleteAllRecords(database.USER_INVITES_TABLE_NAME)
  305. if err != nil {
  306. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending user invites "+err.Error()), "internal"))
  307. return
  308. }
  309. logic.ReturnSuccessResponse(w, r, "cleared all pending user invites")
  310. }
  311. // swagger:route GET /api/v1/user/groups user listUserGroups
  312. //
  313. // Get all user groups.
  314. //
  315. // Schemes: https
  316. //
  317. // Security:
  318. // oauth
  319. //
  320. // Responses:
  321. // 200: userBodyResponse
  322. func listUserGroups(w http.ResponseWriter, r *http.Request) {
  323. groups, err := proLogic.ListUserGroups()
  324. if err != nil {
  325. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  326. Code: http.StatusInternalServerError,
  327. Message: err.Error(),
  328. })
  329. return
  330. }
  331. logic.ReturnSuccessResponseWithJson(w, r, groups, "successfully fetched user groups")
  332. }
  333. // swagger:route GET /api/v1/user/group user getUserGroup
  334. //
  335. // Get user group.
  336. //
  337. // Schemes: https
  338. //
  339. // Security:
  340. // oauth
  341. //
  342. // Responses:
  343. // 200: userBodyResponse
  344. func getUserGroup(w http.ResponseWriter, r *http.Request) {
  345. gid := r.URL.Query().Get("group_id")
  346. if gid == "" {
  347. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
  348. return
  349. }
  350. group, err := proLogic.GetUserGroup(models.UserGroupID(gid))
  351. if err != nil {
  352. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  353. Code: http.StatusInternalServerError,
  354. Message: err.Error(),
  355. })
  356. return
  357. }
  358. logic.ReturnSuccessResponseWithJson(w, r, group, "successfully fetched user group")
  359. }
  360. // swagger:route POST /api/v1/user/group user createUserGroup
  361. //
  362. // Create user groups.
  363. //
  364. // Schemes: https
  365. //
  366. // Security:
  367. // oauth
  368. //
  369. // Responses:
  370. // 200: userBodyResponse
  371. func createUserGroup(w http.ResponseWriter, r *http.Request) {
  372. var userGroupReq models.CreateGroupReq
  373. err := json.NewDecoder(r.Body).Decode(&userGroupReq)
  374. if err != nil {
  375. slog.Error("error decoding request body", "error",
  376. err.Error())
  377. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  378. return
  379. }
  380. err = proLogic.ValidateCreateGroupReq(userGroupReq.Group)
  381. if err != nil {
  382. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  383. return
  384. }
  385. err = proLogic.CreateUserGroup(userGroupReq.Group)
  386. if err != nil {
  387. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  388. return
  389. }
  390. for _, userID := range userGroupReq.Members {
  391. user, err := logic.GetUser(userID)
  392. if err != nil {
  393. continue
  394. }
  395. if len(user.UserGroups) == 0 {
  396. user.UserGroups = make(map[models.UserGroupID]struct{})
  397. }
  398. user.UserGroups[userGroupReq.Group.ID] = struct{}{}
  399. logic.UpsertUser(*user)
  400. }
  401. logic.ReturnSuccessResponseWithJson(w, r, userGroupReq.Group, "created user group")
  402. }
  403. // swagger:route PUT /api/v1/user/group user updateUserGroup
  404. //
  405. // Update user group.
  406. //
  407. // Schemes: https
  408. //
  409. // Security:
  410. // oauth
  411. //
  412. // Responses:
  413. // 200: userBodyResponse
  414. func updateUserGroup(w http.ResponseWriter, r *http.Request) {
  415. var userGroup models.UserGroup
  416. err := json.NewDecoder(r.Body).Decode(&userGroup)
  417. if err != nil {
  418. slog.Error("error decoding request body", "error",
  419. err.Error())
  420. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  421. return
  422. }
  423. // fetch curr group
  424. currUserG, err := proLogic.GetUserGroup(userGroup.ID)
  425. if err != nil {
  426. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  427. return
  428. }
  429. err = proLogic.ValidateUpdateGroupReq(userGroup)
  430. if err != nil {
  431. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  432. return
  433. }
  434. err = proLogic.UpdateUserGroup(userGroup)
  435. if err != nil {
  436. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  437. return
  438. }
  439. // reset configs for service user
  440. go proLogic.UpdatesUserGwAccessOnGrpUpdates(currUserG.NetworkRoles, userGroup.NetworkRoles)
  441. logic.ReturnSuccessResponseWithJson(w, r, userGroup, "updated user group")
  442. }
  443. // swagger:route DELETE /api/v1/user/group user deleteUserGroup
  444. //
  445. // delete user group.
  446. //
  447. // Schemes: https
  448. //
  449. // Security:
  450. // oauth
  451. //
  452. // Responses:
  453. // 200: userBodyResponse
  454. //
  455. // @Summary Delete user group.
  456. // @Router /api/v1/user/group [delete]
  457. // @Tags Users
  458. // @Param group_id param string true "group id required to delete the role"
  459. // @Success 200 {string} string
  460. // @Failure 500 {object} models.ErrorResponse
  461. func deleteUserGroup(w http.ResponseWriter, r *http.Request) {
  462. gid := r.URL.Query().Get("group_id")
  463. if gid == "" {
  464. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
  465. return
  466. }
  467. userG, err := proLogic.GetUserGroup(models.UserGroupID(gid))
  468. if err != nil {
  469. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to fetch group details"), "badrequest"))
  470. return
  471. }
  472. if userG.Default {
  473. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot delete default user group"), "badrequest"))
  474. return
  475. }
  476. err = proLogic.DeleteUserGroup(models.UserGroupID(gid))
  477. if err != nil {
  478. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  479. return
  480. }
  481. go proLogic.UpdatesUserGwAccessOnGrpUpdates(userG.NetworkRoles, make(map[models.NetworkID]map[models.UserRoleID]struct{}))
  482. logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user group")
  483. }
  484. // @Summary lists all user roles.
  485. // @Router /api/v1/user/roles [get]
  486. // @Tags Users
  487. // @Param role_id param string true "roleid required to get the role details"
  488. // @Success 200 {object} []models.UserRolePermissionTemplate
  489. // @Failure 500 {object} models.ErrorResponse
  490. func ListRoles(w http.ResponseWriter, r *http.Request) {
  491. platform := r.URL.Query().Get("platform")
  492. var roles []models.UserRolePermissionTemplate
  493. var err error
  494. if platform == "true" {
  495. roles, err = logic.ListPlatformRoles()
  496. } else {
  497. roles, err = proLogic.ListNetworkRoles()
  498. }
  499. if err != nil {
  500. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  501. Code: http.StatusInternalServerError,
  502. Message: err.Error(),
  503. })
  504. return
  505. }
  506. logic.ReturnSuccessResponseWithJson(w, r, roles, "successfully fetched user roles permission templates")
  507. }
  508. // @Summary Get user role permission template.
  509. // @Router /api/v1/user/role [get]
  510. // @Tags Users
  511. // @Param role_id param string true "roleid required to get the role details"
  512. // @Success 200 {object} models.UserRolePermissionTemplate
  513. // @Failure 500 {object} models.ErrorResponse
  514. func getRole(w http.ResponseWriter, r *http.Request) {
  515. rid := r.URL.Query().Get("role_id")
  516. if rid == "" {
  517. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  518. return
  519. }
  520. role, err := logic.GetRole(models.UserRoleID(rid))
  521. if err != nil {
  522. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  523. Code: http.StatusInternalServerError,
  524. Message: err.Error(),
  525. })
  526. return
  527. }
  528. logic.ReturnSuccessResponseWithJson(w, r, role, "successfully fetched user role permission templates")
  529. }
  530. // @Summary Create user role permission template.
  531. // @Router /api/v1/user/role [post]
  532. // @Tags Users
  533. // @Param body models.UserRolePermissionTemplate true "user role template"
  534. // @Success 200 {object} models.UserRolePermissionTemplate
  535. // @Failure 500 {object} models.ErrorResponse
  536. func createRole(w http.ResponseWriter, r *http.Request) {
  537. var userRole models.UserRolePermissionTemplate
  538. err := json.NewDecoder(r.Body).Decode(&userRole)
  539. if err != nil {
  540. slog.Error("error decoding request body", "error",
  541. err.Error())
  542. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  543. return
  544. }
  545. err = proLogic.ValidateCreateRoleReq(&userRole)
  546. if err != nil {
  547. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  548. return
  549. }
  550. userRole.Default = false
  551. userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
  552. err = proLogic.CreateRole(userRole)
  553. if err != nil {
  554. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  555. return
  556. }
  557. logic.ReturnSuccessResponseWithJson(w, r, userRole, "created user role")
  558. }
  559. // @Summary Update user role permission template.
  560. // @Router /api/v1/user/role [put]
  561. // @Tags Users
  562. // @Param body models.UserRolePermissionTemplate true "user role template"
  563. // @Success 200 {object} userBodyResponse
  564. // @Failure 500 {object} models.ErrorResponse
  565. func updateRole(w http.ResponseWriter, r *http.Request) {
  566. var userRole models.UserRolePermissionTemplate
  567. err := json.NewDecoder(r.Body).Decode(&userRole)
  568. if err != nil {
  569. slog.Error("error decoding request body", "error",
  570. err.Error())
  571. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  572. return
  573. }
  574. currRole, err := logic.GetRole(userRole.ID)
  575. if err != nil {
  576. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  577. return
  578. }
  579. err = proLogic.ValidateUpdateRoleReq(&userRole)
  580. if err != nil {
  581. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  582. return
  583. }
  584. userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
  585. err = proLogic.UpdateRole(userRole)
  586. if err != nil {
  587. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  588. return
  589. }
  590. // reset configs for service user
  591. go proLogic.UpdatesUserGwAccessOnRoleUpdates(currRole.NetworkLevelAccess, userRole.NetworkLevelAccess, string(userRole.NetworkID))
  592. logic.ReturnSuccessResponseWithJson(w, r, userRole, "updated user role")
  593. }
  594. // @Summary Delete user role permission template.
  595. // @Router /api/v1/user/role [delete]
  596. // @Tags Users
  597. // @Param role_id param string true "roleid required to delete the role"
  598. // @Success 200 {string} string
  599. // @Failure 500 {object} models.ErrorResponse
  600. func deleteRole(w http.ResponseWriter, r *http.Request) {
  601. rid := r.URL.Query().Get("role_id")
  602. if rid == "" {
  603. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  604. return
  605. }
  606. role, err := logic.GetRole(models.UserRoleID(rid))
  607. if err != nil {
  608. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  609. return
  610. }
  611. err = proLogic.DeleteRole(models.UserRoleID(rid), false)
  612. if err != nil {
  613. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  614. return
  615. }
  616. go proLogic.UpdatesUserGwAccessOnRoleUpdates(role.NetworkLevelAccess, make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope), role.NetworkID.String())
  617. logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user role")
  618. }
  619. // @Summary Attach user to a remote access gateway
  620. // @Router /api/users/{username}/remote_access_gw/{remote_access_gateway_id} [post]
  621. // @Tags PRO
  622. // @Accept json
  623. // @Produce json
  624. // @Param username path string true "Username"
  625. // @Param remote_access_gateway_id path string true "Remote Access Gateway ID"
  626. // @Success 200 {object} models.ReturnUser
  627. // @Failure 400 {object} models.ErrorResponse
  628. // @Failure 500 {object} models.ErrorResponse
  629. func attachUserToRemoteAccessGw(w http.ResponseWriter, r *http.Request) {
  630. // set header.
  631. w.Header().Set("Content-Type", "application/json")
  632. var params = mux.Vars(r)
  633. username := params["username"]
  634. remoteGwID := params["remote_access_gateway_id"]
  635. if username == "" || remoteGwID == "" {
  636. logic.ReturnErrorResponse(
  637. w,
  638. r,
  639. logic.FormatError(
  640. errors.New("required params `username` and `remote_access_gateway_id`"),
  641. "badrequest",
  642. ),
  643. )
  644. return
  645. }
  646. user, err := logic.GetUser(username)
  647. if err != nil {
  648. slog.Error("failed to fetch user: ", "username", username, "error", err.Error())
  649. logic.ReturnErrorResponse(
  650. w,
  651. r,
  652. logic.FormatError(
  653. fmt.Errorf("failed to fetch user %s, error: %v", username, err),
  654. "badrequest",
  655. ),
  656. )
  657. return
  658. }
  659. if user.PlatformRoleID == models.AdminRole || user.PlatformRoleID == models.SuperAdminRole {
  660. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("superadmins/admins have access to all gateways"), "badrequest"))
  661. return
  662. }
  663. node, err := logic.GetNodeByID(remoteGwID)
  664. if err != nil {
  665. slog.Error("failed to fetch gateway node", "nodeID", remoteGwID, "error", err)
  666. logic.ReturnErrorResponse(
  667. w,
  668. r,
  669. logic.FormatError(
  670. fmt.Errorf("failed to fetch remote access gateway node, error: %v", err),
  671. "badrequest",
  672. ),
  673. )
  674. return
  675. }
  676. if !node.IsIngressGateway {
  677. logic.ReturnErrorResponse(
  678. w,
  679. r,
  680. logic.FormatError(fmt.Errorf("node is not a remote access gateway"), "badrequest"),
  681. )
  682. return
  683. }
  684. if user.RemoteGwIDs == nil {
  685. user.RemoteGwIDs = make(map[string]struct{})
  686. }
  687. user.RemoteGwIDs[node.ID.String()] = struct{}{}
  688. err = logic.UpsertUser(*user)
  689. if err != nil {
  690. slog.Error("failed to update user's gateways", "user", username, "error", err)
  691. logic.ReturnErrorResponse(
  692. w,
  693. r,
  694. logic.FormatError(
  695. fmt.Errorf("failed to fetch remote access gateway node,error: %v", err),
  696. "badrequest",
  697. ),
  698. )
  699. return
  700. }
  701. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  702. }
  703. // @Summary Remove user from a remote access gateway
  704. // @Router /api/users/{username}/remote_access_gw/{remote_access_gateway_id} [delete]
  705. // @Tags PRO
  706. // @Accept json
  707. // @Produce json
  708. // @Param username path string true "Username"
  709. // @Param remote_access_gateway_id path string true "Remote Access Gateway ID"
  710. // @Success 200 {object} models.ReturnUser
  711. // @Failure 400 {object} models.ErrorResponse
  712. // @Failure 500 {object} models.ErrorResponse
  713. func removeUserFromRemoteAccessGW(w http.ResponseWriter, r *http.Request) {
  714. // set header.
  715. w.Header().Set("Content-Type", "application/json")
  716. var params = mux.Vars(r)
  717. username := params["username"]
  718. remoteGwID := params["remote_access_gateway_id"]
  719. if username == "" || remoteGwID == "" {
  720. logic.ReturnErrorResponse(
  721. w,
  722. r,
  723. logic.FormatError(
  724. errors.New("required params `username` and `remote_access_gateway_id`"),
  725. "badrequest",
  726. ),
  727. )
  728. return
  729. }
  730. user, err := logic.GetUser(username)
  731. if err != nil {
  732. logger.Log(0, username, "failed to fetch user: ", err.Error())
  733. logic.ReturnErrorResponse(
  734. w,
  735. r,
  736. logic.FormatError(
  737. fmt.Errorf("failed to fetch user %s, error: %v", username, err),
  738. "badrequest",
  739. ),
  740. )
  741. return
  742. }
  743. delete(user.RemoteGwIDs, remoteGwID)
  744. go func(user models.User, remoteGwID string) {
  745. extclients, err := logic.GetAllExtClients()
  746. if err != nil {
  747. slog.Error("failed to fetch extclients", "error", err)
  748. return
  749. }
  750. for _, extclient := range extclients {
  751. if extclient.OwnerID == user.UserName && remoteGwID == extclient.IngressGatewayID {
  752. err = logic.DeleteExtClientAndCleanup(extclient)
  753. if err != nil {
  754. slog.Error("failed to delete extclient",
  755. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  756. } else {
  757. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  758. slog.Error("error setting ext peers: " + err.Error())
  759. }
  760. }
  761. }
  762. }
  763. if servercfg.IsDNSMode() {
  764. logic.SetDNS()
  765. }
  766. }(*user, remoteGwID)
  767. err = logic.UpsertUser(*user)
  768. if err != nil {
  769. slog.Error("failed to update user gateways", "user", username, "error", err)
  770. logic.ReturnErrorResponse(
  771. w,
  772. r,
  773. logic.FormatError(
  774. errors.New("failed to fetch remote access gaetway node "+err.Error()),
  775. "badrequest",
  776. ),
  777. )
  778. return
  779. }
  780. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  781. }
  782. // @Summary Get Users Remote Access Gw Networks.
  783. // @Router /api/users/{username}/remote_access_gw [get]
  784. // @Tags Users
  785. // @Param username path string true "Username to fetch all the gateways with access"
  786. // @Success 200 {object} map[string][]models.UserRemoteGws
  787. // @Failure 500 {object} models.ErrorResponse
  788. func getUserRemoteAccessNetworks(w http.ResponseWriter, r *http.Request) {
  789. // set header.
  790. w.Header().Set("Content-Type", "application/json")
  791. username := r.Header.Get("user")
  792. user, err := logic.GetUser(username)
  793. if err != nil {
  794. logger.Log(0, username, "failed to fetch user: ", err.Error())
  795. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  796. return
  797. }
  798. userGws := make(map[string][]models.UserRemoteGws)
  799. networks := []models.Network{}
  800. networkMap := make(map[string]struct{})
  801. userGwNodes := proLogic.GetUserRAGNodes(*user)
  802. for _, node := range userGwNodes {
  803. network, err := logic.GetNetwork(node.Network)
  804. if err != nil {
  805. slog.Error("failed to get node network", "error", err)
  806. continue
  807. }
  808. if _, ok := networkMap[network.NetID]; ok {
  809. continue
  810. }
  811. networkMap[network.NetID] = struct{}{}
  812. networks = append(networks, network)
  813. }
  814. slog.Debug("returned user gws", "user", username, "gws", userGws)
  815. logic.ReturnSuccessResponseWithJson(w, r, networks, "fetched user accessible networks")
  816. }
  817. // @Summary Get Users Remote Access Gw Networks.
  818. // @Router /api/users/{username}/remote_access_gw [get]
  819. // @Tags Users
  820. // @Param username path string true "Username to fetch all the gateways with access"
  821. // @Success 200 {object} map[string][]models.UserRemoteGws
  822. // @Failure 500 {object} models.ErrorResponse
  823. func getUserRemoteAccessNetworkGateways(w http.ResponseWriter, r *http.Request) {
  824. // set header.
  825. w.Header().Set("Content-Type", "application/json")
  826. var params = mux.Vars(r)
  827. username := r.Header.Get("user")
  828. user, err := logic.GetUser(username)
  829. if err != nil {
  830. logger.Log(0, username, "failed to fetch user: ", err.Error())
  831. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  832. return
  833. }
  834. network := params["network"]
  835. if network == "" {
  836. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params network"), "badrequest"))
  837. return
  838. }
  839. userGws := []models.UserRAGs{}
  840. userGwNodes := proLogic.GetUserRAGNodes(*user)
  841. for _, node := range userGwNodes {
  842. if node.Network != network {
  843. continue
  844. }
  845. host, err := logic.GetHost(node.HostID.String())
  846. if err != nil {
  847. continue
  848. }
  849. userGws = append(userGws, models.UserRAGs{
  850. GwID: node.ID.String(),
  851. GWName: host.Name,
  852. Network: node.Network,
  853. IsInternetGateway: node.IsInternetGateway,
  854. Metadata: node.Metadata,
  855. })
  856. }
  857. slog.Debug("returned user gws", "user", username, "gws", userGws)
  858. logic.ReturnSuccessResponseWithJson(w, r, userGws, "fetched user accessible gateways in network "+network)
  859. }
  860. // @Summary Get Users Remote Access Gw Networks.
  861. // @Router /api/users/{username}/remote_access_gw [get]
  862. // @Tags Users
  863. // @Param username path string true "Username to fetch all the gateways with access"
  864. // @Success 200 {object} map[string][]models.UserRemoteGws
  865. // @Failure 500 {object} models.ErrorResponse
  866. func getRemoteAccessGatewayConf(w http.ResponseWriter, r *http.Request) {
  867. // set header.
  868. w.Header().Set("Content-Type", "application/json")
  869. var params = mux.Vars(r)
  870. username := r.Header.Get("user")
  871. user, err := logic.GetUser(username)
  872. if err != nil {
  873. logger.Log(0, username, "failed to fetch user: ", err.Error())
  874. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  875. return
  876. }
  877. remoteGwID := params["access_point_id"]
  878. if remoteGwID == "" {
  879. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params access_point_id"), "badrequest"))
  880. return
  881. }
  882. var req models.UserRemoteGwsReq
  883. err = json.NewDecoder(r.Body).Decode(&req)
  884. if err != nil {
  885. slog.Error("error decoding request body: ", "error", err)
  886. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  887. return
  888. }
  889. userGwNodes := proLogic.GetUserRAGNodes(*user)
  890. if _, ok := userGwNodes[remoteGwID]; !ok {
  891. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("access denied"), "forbidden"))
  892. return
  893. }
  894. node, err := logic.GetNodeByID(remoteGwID)
  895. if err != nil {
  896. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch gw node %s, error: %v", remoteGwID, err), "badrequest"))
  897. return
  898. }
  899. host, err := logic.GetHost(node.HostID.String())
  900. if err != nil {
  901. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch gw host %s, error: %v", remoteGwID, err), "badrequest"))
  902. return
  903. }
  904. network, err := logic.GetNetwork(node.Network)
  905. if err != nil {
  906. slog.Error("failed to get node network", "error", err)
  907. }
  908. var userConf models.ExtClient
  909. allextClients, err := logic.GetAllExtClients()
  910. if err != nil {
  911. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  912. return
  913. }
  914. for _, extClient := range allextClients {
  915. if extClient.Network != network.NetID || extClient.IngressGatewayID != node.ID.String() {
  916. continue
  917. }
  918. if extClient.RemoteAccessClientID == req.RemoteAccessClientID && extClient.OwnerID == username {
  919. userConf = extClient
  920. userConf.AllowedIPs = logic.GetExtclientAllowedIPs(extClient)
  921. }
  922. }
  923. if userConf.ClientID == "" {
  924. // create a new conf
  925. userConf.OwnerID = user.UserName
  926. userConf.RemoteAccessClientID = req.RemoteAccessClientID
  927. userConf.IngressGatewayID = node.ID.String()
  928. // set extclient dns to ingressdns if extclient dns is not explicitly set
  929. if (userConf.DNS == "") && (node.IngressDNS != "") {
  930. userConf.DNS = node.IngressDNS
  931. }
  932. userConf.Network = node.Network
  933. host, err := logic.GetHost(node.HostID.String())
  934. if err != nil {
  935. logger.Log(0, r.Header.Get("user"),
  936. fmt.Sprintf("failed to get ingress gateway host for node [%s] info: %v", node.ID, err))
  937. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  938. return
  939. }
  940. listenPort := logic.GetPeerListenPort(host)
  941. if host.EndpointIP.To4() == nil {
  942. userConf.IngressGatewayEndpoint = fmt.Sprintf("[%s]:%d", host.EndpointIPv6.String(), listenPort)
  943. } else {
  944. userConf.IngressGatewayEndpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), listenPort)
  945. }
  946. userConf.Enabled = true
  947. parentNetwork, err := logic.GetNetwork(node.Network)
  948. if err == nil { // check if parent network default ACL is enabled (yes) or not (no)
  949. userConf.Enabled = parentNetwork.DefaultACL == "yes"
  950. }
  951. userConf.Tags = make(map[models.TagID]struct{})
  952. userConf.Tags[models.TagID(fmt.Sprintf("%s.%s", userConf.Network,
  953. models.RemoteAccessTagName))] = struct{}{}
  954. if err = logic.CreateExtClient(&userConf); err != nil {
  955. slog.Error(
  956. "failed to create extclient",
  957. "user",
  958. r.Header.Get("user"),
  959. "network",
  960. node.Network,
  961. "error",
  962. err,
  963. )
  964. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  965. return
  966. }
  967. }
  968. userGw := models.UserRemoteGws{
  969. GwID: node.ID.String(),
  970. GWName: host.Name,
  971. Network: node.Network,
  972. GwClient: userConf,
  973. Connected: true,
  974. IsInternetGateway: node.IsInternetGateway,
  975. GwPeerPublicKey: host.PublicKey.String(),
  976. GwListenPort: logic.GetPeerListenPort(host),
  977. Metadata: node.Metadata,
  978. AllowedEndpoints: getAllowedRagEndpoints(&node, host),
  979. NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
  980. }
  981. slog.Debug("returned user gw config", "user", user.UserName, "gws", userGw)
  982. logic.ReturnSuccessResponseWithJson(w, r, userGw, "fetched user config to gw "+remoteGwID)
  983. }
  984. // @Summary Get Users Remote Access Gw.
  985. // @Router /api/users/{username}/remote_access_gw [get]
  986. // @Tags Users
  987. // @Param username path string true "Username to fetch all the gateways with access"
  988. // @Success 200 {object} map[string][]models.UserRemoteGws
  989. // @Failure 500 {object} models.ErrorResponse
  990. func getUserRemoteAccessGwsV1(w http.ResponseWriter, r *http.Request) {
  991. // set header.
  992. w.Header().Set("Content-Type", "application/json")
  993. var params = mux.Vars(r)
  994. username := params["username"]
  995. if username == "" {
  996. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params username"), "badrequest"))
  997. return
  998. }
  999. user, err := logic.GetUser(username)
  1000. if err != nil {
  1001. logger.Log(0, username, "failed to fetch user: ", err.Error())
  1002. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  1003. return
  1004. }
  1005. remoteAccessClientID := r.URL.Query().Get("remote_access_clientid")
  1006. var req models.UserRemoteGwsReq
  1007. if remoteAccessClientID == "" {
  1008. err := json.NewDecoder(r.Body).Decode(&req)
  1009. if err != nil {
  1010. slog.Error("error decoding request body: ", "error", err)
  1011. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1012. return
  1013. }
  1014. }
  1015. reqFromMobile := r.URL.Query().Get("from_mobile") == "true"
  1016. if req.RemoteAccessClientID == "" && remoteAccessClientID == "" {
  1017. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("remote access client id cannot be empty"), "badrequest"))
  1018. return
  1019. }
  1020. if req.RemoteAccessClientID == "" {
  1021. req.RemoteAccessClientID = remoteAccessClientID
  1022. }
  1023. userGws := make(map[string][]models.UserRemoteGws)
  1024. allextClients, err := logic.GetAllExtClients()
  1025. if err != nil {
  1026. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1027. return
  1028. }
  1029. userGwNodes := proLogic.GetUserRAGNodes(*user)
  1030. for _, extClient := range allextClients {
  1031. node, ok := userGwNodes[extClient.IngressGatewayID]
  1032. if !ok {
  1033. continue
  1034. }
  1035. if extClient.RemoteAccessClientID == req.RemoteAccessClientID && extClient.OwnerID == username {
  1036. host, err := logic.GetHost(node.HostID.String())
  1037. if err != nil {
  1038. continue
  1039. }
  1040. network, err := logic.GetNetwork(node.Network)
  1041. if err != nil {
  1042. slog.Error("failed to get node network", "error", err)
  1043. continue
  1044. }
  1045. gws := userGws[node.Network]
  1046. extClient.AllowedIPs = logic.GetExtclientAllowedIPs(extClient)
  1047. gws = append(gws, models.UserRemoteGws{
  1048. GwID: node.ID.String(),
  1049. GWName: host.Name,
  1050. Network: node.Network,
  1051. GwClient: extClient,
  1052. Connected: true,
  1053. IsInternetGateway: node.IsInternetGateway,
  1054. GwPeerPublicKey: host.PublicKey.String(),
  1055. GwListenPort: logic.GetPeerListenPort(host),
  1056. Metadata: node.Metadata,
  1057. AllowedEndpoints: getAllowedRagEndpoints(&node, host),
  1058. NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
  1059. })
  1060. userGws[node.Network] = gws
  1061. delete(userGwNodes, node.ID.String())
  1062. }
  1063. }
  1064. // add remaining gw nodes to resp
  1065. for gwID := range userGwNodes {
  1066. node, err := logic.GetNodeByID(gwID)
  1067. if err != nil {
  1068. continue
  1069. }
  1070. if !node.IsIngressGateway {
  1071. continue
  1072. }
  1073. if node.PendingDelete {
  1074. continue
  1075. }
  1076. host, err := logic.GetHost(node.HostID.String())
  1077. if err != nil {
  1078. continue
  1079. }
  1080. network, err := logic.GetNetwork(node.Network)
  1081. if err != nil {
  1082. slog.Error("failed to get node network", "error", err)
  1083. }
  1084. gws := userGws[node.Network]
  1085. gws = append(gws, models.UserRemoteGws{
  1086. GwID: node.ID.String(),
  1087. GWName: host.Name,
  1088. Network: node.Network,
  1089. IsInternetGateway: node.IsInternetGateway,
  1090. GwPeerPublicKey: host.PublicKey.String(),
  1091. GwListenPort: logic.GetPeerListenPort(host),
  1092. Metadata: node.Metadata,
  1093. AllowedEndpoints: getAllowedRagEndpoints(&node, host),
  1094. NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
  1095. })
  1096. userGws[node.Network] = gws
  1097. }
  1098. if reqFromMobile {
  1099. // send resp in array format
  1100. userGwsArr := []models.UserRemoteGws{}
  1101. for _, userGwI := range userGws {
  1102. userGwsArr = append(userGwsArr, userGwI...)
  1103. }
  1104. logic.ReturnSuccessResponseWithJson(w, r, userGwsArr, "fetched gateways for user"+username)
  1105. return
  1106. }
  1107. slog.Debug("returned user gws", "user", username, "gws", userGws)
  1108. w.WriteHeader(http.StatusOK)
  1109. json.NewEncoder(w).Encode(userGws)
  1110. }
  1111. // @Summary List users attached to an remote access gateway
  1112. // @Router /api/nodes/{network}/{nodeid}/ingress/users [get]
  1113. // @Tags PRO
  1114. // @Accept json
  1115. // @Produce json
  1116. // @Param ingress_id path string true "Ingress Gateway ID"
  1117. // @Success 200 {array} models.IngressGwUsers
  1118. // @Failure 400 {object} models.ErrorResponse
  1119. // @Failure 500 {object} models.ErrorResponse
  1120. func ingressGatewayUsers(w http.ResponseWriter, r *http.Request) {
  1121. w.Header().Set("Content-Type", "application/json")
  1122. var params = mux.Vars(r)
  1123. ingressID := params["ingress_id"]
  1124. node, err := logic.GetNodeByID(ingressID)
  1125. if err != nil {
  1126. slog.Error("failed to get ingress node", "error", err)
  1127. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1128. return
  1129. }
  1130. gwUsers, err := logic.GetIngressGwUsers(node)
  1131. if err != nil {
  1132. slog.Error(
  1133. "failed to get users on ingress gateway",
  1134. "nodeid",
  1135. ingressID,
  1136. "network",
  1137. node.Network,
  1138. "user",
  1139. r.Header.Get("user"),
  1140. "error",
  1141. err,
  1142. )
  1143. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1144. return
  1145. }
  1146. w.WriteHeader(http.StatusOK)
  1147. json.NewEncoder(w).Encode(gwUsers)
  1148. }
  1149. func getAllowedRagEndpoints(ragNode *models.Node, ragHost *models.Host) []string {
  1150. endpoints := []string{}
  1151. if len(ragHost.EndpointIP) > 0 {
  1152. endpoints = append(endpoints, ragHost.EndpointIP.String())
  1153. }
  1154. if len(ragHost.EndpointIPv6) > 0 {
  1155. endpoints = append(endpoints, ragHost.EndpointIPv6.String())
  1156. }
  1157. if servercfg.IsPro {
  1158. for _, ip := range ragNode.AdditionalRagIps {
  1159. endpoints = append(endpoints, ip.String())
  1160. }
  1161. }
  1162. return endpoints
  1163. }
  1164. // @Summary Get all pending users
  1165. // @Router /api/users_pending [get]
  1166. // @Tags Users
  1167. // @Success 200 {array} models.User
  1168. // @Failure 500 {object} models.ErrorResponse
  1169. func getPendingUsers(w http.ResponseWriter, r *http.Request) {
  1170. // set header.
  1171. w.Header().Set("Content-Type", "application/json")
  1172. users, err := logic.ListPendingUsers()
  1173. if err != nil {
  1174. logger.Log(0, "failed to fetch users: ", err.Error())
  1175. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1176. return
  1177. }
  1178. logic.SortUsers(users[:])
  1179. logger.Log(2, r.Header.Get("user"), "fetched pending users")
  1180. json.NewEncoder(w).Encode(users)
  1181. }
  1182. // @Summary Approve a pending user
  1183. // @Router /api/users_pending/user/{username} [post]
  1184. // @Tags Users
  1185. // @Param username path string true "Username of the pending user to approve"
  1186. // @Success 200 {string} string
  1187. // @Failure 500 {object} models.ErrorResponse
  1188. func approvePendingUser(w http.ResponseWriter, r *http.Request) {
  1189. // set header.
  1190. w.Header().Set("Content-Type", "application/json")
  1191. var params = mux.Vars(r)
  1192. username := params["username"]
  1193. users, err := logic.ListPendingUsers()
  1194. if err != nil {
  1195. logger.Log(0, "failed to fetch users: ", err.Error())
  1196. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1197. return
  1198. }
  1199. for _, user := range users {
  1200. if user.UserName == username {
  1201. var newPass, fetchErr = logic.FetchPassValue("")
  1202. if fetchErr != nil {
  1203. logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
  1204. return
  1205. }
  1206. if err = logic.CreateUser(&models.User{
  1207. UserName: user.UserName,
  1208. Password: newPass,
  1209. PlatformRoleID: models.ServiceUser,
  1210. }); err != nil {
  1211. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to create user: %s", err), "internal"))
  1212. return
  1213. }
  1214. err = logic.DeletePendingUser(username)
  1215. if err != nil {
  1216. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
  1217. return
  1218. }
  1219. break
  1220. }
  1221. }
  1222. logic.ReturnSuccessResponse(w, r, "approved "+username)
  1223. }
  1224. // @Summary Delete a pending user
  1225. // @Router /api/users_pending/user/{username} [delete]
  1226. // @Tags Users
  1227. // @Param username path string true "Username of the pending user to delete"
  1228. // @Success 200 {string} string
  1229. // @Failure 500 {object} models.ErrorResponse
  1230. func deletePendingUser(w http.ResponseWriter, r *http.Request) {
  1231. // set header.
  1232. w.Header().Set("Content-Type", "application/json")
  1233. var params = mux.Vars(r)
  1234. username := params["username"]
  1235. users, err := logic.ListPendingUsers()
  1236. if err != nil {
  1237. logger.Log(0, "failed to fetch users: ", err.Error())
  1238. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1239. return
  1240. }
  1241. for _, user := range users {
  1242. if user.UserName == username {
  1243. err = logic.DeletePendingUser(username)
  1244. if err != nil {
  1245. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
  1246. return
  1247. }
  1248. break
  1249. }
  1250. }
  1251. logic.ReturnSuccessResponse(w, r, "deleted pending "+username)
  1252. }
  1253. // @Summary Delete all pending users
  1254. // @Router /api/users_pending [delete]
  1255. // @Tags Users
  1256. // @Success 200 {string} string
  1257. // @Failure 500 {object} models.ErrorResponse
  1258. func deleteAllPendingUsers(w http.ResponseWriter, r *http.Request) {
  1259. // set header.
  1260. err := database.DeleteAllRecords(database.PENDING_USERS_TABLE_NAME)
  1261. if err != nil {
  1262. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending users "+err.Error()), "internal"))
  1263. return
  1264. }
  1265. logic.ReturnSuccessResponse(w, r, "cleared all pending users")
  1266. }