fix: "unsupported relations: Roles" when using unassign-group on a …

…user moul#329
alterway · Jun 5, 2023 · 49a52c4 · 49a52c4
1 parent 103e8d8
commit 49a52c4
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/pkg/bastion/shell.go b/pkg/bastion/shell.go
@@ -1955,12 +1955,15 @@ GLOBAL OPTIONS:
  tx.Rollback()
  return err
  }
- if err := model.Association("Groups").Append(&appendGroups); err != nil {
+
+ groups := tx.Model(user).Association("Groups")
+
+ if err := groups.Append(&appendGroups); err != nil {
  tx.Rollback()
  return err
  }
  if len(deleteGroups) > 0 {
- if err := model.Association("Groups").Delete(deleteGroups); err != nil {
+ if err := groups.Delete(deleteGroups); err != nil {
  tx.Rollback()
  return err
  }
@@ -1975,12 +1978,15 @@ GLOBAL OPTIONS:
  tx.Rollback()
  return err
  }
- if err := model.Association("Roles").Append(&appendRoles); err != nil {
+
+ roles := tx.Model(user).Association("Groups")
+
+ if err := roles.Append(&appendRoles); err != nil {
  tx.Rollback()
  return err
  }
  if len(deleteRoles) > 0 {
- if err := model.Association("Roles").Delete(deleteRoles); err != nil {
+ if err := roles.Delete(deleteRoles); err != nil {
  tx.Rollback()
  return err
  }