create table if not exists when set account id

affili-ir · Nov 27, 2020 · 1fa344d · 1fa344d
1 parent 6d947e1
commit 1fa344d
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 7 deletions.
diff --git a/classes/Action.php b/classes/Action.php
@@ -5,10 +5,12 @@
 
 require_once 'Woocommerce.php';
 require_once 'ListTable.php';
+require_once 'Installer.php';
 
 
 use AffiliIR\Woocommerce as AffiliIR_Woocommerce;
 use AffiliIR\ListTable as AffiliIR_ListTable;
+use AffiliIR\Installer as AffiliIR_Installer;
 
 class Action
 {
@@ -81,6 +83,8 @@ public function setAccountId()
         $condition = isset($_POST['affili_set_account_id']) && $nonce;
 
         if($condition) {
+            $this->createTableIfNotExists();
+
             $account_id = sanitize_text_field($_POST['account_id']);
             $data = [
                 'name'  => 'account_id',
@@ -338,4 +342,12 @@ public function findCategoryAjax()
         echo json_encode( $return );
         wp_die();
     }
+
+    private function createTableIfNotExists()
+    {
+        $sql        = AffiliIR_Installer::sqlString();
+        $table_name = $this->wpdb->prefix.AffiliIR_Installer::$table;
+
+        maybe_create_table($table_name, $sql);
+    }
 }
diff --git a/classes/Installer.php b/classes/Installer.php
@@ -17,6 +17,19 @@ public static function activation()
             return;
         }
 
+        $sql = self::sqlString();
+
+        dbDelta($sql);
+        update_option(self::$table.'_db_version', $db_version);
+    }
+
+    public static function deactivation()
+    {
+        //
+    }
+
+    public static function sqlString()
+    {
         global $wpdb;
 
         $table_name      = $wpdb->prefix.self::$table;
@@ -32,12 +45,6 @@ public static function activation()
             ) {$charset_collate};"
         ;
 
-        dbDelta($sql);
-        update_option(self::$table.'_db_version', $db_version);
-    }
-
-    public static function deactivation()
-    {
-        //
+        return $sql;
     }
 }