diff --git a/server/authorization/organization.go b/server/authorization/organization.go index 07b0936..2923b89 100644 --- a/server/authorization/organization.go +++ b/server/authorization/organization.go @@ -20,6 +20,14 @@ func isSameOrganization(organizationA viewmodel.Organization, organizationB view } func grabOrgFromUser(user viewmodel.User) (viewmodel.Organization, error) { + if len(user.Profiles) < 1 { + return viewmodel.Organization{}, fmt.Errorf("User has no organizations %v", user) + } + + return user.Profiles[0].Organization, nil +} + +func grabOrgFromUserDirectly(user viewmodel.User) (viewmodel.Organization, error) { if len(user.Organizations) < 1 { return viewmodel.Organization{}, fmt.Errorf("User has no organizations %v", user) } diff --git a/server/authorization/user.go b/server/authorization/user.go index 6e23a4c..fac116f 100644 --- a/server/authorization/user.go +++ b/server/authorization/user.go @@ -21,7 +21,7 @@ func CanCreateUser(currentUser viewmodel.User, updatingUser viewmodel.User) bool return false } - updatingUserOrganization, err := grabOrgFromUser(updatingUser) + updatingUserOrganization, err := grabOrgFromUserDirectly(updatingUser) if err != nil { return false }