Merge pull request #145 from mjvcallibrity/master

Create conditional that checks state to properly display saved / save draft button
unclecheese · Sep 16, 2016 · a703795 · a703795
2 parents 37bdaa8 + 1272872
commit a703795
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/code/buttons/BetterButton_SaveDraft.php b/code/buttons/BetterButton_SaveDraft.php
@@ -41,4 +41,20 @@ public function baseTransform() {
             ->setAttribute('data-icon-alternate', 'addpage')
             ->setAttribute('data-text-alternate', _t('CMSMain.SAVEDRAFT', 'Save draft'));
     }
+
+
+    /** 
+     * Update the UI to reflect unsaved state
+     * @return void
+     */
+    public function transformToButton() {
+        parent::transformToButton();
+
+        if($this->gridFieldRequest->recordIsDeletedFromStage()) {
+            $this->addExtraClass('ss-ui-alternate');
+        }
+
+        return $this;
+    }
+
 }