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
 								}