users.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. package logic
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "sort"
  6. "github.com/gravitl/netmaker/database"
  7. "github.com/gravitl/netmaker/models"
  8. )
  9. // GetUser - gets a user
  10. // TODO support "masteradmin"
  11. func GetUser(username string) (*models.User, error) {
  12. var user models.User
  13. record, err := database.FetchRecord(database.USERS_TABLE_NAME, username)
  14. if err != nil {
  15. return &user, err
  16. }
  17. if err = json.Unmarshal([]byte(record), &user); err != nil {
  18. return &models.User{}, err
  19. }
  20. return &user, err
  21. }
  22. // GetReturnUser - gets a user
  23. func GetReturnUser(username string) (models.ReturnUser, error) {
  24. var user models.ReturnUser
  25. record, err := database.FetchRecord(database.USERS_TABLE_NAME, username)
  26. if err != nil {
  27. return user, err
  28. }
  29. if err = json.Unmarshal([]byte(record), &user); err != nil {
  30. return models.ReturnUser{}, err
  31. }
  32. return user, err
  33. }
  34. // ToReturnUser - gets a user as a return user
  35. func ToReturnUser(user models.User) models.ReturnUser {
  36. return models.ReturnUser{
  37. UserName: user.UserName,
  38. IsSuperAdmin: user.IsSuperAdmin,
  39. IsAdmin: user.IsAdmin,
  40. RemoteGwIDs: user.RemoteGwIDs,
  41. }
  42. }
  43. // SetUserDefaults - sets the defaults of a user to avoid empty fields
  44. func SetUserDefaults(user *models.User) {
  45. if user.RemoteGwIDs == nil {
  46. user.RemoteGwIDs = make(map[string]struct{})
  47. }
  48. }
  49. // SortUsers - Sorts slice of Users by username
  50. func SortUsers(unsortedUsers []models.ReturnUser) {
  51. sort.Slice(unsortedUsers, func(i, j int) bool {
  52. return unsortedUsers[i].UserName < unsortedUsers[j].UserName
  53. })
  54. }
  55. // GetSuperAdmin - fetches superadmin user
  56. func GetSuperAdmin() (models.ReturnUser, error) {
  57. users, err := GetUsers()
  58. if err != nil {
  59. return models.ReturnUser{}, err
  60. }
  61. for _, user := range users {
  62. if user.IsSuperAdmin {
  63. return user, nil
  64. }
  65. }
  66. return models.ReturnUser{}, errors.New("superadmin not found")
  67. }