user.go 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287
  1. package controller
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "net/http"
  7. "net/url"
  8. "github.com/gorilla/mux"
  9. "github.com/gorilla/websocket"
  10. "github.com/gravitl/netmaker/auth"
  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. "github.com/gravitl/netmaker/servercfg"
  17. "golang.org/x/exp/slog"
  18. )
  19. var (
  20. upgrader = websocket.Upgrader{}
  21. )
  22. func userHandlers(r *mux.Router) {
  23. r.HandleFunc("/api/users/adm/hassuperadmin", hasSuperAdmin).Methods(http.MethodGet)
  24. r.HandleFunc("/api/users/adm/createsuperadmin", createSuperAdmin).Methods(http.MethodPost)
  25. r.HandleFunc("/api/users/adm/transfersuperadmin/{username}", logic.SecurityCheck(true, http.HandlerFunc(transferSuperAdmin))).Methods(http.MethodPost)
  26. r.HandleFunc("/api/users/adm/authenticate", authenticateUser).Methods(http.MethodPost)
  27. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, http.HandlerFunc(updateUser))).Methods(http.MethodPut)
  28. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, checkFreeTierLimits(limitChoiceUsers, http.HandlerFunc(createUser)))).Methods(http.MethodPost)
  29. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, http.HandlerFunc(deleteUser))).Methods(http.MethodDelete)
  30. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUser)))).Methods(http.MethodGet)
  31. r.HandleFunc("/api/users", logic.SecurityCheck(true, http.HandlerFunc(getUsers))).Methods(http.MethodGet)
  32. r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(getPendingUsers))).Methods(http.MethodGet)
  33. r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(deleteAllPendingUsers))).Methods(http.MethodDelete)
  34. r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(deletePendingUser))).Methods(http.MethodDelete)
  35. r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(approvePendingUser))).Methods(http.MethodPost)
  36. // User Role Handlers
  37. r.HandleFunc("/api/v1/users/roles", logic.SecurityCheck(true, http.HandlerFunc(listRoles))).Methods(http.MethodGet)
  38. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(getRole))).Methods(http.MethodGet)
  39. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(createRole))).Methods(http.MethodPost)
  40. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(updateRole))).Methods(http.MethodPut)
  41. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(deleteRole))).Methods(http.MethodDelete)
  42. // User Group Handlers
  43. r.HandleFunc("/api/v1/users/groups", logic.SecurityCheck(true, http.HandlerFunc(listUserGroups))).Methods(http.MethodGet)
  44. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(getUserGroup))).Methods(http.MethodGet)
  45. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(createUserGroup))).Methods(http.MethodPost)
  46. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(updateUserGroup))).Methods(http.MethodPut)
  47. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(deleteUserGroup))).Methods(http.MethodDelete)
  48. // User Invite Handlers
  49. r.HandleFunc("/api/v1/users/invite", userInviteVerify).Methods(http.MethodGet)
  50. r.HandleFunc("/api/v1/users/invite-signup", userInviteSignUp).Methods(http.MethodPost)
  51. r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(inviteUsers))).Methods(http.MethodPost)
  52. r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(listUserInvites))).Methods(http.MethodGet)
  53. r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(deleteUserInvite))).Methods(http.MethodDelete)
  54. r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(deleteAllUserInvites))).Methods(http.MethodDelete)
  55. }
  56. // swagger:route GET /api/v1/user/groups user listUserGroups
  57. //
  58. // Get all user groups.
  59. //
  60. // Schemes: https
  61. //
  62. // Security:
  63. // oauth
  64. //
  65. // Responses:
  66. // 200: userBodyResponse
  67. func listUserGroups(w http.ResponseWriter, r *http.Request) {
  68. groups, err := logic.ListUserGroups()
  69. if err != nil {
  70. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  71. Code: http.StatusInternalServerError,
  72. Message: err.Error(),
  73. })
  74. return
  75. }
  76. logic.ReturnSuccessResponseWithJson(w, r, groups, "successfully fetched user groups")
  77. }
  78. // swagger:route GET /api/v1/user/group user getUserGroup
  79. //
  80. // Get user group.
  81. //
  82. // Schemes: https
  83. //
  84. // Security:
  85. // oauth
  86. //
  87. // Responses:
  88. // 200: userBodyResponse
  89. func getUserGroup(w http.ResponseWriter, r *http.Request) {
  90. var params = mux.Vars(r)
  91. gid := params["group_id"]
  92. if gid == "" {
  93. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
  94. return
  95. }
  96. group, err := logic.GetUserGroup(models.UserGroupID(gid))
  97. if err != nil {
  98. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  99. Code: http.StatusInternalServerError,
  100. Message: err.Error(),
  101. })
  102. return
  103. }
  104. logic.ReturnSuccessResponseWithJson(w, r, group, "successfully fetched user group")
  105. }
  106. // swagger:route POST /api/v1/user/group user createUserGroup
  107. //
  108. // Create user groups.
  109. //
  110. // Schemes: https
  111. //
  112. // Security:
  113. // oauth
  114. //
  115. // Responses:
  116. // 200: userBodyResponse
  117. func createUserGroup(w http.ResponseWriter, r *http.Request) {
  118. var userGroup models.UserGroup
  119. err := json.NewDecoder(r.Body).Decode(&userGroup)
  120. if err != nil {
  121. slog.Error("error decoding request body", "error",
  122. err.Error())
  123. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  124. return
  125. }
  126. err = logic.ValidateCreateGroupReq(userGroup)
  127. if err != nil {
  128. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  129. return
  130. }
  131. err = logic.CreateUserGroup(userGroup)
  132. if err != nil {
  133. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  134. return
  135. }
  136. logic.ReturnSuccessResponseWithJson(w, r, userGroup, "created user group")
  137. }
  138. // swagger:route PUT /api/v1/user/group user updateUserGroup
  139. //
  140. // Update user group.
  141. //
  142. // Schemes: https
  143. //
  144. // Security:
  145. // oauth
  146. //
  147. // Responses:
  148. // 200: userBodyResponse
  149. func updateUserGroup(w http.ResponseWriter, r *http.Request) {
  150. var userGroup models.UserGroup
  151. err := json.NewDecoder(r.Body).Decode(&userGroup)
  152. if err != nil {
  153. slog.Error("error decoding request body", "error",
  154. err.Error())
  155. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  156. return
  157. }
  158. err = logic.ValidateUpdateGroupReq(userGroup)
  159. if err != nil {
  160. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  161. return
  162. }
  163. err = logic.UpdateUserGroup(userGroup)
  164. if err != nil {
  165. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  166. return
  167. }
  168. logic.ReturnSuccessResponseWithJson(w, r, userGroup, "updated user group")
  169. }
  170. // swagger:route DELETE /api/v1/user/group user deleteUserGroup
  171. //
  172. // delete user group.
  173. //
  174. // Schemes: https
  175. //
  176. // Security:
  177. // oauth
  178. //
  179. // Responses:
  180. // 200: userBodyResponse
  181. func deleteUserGroup(w http.ResponseWriter, r *http.Request) {
  182. var params = mux.Vars(r)
  183. gid := params["group_id"]
  184. if gid == "" {
  185. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  186. return
  187. }
  188. err := logic.DeleteUserGroup(models.UserGroupID(gid))
  189. if err != nil {
  190. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  191. return
  192. }
  193. logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user group")
  194. }
  195. // swagger:route GET /api/v1/user/roles user listRoles
  196. //
  197. // lists all user roles.
  198. //
  199. // Schemes: https
  200. //
  201. // Security:
  202. // oauth
  203. //
  204. // Responses:
  205. // 200: userBodyResponse
  206. func listRoles(w http.ResponseWriter, r *http.Request) {
  207. roles, err := logic.ListRoles()
  208. if err != nil {
  209. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  210. Code: http.StatusInternalServerError,
  211. Message: err.Error(),
  212. })
  213. return
  214. }
  215. logic.ReturnSuccessResponseWithJson(w, r, roles, "successfully fetched user roles permission templates")
  216. }
  217. // swagger:route GET /api/v1/user/role user getRole
  218. //
  219. // Get user role permission templates.
  220. //
  221. // Schemes: https
  222. //
  223. // Security:
  224. // oauth
  225. //
  226. // Responses:
  227. // 200: userBodyResponse
  228. func getRole(w http.ResponseWriter, r *http.Request) {
  229. var params = mux.Vars(r)
  230. rid := params["role_id"]
  231. if rid == "" {
  232. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  233. return
  234. }
  235. role, err := logic.GetRole(models.UserRole(rid))
  236. if err != nil {
  237. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  238. Code: http.StatusInternalServerError,
  239. Message: err.Error(),
  240. })
  241. return
  242. }
  243. logic.ReturnSuccessResponseWithJson(w, r, role, "successfully fetched user role permission templates")
  244. }
  245. // swagger:route POST /api/v1/user/role user createRole
  246. //
  247. // Create user role permission template.
  248. //
  249. // Schemes: https
  250. //
  251. // Security:
  252. // oauth
  253. //
  254. // Responses:
  255. // 200: userBodyResponse
  256. func createRole(w http.ResponseWriter, r *http.Request) {
  257. var userRole models.UserRolePermissionTemplate
  258. err := json.NewDecoder(r.Body).Decode(&userRole)
  259. if err != nil {
  260. slog.Error("error decoding request body", "error",
  261. err.Error())
  262. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  263. return
  264. }
  265. err = logic.ValidateCreateRoleReq(userRole)
  266. if err != nil {
  267. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  268. return
  269. }
  270. userRole.Default = false
  271. userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
  272. err = logic.CreateRole(userRole)
  273. if err != nil {
  274. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  275. return
  276. }
  277. logic.ReturnSuccessResponseWithJson(w, r, userRole, "created user role")
  278. }
  279. // swagger:route PUT /api/v1/user/role user updateRole
  280. //
  281. // Update user role permission template.
  282. //
  283. // Schemes: https
  284. //
  285. // Security:
  286. // oauth
  287. //
  288. // Responses:
  289. // 200: userBodyResponse
  290. func updateRole(w http.ResponseWriter, r *http.Request) {
  291. var userRole models.UserRolePermissionTemplate
  292. err := json.NewDecoder(r.Body).Decode(&userRole)
  293. if err != nil {
  294. slog.Error("error decoding request body", "error",
  295. err.Error())
  296. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  297. return
  298. }
  299. err = logic.ValidateUpdateRoleReq(userRole)
  300. if err != nil {
  301. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  302. return
  303. }
  304. userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
  305. err = logic.UpdateRole(userRole)
  306. if err != nil {
  307. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  308. return
  309. }
  310. logic.ReturnSuccessResponseWithJson(w, r, userRole, "updated user role")
  311. }
  312. // swagger:route DELETE /api/v1/user/role user deleteRole
  313. //
  314. // Delete user role permission template.
  315. //
  316. // Schemes: https
  317. //
  318. // Security:
  319. // oauth
  320. //
  321. // Responses:
  322. // 200: userBodyResponse
  323. func deleteRole(w http.ResponseWriter, r *http.Request) {
  324. var params = mux.Vars(r)
  325. rid := params["role_id"]
  326. if rid == "" {
  327. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  328. return
  329. }
  330. err := logic.DeleteRole(models.UserRole(rid))
  331. if err != nil {
  332. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  333. return
  334. }
  335. logic.ReturnSuccessResponseWithJson(w, r, nil, "created user role")
  336. }
  337. // swagger:route POST /api/users/adm/authenticate authenticate authenticateUser
  338. //
  339. // User authenticates using its password and retrieves a JWT for authorization.
  340. //
  341. // Schemes: https
  342. //
  343. // Security:
  344. // oauth
  345. //
  346. // Responses:
  347. // 200: successResponse
  348. func authenticateUser(response http.ResponseWriter, request *http.Request) {
  349. // Auth request consists of Mac Address and Password (from node that is authorizing
  350. // in case of Master, auth is ignored and mac is set to "mastermac"
  351. var authRequest models.UserAuthParams
  352. var errorResponse = models.ErrorResponse{
  353. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  354. }
  355. if !servercfg.IsBasicAuthEnabled() {
  356. logic.ReturnErrorResponse(response, request, logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"))
  357. return
  358. }
  359. decoder := json.NewDecoder(request.Body)
  360. decoderErr := decoder.Decode(&authRequest)
  361. defer request.Body.Close()
  362. if decoderErr != nil {
  363. logger.Log(0, "error decoding request body: ",
  364. decoderErr.Error())
  365. logic.ReturnErrorResponse(response, request, errorResponse)
  366. return
  367. }
  368. if val := request.Header.Get("From-Ui"); val == "true" {
  369. // request came from UI, if normal user block Login
  370. user, err := logic.GetUser(authRequest.UserName)
  371. if err != nil {
  372. logger.Log(0, authRequest.UserName, "user validation failed: ",
  373. err.Error())
  374. logic.ReturnErrorResponse(response, request, logic.FormatError(err, "unauthorized"))
  375. return
  376. }
  377. role, err := logic.GetRole(user.PlatformRoleID)
  378. if err != nil {
  379. logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("access denied to dashboard"), "unauthorized"))
  380. return
  381. }
  382. if role.DenyDashboardAccess {
  383. logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("access denied to dashboard"), "unauthorized"))
  384. return
  385. }
  386. }
  387. username := authRequest.UserName
  388. jwt, err := logic.VerifyAuthRequest(authRequest)
  389. if err != nil {
  390. logger.Log(0, username, "user validation failed: ",
  391. err.Error())
  392. logic.ReturnErrorResponse(response, request, logic.FormatError(err, "badrequest"))
  393. return
  394. }
  395. if jwt == "" {
  396. // very unlikely that err is !nil and no jwt returned, but handle it anyways.
  397. logger.Log(0, username, "jwt token is empty")
  398. logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("no token returned"), "internal"))
  399. return
  400. }
  401. var successResponse = models.SuccessResponse{
  402. Code: http.StatusOK,
  403. Message: "W1R3: Device " + username + " Authorized",
  404. Response: models.SuccessfulUserLoginResponse{
  405. AuthToken: jwt,
  406. UserName: username,
  407. },
  408. }
  409. // Send back the JWT
  410. successJSONResponse, jsonError := json.Marshal(successResponse)
  411. if jsonError != nil {
  412. logger.Log(0, username,
  413. "error marshalling resp: ", jsonError.Error())
  414. logic.ReturnErrorResponse(response, request, errorResponse)
  415. return
  416. }
  417. logger.Log(2, username, "was authenticated")
  418. response.Header().Set("Content-Type", "application/json")
  419. response.Write(successJSONResponse)
  420. go func() {
  421. if servercfg.IsPro && servercfg.GetRacAutoDisable() {
  422. // enable all associeated clients for the user
  423. clients, err := logic.GetAllExtClients()
  424. if err != nil {
  425. slog.Error("error getting clients: ", "error", err)
  426. return
  427. }
  428. for _, client := range clients {
  429. if client.OwnerID == username && !client.Enabled {
  430. slog.Info(fmt.Sprintf("enabling ext client %s for user %s due to RAC autodisabling feature", client.ClientID, client.OwnerID))
  431. if newClient, err := logic.ToggleExtClientConnectivity(&client, true); err != nil {
  432. slog.Error("error enabling ext client in RAC autodisable hook", "error", err)
  433. continue // dont return but try for other clients
  434. } else {
  435. // publish peer update to ingress gateway
  436. if ingressNode, err := logic.GetNodeByID(newClient.IngressGatewayID); err == nil {
  437. if err = mq.PublishPeerUpdate(false); err != nil {
  438. slog.Error("error updating ext clients on", "ingress", ingressNode.ID.String(), "err", err.Error())
  439. }
  440. }
  441. }
  442. }
  443. }
  444. }
  445. }()
  446. }
  447. // swagger:route GET /api/users/adm/hassuperadmin user hasSuperAdmin
  448. //
  449. // Checks whether the server has an admin.
  450. //
  451. // Schemes: https
  452. //
  453. // Security:
  454. // oauth
  455. //
  456. // Responses:
  457. // 200: hasAdmin
  458. func hasSuperAdmin(w http.ResponseWriter, r *http.Request) {
  459. w.Header().Set("Content-Type", "application/json")
  460. hasSuperAdmin, err := logic.HasSuperAdmin()
  461. if err != nil {
  462. logger.Log(0, "failed to check for admin: ", err.Error())
  463. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  464. return
  465. }
  466. json.NewEncoder(w).Encode(hasSuperAdmin)
  467. }
  468. // swagger:route GET /api/users/{username} user getUser
  469. //
  470. // Get an individual user.
  471. //
  472. // Schemes: https
  473. //
  474. // Security:
  475. // oauth
  476. //
  477. // Responses:
  478. // 200: userBodyResponse
  479. func getUser(w http.ResponseWriter, r *http.Request) {
  480. // set header.
  481. w.Header().Set("Content-Type", "application/json")
  482. var params = mux.Vars(r)
  483. usernameFetched := params["username"]
  484. user, err := logic.GetReturnUser(usernameFetched)
  485. if err != nil {
  486. logger.Log(0, usernameFetched, "failed to fetch user: ", err.Error())
  487. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  488. return
  489. }
  490. logger.Log(2, r.Header.Get("user"), "fetched user", usernameFetched)
  491. json.NewEncoder(w).Encode(user)
  492. }
  493. // swagger:route GET /api/users user getUsers
  494. //
  495. // Get all users.
  496. //
  497. // Schemes: https
  498. //
  499. // Security:
  500. // oauth
  501. //
  502. // Responses:
  503. // 200: userBodyResponse
  504. func getUsers(w http.ResponseWriter, r *http.Request) {
  505. // set header.
  506. w.Header().Set("Content-Type", "application/json")
  507. users, err := logic.GetUsers()
  508. if err != nil {
  509. logger.Log(0, "failed to fetch users: ", err.Error())
  510. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  511. return
  512. }
  513. logic.SortUsers(users[:])
  514. logger.Log(2, r.Header.Get("user"), "fetched users")
  515. json.NewEncoder(w).Encode(users)
  516. }
  517. // swagger:route POST /api/users/adm/createsuperadmin user createAdmin
  518. //
  519. // Make a user an admin.
  520. //
  521. // Schemes: https
  522. //
  523. // Security:
  524. // oauth
  525. //
  526. // Responses:
  527. // 200: userBodyResponse
  528. func createSuperAdmin(w http.ResponseWriter, r *http.Request) {
  529. w.Header().Set("Content-Type", "application/json")
  530. var u models.User
  531. err := json.NewDecoder(r.Body).Decode(&u)
  532. if err != nil {
  533. slog.Error("error decoding request body", "error", err.Error())
  534. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  535. return
  536. }
  537. if !servercfg.IsBasicAuthEnabled() {
  538. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"))
  539. return
  540. }
  541. err = logic.CreateSuperAdmin(&u)
  542. if err != nil {
  543. slog.Error("failed to create admin", "error", err.Error())
  544. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  545. return
  546. }
  547. logger.Log(1, u.UserName, "was made a super admin")
  548. json.NewEncoder(w).Encode(logic.ToReturnUser(u))
  549. }
  550. // swagger:route POST /api/users/adm/transfersuperadmin user transferSuperAdmin
  551. //
  552. // Transfers superadmin role to an admin user.
  553. //
  554. // Schemes: https
  555. //
  556. // Security:
  557. // oauth
  558. //
  559. // Responses:
  560. // 200: userBodyResponse
  561. func transferSuperAdmin(w http.ResponseWriter, r *http.Request) {
  562. w.Header().Set("Content-Type", "application/json")
  563. caller, err := logic.GetUser(r.Header.Get("user"))
  564. if err != nil {
  565. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  566. }
  567. if !caller.IsSuperAdmin {
  568. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can assign the superadmin role to another user"), "forbidden"))
  569. return
  570. }
  571. var params = mux.Vars(r)
  572. username := params["username"]
  573. u, err := logic.GetUser(username)
  574. if err != nil {
  575. slog.Error("error getting user", "user", u.UserName, "error", err.Error())
  576. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  577. return
  578. }
  579. if !u.IsAdmin {
  580. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only admins can be promoted to superadmin role"), "forbidden"))
  581. return
  582. }
  583. if !servercfg.IsBasicAuthEnabled() {
  584. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"))
  585. return
  586. }
  587. u.IsSuperAdmin = true
  588. u.IsAdmin = false
  589. err = logic.UpsertUser(*u)
  590. if err != nil {
  591. slog.Error("error updating user to superadmin: ", "user", u.UserName, "error", err.Error())
  592. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  593. return
  594. }
  595. caller.IsSuperAdmin = false
  596. caller.IsAdmin = true
  597. err = logic.UpsertUser(*caller)
  598. if err != nil {
  599. slog.Error("error demoting user to admin: ", "user", caller.UserName, "error", err.Error())
  600. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  601. return
  602. }
  603. slog.Info("user was made a super admin", "user", u.UserName)
  604. json.NewEncoder(w).Encode(logic.ToReturnUser(*u))
  605. }
  606. // swagger:route POST /api/users/{username} user createUser
  607. //
  608. // Create a user.
  609. //
  610. // Schemes: https
  611. //
  612. // Security:
  613. // oauth
  614. //
  615. // Responses:
  616. // 200: userBodyResponse
  617. func createUser(w http.ResponseWriter, r *http.Request) {
  618. w.Header().Set("Content-Type", "application/json")
  619. caller, err := logic.GetUser(r.Header.Get("user"))
  620. if err != nil {
  621. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  622. return
  623. }
  624. callerUserRole, err := logic.GetRole(caller.PlatformRoleID)
  625. if err != nil {
  626. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  627. return
  628. }
  629. var user models.User
  630. err = json.NewDecoder(r.Body).Decode(&user)
  631. if err != nil {
  632. logger.Log(0, user.UserName, "error decoding request body: ",
  633. err.Error())
  634. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  635. return
  636. }
  637. uniqueGroupsPlatformRole := make(map[models.UserRole]struct{})
  638. for groupID := range user.UserGroups {
  639. userG, err := logic.GetUserGroup(groupID)
  640. if err != nil {
  641. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  642. return
  643. }
  644. uniqueGroupsPlatformRole[userG.PlatformRole] = struct{}{}
  645. }
  646. if len(uniqueGroupsPlatformRole) > 1 {
  647. err = errors.New("only groups with same platform role can be assigned to an user")
  648. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  649. return
  650. }
  651. userRole, err := logic.GetRole(user.PlatformRoleID)
  652. if err != nil {
  653. err = errors.New("error fetching role " + user.PlatformRoleID.String() + " " + err.Error())
  654. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  655. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  656. return
  657. }
  658. if userRole.ID == models.SuperAdminRole {
  659. err = errors.New("additional superadmins cannot be created")
  660. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  661. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  662. return
  663. }
  664. if callerUserRole.ID != models.SuperAdminRole && user.IsAdmin {
  665. err = errors.New("only superadmin can create admin users")
  666. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  667. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  668. return
  669. }
  670. if !servercfg.IsPro && !user.IsAdmin {
  671. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("non-admins users can only be created on Pro version"), "forbidden"))
  672. return
  673. }
  674. if userRole.ID == models.AdminRole {
  675. user.IsAdmin = true
  676. }
  677. err = logic.CreateUser(&user)
  678. if err != nil {
  679. slog.Error("error creating new user: ", "user", user.UserName, "error", err.Error())
  680. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  681. return
  682. }
  683. logic.DeleteUserInvite(user.UserName)
  684. logic.DeletePendingUser(user.UserName)
  685. slog.Info("user was created", "username", user.UserName)
  686. json.NewEncoder(w).Encode(logic.ToReturnUser(user))
  687. }
  688. // swagger:route PUT /api/users/{username} user updateUser
  689. //
  690. // Update a user.
  691. //
  692. // Schemes: https
  693. //
  694. // Security:
  695. // oauth
  696. //
  697. // Responses:
  698. // 200: userBodyResponse
  699. func updateUser(w http.ResponseWriter, r *http.Request) {
  700. w.Header().Set("Content-Type", "application/json")
  701. var params = mux.Vars(r)
  702. // start here
  703. var caller *models.User
  704. var err error
  705. var ismaster bool
  706. if r.Header.Get("user") == logic.MasterUser {
  707. ismaster = true
  708. } else {
  709. caller, err = logic.GetUser(r.Header.Get("user"))
  710. if err != nil {
  711. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  712. }
  713. }
  714. username := params["username"]
  715. user, err := logic.GetUser(username)
  716. if err != nil {
  717. logger.Log(0, username,
  718. "failed to update user info: ", err.Error())
  719. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  720. return
  721. }
  722. var userchange models.User
  723. // we decode our body request params
  724. err = json.NewDecoder(r.Body).Decode(&userchange)
  725. if err != nil {
  726. slog.Error("failed to decode body", "error ", err.Error())
  727. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  728. return
  729. }
  730. if user.UserName != userchange.UserName {
  731. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user in param and request body not matching"), "badrequest"))
  732. return
  733. }
  734. selfUpdate := false
  735. if !ismaster && caller.UserName == user.UserName {
  736. selfUpdate = true
  737. }
  738. if !ismaster && !selfUpdate {
  739. if caller.IsAdmin && user.IsSuperAdmin {
  740. slog.Error("non-superadmin user", "caller", caller.UserName, "attempted to update superadmin user", username)
  741. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
  742. return
  743. }
  744. if !caller.IsAdmin && !caller.IsSuperAdmin {
  745. slog.Error("operation not allowed", "caller", caller.UserName, "attempted to update user", username)
  746. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
  747. return
  748. }
  749. if caller.IsAdmin && user.IsAdmin {
  750. slog.Error("admin user cannot update another admin", "caller", caller.UserName, "attempted to update admin user", username)
  751. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("admin user cannot update another admin"), "forbidden"))
  752. return
  753. }
  754. if caller.IsAdmin && userchange.IsAdmin {
  755. err = errors.New("admin user cannot update role of an another user to admin")
  756. slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
  757. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  758. return
  759. }
  760. }
  761. if !ismaster && selfUpdate {
  762. if user.IsAdmin != userchange.IsAdmin || user.IsSuperAdmin != userchange.IsSuperAdmin {
  763. slog.Error("user cannot change his own role", "caller", caller.UserName, "attempted to update user role", username)
  764. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user not allowed to self assign role"), "forbidden"))
  765. return
  766. }
  767. }
  768. if ismaster {
  769. if !user.IsSuperAdmin && userchange.IsSuperAdmin {
  770. slog.Error("operation not allowed", "caller", logic.MasterUser, "attempted to update user role to superadmin", username)
  771. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("attempted to update user role to superadmin"), "forbidden"))
  772. return
  773. }
  774. }
  775. if auth.IsOauthUser(user) == nil && userchange.Password != "" {
  776. err := fmt.Errorf("cannot update user's password for an oauth user %s", username)
  777. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  778. return
  779. }
  780. user, err = logic.UpdateUser(&userchange, user)
  781. if err != nil {
  782. logger.Log(0, username,
  783. "failed to update user info: ", err.Error())
  784. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  785. return
  786. }
  787. logger.Log(1, username, "was updated")
  788. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  789. }
  790. // swagger:route DELETE /api/users/{username} user deleteUser
  791. //
  792. // Delete a user.
  793. //
  794. // Schemes: https
  795. //
  796. // Security:
  797. // oauth
  798. //
  799. // Responses:
  800. // 200: userBodyResponse
  801. func deleteUser(w http.ResponseWriter, r *http.Request) {
  802. // Set header
  803. w.Header().Set("Content-Type", "application/json")
  804. // get params
  805. var params = mux.Vars(r)
  806. caller, err := logic.GetUser(r.Header.Get("user"))
  807. if err != nil {
  808. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  809. }
  810. username := params["username"]
  811. user, err := logic.GetUser(username)
  812. if err != nil {
  813. logger.Log(0, username,
  814. "failed to update user info: ", err.Error())
  815. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  816. return
  817. }
  818. if user.IsSuperAdmin {
  819. slog.Error(
  820. "failed to delete user: ", "user", username, "error", "superadmin cannot be deleted")
  821. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("superadmin cannot be deleted"), "internal"))
  822. return
  823. }
  824. if !caller.IsSuperAdmin {
  825. if caller.IsAdmin && user.IsAdmin {
  826. slog.Error(
  827. "failed to delete user: ", "user", username, "error", "admin cannot delete another admin user, including oneself")
  828. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("admin cannot delete another admin user, including oneself"), "internal"))
  829. return
  830. }
  831. }
  832. success, err := logic.DeleteUser(username)
  833. if err != nil {
  834. logger.Log(0, username,
  835. "failed to delete user: ", err.Error())
  836. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  837. return
  838. } else if !success {
  839. err := errors.New("delete unsuccessful")
  840. logger.Log(0, username, err.Error())
  841. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  842. return
  843. }
  844. // check and delete extclient with this ownerID
  845. go func() {
  846. extclients, err := logic.GetAllExtClients()
  847. if err != nil {
  848. slog.Error("failed to get extclients", "error", err)
  849. return
  850. }
  851. for _, extclient := range extclients {
  852. if extclient.OwnerID == user.UserName {
  853. err = logic.DeleteExtClient(extclient.Network, extclient.ClientID)
  854. if err != nil {
  855. slog.Error("failed to delete extclient",
  856. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  857. }
  858. }
  859. }
  860. if servercfg.IsDNSMode() {
  861. logic.SetDNS()
  862. }
  863. }()
  864. logger.Log(1, username, "was deleted")
  865. json.NewEncoder(w).Encode(params["username"] + " deleted.")
  866. }
  867. // Called when vpn client dials in to start the auth flow and first stage is to get register URL itself
  868. func socketHandler(w http.ResponseWriter, r *http.Request) {
  869. // Upgrade our raw HTTP connection to a websocket based one
  870. conn, err := upgrader.Upgrade(w, r, nil)
  871. if err != nil {
  872. logger.Log(0, "error during connection upgrade for node sign-in:", err.Error())
  873. return
  874. }
  875. if conn == nil {
  876. logger.Log(0, "failed to establish web-socket connection during node sign-in")
  877. return
  878. }
  879. // Start handling the session
  880. go auth.SessionHandler(conn)
  881. }
  882. // swagger:route GET /api/users_pending user getPendingUsers
  883. //
  884. // Get all pending users.
  885. //
  886. // Schemes: https
  887. //
  888. // Security:
  889. // oauth
  890. //
  891. // Responses:
  892. // 200: userBodyResponse
  893. func getPendingUsers(w http.ResponseWriter, r *http.Request) {
  894. // set header.
  895. w.Header().Set("Content-Type", "application/json")
  896. users, err := logic.ListPendingUsers()
  897. if err != nil {
  898. logger.Log(0, "failed to fetch users: ", err.Error())
  899. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  900. return
  901. }
  902. logic.SortUsers(users[:])
  903. logger.Log(2, r.Header.Get("user"), "fetched pending users")
  904. json.NewEncoder(w).Encode(users)
  905. }
  906. // swagger:route POST /api/users_pending/user/{username} user approvePendingUser
  907. //
  908. // approve pending user.
  909. //
  910. // Schemes: https
  911. //
  912. // Security:
  913. // oauth
  914. //
  915. // Responses:
  916. // 200: userBodyResponse
  917. func approvePendingUser(w http.ResponseWriter, r *http.Request) {
  918. // set header.
  919. w.Header().Set("Content-Type", "application/json")
  920. var params = mux.Vars(r)
  921. username := params["username"]
  922. users, err := logic.ListPendingUsers()
  923. if err != nil {
  924. logger.Log(0, "failed to fetch users: ", err.Error())
  925. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  926. return
  927. }
  928. for _, user := range users {
  929. if user.UserName == username {
  930. var newPass, fetchErr = auth.FetchPassValue("")
  931. if fetchErr != nil {
  932. logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
  933. return
  934. }
  935. if err = logic.CreateUser(&models.User{
  936. UserName: user.UserName,
  937. Password: newPass,
  938. }); err != nil {
  939. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to create user: %s", err), "internal"))
  940. return
  941. }
  942. err = logic.DeletePendingUser(username)
  943. if err != nil {
  944. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
  945. return
  946. }
  947. break
  948. }
  949. }
  950. logic.ReturnSuccessResponse(w, r, "approved "+username)
  951. }
  952. // swagger:route DELETE /api/users_pending/user/{username} user deletePendingUser
  953. //
  954. // delete pending user.
  955. //
  956. // Schemes: https
  957. //
  958. // Security:
  959. // oauth
  960. //
  961. // Responses:
  962. // 200: userBodyResponse
  963. func deletePendingUser(w http.ResponseWriter, r *http.Request) {
  964. // set header.
  965. w.Header().Set("Content-Type", "application/json")
  966. var params = mux.Vars(r)
  967. username := params["username"]
  968. users, err := logic.ListPendingUsers()
  969. if err != nil {
  970. logger.Log(0, "failed to fetch users: ", err.Error())
  971. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  972. return
  973. }
  974. for _, user := range users {
  975. if user.UserName == username {
  976. err = logic.DeletePendingUser(username)
  977. if err != nil {
  978. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
  979. return
  980. }
  981. break
  982. }
  983. }
  984. logic.ReturnSuccessResponse(w, r, "deleted pending "+username)
  985. }
  986. // swagger:route DELETE /api/users_pending/{username}/pending user deleteAllPendingUsers
  987. //
  988. // delete all pending users.
  989. //
  990. // Schemes: https
  991. //
  992. // Security:
  993. // oauth
  994. //
  995. // Responses:
  996. // 200: userBodyResponse
  997. func deleteAllPendingUsers(w http.ResponseWriter, r *http.Request) {
  998. // set header.
  999. err := database.DeleteAllRecords(database.PENDING_USERS_TABLE_NAME)
  1000. if err != nil {
  1001. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending users "+err.Error()), "internal"))
  1002. return
  1003. }
  1004. logic.ReturnSuccessResponse(w, r, "cleared all pending users")
  1005. }
  1006. // swagger:route POST /api/v1/users/invite-signup user userInviteSignUp
  1007. //
  1008. // user signup via invite.
  1009. //
  1010. // Schemes: https
  1011. //
  1012. // Responses:
  1013. // 200: ReturnSuccessResponse
  1014. func userInviteSignUp(w http.ResponseWriter, r *http.Request) {
  1015. email, _ := url.QueryUnescape(r.URL.Query().Get("email"))
  1016. code, _ := url.QueryUnescape(r.URL.Query().Get("code"))
  1017. in, err := logic.GetUserInvite(email)
  1018. if err != nil {
  1019. logger.Log(0, "failed to fetch users: ", err.Error())
  1020. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1021. return
  1022. }
  1023. if code != in.InviteCode {
  1024. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid invite code"), "badrequest"))
  1025. return
  1026. }
  1027. // check if user already exists
  1028. _, err = logic.GetUser(email)
  1029. if err == nil {
  1030. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user already exists"), "badrequest"))
  1031. return
  1032. }
  1033. var user models.User
  1034. err = json.NewDecoder(r.Body).Decode(&user)
  1035. if err != nil {
  1036. logger.Log(0, user.UserName, "error decoding request body: ",
  1037. err.Error())
  1038. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1039. return
  1040. }
  1041. if user.UserName != email {
  1042. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username not matching with invite"), "badrequest"))
  1043. return
  1044. }
  1045. if user.Password == "" {
  1046. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("password cannot be empty"), "badrequest"))
  1047. return
  1048. }
  1049. for _, inviteGroupID := range in.Groups {
  1050. userG, err := logic.GetUserGroup(inviteGroupID)
  1051. if err != nil {
  1052. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("error fetching group id "+inviteGroupID.String()), "badrequest"))
  1053. return
  1054. }
  1055. user.PlatformRoleID = userG.PlatformRole
  1056. user.UserGroups[inviteGroupID] = struct{}{}
  1057. }
  1058. if user.PlatformRoleID == models.AdminRole {
  1059. user.IsAdmin = true
  1060. }
  1061. user.NetworkRoles = make(map[models.NetworkID]map[models.UserRole]struct{})
  1062. user.IsSuperAdmin = false
  1063. err = logic.CreateUser(&user)
  1064. if err != nil {
  1065. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1066. return
  1067. }
  1068. // delete invite
  1069. logic.DeleteUserInvite(email)
  1070. logic.DeletePendingUser(email)
  1071. logic.ReturnSuccessResponse(w, r, "created user successfully "+email)
  1072. }
  1073. // swagger:route GET /api/v1/users/invite user userInviteVerify
  1074. //
  1075. // verfies user invite.
  1076. //
  1077. // Schemes: https
  1078. //
  1079. // Responses:
  1080. // 200: ReturnSuccessResponse
  1081. func userInviteVerify(w http.ResponseWriter, r *http.Request) {
  1082. email, _ := url.QueryUnescape(r.URL.Query().Get("email"))
  1083. code, _ := url.QueryUnescape(r.URL.Query().Get("code"))
  1084. logger.Log(0, "EMAIL", email, "CODE", code)
  1085. err := logic.ValidateAndApproveUserInvite(email, code)
  1086. if err != nil {
  1087. logger.Log(0, "failed to fetch users: ", err.Error())
  1088. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1089. return
  1090. }
  1091. logic.ReturnSuccessResponse(w, r, "invite is valid")
  1092. }
  1093. // swagger:route POST /api/v1/users/invite user inviteUsers
  1094. //
  1095. // invite users.
  1096. //
  1097. // Schemes: https
  1098. //
  1099. // Security:
  1100. // oauth
  1101. //
  1102. // Responses:
  1103. // 200: userBodyResponse
  1104. func inviteUsers(w http.ResponseWriter, r *http.Request) {
  1105. var inviteReq models.InviteUsersReq
  1106. err := json.NewDecoder(r.Body).Decode(&inviteReq)
  1107. if err != nil {
  1108. slog.Error("error decoding request body", "error",
  1109. err.Error())
  1110. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1111. return
  1112. }
  1113. //validate Req
  1114. uniqueGroupsPlatformRole := make(map[models.UserRole]struct{})
  1115. for _, groupID := range inviteReq.Groups {
  1116. userG, err := logic.GetUserGroup(groupID)
  1117. if err != nil {
  1118. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1119. return
  1120. }
  1121. uniqueGroupsPlatformRole[userG.PlatformRole] = struct{}{}
  1122. }
  1123. if len(uniqueGroupsPlatformRole) > 1 {
  1124. err = errors.New("only groups with same platform role can be assigned to an user")
  1125. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1126. return
  1127. }
  1128. for _, inviteeEmail := range inviteReq.UserEmails {
  1129. // check if user with email exists, then ignore
  1130. _, err := logic.GetUser(inviteeEmail)
  1131. if err == nil {
  1132. // user exists already, so ignore
  1133. continue
  1134. }
  1135. invite := models.UserInvite{
  1136. Email: inviteeEmail,
  1137. Groups: inviteReq.Groups,
  1138. InviteCode: logic.RandomString(8),
  1139. }
  1140. err = logic.InsertUserInvite(invite)
  1141. if err != nil {
  1142. slog.Error("failed to insert invite for user", "email", invite.Email, "error", err)
  1143. }
  1144. // notify user with magic link
  1145. go func(invite models.UserInvite) {
  1146. err = logic.SendInviteEmail(invite)
  1147. if err != nil {
  1148. slog.Error("failed to send email invite", "user", invite.Email, "error", err)
  1149. }
  1150. }(invite)
  1151. }
  1152. }
  1153. // swagger:route GET /api/v1/users/invites user listUserInvites
  1154. //
  1155. // lists all pending invited users.
  1156. //
  1157. // Schemes: https
  1158. //
  1159. // Security:
  1160. // oauth
  1161. //
  1162. // Responses:
  1163. // 200: ReturnSuccessResponseWithJson
  1164. func listUserInvites(w http.ResponseWriter, r *http.Request) {
  1165. usersInvites, err := logic.ListUserInvites()
  1166. if err != nil {
  1167. logger.Log(0, "failed to fetch users: ", err.Error())
  1168. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1169. return
  1170. }
  1171. logic.ReturnSuccessResponseWithJson(w, r, usersInvites, "fetched pending user invites")
  1172. }
  1173. // swagger:route DELETE /api/v1/users/invite user deleteUserInvite
  1174. //
  1175. // delete pending invite.
  1176. //
  1177. // Schemes: https
  1178. //
  1179. // Security:
  1180. // oauth
  1181. //
  1182. // Responses:
  1183. // 200: ReturnSuccessResponse
  1184. func deleteUserInvite(w http.ResponseWriter, r *http.Request) {
  1185. var params = mux.Vars(r)
  1186. username := params["invitee_email"]
  1187. err := logic.DeleteUserInvite(username)
  1188. if err != nil {
  1189. logger.Log(0, "failed to delete user invite: ", username, err.Error())
  1190. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1191. return
  1192. }
  1193. logic.ReturnSuccessResponse(w, r, "deleted user invite")
  1194. }
  1195. // swagger:route DELETE /api/v1/users/invites user deleteAllUserInvites
  1196. //
  1197. // deletes all pending invites.
  1198. //
  1199. // Schemes: https
  1200. //
  1201. // Security:
  1202. // oauth
  1203. //
  1204. // Responses:
  1205. // 200: ReturnSuccessResponse
  1206. func deleteAllUserInvites(w http.ResponseWriter, r *http.Request) {
  1207. err := database.DeleteAllRecords(database.USER_INVITES_TABLE_NAME)
  1208. if err != nil {
  1209. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending user invites "+err.Error()), "internal"))
  1210. return
  1211. }
  1212. logic.ReturnSuccessResponse(w, r, "cleared all pending user invites")
  1213. }