users.go 60 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910
  1. package controllers
  2. import (
  3. "context"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "net/http"
  8. "net/url"
  9. "strings"
  10. "time"
  11. "github.com/gravitl/netmaker/pro/idp"
  12. "github.com/gravitl/netmaker/pro/idp/azure"
  13. "github.com/gravitl/netmaker/pro/idp/google"
  14. "github.com/gravitl/netmaker/pro/idp/okta"
  15. "github.com/google/uuid"
  16. "github.com/gorilla/mux"
  17. "github.com/gravitl/netmaker/database"
  18. "github.com/gravitl/netmaker/logger"
  19. "github.com/gravitl/netmaker/logic"
  20. "github.com/gravitl/netmaker/models"
  21. "github.com/gravitl/netmaker/mq"
  22. proAuth "github.com/gravitl/netmaker/pro/auth"
  23. "github.com/gravitl/netmaker/pro/email"
  24. proLogic "github.com/gravitl/netmaker/pro/logic"
  25. "github.com/gravitl/netmaker/servercfg"
  26. "github.com/gravitl/netmaker/utils"
  27. "golang.org/x/exp/slog"
  28. )
  29. func UserHandlers(r *mux.Router) {
  30. r.HandleFunc("/api/oauth/login", proAuth.HandleAuthLogin).Methods(http.MethodGet)
  31. r.HandleFunc("/api/oauth/callback", proAuth.HandleAuthCallback).Methods(http.MethodGet)
  32. r.HandleFunc("/api/oauth/headless", proAuth.HandleHeadlessSSO)
  33. r.HandleFunc("/api/oauth/register/{regKey}", proAuth.RegisterHostSSO).Methods(http.MethodGet)
  34. // User Role Handlers
  35. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(getRole))).Methods(http.MethodGet)
  36. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(createRole))).Methods(http.MethodPost)
  37. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(updateRole))).Methods(http.MethodPut)
  38. r.HandleFunc("/api/v1/users/role", logic.SecurityCheck(true, http.HandlerFunc(deleteRole))).Methods(http.MethodDelete)
  39. // User Group Handlers
  40. r.HandleFunc("/api/v1/users/groups", logic.SecurityCheck(true, http.HandlerFunc(listUserGroups))).Methods(http.MethodGet)
  41. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(getUserGroup))).Methods(http.MethodGet)
  42. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(createUserGroup))).Methods(http.MethodPost)
  43. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(updateUserGroup))).Methods(http.MethodPut)
  44. r.HandleFunc("/api/v1/users/group", logic.SecurityCheck(true, http.HandlerFunc(deleteUserGroup))).Methods(http.MethodDelete)
  45. // User Invite Handlers
  46. r.HandleFunc("/api/v1/users/invite", userInviteVerify).Methods(http.MethodGet)
  47. r.HandleFunc("/api/v1/users/invite-signup", userInviteSignUp).Methods(http.MethodPost)
  48. r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(inviteUsers))).Methods(http.MethodPost)
  49. r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(listUserInvites))).Methods(http.MethodGet)
  50. r.HandleFunc("/api/v1/users/invite", logic.SecurityCheck(true, http.HandlerFunc(deleteUserInvite))).Methods(http.MethodDelete)
  51. r.HandleFunc("/api/v1/users/invites", logic.SecurityCheck(true, http.HandlerFunc(deleteAllUserInvites))).Methods(http.MethodDelete)
  52. r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(getPendingUsers))).Methods(http.MethodGet)
  53. r.HandleFunc("/api/users_pending", logic.SecurityCheck(true, http.HandlerFunc(deleteAllPendingUsers))).Methods(http.MethodDelete)
  54. r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(deletePendingUser))).Methods(http.MethodDelete)
  55. r.HandleFunc("/api/users_pending/user/{username}", logic.SecurityCheck(true, http.HandlerFunc(approvePendingUser))).Methods(http.MethodPost)
  56. r.HandleFunc("/api/users/{username}/remote_access_gw/{remote_access_gateway_id}", logic.SecurityCheck(true, http.HandlerFunc(attachUserToRemoteAccessGw))).Methods(http.MethodPost)
  57. r.HandleFunc("/api/users/{username}/remote_access_gw/{remote_access_gateway_id}", logic.SecurityCheck(true, http.HandlerFunc(removeUserFromRemoteAccessGW))).Methods(http.MethodDelete)
  58. r.HandleFunc("/api/users/{username}/remote_access_gw", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUserRemoteAccessGwsV1)))).Methods(http.MethodGet)
  59. r.HandleFunc("/api/users/ingress/{ingress_id}", logic.SecurityCheck(true, http.HandlerFunc(ingressGatewayUsers))).Methods(http.MethodGet)
  60. r.HandleFunc("/api/idp/sync", logic.SecurityCheck(true, http.HandlerFunc(syncIDP))).Methods(http.MethodPost)
  61. r.HandleFunc("/api/idp/sync/test", logic.SecurityCheck(true, http.HandlerFunc(testIDPSync))).Methods(http.MethodPost)
  62. r.HandleFunc("/api/idp/sync/status", logic.SecurityCheck(true, http.HandlerFunc(getIDPSyncStatus))).Methods(http.MethodGet)
  63. r.HandleFunc("/api/idp", logic.SecurityCheck(true, http.HandlerFunc(removeIDPIntegration))).Methods(http.MethodDelete)
  64. }
  65. // swagger:route POST /api/v1/users/invite-signup user userInviteSignUp
  66. //
  67. // user signup via invite.
  68. //
  69. // Schemes: https
  70. //
  71. // Responses:
  72. // 200: ReturnSuccessResponse
  73. func userInviteSignUp(w http.ResponseWriter, r *http.Request) {
  74. email := r.URL.Query().Get("email")
  75. code := r.URL.Query().Get("invite_code")
  76. in, err := logic.GetUserInvite(email)
  77. if err != nil {
  78. logger.Log(0, "failed to fetch users: ", err.Error())
  79. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  80. return
  81. }
  82. if code != in.InviteCode {
  83. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid invite code"), "badrequest"))
  84. return
  85. }
  86. // check if user already exists
  87. _, err = logic.GetUser(email)
  88. if err == nil {
  89. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user already exists"), "badrequest"))
  90. return
  91. }
  92. var user models.User
  93. err = json.NewDecoder(r.Body).Decode(&user)
  94. if err != nil {
  95. logger.Log(0, user.UserName, "error decoding request body: ",
  96. err.Error())
  97. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  98. return
  99. }
  100. if user.UserName != email {
  101. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username not matching with invite"), "badrequest"))
  102. return
  103. }
  104. if user.Password == "" {
  105. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("password cannot be empty"), "badrequest"))
  106. return
  107. }
  108. user.UserGroups = in.UserGroups
  109. user.PlatformRoleID = models.UserRoleID(in.PlatformRoleID)
  110. if user.PlatformRoleID == "" {
  111. user.PlatformRoleID = models.ServiceUser
  112. }
  113. user.NetworkRoles = in.NetworkRoles
  114. err = logic.CreateUser(&user)
  115. if err != nil {
  116. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  117. return
  118. }
  119. // delete invite
  120. logic.DeleteUserInvite(email)
  121. logic.DeletePendingUser(email)
  122. w.Header().Set("Access-Control-Allow-Origin", "*")
  123. logic.ReturnSuccessResponse(w, r, "created user successfully "+email)
  124. }
  125. // swagger:route GET /api/v1/users/invite user userInviteVerify
  126. //
  127. // verfies user invite.
  128. //
  129. // Schemes: https
  130. //
  131. // Responses:
  132. // 200: ReturnSuccessResponse
  133. func userInviteVerify(w http.ResponseWriter, r *http.Request) {
  134. email := r.URL.Query().Get("email")
  135. code := r.URL.Query().Get("invite_code")
  136. err := logic.ValidateAndApproveUserInvite(email, code)
  137. if err != nil {
  138. logger.Log(0, "failed to fetch users: ", err.Error())
  139. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  140. return
  141. }
  142. logic.ReturnSuccessResponse(w, r, "invite is valid")
  143. }
  144. // swagger:route POST /api/v1/users/invite user inviteUsers
  145. //
  146. // invite users.
  147. //
  148. // Schemes: https
  149. //
  150. // Security:
  151. // oauth
  152. //
  153. // Responses:
  154. // 200: userBodyResponse
  155. func inviteUsers(w http.ResponseWriter, r *http.Request) {
  156. var inviteReq models.InviteUsersReq
  157. err := json.NewDecoder(r.Body).Decode(&inviteReq)
  158. if err != nil {
  159. slog.Error("error decoding request body", "error",
  160. err.Error())
  161. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  162. return
  163. }
  164. callerUserName := r.Header.Get("user")
  165. if r.Header.Get("ismaster") != "yes" {
  166. caller, err := logic.GetUser(callerUserName)
  167. if err != nil {
  168. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
  169. return
  170. }
  171. if inviteReq.PlatformRoleID == models.SuperAdminRole.String() {
  172. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("super admin cannot be invited"), "badrequest"))
  173. return
  174. }
  175. if inviteReq.PlatformRoleID == "" {
  176. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("platform role id cannot be empty"), "badrequest"))
  177. return
  178. }
  179. if (inviteReq.PlatformRoleID == models.AdminRole.String() ||
  180. inviteReq.PlatformRoleID == models.SuperAdminRole.String()) && caller.PlatformRoleID != models.SuperAdminRole {
  181. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can invite admin users"), "forbidden"))
  182. return
  183. }
  184. }
  185. //validate Req
  186. err = proLogic.IsGroupsValid(inviteReq.UserGroups)
  187. if err != nil {
  188. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  189. return
  190. }
  191. err = proLogic.IsNetworkRolesValid(inviteReq.NetworkRoles)
  192. if err != nil {
  193. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  194. return
  195. }
  196. // check platform role
  197. _, err = logic.GetRole(models.UserRoleID(inviteReq.PlatformRoleID))
  198. if err != nil {
  199. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  200. return
  201. }
  202. for _, inviteeEmail := range inviteReq.UserEmails {
  203. inviteeEmail = strings.ToLower(inviteeEmail)
  204. // check if user with email exists, then ignore
  205. if !email.IsValid(inviteeEmail) {
  206. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid email "+inviteeEmail), "badrequest"))
  207. return
  208. }
  209. _, err := logic.GetUser(inviteeEmail)
  210. if err == nil {
  211. // user exists already, so ignore
  212. continue
  213. }
  214. invite := models.UserInvite{
  215. Email: inviteeEmail,
  216. PlatformRoleID: inviteReq.PlatformRoleID,
  217. UserGroups: inviteReq.UserGroups,
  218. NetworkRoles: inviteReq.NetworkRoles,
  219. InviteCode: logic.RandomString(8),
  220. }
  221. frontendURL := strings.TrimSuffix(servercfg.GetFrontendURL(), "/")
  222. if frontendURL == "" {
  223. frontendURL = fmt.Sprintf("https://dashboard.%s", servercfg.GetNmBaseDomain())
  224. }
  225. u, err := url.Parse(fmt.Sprintf("%s/invite?email=%s&invite_code=%s",
  226. frontendURL, url.QueryEscape(invite.Email), url.QueryEscape(invite.InviteCode)))
  227. if err != nil {
  228. slog.Error("failed to parse to invite url", "error", err)
  229. return
  230. }
  231. if servercfg.DeployedByOperator() {
  232. u, err = url.Parse(fmt.Sprintf("%s/invite?tenant_id=%s&email=%s&invite_code=%s",
  233. proLogic.GetAccountsUIHost(), url.QueryEscape(servercfg.GetNetmakerTenantID()), url.QueryEscape(invite.Email), url.QueryEscape(invite.InviteCode)))
  234. if err != nil {
  235. slog.Error("failed to parse to invite url", "error", err)
  236. return
  237. }
  238. }
  239. invite.InviteURL = u.String()
  240. err = logic.InsertUserInvite(invite)
  241. if err != nil {
  242. slog.Error("failed to insert invite for user", "email", invite.Email, "error", err)
  243. }
  244. logic.LogEvent(&models.Event{
  245. Action: models.Create,
  246. Source: models.Subject{
  247. ID: callerUserName,
  248. Name: callerUserName,
  249. Type: models.UserSub,
  250. Info: invite,
  251. },
  252. TriggeredBy: callerUserName,
  253. Target: models.Subject{
  254. ID: inviteeEmail,
  255. Name: inviteeEmail,
  256. Type: models.UserInviteSub,
  257. },
  258. Origin: models.Dashboard,
  259. })
  260. // notify user with magic link
  261. go func(invite models.UserInvite) {
  262. // Set E-Mail body. You can set plain text or html with text/html
  263. e := email.UserInvitedMail{
  264. BodyBuilder: &email.EmailBodyBuilderWithH1HeadlineAndImage{},
  265. InviteURL: invite.InviteURL,
  266. PlatformRoleID: invite.PlatformRoleID,
  267. }
  268. n := email.Notification{
  269. RecipientMail: invite.Email,
  270. }
  271. err = email.GetClient().SendEmail(context.Background(), n, e)
  272. if err != nil {
  273. slog.Error("failed to send email invite", "user", invite.Email, "error", err)
  274. }
  275. }(invite)
  276. }
  277. logic.ReturnSuccessResponse(w, r, "triggered user invites")
  278. }
  279. // swagger:route GET /api/v1/users/invites user listUserInvites
  280. //
  281. // lists all pending invited users.
  282. //
  283. // Schemes: https
  284. //
  285. // Security:
  286. // oauth
  287. //
  288. // Responses:
  289. // 200: ReturnSuccessResponseWithJson
  290. func listUserInvites(w http.ResponseWriter, r *http.Request) {
  291. usersInvites, err := logic.ListUserInvites()
  292. if err != nil {
  293. logger.Log(0, "failed to fetch users: ", err.Error())
  294. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  295. return
  296. }
  297. logic.ReturnSuccessResponseWithJson(w, r, usersInvites, "fetched pending user invites")
  298. }
  299. // swagger:route DELETE /api/v1/users/invite user deleteUserInvite
  300. //
  301. // delete pending invite.
  302. //
  303. // Schemes: https
  304. //
  305. // Security:
  306. // oauth
  307. //
  308. // Responses:
  309. // 200: ReturnSuccessResponse
  310. func deleteUserInvite(w http.ResponseWriter, r *http.Request) {
  311. email := r.URL.Query().Get("invitee_email")
  312. err := logic.DeleteUserInvite(email)
  313. if err != nil {
  314. logger.Log(0, "failed to delete user invite: ", email, err.Error())
  315. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  316. return
  317. }
  318. logic.LogEvent(&models.Event{
  319. Action: models.Delete,
  320. Source: models.Subject{
  321. ID: r.Header.Get("user"),
  322. Name: r.Header.Get("user"),
  323. Type: models.UserSub,
  324. },
  325. TriggeredBy: r.Header.Get("user"),
  326. Target: models.Subject{
  327. ID: email,
  328. Name: email,
  329. Type: models.UserInviteSub,
  330. },
  331. Origin: models.Dashboard,
  332. })
  333. logic.ReturnSuccessResponse(w, r, "deleted user invite")
  334. }
  335. // swagger:route DELETE /api/v1/users/invites user deleteAllUserInvites
  336. //
  337. // deletes all pending invites.
  338. //
  339. // Schemes: https
  340. //
  341. // Security:
  342. // oauth
  343. //
  344. // Responses:
  345. // 200: ReturnSuccessResponse
  346. func deleteAllUserInvites(w http.ResponseWriter, r *http.Request) {
  347. err := database.DeleteAllRecords(database.USER_INVITES_TABLE_NAME)
  348. if err != nil {
  349. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending user invites "+err.Error()), "internal"))
  350. return
  351. }
  352. logic.LogEvent(&models.Event{
  353. Action: models.DeleteAll,
  354. Source: models.Subject{
  355. ID: r.Header.Get("user"),
  356. Name: r.Header.Get("user"),
  357. Type: models.UserSub,
  358. },
  359. TriggeredBy: r.Header.Get("user"),
  360. Target: models.Subject{
  361. ID: "All Invites",
  362. Name: "All Invites",
  363. Type: models.UserInviteSub,
  364. },
  365. Origin: models.Dashboard,
  366. })
  367. logic.ReturnSuccessResponse(w, r, "cleared all pending user invites")
  368. }
  369. // swagger:route GET /api/v1/user/groups user listUserGroups
  370. //
  371. // Get all user groups.
  372. //
  373. // Schemes: https
  374. //
  375. // Security:
  376. // oauth
  377. //
  378. // Responses:
  379. // 200: userBodyResponse
  380. func listUserGroups(w http.ResponseWriter, r *http.Request) {
  381. groups, err := proLogic.ListUserGroups()
  382. if err != nil {
  383. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  384. Code: http.StatusInternalServerError,
  385. Message: err.Error(),
  386. })
  387. return
  388. }
  389. logic.ReturnSuccessResponseWithJson(w, r, groups, "successfully fetched user groups")
  390. }
  391. // swagger:route GET /api/v1/user/group user getUserGroup
  392. //
  393. // Get user group.
  394. //
  395. // Schemes: https
  396. //
  397. // Security:
  398. // oauth
  399. //
  400. // Responses:
  401. // 200: userBodyResponse
  402. func getUserGroup(w http.ResponseWriter, r *http.Request) {
  403. gid := r.URL.Query().Get("group_id")
  404. if gid == "" {
  405. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
  406. return
  407. }
  408. group, err := proLogic.GetUserGroup(models.UserGroupID(gid))
  409. if err != nil {
  410. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  411. Code: http.StatusInternalServerError,
  412. Message: err.Error(),
  413. })
  414. return
  415. }
  416. logic.ReturnSuccessResponseWithJson(w, r, group, "successfully fetched user group")
  417. }
  418. // swagger:route POST /api/v1/user/group user createUserGroup
  419. //
  420. // Create user groups.
  421. //
  422. // Schemes: https
  423. //
  424. // Security:
  425. // oauth
  426. //
  427. // Responses:
  428. // 200: userBodyResponse
  429. func createUserGroup(w http.ResponseWriter, r *http.Request) {
  430. var userGroupReq models.CreateGroupReq
  431. err := json.NewDecoder(r.Body).Decode(&userGroupReq)
  432. if err != nil {
  433. slog.Error("error decoding request body", "error",
  434. err.Error())
  435. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  436. return
  437. }
  438. err = proLogic.ValidateCreateGroupReq(userGroupReq.Group)
  439. if err != nil {
  440. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  441. return
  442. }
  443. err = proLogic.CreateUserGroup(&userGroupReq.Group)
  444. if err != nil {
  445. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  446. return
  447. }
  448. for networkID := range userGroupReq.Group.NetworkRoles {
  449. network, err := logic.GetNetwork(networkID.String())
  450. if err != nil {
  451. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  452. return
  453. }
  454. acl := models.Acl{
  455. ID: uuid.New().String(),
  456. Name: fmt.Sprintf("%s group", userGroupReq.Group.Name),
  457. MetaData: "This Policy allows user group to communicate with all gateways",
  458. Default: false,
  459. ServiceType: models.Any,
  460. NetworkID: models.NetworkID(network.NetID),
  461. Proto: models.ALL,
  462. RuleType: models.UserPolicy,
  463. Src: []models.AclPolicyTag{
  464. {
  465. ID: models.UserGroupAclID,
  466. Value: userGroupReq.Group.ID.String(),
  467. },
  468. },
  469. Dst: []models.AclPolicyTag{
  470. {
  471. ID: models.NodeTagID,
  472. Value: fmt.Sprintf("%s.%s", models.NetworkID(network.NetID), models.GwTagName),
  473. }},
  474. AllowedDirection: models.TrafficDirectionUni,
  475. Enabled: true,
  476. CreatedBy: "auto",
  477. CreatedAt: time.Now().UTC(),
  478. }
  479. err = logic.InsertAcl(acl)
  480. if err != nil {
  481. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  482. return
  483. }
  484. }
  485. for _, userID := range userGroupReq.Members {
  486. user, err := logic.GetUser(userID)
  487. if err != nil {
  488. continue
  489. }
  490. if len(user.UserGroups) == 0 {
  491. user.UserGroups = make(map[models.UserGroupID]struct{})
  492. }
  493. user.UserGroups[userGroupReq.Group.ID] = struct{}{}
  494. logic.UpsertUser(*user)
  495. }
  496. logic.LogEvent(&models.Event{
  497. Action: models.Create,
  498. Source: models.Subject{
  499. ID: r.Header.Get("user"),
  500. Name: r.Header.Get("user"),
  501. Type: models.UserSub,
  502. },
  503. TriggeredBy: r.Header.Get("user"),
  504. Target: models.Subject{
  505. ID: userGroupReq.Group.ID.String(),
  506. Name: userGroupReq.Group.Name,
  507. Type: models.UserGroupSub,
  508. },
  509. Origin: models.Dashboard,
  510. })
  511. logic.ReturnSuccessResponseWithJson(w, r, userGroupReq.Group, "created user group")
  512. }
  513. // swagger:route PUT /api/v1/user/group user updateUserGroup
  514. //
  515. // Update user group.
  516. //
  517. // Schemes: https
  518. //
  519. // Security:
  520. // oauth
  521. //
  522. // Responses:
  523. // 200: userBodyResponse
  524. func updateUserGroup(w http.ResponseWriter, r *http.Request) {
  525. var userGroup models.UserGroup
  526. err := json.NewDecoder(r.Body).Decode(&userGroup)
  527. if err != nil {
  528. slog.Error("error decoding request body", "error",
  529. err.Error())
  530. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  531. return
  532. }
  533. // fetch curr group
  534. currUserG, err := proLogic.GetUserGroup(userGroup.ID)
  535. if err != nil {
  536. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  537. return
  538. }
  539. if currUserG.Default {
  540. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update default user group"), "badrequest"))
  541. return
  542. }
  543. err = proLogic.ValidateUpdateGroupReq(userGroup)
  544. if err != nil {
  545. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  546. return
  547. }
  548. userGroup.ExternalIdentityProviderID = currUserG.ExternalIdentityProviderID
  549. err = proLogic.UpdateUserGroup(userGroup)
  550. if err != nil {
  551. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  552. return
  553. }
  554. logic.LogEvent(&models.Event{
  555. Action: models.Update,
  556. Source: models.Subject{
  557. ID: r.Header.Get("user"),
  558. Name: r.Header.Get("user"),
  559. Type: models.UserSub,
  560. },
  561. TriggeredBy: r.Header.Get("user"),
  562. Target: models.Subject{
  563. ID: userGroup.ID.String(),
  564. Name: userGroup.Name,
  565. Type: models.UserGroupSub,
  566. },
  567. Diff: models.Diff{
  568. Old: currUserG,
  569. New: userGroup,
  570. },
  571. Origin: models.Dashboard,
  572. })
  573. go func() {
  574. networksAdded := make([]models.NetworkID, 0)
  575. networksRemoved := make([]models.NetworkID, 0)
  576. for networkID := range userGroup.NetworkRoles {
  577. if _, ok := currUserG.NetworkRoles[networkID]; !ok {
  578. networksAdded = append(networksAdded, networkID)
  579. }
  580. }
  581. for networkID := range currUserG.NetworkRoles {
  582. if _, ok := userGroup.NetworkRoles[networkID]; !ok {
  583. networksRemoved = append(networksRemoved, networkID)
  584. }
  585. }
  586. for _, networkID := range networksAdded {
  587. // ensure the network exists.
  588. network, err := logic.GetNetwork(networkID.String())
  589. if err != nil {
  590. continue
  591. }
  592. // insert acl if the network is added to the group.
  593. acl := models.Acl{
  594. ID: uuid.New().String(),
  595. Name: fmt.Sprintf("%s group", userGroup.Name),
  596. MetaData: "This Policy allows user group to communicate with all gateways",
  597. Default: false,
  598. ServiceType: models.Any,
  599. NetworkID: models.NetworkID(network.NetID),
  600. Proto: models.ALL,
  601. RuleType: models.UserPolicy,
  602. Src: []models.AclPolicyTag{
  603. {
  604. ID: models.UserGroupAclID,
  605. Value: userGroup.ID.String(),
  606. },
  607. },
  608. Dst: []models.AclPolicyTag{
  609. {
  610. ID: models.NodeTagID,
  611. Value: fmt.Sprintf("%s.%s", models.NetworkID(network.NetID), models.GwTagName),
  612. }},
  613. AllowedDirection: models.TrafficDirectionUni,
  614. Enabled: true,
  615. CreatedBy: "auto",
  616. CreatedAt: time.Now().UTC(),
  617. }
  618. _ = logic.InsertAcl(acl)
  619. }
  620. // since this group doesn't have a role for this network,
  621. // there is no point in having this group as src in any
  622. // of the network's acls.
  623. for _, networkID := range networksRemoved {
  624. acls, err := logic.ListAclsByNetwork(networkID)
  625. if err != nil {
  626. continue
  627. }
  628. for _, acl := range acls {
  629. var hasGroupSrc bool
  630. newAclSrc := make([]models.AclPolicyTag, 0)
  631. for _, src := range acl.Src {
  632. if src.ID == models.UserGroupAclID && src.Value == userGroup.ID.String() {
  633. hasGroupSrc = true
  634. } else {
  635. newAclSrc = append(newAclSrc, src)
  636. }
  637. }
  638. if hasGroupSrc {
  639. if len(newAclSrc) == 0 {
  640. // no other src exists, delete acl.
  641. _ = logic.DeleteAcl(acl)
  642. } else {
  643. // other sources exist, update acl.
  644. acl.Src = newAclSrc
  645. _ = logic.UpsertAcl(acl)
  646. }
  647. }
  648. }
  649. }
  650. }()
  651. // reset configs for service user
  652. go proLogic.UpdatesUserGwAccessOnGrpUpdates(currUserG.NetworkRoles, userGroup.NetworkRoles)
  653. logic.ReturnSuccessResponseWithJson(w, r, userGroup, "updated user group")
  654. }
  655. // swagger:route DELETE /api/v1/user/group user deleteUserGroup
  656. //
  657. // delete user group.
  658. //
  659. // Schemes: https
  660. //
  661. // Security:
  662. // oauth
  663. //
  664. // Responses:
  665. // 200: userBodyResponse
  666. //
  667. // @Summary Delete user group.
  668. // @Router /api/v1/user/group [delete]
  669. // @Tags Users
  670. // @Param group_id query string true "group id required to delete the role"
  671. // @Success 200 {string} string
  672. // @Failure 500 {object} models.ErrorResponse
  673. func deleteUserGroup(w http.ResponseWriter, r *http.Request) {
  674. gid := r.URL.Query().Get("group_id")
  675. if gid == "" {
  676. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("group id is required"), "badrequest"))
  677. return
  678. }
  679. userG, err := proLogic.GetUserGroup(models.UserGroupID(gid))
  680. if err != nil {
  681. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to fetch group details"), "badrequest"))
  682. return
  683. }
  684. if userG.Default {
  685. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot delete default user group"), "badrequest"))
  686. return
  687. }
  688. err = proLogic.DeleteUserGroup(models.UserGroupID(gid))
  689. if err != nil {
  690. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  691. return
  692. }
  693. logic.LogEvent(&models.Event{
  694. Action: models.Delete,
  695. Source: models.Subject{
  696. ID: r.Header.Get("user"),
  697. Name: r.Header.Get("user"),
  698. Type: models.UserSub,
  699. },
  700. TriggeredBy: r.Header.Get("user"),
  701. Target: models.Subject{
  702. ID: userG.ID.String(),
  703. Name: userG.Name,
  704. Type: models.UserGroupSub,
  705. },
  706. Origin: models.Dashboard,
  707. })
  708. go func() {
  709. for networkID := range userG.NetworkRoles {
  710. acls, err := logic.ListAclsByNetwork(networkID)
  711. if err != nil {
  712. continue
  713. }
  714. for _, acl := range acls {
  715. var hasGroupSrc bool
  716. newAclSrc := make([]models.AclPolicyTag, 0)
  717. for _, src := range acl.Src {
  718. if src.ID == models.UserGroupAclID && src.Value == userG.ID.String() {
  719. hasGroupSrc = true
  720. } else {
  721. newAclSrc = append(newAclSrc, src)
  722. }
  723. }
  724. if hasGroupSrc {
  725. if len(newAclSrc) == 0 {
  726. // no other src exists, delete acl.
  727. _ = logic.DeleteAcl(acl)
  728. } else {
  729. // other sources exist, update acl.
  730. acl.Src = newAclSrc
  731. _ = logic.UpsertAcl(acl)
  732. }
  733. }
  734. }
  735. }
  736. }()
  737. go proLogic.UpdatesUserGwAccessOnGrpUpdates(userG.NetworkRoles, make(map[models.NetworkID]map[models.UserRoleID]struct{}))
  738. logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user group")
  739. }
  740. // @Summary lists all user roles.
  741. // @Router /api/v1/user/roles [get]
  742. // @Tags Users
  743. // @Param role_id query string true "roleid required to get the role details"
  744. // @Success 200 {object} []models.UserRolePermissionTemplate
  745. // @Failure 500 {object} models.ErrorResponse
  746. func ListRoles(w http.ResponseWriter, r *http.Request) {
  747. platform := r.URL.Query().Get("platform")
  748. var roles []models.UserRolePermissionTemplate
  749. var err error
  750. if platform == "true" {
  751. roles, err = logic.ListPlatformRoles()
  752. } else {
  753. roles, err = proLogic.ListNetworkRoles()
  754. }
  755. if err != nil {
  756. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  757. Code: http.StatusInternalServerError,
  758. Message: err.Error(),
  759. })
  760. return
  761. }
  762. logic.ReturnSuccessResponseWithJson(w, r, roles, "successfully fetched user roles permission templates")
  763. }
  764. // @Summary Get user role permission template.
  765. // @Router /api/v1/user/role [get]
  766. // @Tags Users
  767. // @Param role_id query string true "roleid required to get the role details"
  768. // @Success 200 {object} models.UserRolePermissionTemplate
  769. // @Failure 500 {object} models.ErrorResponse
  770. func getRole(w http.ResponseWriter, r *http.Request) {
  771. rid := r.URL.Query().Get("role_id")
  772. if rid == "" {
  773. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  774. return
  775. }
  776. role, err := logic.GetRole(models.UserRoleID(rid))
  777. if err != nil {
  778. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  779. Code: http.StatusInternalServerError,
  780. Message: err.Error(),
  781. })
  782. return
  783. }
  784. logic.ReturnSuccessResponseWithJson(w, r, role, "successfully fetched user role permission templates")
  785. }
  786. // @Summary Create user role permission template.
  787. // @Router /api/v1/user/role [post]
  788. // @Tags Users
  789. // @Param body body models.UserRolePermissionTemplate true "user role template"
  790. // @Success 200 {object} models.UserRolePermissionTemplate
  791. // @Failure 500 {object} models.ErrorResponse
  792. func createRole(w http.ResponseWriter, r *http.Request) {
  793. var userRole models.UserRolePermissionTemplate
  794. err := json.NewDecoder(r.Body).Decode(&userRole)
  795. if err != nil {
  796. slog.Error("error decoding request body", "error",
  797. err.Error())
  798. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  799. return
  800. }
  801. err = proLogic.ValidateCreateRoleReq(&userRole)
  802. if err != nil {
  803. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  804. return
  805. }
  806. userRole.Default = false
  807. userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
  808. err = proLogic.CreateRole(userRole)
  809. if err != nil {
  810. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  811. return
  812. }
  813. logic.LogEvent(&models.Event{
  814. Action: models.Create,
  815. Source: models.Subject{
  816. ID: r.Header.Get("user"),
  817. Name: r.Header.Get("user"),
  818. Type: models.UserSub,
  819. },
  820. TriggeredBy: r.Header.Get("user"),
  821. Target: models.Subject{
  822. ID: userRole.ID.String(),
  823. Name: userRole.Name,
  824. Type: models.UserRoleSub,
  825. },
  826. Origin: models.ClientApp,
  827. })
  828. logic.ReturnSuccessResponseWithJson(w, r, userRole, "created user role")
  829. }
  830. // @Summary Update user role permission template.
  831. // @Router /api/v1/user/role [put]
  832. // @Tags Users
  833. // @Param body body models.UserRolePermissionTemplate true "user role template"
  834. // @Success 200 {object} models.UserRolePermissionTemplate
  835. // @Failure 500 {object} models.ErrorResponse
  836. func updateRole(w http.ResponseWriter, r *http.Request) {
  837. var userRole models.UserRolePermissionTemplate
  838. err := json.NewDecoder(r.Body).Decode(&userRole)
  839. if err != nil {
  840. slog.Error("error decoding request body", "error",
  841. err.Error())
  842. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  843. return
  844. }
  845. currRole, err := logic.GetRole(userRole.ID)
  846. if err != nil {
  847. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  848. return
  849. }
  850. err = proLogic.ValidateUpdateRoleReq(&userRole)
  851. if err != nil {
  852. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  853. return
  854. }
  855. userRole.GlobalLevelAccess = make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope)
  856. err = proLogic.UpdateRole(userRole)
  857. if err != nil {
  858. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  859. return
  860. }
  861. logic.LogEvent(&models.Event{
  862. Action: models.Update,
  863. Source: models.Subject{
  864. ID: r.Header.Get("user"),
  865. Name: r.Header.Get("user"),
  866. Type: models.UserSub,
  867. },
  868. TriggeredBy: r.Header.Get("user"),
  869. Target: models.Subject{
  870. ID: userRole.ID.String(),
  871. Name: userRole.Name,
  872. Type: models.UserRoleSub,
  873. },
  874. Diff: models.Diff{
  875. Old: currRole,
  876. New: userRole,
  877. },
  878. Origin: models.Dashboard,
  879. })
  880. // reset configs for service user
  881. go proLogic.UpdatesUserGwAccessOnRoleUpdates(currRole.NetworkLevelAccess, userRole.NetworkLevelAccess, string(userRole.NetworkID))
  882. logic.ReturnSuccessResponseWithJson(w, r, userRole, "updated user role")
  883. }
  884. // @Summary Delete user role permission template.
  885. // @Router /api/v1/user/role [delete]
  886. // @Tags Users
  887. // @Param role_id query string true "roleid required to delete the role"
  888. // @Success 200 {string} string
  889. // @Failure 500 {object} models.ErrorResponse
  890. func deleteRole(w http.ResponseWriter, r *http.Request) {
  891. rid := r.URL.Query().Get("role_id")
  892. if rid == "" {
  893. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  894. return
  895. }
  896. role, err := logic.GetRole(models.UserRoleID(rid))
  897. if err != nil {
  898. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("role is required"), "badrequest"))
  899. return
  900. }
  901. err = proLogic.DeleteRole(models.UserRoleID(rid), false)
  902. if err != nil {
  903. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  904. return
  905. }
  906. logic.LogEvent(&models.Event{
  907. Action: models.Delete,
  908. Source: models.Subject{
  909. ID: r.Header.Get("user"),
  910. Name: r.Header.Get("user"),
  911. Type: models.UserSub,
  912. },
  913. TriggeredBy: r.Header.Get("user"),
  914. Target: models.Subject{
  915. ID: role.ID.String(),
  916. Name: role.Name,
  917. Type: models.UserRoleSub,
  918. },
  919. Origin: models.Dashboard,
  920. })
  921. go proLogic.UpdatesUserGwAccessOnRoleUpdates(role.NetworkLevelAccess, make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope), role.NetworkID.String())
  922. logic.ReturnSuccessResponseWithJson(w, r, nil, "deleted user role")
  923. }
  924. // @Summary Attach user to a remote access gateway
  925. // @Router /api/users/{username}/remote_access_gw/{remote_access_gateway_id} [post]
  926. // @Tags PRO
  927. // @Accept json
  928. // @Produce json
  929. // @Param username path string true "Username"
  930. // @Param remote_access_gateway_id path string true "Remote Access Gateway ID"
  931. // @Success 200 {object} models.ReturnUser
  932. // @Failure 400 {object} models.ErrorResponse
  933. // @Failure 500 {object} models.ErrorResponse
  934. func attachUserToRemoteAccessGw(w http.ResponseWriter, r *http.Request) {
  935. // set header.
  936. w.Header().Set("Content-Type", "application/json")
  937. var params = mux.Vars(r)
  938. username := params["username"]
  939. remoteGwID := params["remote_access_gateway_id"]
  940. if username == "" || remoteGwID == "" {
  941. logic.ReturnErrorResponse(
  942. w,
  943. r,
  944. logic.FormatError(
  945. errors.New("required params `username` and `remote_access_gateway_id`"),
  946. "badrequest",
  947. ),
  948. )
  949. return
  950. }
  951. user, err := logic.GetUser(username)
  952. if err != nil {
  953. slog.Error("failed to fetch user: ", "username", username, "error", err.Error())
  954. logic.ReturnErrorResponse(
  955. w,
  956. r,
  957. logic.FormatError(
  958. fmt.Errorf("failed to fetch user %s, error: %v", username, err),
  959. "badrequest",
  960. ),
  961. )
  962. return
  963. }
  964. if user.PlatformRoleID == models.AdminRole || user.PlatformRoleID == models.SuperAdminRole {
  965. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("superadmins/admins have access to all gateways"), "badrequest"))
  966. return
  967. }
  968. node, err := logic.GetNodeByID(remoteGwID)
  969. if err != nil {
  970. slog.Error("failed to fetch gateway node", "nodeID", remoteGwID, "error", err)
  971. logic.ReturnErrorResponse(
  972. w,
  973. r,
  974. logic.FormatError(
  975. fmt.Errorf("failed to fetch remote access gateway node, error: %v", err),
  976. "badrequest",
  977. ),
  978. )
  979. return
  980. }
  981. if !node.IsIngressGateway {
  982. logic.ReturnErrorResponse(
  983. w,
  984. r,
  985. logic.FormatError(fmt.Errorf("node is not a remote access gateway"), "badrequest"),
  986. )
  987. return
  988. }
  989. if user.RemoteGwIDs == nil {
  990. user.RemoteGwIDs = make(map[string]struct{})
  991. }
  992. user.RemoteGwIDs[node.ID.String()] = struct{}{}
  993. err = logic.UpsertUser(*user)
  994. if err != nil {
  995. slog.Error("failed to update user's gateways", "user", username, "error", err)
  996. logic.ReturnErrorResponse(
  997. w,
  998. r,
  999. logic.FormatError(
  1000. fmt.Errorf("failed to fetch remote access gateway node,error: %v", err),
  1001. "badrequest",
  1002. ),
  1003. )
  1004. return
  1005. }
  1006. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  1007. }
  1008. // @Summary Remove user from a remote access gateway
  1009. // @Router /api/users/{username}/remote_access_gw/{remote_access_gateway_id} [delete]
  1010. // @Tags PRO
  1011. // @Accept json
  1012. // @Produce json
  1013. // @Param username path string true "Username"
  1014. // @Param remote_access_gateway_id path string true "Remote Access Gateway ID"
  1015. // @Success 200 {object} models.ReturnUser
  1016. // @Failure 400 {object} models.ErrorResponse
  1017. // @Failure 500 {object} models.ErrorResponse
  1018. func removeUserFromRemoteAccessGW(w http.ResponseWriter, r *http.Request) {
  1019. // set header.
  1020. w.Header().Set("Content-Type", "application/json")
  1021. var params = mux.Vars(r)
  1022. username := params["username"]
  1023. remoteGwID := params["remote_access_gateway_id"]
  1024. if username == "" || remoteGwID == "" {
  1025. logic.ReturnErrorResponse(
  1026. w,
  1027. r,
  1028. logic.FormatError(
  1029. errors.New("required params `username` and `remote_access_gateway_id`"),
  1030. "badrequest",
  1031. ),
  1032. )
  1033. return
  1034. }
  1035. user, err := logic.GetUser(username)
  1036. if err != nil {
  1037. logger.Log(0, username, "failed to fetch user: ", err.Error())
  1038. logic.ReturnErrorResponse(
  1039. w,
  1040. r,
  1041. logic.FormatError(
  1042. fmt.Errorf("failed to fetch user %s, error: %v", username, err),
  1043. "badrequest",
  1044. ),
  1045. )
  1046. return
  1047. }
  1048. delete(user.RemoteGwIDs, remoteGwID)
  1049. go func(user models.User, remoteGwID string) {
  1050. extclients, err := logic.GetAllExtClients()
  1051. if err != nil {
  1052. slog.Error("failed to fetch extclients", "error", err)
  1053. return
  1054. }
  1055. for _, extclient := range extclients {
  1056. if extclient.OwnerID == user.UserName && remoteGwID == extclient.IngressGatewayID {
  1057. err = logic.DeleteExtClientAndCleanup(extclient)
  1058. if err != nil {
  1059. slog.Error("failed to delete extclient",
  1060. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  1061. } else {
  1062. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1063. slog.Error("error setting ext peers: " + err.Error())
  1064. }
  1065. }
  1066. }
  1067. }
  1068. if servercfg.IsDNSMode() {
  1069. logic.SetDNS()
  1070. }
  1071. }(*user, remoteGwID)
  1072. err = logic.UpsertUser(*user)
  1073. if err != nil {
  1074. slog.Error("failed to update user gateways", "user", username, "error", err)
  1075. logic.ReturnErrorResponse(
  1076. w,
  1077. r,
  1078. logic.FormatError(
  1079. errors.New("failed to fetch remote access gaetway node "+err.Error()),
  1080. "badrequest",
  1081. ),
  1082. )
  1083. return
  1084. }
  1085. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  1086. }
  1087. // @Summary Get Users Remote Access Gw Networks.
  1088. // @Router /api/users/{username}/remote_access_gw [get]
  1089. // @Tags Users
  1090. // @Param username path string true "Username to fetch all the gateways with access"
  1091. // @Success 200 {object} map[string][]models.UserRemoteGws
  1092. // @Failure 500 {object} models.ErrorResponse
  1093. func getUserRemoteAccessNetworks(w http.ResponseWriter, r *http.Request) {
  1094. // set header.
  1095. w.Header().Set("Content-Type", "application/json")
  1096. username := r.Header.Get("user")
  1097. user, err := logic.GetUser(username)
  1098. if err != nil {
  1099. logger.Log(0, username, "failed to fetch user: ", err.Error())
  1100. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  1101. return
  1102. }
  1103. userGws := make(map[string][]models.UserRemoteGws)
  1104. networks := []models.Network{}
  1105. networkMap := make(map[string]struct{})
  1106. userGwNodes := proLogic.GetUserRAGNodes(*user)
  1107. for _, node := range userGwNodes {
  1108. network, err := logic.GetNetwork(node.Network)
  1109. if err != nil {
  1110. slog.Error("failed to get node network", "error", err)
  1111. continue
  1112. }
  1113. if _, ok := networkMap[network.NetID]; ok {
  1114. continue
  1115. }
  1116. networkMap[network.NetID] = struct{}{}
  1117. networks = append(networks, network)
  1118. }
  1119. slog.Debug("returned user gws", "user", username, "gws", userGws)
  1120. logic.ReturnSuccessResponseWithJson(w, r, networks, "fetched user accessible networks")
  1121. }
  1122. // @Summary Get Users Remote Access Gw Networks.
  1123. // @Router /api/users/{username}/remote_access_gw [get]
  1124. // @Tags Users
  1125. // @Param username path string true "Username to fetch all the gateways with access"
  1126. // @Success 200 {object} map[string][]models.UserRemoteGws
  1127. // @Failure 500 {object} models.ErrorResponse
  1128. func getUserRemoteAccessNetworkGateways(w http.ResponseWriter, r *http.Request) {
  1129. // set header.
  1130. w.Header().Set("Content-Type", "application/json")
  1131. var params = mux.Vars(r)
  1132. username := r.Header.Get("user")
  1133. user, err := logic.GetUser(username)
  1134. if err != nil {
  1135. logger.Log(0, username, "failed to fetch user: ", err.Error())
  1136. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  1137. return
  1138. }
  1139. network := params["network"]
  1140. if network == "" {
  1141. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params network"), "badrequest"))
  1142. return
  1143. }
  1144. userGws := []models.UserRAGs{}
  1145. userGwNodes := proLogic.GetUserRAGNodes(*user)
  1146. for _, node := range userGwNodes {
  1147. if node.Network != network {
  1148. continue
  1149. }
  1150. host, err := logic.GetHost(node.HostID.String())
  1151. if err != nil {
  1152. continue
  1153. }
  1154. userGws = append(userGws, models.UserRAGs{
  1155. GwID: node.ID.String(),
  1156. GWName: host.Name,
  1157. Network: node.Network,
  1158. IsInternetGateway: node.IsInternetGateway,
  1159. Metadata: node.Metadata,
  1160. })
  1161. }
  1162. slog.Debug("returned user gws", "user", username, "gws", userGws)
  1163. logic.ReturnSuccessResponseWithJson(w, r, userGws, "fetched user accessible gateways in network "+network)
  1164. }
  1165. // @Summary Get Users Remote Access Gw Networks.
  1166. // @Router /api/users/{username}/remote_access_gw [get]
  1167. // @Tags Users
  1168. // @Param username path string true "Username to fetch all the gateways with access"
  1169. // @Success 200 {object} map[string][]models.UserRemoteGws
  1170. // @Failure 500 {object} models.ErrorResponse
  1171. func getRemoteAccessGatewayConf(w http.ResponseWriter, r *http.Request) {
  1172. // set header.
  1173. w.Header().Set("Content-Type", "application/json")
  1174. var params = mux.Vars(r)
  1175. username := r.Header.Get("user")
  1176. user, err := logic.GetUser(username)
  1177. if err != nil {
  1178. logger.Log(0, username, "failed to fetch user: ", err.Error())
  1179. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  1180. return
  1181. }
  1182. remoteGwID := params["access_point_id"]
  1183. if remoteGwID == "" {
  1184. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params access_point_id"), "badrequest"))
  1185. return
  1186. }
  1187. var req models.UserRemoteGwsReq
  1188. err = json.NewDecoder(r.Body).Decode(&req)
  1189. if err != nil {
  1190. slog.Error("error decoding request body: ", "error", err)
  1191. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1192. return
  1193. }
  1194. userGwNodes := proLogic.GetUserRAGNodes(*user)
  1195. if _, ok := userGwNodes[remoteGwID]; !ok {
  1196. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("access denied"), "forbidden"))
  1197. return
  1198. }
  1199. node, err := logic.GetNodeByID(remoteGwID)
  1200. if err != nil {
  1201. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch gw node %s, error: %v", remoteGwID, err), "badrequest"))
  1202. return
  1203. }
  1204. host, err := logic.GetHost(node.HostID.String())
  1205. if err != nil {
  1206. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch gw host %s, error: %v", remoteGwID, err), "badrequest"))
  1207. return
  1208. }
  1209. network, err := logic.GetNetwork(node.Network)
  1210. if err != nil {
  1211. slog.Error("failed to get node network", "error", err)
  1212. }
  1213. var userConf models.ExtClient
  1214. allextClients, err := logic.GetAllExtClients()
  1215. if err != nil {
  1216. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1217. return
  1218. }
  1219. for _, extClient := range allextClients {
  1220. if extClient.Network != network.NetID || extClient.IngressGatewayID != node.ID.String() {
  1221. continue
  1222. }
  1223. if extClient.RemoteAccessClientID == req.RemoteAccessClientID && extClient.OwnerID == username {
  1224. userConf = extClient
  1225. userConf.AllowedIPs = logic.GetExtclientAllowedIPs(extClient)
  1226. }
  1227. }
  1228. if userConf.ClientID == "" {
  1229. // create a new conf
  1230. userConf.OwnerID = user.UserName
  1231. userConf.RemoteAccessClientID = req.RemoteAccessClientID
  1232. userConf.IngressGatewayID = node.ID.String()
  1233. // set extclient dns to ingressdns if extclient dns is not explicitly set
  1234. if (userConf.DNS == "") && (node.IngressDNS != "") {
  1235. userConf.DNS = node.IngressDNS
  1236. }
  1237. userConf.Network = node.Network
  1238. host, err := logic.GetHost(node.HostID.String())
  1239. if err != nil {
  1240. logger.Log(0, r.Header.Get("user"),
  1241. fmt.Sprintf("failed to get ingress gateway host for node [%s] info: %v", node.ID, err))
  1242. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1243. return
  1244. }
  1245. listenPort := logic.GetPeerListenPort(host)
  1246. if host.EndpointIP.To4() == nil {
  1247. userConf.IngressGatewayEndpoint = fmt.Sprintf("[%s]:%d", host.EndpointIPv6.String(), listenPort)
  1248. } else {
  1249. userConf.IngressGatewayEndpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), listenPort)
  1250. }
  1251. userConf.Enabled = true
  1252. parentNetwork, err := logic.GetNetwork(node.Network)
  1253. if err == nil { // check if parent network default ACL is enabled (yes) or not (no)
  1254. userConf.Enabled = parentNetwork.DefaultACL == "yes"
  1255. }
  1256. userConf.Tags = make(map[models.TagID]struct{})
  1257. // userConf.Tags[models.TagID(fmt.Sprintf("%s.%s", userConf.Network,
  1258. // models.RemoteAccessTagName))] = struct{}{}
  1259. if err = logic.CreateExtClient(&userConf); err != nil {
  1260. slog.Error(
  1261. "failed to create extclient",
  1262. "user",
  1263. r.Header.Get("user"),
  1264. "network",
  1265. node.Network,
  1266. "error",
  1267. err,
  1268. )
  1269. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1270. return
  1271. }
  1272. }
  1273. userGw := models.UserRemoteGws{
  1274. GwID: node.ID.String(),
  1275. GWName: host.Name,
  1276. Network: node.Network,
  1277. GwClient: userConf,
  1278. Connected: true,
  1279. IsInternetGateway: node.IsInternetGateway,
  1280. GwPeerPublicKey: host.PublicKey.String(),
  1281. GwListenPort: logic.GetPeerListenPort(host),
  1282. Metadata: node.Metadata,
  1283. AllowedEndpoints: getAllowedRagEndpoints(&node, host),
  1284. NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
  1285. DnsAddress: node.IngressDNS,
  1286. Addresses: utils.NoEmptyStringToCsv(node.Address.String(), node.Address6.String()),
  1287. }
  1288. slog.Debug("returned user gw config", "user", user.UserName, "gws", userGw)
  1289. logic.ReturnSuccessResponseWithJson(w, r, userGw, "fetched user config to gw "+remoteGwID)
  1290. }
  1291. // @Summary Get Users Remote Access Gw.
  1292. // @Router /api/users/{username}/remote_access_gw [get]
  1293. // @Tags Users
  1294. // @Param username path string true "Username to fetch all the gateways with access"
  1295. // @Success 200 {object} map[string][]models.UserRemoteGws
  1296. // @Failure 500 {object} models.ErrorResponse
  1297. func getUserRemoteAccessGwsV1(w http.ResponseWriter, r *http.Request) {
  1298. // set header.
  1299. w.Header().Set("Content-Type", "application/json")
  1300. var params = mux.Vars(r)
  1301. username := params["username"]
  1302. if username == "" {
  1303. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("required params username"), "badrequest"))
  1304. return
  1305. }
  1306. user, err := logic.GetUser(username)
  1307. if err != nil {
  1308. logger.Log(0, username, "failed to fetch user: ", err.Error())
  1309. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to fetch user %s, error: %v", username, err), "badrequest"))
  1310. return
  1311. }
  1312. deviceID := r.URL.Query().Get("device_id")
  1313. remoteAccessClientID := r.URL.Query().Get("remote_access_clientid")
  1314. var req models.UserRemoteGwsReq
  1315. if remoteAccessClientID == "" {
  1316. err := json.NewDecoder(r.Body).Decode(&req)
  1317. if err != nil {
  1318. slog.Error("error decoding request body: ", "error", err)
  1319. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1320. return
  1321. }
  1322. }
  1323. reqFromMobile := r.URL.Query().Get("from_mobile") == "true"
  1324. if req.RemoteAccessClientID == "" && remoteAccessClientID == "" {
  1325. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("remote access client id cannot be empty"), "badrequest"))
  1326. return
  1327. }
  1328. if req.RemoteAccessClientID == "" {
  1329. req.RemoteAccessClientID = remoteAccessClientID
  1330. }
  1331. userGws := make(map[string][]models.UserRemoteGws)
  1332. allextClients, err := logic.GetAllExtClients()
  1333. if err != nil {
  1334. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1335. return
  1336. }
  1337. userGwNodes := proLogic.GetUserRAGNodes(*user)
  1338. userExtClients := make(map[string][]models.ExtClient)
  1339. // group all extclients of the requesting user by ingress
  1340. // gateway.
  1341. for _, extClient := range allextClients {
  1342. // filter our extclients that don't belong to this user.
  1343. if extClient.OwnerID != username {
  1344. continue
  1345. }
  1346. _, ok := userExtClients[extClient.IngressGatewayID]
  1347. if !ok {
  1348. userExtClients[extClient.IngressGatewayID] = []models.ExtClient{}
  1349. }
  1350. userExtClients[extClient.IngressGatewayID] = append(userExtClients[extClient.IngressGatewayID], extClient)
  1351. }
  1352. for ingressGatewayID, extClients := range userExtClients {
  1353. node, ok := userGwNodes[ingressGatewayID]
  1354. if !ok {
  1355. continue
  1356. }
  1357. var gwClient models.ExtClient
  1358. var found bool
  1359. if deviceID != "" {
  1360. for _, extClient := range extClients {
  1361. if extClient.DeviceID == deviceID {
  1362. gwClient = extClient
  1363. found = true
  1364. break
  1365. }
  1366. }
  1367. }
  1368. if !found {
  1369. // TODO: prevent ip clashes.
  1370. if len(extClients) > 0 {
  1371. gwClient = extClients[0]
  1372. }
  1373. }
  1374. host, err := logic.GetHost(node.HostID.String())
  1375. if err != nil {
  1376. continue
  1377. }
  1378. network, err := logic.GetNetwork(node.Network)
  1379. if err != nil {
  1380. slog.Error("failed to get node network", "error", err)
  1381. continue
  1382. }
  1383. nodesWithStatus := logic.AddStatusToNodes([]models.Node{node}, false)
  1384. if len(nodesWithStatus) > 0 {
  1385. node = nodesWithStatus[0]
  1386. }
  1387. gws := userGws[node.Network]
  1388. if gwClient.DNS == "" {
  1389. gwClient.DNS = node.IngressDNS
  1390. }
  1391. gwClient.IngressGatewayEndpoint = utils.GetExtClientEndpoint(
  1392. host.EndpointIP,
  1393. host.EndpointIPv6,
  1394. logic.GetPeerListenPort(host),
  1395. )
  1396. gwClient.AllowedIPs = logic.GetExtclientAllowedIPs(gwClient)
  1397. gw := models.UserRemoteGws{
  1398. GwID: node.ID.String(),
  1399. GWName: host.Name,
  1400. Network: node.Network,
  1401. GwClient: gwClient,
  1402. Connected: true,
  1403. IsInternetGateway: node.IsInternetGateway,
  1404. GwPeerPublicKey: host.PublicKey.String(),
  1405. GwListenPort: logic.GetPeerListenPort(host),
  1406. Metadata: node.Metadata,
  1407. AllowedEndpoints: getAllowedRagEndpoints(&node, host),
  1408. NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
  1409. Status: node.Status,
  1410. DnsAddress: node.IngressDNS,
  1411. Addresses: utils.NoEmptyStringToCsv(node.Address.String(), node.Address6.String()),
  1412. }
  1413. if !node.IsInternetGateway {
  1414. hNs := logic.GetNameserversForHost(host)
  1415. for _, nsI := range hNs {
  1416. gw.MatchDomains = append(gw.MatchDomains, nsI.MatchDomain)
  1417. }
  1418. }
  1419. gws = append(gws, gw)
  1420. userGws[node.Network] = gws
  1421. delete(userGwNodes, node.ID.String())
  1422. }
  1423. // add remaining gw nodes to resp
  1424. for gwID := range userGwNodes {
  1425. node, err := logic.GetNodeByID(gwID)
  1426. if err != nil {
  1427. continue
  1428. }
  1429. if !node.IsIngressGateway {
  1430. continue
  1431. }
  1432. if node.PendingDelete {
  1433. continue
  1434. }
  1435. host, err := logic.GetHost(node.HostID.String())
  1436. if err != nil {
  1437. continue
  1438. }
  1439. nodesWithStatus := logic.AddStatusToNodes([]models.Node{node}, false)
  1440. if len(nodesWithStatus) > 0 {
  1441. node = nodesWithStatus[0]
  1442. }
  1443. network, err := logic.GetNetwork(node.Network)
  1444. if err != nil {
  1445. slog.Error("failed to get node network", "error", err)
  1446. }
  1447. gws := userGws[node.Network]
  1448. gws = append(gws, models.UserRemoteGws{
  1449. GwID: node.ID.String(),
  1450. GWName: host.Name,
  1451. Network: node.Network,
  1452. IsInternetGateway: node.IsInternetGateway,
  1453. GwPeerPublicKey: host.PublicKey.String(),
  1454. GwListenPort: logic.GetPeerListenPort(host),
  1455. Metadata: node.Metadata,
  1456. AllowedEndpoints: getAllowedRagEndpoints(&node, host),
  1457. NetworkAddresses: []string{network.AddressRange, network.AddressRange6},
  1458. Status: node.Status,
  1459. DnsAddress: node.IngressDNS,
  1460. Addresses: utils.NoEmptyStringToCsv(node.Address.String(), node.Address6.String()),
  1461. })
  1462. userGws[node.Network] = gws
  1463. }
  1464. if reqFromMobile {
  1465. // send resp in array format
  1466. userGwsArr := []models.UserRemoteGws{}
  1467. for _, userGwI := range userGws {
  1468. userGwsArr = append(userGwsArr, userGwI...)
  1469. }
  1470. logic.ReturnSuccessResponseWithJson(w, r, userGwsArr, "fetched gateways for user"+username)
  1471. return
  1472. }
  1473. slog.Debug("returned user gws", "user", username, "gws", userGws)
  1474. w.WriteHeader(http.StatusOK)
  1475. json.NewEncoder(w).Encode(userGws)
  1476. }
  1477. // @Summary List users attached to an remote access gateway
  1478. // @Router /api/nodes/{network}/{nodeid}/ingress/users [get]
  1479. // @Tags PRO
  1480. // @Accept json
  1481. // @Produce json
  1482. // @Param ingress_id path string true "Ingress Gateway ID"
  1483. // @Success 200 {array} models.IngressGwUsers
  1484. // @Failure 400 {object} models.ErrorResponse
  1485. // @Failure 500 {object} models.ErrorResponse
  1486. func ingressGatewayUsers(w http.ResponseWriter, r *http.Request) {
  1487. w.Header().Set("Content-Type", "application/json")
  1488. var params = mux.Vars(r)
  1489. ingressID := params["ingress_id"]
  1490. node, err := logic.GetNodeByID(ingressID)
  1491. if err != nil {
  1492. slog.Error("failed to get ingress node", "error", err)
  1493. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1494. return
  1495. }
  1496. gwUsers, err := logic.GetIngressGwUsers(node)
  1497. if err != nil {
  1498. slog.Error(
  1499. "failed to get users on ingress gateway",
  1500. "nodeid",
  1501. ingressID,
  1502. "network",
  1503. node.Network,
  1504. "user",
  1505. r.Header.Get("user"),
  1506. "error",
  1507. err,
  1508. )
  1509. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1510. return
  1511. }
  1512. w.WriteHeader(http.StatusOK)
  1513. json.NewEncoder(w).Encode(gwUsers)
  1514. }
  1515. func getAllowedRagEndpoints(ragNode *models.Node, ragHost *models.Host) []string {
  1516. endpoints := []string{}
  1517. if len(ragHost.EndpointIP) > 0 {
  1518. endpoints = append(endpoints, ragHost.EndpointIP.String())
  1519. }
  1520. if len(ragHost.EndpointIPv6) > 0 {
  1521. endpoints = append(endpoints, ragHost.EndpointIPv6.String())
  1522. }
  1523. if servercfg.IsPro {
  1524. for _, ip := range ragNode.AdditionalRagIps {
  1525. endpoints = append(endpoints, ip.String())
  1526. }
  1527. }
  1528. return endpoints
  1529. }
  1530. // @Summary Get all pending users
  1531. // @Router /api/users_pending [get]
  1532. // @Tags Users
  1533. // @Success 200 {array} models.User
  1534. // @Failure 500 {object} models.ErrorResponse
  1535. func getPendingUsers(w http.ResponseWriter, r *http.Request) {
  1536. // set header.
  1537. w.Header().Set("Content-Type", "application/json")
  1538. users, err := logic.ListPendingReturnUsers()
  1539. if err != nil {
  1540. logger.Log(0, "failed to fetch users: ", err.Error())
  1541. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1542. return
  1543. }
  1544. logic.SortUsers(users[:])
  1545. logger.Log(2, r.Header.Get("user"), "fetched pending users")
  1546. json.NewEncoder(w).Encode(users)
  1547. }
  1548. // @Summary Approve a pending user
  1549. // @Router /api/users_pending/user/{username} [post]
  1550. // @Tags Users
  1551. // @Param username path string true "Username of the pending user to approve"
  1552. // @Success 200 {string} string
  1553. // @Failure 500 {object} models.ErrorResponse
  1554. func approvePendingUser(w http.ResponseWriter, r *http.Request) {
  1555. // set header.
  1556. w.Header().Set("Content-Type", "application/json")
  1557. var params = mux.Vars(r)
  1558. username := params["username"]
  1559. users, err := logic.ListPendingUsers()
  1560. if err != nil {
  1561. logger.Log(0, "failed to fetch users: ", err.Error())
  1562. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1563. return
  1564. }
  1565. for _, user := range users {
  1566. if user.UserName == username {
  1567. var newPass, fetchErr = logic.FetchPassValue("")
  1568. if fetchErr != nil {
  1569. logic.ReturnErrorResponse(w, r, logic.FormatError(fetchErr, "internal"))
  1570. return
  1571. }
  1572. if err = logic.CreateUser(&models.User{
  1573. UserName: user.UserName,
  1574. ExternalIdentityProviderID: user.ExternalIdentityProviderID,
  1575. Password: newPass,
  1576. AuthType: user.AuthType,
  1577. PlatformRoleID: models.ServiceUser,
  1578. }); err != nil {
  1579. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to create user: %s", err), "internal"))
  1580. return
  1581. }
  1582. err = logic.DeletePendingUser(username)
  1583. if err != nil {
  1584. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
  1585. return
  1586. }
  1587. break
  1588. }
  1589. }
  1590. logic.LogEvent(&models.Event{
  1591. Action: models.Create,
  1592. Source: models.Subject{
  1593. ID: r.Header.Get("user"),
  1594. Name: r.Header.Get("user"),
  1595. Type: models.UserSub,
  1596. },
  1597. TriggeredBy: r.Header.Get("user"),
  1598. Target: models.Subject{
  1599. ID: username,
  1600. Name: username,
  1601. Type: models.PendingUserSub,
  1602. },
  1603. Origin: models.Dashboard,
  1604. })
  1605. logic.ReturnSuccessResponse(w, r, "approved "+username)
  1606. }
  1607. // @Summary Delete a pending user
  1608. // @Router /api/users_pending/user/{username} [delete]
  1609. // @Tags Users
  1610. // @Param username path string true "Username of the pending user to delete"
  1611. // @Success 200 {string} string
  1612. // @Failure 500 {object} models.ErrorResponse
  1613. func deletePendingUser(w http.ResponseWriter, r *http.Request) {
  1614. // set header.
  1615. w.Header().Set("Content-Type", "application/json")
  1616. var params = mux.Vars(r)
  1617. username := params["username"]
  1618. users, err := logic.ListPendingReturnUsers()
  1619. if err != nil {
  1620. logger.Log(0, "failed to fetch users: ", err.Error())
  1621. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1622. return
  1623. }
  1624. for _, user := range users {
  1625. if user.UserName == username {
  1626. err = logic.DeletePendingUser(username)
  1627. if err != nil {
  1628. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete pending user: %s", err), "internal"))
  1629. return
  1630. }
  1631. break
  1632. }
  1633. }
  1634. logic.LogEvent(&models.Event{
  1635. Action: models.Delete,
  1636. Source: models.Subject{
  1637. ID: r.Header.Get("user"),
  1638. Name: r.Header.Get("user"),
  1639. Type: models.UserSub,
  1640. },
  1641. TriggeredBy: r.Header.Get("user"),
  1642. Target: models.Subject{
  1643. ID: username,
  1644. Name: username,
  1645. Type: models.PendingUserSub,
  1646. },
  1647. Origin: models.Dashboard,
  1648. })
  1649. logic.ReturnSuccessResponse(w, r, "deleted pending "+username)
  1650. }
  1651. // @Summary Delete all pending users
  1652. // @Router /api/users_pending [delete]
  1653. // @Tags Users
  1654. // @Success 200 {string} string
  1655. // @Failure 500 {object} models.ErrorResponse
  1656. func deleteAllPendingUsers(w http.ResponseWriter, r *http.Request) {
  1657. // set header.
  1658. err := database.DeleteAllRecords(database.PENDING_USERS_TABLE_NAME)
  1659. if err != nil {
  1660. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failed to delete all pending users "+err.Error()), "internal"))
  1661. return
  1662. }
  1663. logic.LogEvent(&models.Event{
  1664. Action: models.DeleteAll,
  1665. Source: models.Subject{
  1666. ID: r.Header.Get("user"),
  1667. Name: r.Header.Get("user"),
  1668. Type: models.UserSub,
  1669. },
  1670. TriggeredBy: r.Header.Get("user"),
  1671. Target: models.Subject{
  1672. ID: r.Header.Get("user"),
  1673. Name: r.Header.Get("user"),
  1674. Type: models.PendingUserSub,
  1675. },
  1676. Origin: models.Dashboard,
  1677. })
  1678. logic.ReturnSuccessResponse(w, r, "cleared all pending users")
  1679. }
  1680. // @Summary Sync users and groups from idp.
  1681. // @Router /api/idp/sync [post]
  1682. // @Tags IDP
  1683. // @Success 200 {object} models.SuccessResponse
  1684. func syncIDP(w http.ResponseWriter, r *http.Request) {
  1685. go func() {
  1686. err := proAuth.SyncFromIDP()
  1687. if err != nil {
  1688. logger.Log(0, "failed to sync from idp: ", err.Error())
  1689. } else {
  1690. logger.Log(0, "sync from idp complete")
  1691. }
  1692. }()
  1693. logic.ReturnSuccessResponse(w, r, "starting sync from idp")
  1694. }
  1695. // @Summary Test IDP Sync Credentials.
  1696. // @Router /api/idp/sync/test [post]
  1697. // @Tags IDP
  1698. // @Success 200 {object} models.SuccessResponse
  1699. // @Failure 400 {object} models.ErrorResponse
  1700. func testIDPSync(w http.ResponseWriter, r *http.Request) {
  1701. var req models.IDPSyncTestRequest
  1702. err := json.NewDecoder(r.Body).Decode(&req)
  1703. if err != nil {
  1704. err = fmt.Errorf("failed to decode request body: %v", err)
  1705. logger.Log(0, err.Error())
  1706. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1707. return
  1708. }
  1709. var idpClient idp.Client
  1710. switch req.AuthProvider {
  1711. case "google":
  1712. idpClient, err = google.NewGoogleWorkspaceClient(req.GoogleAdminEmail, req.GoogleSACredsJson)
  1713. if err != nil {
  1714. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1715. return
  1716. }
  1717. case "azure-ad":
  1718. idpClient = azure.NewAzureEntraIDClient(req.ClientID, req.ClientSecret, req.AzureTenantID)
  1719. case "okta":
  1720. idpClient, err = okta.NewOktaClient(req.OktaOrgURL, req.OktaAPIToken)
  1721. if err != nil {
  1722. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1723. return
  1724. }
  1725. default:
  1726. err = fmt.Errorf("invalid auth provider: %s", req.AuthProvider)
  1727. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1728. return
  1729. }
  1730. err = idpClient.Verify()
  1731. if err != nil {
  1732. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1733. return
  1734. }
  1735. logic.ReturnSuccessResponse(w, r, "idp sync test successful")
  1736. }
  1737. // @Summary Gets idp sync status.
  1738. // @Router /api/idp/sync/status [get]
  1739. // @Tags IDP
  1740. // @Success 200 {object} models.SuccessResponse
  1741. func getIDPSyncStatus(w http.ResponseWriter, r *http.Request) {
  1742. logic.ReturnSuccessResponseWithJson(w, r, proAuth.GetIDPSyncStatus(), "idp sync status retrieved")
  1743. }
  1744. // @Summary Remove idp integration.
  1745. // @Router /api/idp [delete]
  1746. // @Tags IDP
  1747. // @Success 200 {object} models.SuccessResponse
  1748. // @Failure 500 {object} models.ErrorResponse
  1749. func removeIDPIntegration(w http.ResponseWriter, r *http.Request) {
  1750. superAdmin, err := logic.GetSuperAdmin()
  1751. if err != nil {
  1752. logic.ReturnErrorResponse(
  1753. w,
  1754. r,
  1755. logic.FormatError(fmt.Errorf("failed to get superadmin: %v", err), "internal"),
  1756. )
  1757. return
  1758. }
  1759. if superAdmin.AuthType == models.OAuth {
  1760. logic.ReturnErrorResponse(
  1761. w,
  1762. r,
  1763. logic.FormatError(fmt.Errorf("cannot remove idp integration with superadmin oauth user"), "badrequest"),
  1764. )
  1765. return
  1766. }
  1767. settings := logic.GetServerSettings()
  1768. settings.AuthProvider = ""
  1769. settings.OIDCIssuer = ""
  1770. settings.ClientID = ""
  1771. settings.ClientSecret = ""
  1772. settings.SyncEnabled = false
  1773. settings.GoogleAdminEmail = ""
  1774. settings.GoogleSACredsJson = ""
  1775. settings.AzureTenant = ""
  1776. settings.UserFilters = nil
  1777. settings.GroupFilters = nil
  1778. err = logic.UpsertServerSettings(settings)
  1779. if err != nil {
  1780. logic.ReturnErrorResponse(
  1781. w,
  1782. r,
  1783. logic.FormatError(fmt.Errorf("failed to remove idp integration: %v", err), "internal"),
  1784. )
  1785. return
  1786. }
  1787. proAuth.ResetAuthProvider()
  1788. proAuth.ResetIDPSyncHook()
  1789. go func() {
  1790. err := proAuth.SyncFromIDP()
  1791. if err != nil {
  1792. logger.Log(0, "failed to sync from idp: ", err.Error())
  1793. } else {
  1794. logger.Log(0, "sync from idp complete")
  1795. }
  1796. }()
  1797. logic.ReturnSuccessResponse(w, r, "removed idp integration successfully")
  1798. }