networkHttpController.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745
  1. package controller
  2. import (
  3. "context"
  4. "encoding/base64"
  5. "encoding/json"
  6. "errors"
  7. "fmt"
  8. "net/http"
  9. "strings"
  10. "time"
  11. "github.com/go-playground/validator/v10"
  12. "github.com/gorilla/mux"
  13. "github.com/gravitl/netmaker/functions"
  14. "github.com/gravitl/netmaker/models"
  15. "github.com/gravitl/netmaker/mongoconn"
  16. "github.com/gravitl/netmaker/servercfg"
  17. "go.mongodb.org/mongo-driver/bson"
  18. "go.mongodb.org/mongo-driver/mongo"
  19. "go.mongodb.org/mongo-driver/mongo/options"
  20. )
  21. func networkHandlers(r *mux.Router) {
  22. r.HandleFunc("/api/networks", securityCheck(http.HandlerFunc(getNetworks))).Methods("GET")
  23. r.HandleFunc("/api/networks", securityCheck(http.HandlerFunc(createNetwork))).Methods("POST")
  24. r.HandleFunc("/api/networks/{networkname}", securityCheck(http.HandlerFunc(getNetwork))).Methods("GET")
  25. r.HandleFunc("/api/networks/{networkname}", securityCheck(http.HandlerFunc(updateNetwork))).Methods("PUT")
  26. r.HandleFunc("/api/networks/{networkname}", securityCheck(http.HandlerFunc(deleteNetwork))).Methods("DELETE")
  27. r.HandleFunc("/api/networks/{networkname}/keyupdate", securityCheck(http.HandlerFunc(keyUpdate))).Methods("POST")
  28. r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(http.HandlerFunc(createAccessKey))).Methods("POST")
  29. r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(http.HandlerFunc(getAccessKeys))).Methods("GET")
  30. r.HandleFunc("/api/networks/{networkname}/signuptoken", securityCheck(http.HandlerFunc(getSignupToken))).Methods("GET")
  31. r.HandleFunc("/api/networks/{networkname}/keys/{name}", securityCheck(http.HandlerFunc(deleteAccessKey))).Methods("DELETE")
  32. }
  33. //Security check is middleware for every function and just checks to make sure that its the master calling
  34. //Only admin should have access to all these network-level actions
  35. //or maybe some Users once implemented
  36. func securityCheck(next http.Handler) http.HandlerFunc {
  37. return func(w http.ResponseWriter, r *http.Request) {
  38. var errorResponse = models.ErrorResponse{
  39. Code: http.StatusUnauthorized, Message: "W1R3: It's not you it's me.",
  40. }
  41. var params = mux.Vars(r)
  42. bearerToken := r.Header.Get("Authorization")
  43. err := SecurityCheck(params["networkname"], bearerToken)
  44. if err != nil {
  45. errorResponse.Message = err.Error()
  46. returnErrorResponse(w, r, errorResponse)
  47. return
  48. }
  49. next.ServeHTTP(w, r)
  50. }
  51. }
  52. func SecurityCheck(netname, token string) error {
  53. hasnetwork := netname != ""
  54. networkexists, err := functions.NetworkExists(netname)
  55. if err != nil {
  56. return err
  57. }
  58. if hasnetwork && !networkexists {
  59. //errorResponse = models.ErrorResponse{
  60. // Code: http.StatusNotFound, Message: "W1R3: This network does not exist.",
  61. //}
  62. //returnErrorResponse(w, r, errorResponse)
  63. return errors.New("This network does not exist")
  64. }
  65. var hasBearer = true
  66. var tokenSplit = strings.Split(token, " ")
  67. var authToken = ""
  68. if len(tokenSplit) < 2 {
  69. hasBearer = false
  70. } else {
  71. authToken = tokenSplit[1]
  72. }
  73. //all endpoints here require master so not as complicated
  74. //still might not be a good way of doing this
  75. if !hasBearer || !authenticateMaster(authToken) {
  76. //errorResponse = models.ErrorResponse{
  77. // Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
  78. // }
  79. // returnErrorResponse(w, r, errorResponse)
  80. return errors.New("You are unauthorized to access this endpoint")
  81. } //else {
  82. return nil
  83. }
  84. //Consider a more secure way of setting master key
  85. func authenticateMaster(tokenString string) bool {
  86. if tokenString == servercfg.GetMasterKey() {
  87. return true
  88. }
  89. return false
  90. }
  91. //simple get all networks function
  92. func getNetworks(w http.ResponseWriter, r *http.Request) {
  93. networks, err := functions.ListNetworks()
  94. if err != nil {
  95. returnErrorResponse(w, r, formatError(err, "internal"))
  96. return
  97. } else {
  98. w.WriteHeader(http.StatusOK)
  99. json.NewEncoder(w).Encode(networks)
  100. return
  101. }
  102. }
  103. func ValidateNetworkUpdate(network models.NetworkUpdate) error {
  104. v := validator.New()
  105. // _ = v.RegisterValidation("addressrange_valid", func(fl validator.FieldLevel) bool {
  106. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  107. // return isvalid
  108. // })
  109. //_ = v.RegisterValidation("addressrange6_valid", func(fl validator.FieldLevel) bool {
  110. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  111. // return isvalid
  112. // })
  113. // _ = v.RegisterValidation("localrange_valid", func(fl validator.FieldLevel) bool {
  114. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  115. // return isvalid
  116. // })
  117. // _ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
  118. // return true
  119. // })
  120. // _ = v.RegisterValidation("displayname_unique", func(fl validator.FieldLevel) bool {
  121. // return true
  122. // })
  123. err := v.Struct(network)
  124. if err != nil {
  125. for _, e := range err.(validator.ValidationErrors) {
  126. fmt.Println(e)
  127. }
  128. }
  129. return err
  130. }
  131. func ValidateNetworkCreate(network models.Network) error {
  132. v := validator.New()
  133. // _ = v.RegisterValidation("addressrange_valid", func(fl validator.FieldLevel) bool {
  134. // isvalid := functions.IsIpCIDR(fl.Field().String())
  135. // return isvalid
  136. // })
  137. _ = v.RegisterValidation("addressrange6_valid", func(fl validator.FieldLevel) bool {
  138. isvalid := true
  139. if *network.IsDualStack {
  140. isvalid = functions.IsIpCIDR(fl.Field().String())
  141. }
  142. return isvalid
  143. })
  144. //
  145. // _ = v.RegisterValidation("localrange_valid", func(fl validator.FieldLevel) bool {
  146. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  147. // return isvalid
  148. // })
  149. //
  150. _ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
  151. isFieldUnique, _ := functions.IsNetworkNameUnique(fl.Field().String())
  152. // inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
  153. return isFieldUnique
  154. })
  155. //
  156. _ = v.RegisterValidation("displayname_unique", func(fl validator.FieldLevel) bool {
  157. isFieldUnique, _ := functions.IsNetworkDisplayNameUnique(fl.Field().String())
  158. return isFieldUnique
  159. })
  160. err := v.Struct(network)
  161. if err != nil {
  162. for _, e := range err.(validator.ValidationErrors) {
  163. fmt.Println(e)
  164. }
  165. }
  166. return err
  167. }
  168. //Simple get network function
  169. func getNetwork(w http.ResponseWriter, r *http.Request) {
  170. // set header.
  171. w.Header().Set("Content-Type", "application/json")
  172. var params = mux.Vars(r)
  173. netname := params["networkname"]
  174. network, err := GetNetwork(netname)
  175. if err != nil {
  176. returnErrorResponse(w, r, formatError(err, "internal"))
  177. return
  178. }
  179. w.WriteHeader(http.StatusOK)
  180. json.NewEncoder(w).Encode(network)
  181. }
  182. func GetNetwork(name string) (models.Network, error) {
  183. var network models.Network
  184. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  185. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  186. filter := bson.M{"netid": name}
  187. err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&network)
  188. defer cancel()
  189. if err != nil {
  190. return models.Network{}, err
  191. }
  192. return network, nil
  193. }
  194. func keyUpdate(w http.ResponseWriter, r *http.Request) {
  195. w.Header().Set("Content-Type", "application/json")
  196. var params = mux.Vars(r)
  197. netname := params["networkname"]
  198. network, err := KeyUpdate(netname)
  199. if err != nil {
  200. returnErrorResponse(w, r, formatError(err, "internal"))
  201. return
  202. }
  203. w.WriteHeader(http.StatusOK)
  204. json.NewEncoder(w).Encode(network)
  205. }
  206. func KeyUpdate(netname string) (models.Network, error) {
  207. network, err := functions.GetParentNetwork(netname)
  208. if err != nil {
  209. return models.Network{}, err
  210. }
  211. network.KeyUpdateTimeStamp = time.Now().Unix()
  212. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  213. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  214. filter := bson.M{"netid": netname}
  215. // prepare update model.
  216. update := bson.D{
  217. {"$set", bson.D{
  218. {"addressrange", network.AddressRange},
  219. {"addressrange6", network.AddressRange6},
  220. {"displayname", network.DisplayName},
  221. {"defaultlistenport", network.DefaultListenPort},
  222. {"defaultpostup", network.DefaultPostUp},
  223. {"defaultpostdown", network.DefaultPostDown},
  224. {"defaultkeepalive", network.DefaultKeepalive},
  225. {"keyupdatetimestamp", network.KeyUpdateTimeStamp},
  226. {"defaultsaveconfig", network.DefaultSaveConfig},
  227. {"defaultinterface", network.DefaultInterface},
  228. {"nodeslastmodified", network.NodesLastModified},
  229. {"networklastmodified", network.NetworkLastModified},
  230. {"allowmanualsignup", network.AllowManualSignUp},
  231. {"checkininterval", network.DefaultCheckInInterval},
  232. }},
  233. }
  234. err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  235. defer cancel()
  236. if err != nil {
  237. return models.Network{}, err
  238. }
  239. return network, nil
  240. }
  241. //Update a network
  242. func AlertNetwork(netid string) error {
  243. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  244. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  245. filter := bson.M{"netid": netid}
  246. var network models.Network
  247. network, err := functions.GetParentNetwork(netid)
  248. if err != nil {
  249. return err
  250. }
  251. updatetime := time.Now().Unix()
  252. update := bson.D{
  253. {"$set", bson.D{
  254. {"nodeslastmodified", updatetime},
  255. {"networklastmodified", updatetime},
  256. }},
  257. }
  258. err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  259. defer cancel()
  260. return err
  261. }
  262. //Update a network
  263. func updateNetwork(w http.ResponseWriter, r *http.Request) {
  264. w.Header().Set("Content-Type", "application/json")
  265. var params = mux.Vars(r)
  266. var network models.Network
  267. network, err := functions.GetParentNetwork(params["networkname"])
  268. if err != nil {
  269. returnErrorResponse(w, r, formatError(err, "internal"))
  270. return
  271. }
  272. var networkChange models.NetworkUpdate
  273. _ = json.NewDecoder(r.Body).Decode(&networkChange)
  274. if networkChange.AddressRange == "" {
  275. networkChange.AddressRange = network.AddressRange
  276. }
  277. if networkChange.AddressRange6 == "" {
  278. networkChange.AddressRange6 = network.AddressRange6
  279. }
  280. if networkChange.NetID == "" {
  281. networkChange.NetID = network.NetID
  282. }
  283. err = ValidateNetworkUpdate(networkChange)
  284. if err != nil {
  285. returnErrorResponse(w, r, formatError(err, "badrequest"))
  286. return
  287. }
  288. returnednetwork, err := UpdateNetwork(networkChange, network)
  289. if err != nil {
  290. returnErrorResponse(w, r, formatError(err, "badrequest"))
  291. return
  292. }
  293. w.WriteHeader(http.StatusOK)
  294. json.NewEncoder(w).Encode(returnednetwork)
  295. }
  296. func UpdateNetwork(networkChange models.NetworkUpdate, network models.Network) (models.Network, error) {
  297. //NOTE: Network.NetID is intentionally NOT editable. It acts as a static ID for the network.
  298. //DisplayName can be changed instead, which is what shows on the front end
  299. if networkChange.NetID != network.NetID {
  300. return models.Network{}, errors.New("NetID is not editable")
  301. }
  302. haschange := false
  303. hasrangeupdate := false
  304. haslocalrangeupdate := false
  305. if networkChange.AddressRange != "" {
  306. haschange = true
  307. hasrangeupdate = true
  308. network.AddressRange = networkChange.AddressRange
  309. }
  310. if networkChange.LocalRange != "" {
  311. haschange = true
  312. haslocalrangeupdate = true
  313. network.LocalRange = networkChange.LocalRange
  314. }
  315. if networkChange.IsLocal != nil {
  316. network.IsLocal = networkChange.IsLocal
  317. }
  318. if networkChange.IsDualStack != nil {
  319. network.IsDualStack = networkChange.IsDualStack
  320. }
  321. if networkChange.DefaultListenPort != 0 {
  322. network.DefaultListenPort = networkChange.DefaultListenPort
  323. haschange = true
  324. }
  325. if networkChange.DefaultPostDown != "" {
  326. network.DefaultPostDown = networkChange.DefaultPostDown
  327. haschange = true
  328. }
  329. if networkChange.DefaultInterface != "" {
  330. network.DefaultInterface = networkChange.DefaultInterface
  331. haschange = true
  332. }
  333. if networkChange.DefaultPostUp != "" {
  334. network.DefaultPostUp = networkChange.DefaultPostUp
  335. haschange = true
  336. }
  337. if networkChange.DefaultKeepalive != 0 {
  338. network.DefaultKeepalive = networkChange.DefaultKeepalive
  339. haschange = true
  340. }
  341. if networkChange.DisplayName != "" {
  342. network.DisplayName = networkChange.DisplayName
  343. haschange = true
  344. }
  345. if networkChange.DefaultCheckInInterval != 0 {
  346. network.DefaultCheckInInterval = networkChange.DefaultCheckInInterval
  347. haschange = true
  348. }
  349. if networkChange.AllowManualSignUp != nil {
  350. network.AllowManualSignUp = networkChange.AllowManualSignUp
  351. haschange = true
  352. }
  353. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  354. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  355. filter := bson.M{"netid": network.NetID}
  356. if haschange {
  357. network.SetNetworkLastModified()
  358. }
  359. // prepare update model.
  360. update := bson.D{
  361. {"$set", bson.D{
  362. {"addressrange", network.AddressRange},
  363. {"addressrange6", network.AddressRange6},
  364. {"displayname", network.DisplayName},
  365. {"defaultlistenport", network.DefaultListenPort},
  366. {"defaultpostup", network.DefaultPostUp},
  367. {"defaultpostdown", network.DefaultPostDown},
  368. {"defaultkeepalive", network.DefaultKeepalive},
  369. {"defaultsaveconfig", network.DefaultSaveConfig},
  370. {"defaultinterface", network.DefaultInterface},
  371. {"nodeslastmodified", network.NodesLastModified},
  372. {"networklastmodified", network.NetworkLastModified},
  373. {"allowmanualsignup", network.AllowManualSignUp},
  374. {"localrange", network.LocalRange},
  375. {"islocal", network.IsLocal},
  376. {"isdualstack", network.IsDualStack},
  377. {"checkininterval", network.DefaultCheckInInterval},
  378. }},
  379. }
  380. err := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  381. defer cancel()
  382. if err != nil {
  383. return models.Network{}, err
  384. }
  385. //Cycles through nodes and gives them new IP's based on the new range
  386. //Pretty cool, but also pretty inefficient currently
  387. if hasrangeupdate {
  388. err = functions.UpdateNetworkNodeAddresses(network.NetID)
  389. if err != nil {
  390. return models.Network{}, err
  391. }
  392. }
  393. if haslocalrangeupdate {
  394. err = functions.UpdateNetworkPrivateAddresses(network.NetID)
  395. if err != nil {
  396. return models.Network{}, err
  397. }
  398. }
  399. returnnetwork, err := functions.GetParentNetwork(network.NetID)
  400. if err != nil {
  401. return models.Network{}, err
  402. }
  403. return returnnetwork, nil
  404. }
  405. //Delete a network
  406. //Will stop you if there's any nodes associated
  407. func deleteNetwork(w http.ResponseWriter, r *http.Request) {
  408. // Set header
  409. w.Header().Set("Content-Type", "application/json")
  410. var params = mux.Vars(r)
  411. network := params["networkname"]
  412. count, err := DeleteNetwork(network)
  413. if err != nil {
  414. returnErrorResponse(w, r, formatError(err, "badrequest"))
  415. return
  416. }
  417. w.WriteHeader(http.StatusOK)
  418. json.NewEncoder(w).Encode(count)
  419. }
  420. func DeleteNetwork(network string) (*mongo.DeleteResult, error) {
  421. none := &mongo.DeleteResult{}
  422. nodecount, err := functions.GetNetworkNodeNumber(network)
  423. if err != nil {
  424. //returnErrorResponse(w, r, formatError(err, "internal"))
  425. return none, err
  426. } else if nodecount > 0 {
  427. //errorResponse := models.ErrorResponse{
  428. // Code: http.StatusForbidden, Message: "W1R3: Node check failed. All nodes must be deleted before deleting network.",
  429. //}
  430. //returnErrorResponse(w, r, errorResponse)
  431. return none, errors.New("Node check failed. All nodes must be deleted before deleting network")
  432. }
  433. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  434. filter := bson.M{"netid": network}
  435. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  436. deleteResult, err := collection.DeleteOne(ctx, filter)
  437. defer cancel()
  438. if err != nil {
  439. //returnErrorResponse(w, r, formatError(err, "internal"))
  440. return none, err
  441. }
  442. return deleteResult, nil
  443. }
  444. //Create a network
  445. //Pretty simple
  446. func createNetwork(w http.ResponseWriter, r *http.Request) {
  447. w.Header().Set("Content-Type", "application/json")
  448. var network models.Network
  449. // we decode our body request params
  450. err := json.NewDecoder(r.Body).Decode(&network)
  451. if err != nil {
  452. returnErrorResponse(w, r, formatError(err, "internal"))
  453. return
  454. }
  455. err = CreateNetwork(network)
  456. if err != nil {
  457. returnErrorResponse(w, r, formatError(err, "badrequest"))
  458. return
  459. }
  460. w.WriteHeader(http.StatusOK)
  461. //json.NewEncoder(w).Encode(result)
  462. }
  463. func CreateNetwork(network models.Network) error {
  464. //TODO: Not really doing good validation here. Same as createNode, updateNode, and updateNetwork
  465. //Need to implement some better validation across the board
  466. if network.IsLocal == nil {
  467. falsevar := false
  468. network.IsLocal = &falsevar
  469. }
  470. if network.IsDualStack == nil {
  471. falsevar := false
  472. network.IsDualStack = &falsevar
  473. }
  474. err := ValidateNetworkCreate(network)
  475. if err != nil {
  476. //returnErrorResponse(w, r, formatError(err, "badrequest"))
  477. return err
  478. }
  479. network.SetDefaults()
  480. network.SetNodesLastModified()
  481. network.SetNetworkLastModified()
  482. network.KeyUpdateTimeStamp = time.Now().Unix()
  483. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  484. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  485. // insert our network into the network table
  486. _, err = collection.InsertOne(ctx, network)
  487. defer cancel()
  488. if err != nil {
  489. return err
  490. }
  491. return nil
  492. }
  493. // BEGIN KEY MANAGEMENT SECTION
  494. //TODO: Very little error handling
  495. //accesskey is created as a json string inside the Network collection item in mongo
  496. func createAccessKey(w http.ResponseWriter, r *http.Request) {
  497. w.Header().Set("Content-Type", "application/json")
  498. var params = mux.Vars(r)
  499. var accesskey models.AccessKey
  500. //start here
  501. network, err := functions.GetParentNetwork(params["networkname"])
  502. if err != nil {
  503. returnErrorResponse(w, r, formatError(err, "internal"))
  504. return
  505. }
  506. err = json.NewDecoder(r.Body).Decode(&accesskey)
  507. if err != nil {
  508. returnErrorResponse(w, r, formatError(err, "internal"))
  509. return
  510. }
  511. key, err := CreateAccessKey(accesskey, network)
  512. if err != nil {
  513. returnErrorResponse(w, r, formatError(err, "badrequest"))
  514. return
  515. }
  516. w.WriteHeader(http.StatusOK)
  517. json.NewEncoder(w).Encode(key)
  518. //w.Write([]byte(accesskey.AccessString))
  519. }
  520. func CreateAccessKey(accesskey models.AccessKey, network models.Network) (models.AccessKey, error) {
  521. fmt.Println(accesskey)
  522. if accesskey.Name == "" {
  523. accesskey.Name = functions.GenKeyName()
  524. }
  525. if accesskey.Value == "" {
  526. accesskey.Value = functions.GenKey()
  527. }
  528. if accesskey.Uses == 0 {
  529. accesskey.Uses = 1
  530. }
  531. for _, key := range network.AccessKeys {
  532. if key.Name == accesskey.Name {
  533. return models.AccessKey{}, errors.New("Duplicate AccessKey Name")
  534. }
  535. }
  536. privAddr := ""
  537. if network.IsLocal != nil {
  538. if *network.IsLocal {
  539. privAddr = network.LocalRange
  540. }
  541. }
  542. netID := network.NetID
  543. address := servercfg.GetGRPCHost() + ":" + servercfg.GetGRPCPort()
  544. accessstringdec := address + "|" + netID + "|" + accesskey.Value + "|" + privAddr
  545. accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
  546. //validate accesskey
  547. v := validator.New()
  548. err := v.Struct(accesskey)
  549. if err != nil {
  550. for _, e := range err.(validator.ValidationErrors) {
  551. fmt.Println(e)
  552. }
  553. return models.AccessKey{}, err
  554. }
  555. network.AccessKeys = append(network.AccessKeys, accesskey)
  556. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  557. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  558. // Create filter
  559. filter := bson.M{"netid": network.NetID}
  560. // Read update model from body request
  561. fmt.Println("Adding key to " + network.NetID)
  562. // prepare update model.
  563. update := bson.D{
  564. {"$set", bson.D{
  565. {"accesskeys", network.AccessKeys},
  566. }},
  567. }
  568. err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  569. defer cancel()
  570. if err != nil {
  571. //returnErrorResponse(w, r, formatError(err, "internal"))
  572. return models.AccessKey{}, err
  573. }
  574. return accesskey, nil
  575. }
  576. func GetSignupToken(netID string) (models.AccessKey, error) {
  577. var accesskey models.AccessKey
  578. address := servercfg.GetGRPCHost() + ":" + servercfg.GetGRPCPort()
  579. accessstringdec := address + "|" + netID + "|" + "" + "|"
  580. accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
  581. return accesskey, nil
  582. }
  583. func getSignupToken(w http.ResponseWriter, r *http.Request) {
  584. w.Header().Set("Content-Type", "application/json")
  585. var params = mux.Vars(r)
  586. netID := params["networkname"]
  587. token, err := GetSignupToken(netID)
  588. if err != nil {
  589. returnErrorResponse(w, r, formatError(err, "internal"))
  590. return
  591. }
  592. w.WriteHeader(http.StatusOK)
  593. json.NewEncoder(w).Encode(token)
  594. }
  595. //pretty simple get
  596. func getAccessKeys(w http.ResponseWriter, r *http.Request) {
  597. w.Header().Set("Content-Type", "application/json")
  598. var params = mux.Vars(r)
  599. network := params["networkname"]
  600. keys, err := GetKeys(network)
  601. if err != nil {
  602. returnErrorResponse(w, r, formatError(err, "internal"))
  603. return
  604. }
  605. w.WriteHeader(http.StatusOK)
  606. json.NewEncoder(w).Encode(keys)
  607. }
  608. func GetKeys(net string) ([]models.AccessKey, error) {
  609. var network models.Network
  610. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  611. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  612. filter := bson.M{"netid": net}
  613. err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&network)
  614. defer cancel()
  615. if err != nil {
  616. return []models.AccessKey{}, err
  617. }
  618. return network.AccessKeys, nil
  619. }
  620. //delete key. Has to do a little funky logic since it's not a collection item
  621. func deleteAccessKey(w http.ResponseWriter, r *http.Request) {
  622. w.Header().Set("Content-Type", "application/json")
  623. var params = mux.Vars(r)
  624. keyname := params["name"]
  625. netname := params["networkname"]
  626. err := DeleteKey(keyname, netname)
  627. if err != nil {
  628. returnErrorResponse(w, r, formatError(err, "badrequest"))
  629. return
  630. }
  631. w.WriteHeader(http.StatusOK)
  632. }
  633. func DeleteKey(keyname, netname string) error {
  634. network, err := functions.GetParentNetwork(netname)
  635. if err != nil {
  636. return err
  637. }
  638. //basically, turn the list of access keys into the list of access keys before and after the item
  639. //have not done any error handling for if there's like...1 item. I think it works? need to test.
  640. found := false
  641. var updatedKeys []models.AccessKey
  642. for _, currentkey := range network.AccessKeys {
  643. if currentkey.Name == keyname {
  644. found = true
  645. } else {
  646. updatedKeys = append(updatedKeys, currentkey)
  647. }
  648. }
  649. if !found {
  650. return errors.New("key " + keyname + " does not exist")
  651. }
  652. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  653. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  654. // Create filter
  655. filter := bson.M{"netid": netname}
  656. // prepare update model.
  657. update := bson.D{
  658. {"$set", bson.D{
  659. {"accesskeys", updatedKeys},
  660. }},
  661. }
  662. err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  663. defer cancel()
  664. if err != nil {
  665. return err
  666. }
  667. return nil
  668. }