Merge pull request #144 from dnadesign/master

Check for subclasses of Versioned
unclecheese · Sep 5, 2016 · 37bdaa8 · 37bdaa8
2 parents a7260e6 + 23429cf
commit 37bdaa8
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/code/extensions/BetterButtonDataObject.php b/code/extensions/BetterButtonDataObject.php
@@ -196,9 +196,18 @@ protected function createButtonGroup($groupName) {
      * @return boolean
      */
     public function checkVersioned() {
-        return $this->owner->config()->better_buttons_versioned_enabled &&
-               $this->owner->hasExtension('Versioned') &&
-               count($this->owner->getVersionedStages()) > 1;
+        $isVersioned = false;
+
+        foreach($this->owner->getExtensionInstances() as $extension) {
+            if($extension instanceof Versioned) {
+                $isVersioned = true;
+                break;
+            }
+        }
+
+        return $isVersioned &&
+            $this->owner->config()->better_buttons_versioned_enabled &&
+            count($this->owner->getVersionedStages()) > 1;
     }