mark owner object as changed when attributes are changed

fromholdio · Jun 16, 2023 · d2b2350 · d2b2350
1 parent 99a47d0
commit d2b2350
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/Extensions/Attributable.php b/src/Extensions/Attributable.php
@@ -10,6 +10,7 @@
 use SilverStripe\Forms\HiddenField;
 use SilverStripe\ORM\DataExtension;
 use SilverStripe\ORM\DataObject;
+use SilverStripe\Versioned\Versioned;
 
 class Attributable extends DataExtension
 {
@@ -147,6 +148,12 @@ public function detachAttribute($attribute)
         $existing = Attribution::get_from_pair($this->owner, $attribute);
         if ($existing) {
             $this->owner->Attributions()->remove($existing);
+            // mark owner as changed
+            if ($this->owner->hasExtension(Versioned::class)) {
+                $this->owner->writeToStage(Versioned::DRAFT);
+            } else {
+                $this->owner->write();
+            }
         }
     }
 
@@ -175,6 +182,12 @@ public function attachAttribute($attribute)
         $attribution->write();
 
         $this->owner->Attributions()->add($attribution);
+        // mark owner as changed
+        if ($this->owner->hasExtension(Versioned::class)) {
+            $this->owner->writeToStage(Versioned::DRAFT);
+        } else {
+            $this->owner->write();
+        }
 
         return $attribution;
     }