FIX Do not show TOTPSecret until the member has been initially saved

elliot-sawyer · Nov 5, 2018 · 0ca0af7 · 0ca0af7
1 parent bc3baa2
commit 0ca0af7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/Extensions/MemberExtension.php b/src/Extensions/MemberExtension.php
@@ -25,7 +25,7 @@ class MemberExtension extends DataExtension
      * @var array
      */
     private static $db = [
-        'TOTPSecret' => 'Varchar(1024)'
+        'TOTPSecret' => 'Varchar(1024)',
     ];
 
     /**
@@ -48,6 +48,10 @@ public function onBeforeWrite()
      */
     public function updateCMSFields(FieldList $fields)
     {
+        if (!$this->owner->exists()) {
+            $fields->removeByName('TOTPSecret');
+        }
+
         if (strlen($this->owner->TOTPSecret)) {
             $qrcodeURI = $this->GoogleAuthenticatorQRCode();
             $fields->addFieldToTab('Root.Main', ToggleCompositeField::create(