Added Foreign key and Added Primary key

rootrus · web-flow · commit c55cdf81ffaa · 2020-11-30T05:16:15.000+13:00
- Once export, it will check the database if there is any foreign key for that specific database and table . If exist then it will format it into migration format and out put as add_fk_to_table.php.
- If columnname id, is primary and auto increment then it will set as $table-&gt;id();
diff --git a/ExportToLaravelMigration.spBundle/MigrationParser.php b/ExportToLaravelMigration.spBundle/MigrationParser.php
@@ -76,6 +76,16 @@ class MigrationParser
      */
     protected $tableCharsetAndCollationFile;
 
+    /**
+     * @var string
+     */
+    protected $foreignStructureFile;
+
+    /**
+     * @var string
+     */
+    protected $hasForeign;
+
     /**
      * MigrationParser constructor.
      *
@@ -84,14 +94,18 @@ class MigrationParser
      * @param string $keysFile
      * @param string $constraintsFile
      * @param string $tableCharsetAndCollationFile
+     * @param string $foreignStructureFile
+     * @param string $hasForeign
      */
-    public function __construct($tableName, $structureFile, $keysFile, $constraintsFile, $tableCharsetAndCollationFile)
+    public function __construct($tableName, $structureFile, $keysFile, $constraintsFile, $tableCharsetAndCollationFile, $foreignStructureFile, $hasForeign)
     {
         $this->tableName = $tableName;
         $this->structureFile = $structureFile;
         $this->keysFile = $keysFile;
         $this->constraintsFile = $constraintsFile;
         $this->tableCharsetAndCollationFile = $tableCharsetAndCollationFile;
+        $this->foreignStructureFile = $foreignStructureFile;
+        $this->hasForeign = $hasForeign;
     }
 
     public function makeMigration()
@@ -110,40 +124,94 @@ public function makeMigration()
         $constraints = trim(implode($eol . $indent12, $this->formatConstraints())) . $eol;
         $tableCollationAndCharset = trim(implode($eol . $indent12, $this->formatTableCollationAndCharset())) . $eol;
         $extras = trim(implode($eol . $indent8, $this->formatExtras())) . $eol;
+        $foreign = trim(implode($eol . $indent12, $this->formatForeign())) . $eol;
+        $foreignDrop = trim(implode($eol . $indent12, $this->formatForeignDrop())) . $eol;
+        if ($this->hasForeign == "true") {
+            $className = 'AddFkTo' . $this->studly($this->tableName) . 'Table';
+            $output = file_get_contents(__DIR__ . '/table.stub');
+            $output = str_replace(
+                [
+                    ':VERSION:',
+                    'DummyClass',
+                    'DummyTable',
+                    "// foreign\n",
+                    "// foreignDrop\n",                
+                ],
+                [
+                    $this->version,
+                    $className,
+                    $this->tableName,
+                    $foreign,
+                    $foreignDrop,
+                ],
+                $output
+            );
+        } else {
+            $className = 'Create' . $this->studly($this->tableName) . 'Table';
+            $output = file_get_contents(__DIR__ . '/create.stub');
+            $output = str_replace(
+                [
+                    ':VERSION:',
+                    'DummyClass',
+                    'DummyTable',
+                    "// structure\n",
+                    "// keys\n",
+                    "// constraints\n",
+                    "// tableCollationAndCharset\n",
+                    "// extras\n",
+                ],
+                [
+                    $this->version,
+                    $className,
+                    $this->tableName,
+                    $structure,
+                    $keys,
+                    $constraints,
+                    $tableCollationAndCharset,
+                    $extras,
+                ],
+                $output
+            );
+        }
 
-        $output = file_get_contents(__DIR__ . '/create.stub');
-
-        $className = 'Create' . $this->studly($this->tableName) . 'Table';
-
-        $output = str_replace(
-            [
-                ':VERSION:',
-                'DummyClass',
-                'DummyTable',
-                "// structure\n",
-                "// keys\n",
-                "// constraints\n",
-                "// tableCollationAndCharset\n",
-                "// extras\n",
-            ],
-            [
-                $this->version,
-                $className,
-                $this->tableName,
-                $structure,
-                $keys,
-                $constraints,
-                $tableCollationAndCharset,
-                $extras,
-            ],
-            $output
-        );
 
         $output = preg_replace("/^(\s*\R){2,}/m", "\n", $output);
 
         return $output;
     }
 
+    public function formatForeign()
+    {
+        $fields = [];
+
+        $rows = file($this->foreignStructureFile);
+        array_shift($rows);
+
+        foreach ($rows as $row) {
+            list($table, $colName, $constName, $refTable, $refColumnName) = explode("\t",
+                $row, 5);
+            $fields[] = '$table->unsignedBigInteger(\''. trim($colName) .'\')->change();';
+            $fields[] = '$table->foreign(\''. trim($colName) .'\')->references(\''. trim($refColumnName) .'\')->on(\''. trim($refTable) .'\');';
+        }
+        return array_filter($fields);
+    }
+
+    public function formatForeignDrop()
+    {
+        $fields = [];
+
+        $rows = file($this->foreignStructureFile);
+        array_shift($rows);
+
+        foreach ($rows as $row) {
+            list($table, $colName, $constName, $refTable, $refColumnName) = explode("\t",
+                $row, 5);
+            //$table->dropForeign('local_table_foreign_id_foreign');
+            $fields[] = '$table->dropForeign(\''. trim($table) .'_'. trim($colName) .'_foreign\');';
+        }
+        return array_filter($fields);
+    }
+
     protected function studly($value)
     {
         $value = ucwords(str_replace(['-', '_'], ' ', $value));
@@ -242,6 +310,24 @@ public function buildStructure()
             $this->structure['remember_token']['nullable'] = false;
             $this->structure['remember_token']['field'] = null;
         }
+
+        // look for id primary key
+
+        if (
+            array_key_exists('id', $this->structure)
+            && $this->structure['id']['method'] === 'integer'
+            && $this->structure['id']['autoIncrement'] === true
+            && $this->structure['id']['args'] === null
+        ) {
+            $this->structure['id']['method'] = 'id';
+            $this->structure['id']['args'] = null;
+            $this->structure['id']['default'] = null;
+            $this->structure['id']['nullable'] = false;
+            $this->structure['id']['field'] = null;
+            $this->structure['id']['comment'] = null;
+            $this->structure['id']['autoIncrement'] = null;
+            $this->structure['id']['unsigned'] = null;
+        }
     }
 
     public function formatStructure()
@@ -359,7 +445,7 @@ public function buildKeys()
                 $primaryColumn = reset($primary['columns']);
                 $field = $this->structure[$primaryColumn];
                 // and that column is an "increments" field ...
-                if (isset($field['args']['autoIncrement']) && $field['args']['autoIncrement'] === 'true') {
+                if (isset($field['autoIncrement']) && $field['autoIncrement'] == '1') {
                     // then don't build the primary key, since Laravel takes care of it
                     unset($this->keys['PRIMARY']);
                 }
@@ -375,7 +461,7 @@ public function formatKeys()
             $columns = $this->escapeArray($data['columns']);
 
             if ($field === 'PRIMARY') {
-                $fields[$field] = sprintf('$table->primary(%s);', $columns);
+                //$fields[$field] = sprintf('$table->primary(%s);', $columns);
             } else {
                 $fields[$field] = sprintf('$table->%s(%s, \'%s\');',
                     $data['method'],
diff --git a/ExportToLaravelMigration.spBundle/parse.php b/ExportToLaravelMigration.spBundle/parse.php
@@ -1,15 +1,19 @@
 <?php
-
 require_once(__DIR__ . '/MigrationParser.php');
-
 $tableName = $argv[1];
+$hasForeign = "false";
+if (isset($argv[2])) {
+	$hasForeign = "true";
+}
 
 $m = new MigrationParser(
     $tableName,
     __DIR__ . '/rowsStructure.tsv',
     __DIR__ . '/rowsKeys.tsv',
     __DIR__ . '/rowsConstraints.tsv',
-    __DIR__ . '/rowsTableCharsetAndCollation.tsv'
+    __DIR__ . '/rowsTableCharsetAndCollation.tsv',
+    __DIR__ . '/rowsForeignStructure.tsv',
+    $hasForeign
 );
 
 echo $m->makeMigration();
diff --git a/ExportToLaravelMigration.spBundle/parse.sh b/ExportToLaravelMigration.spBundle/parse.sh
@@ -117,7 +117,7 @@ do
     # get character set and collation name
     echo "
     SELECT
-	    CHARACTER_SET_NAME, TABLE_COLLATION
+        CHARACTER_SET_NAME, TABLE_COLLATION
     FROM
       information_schema.TABLES, information_schema.COLLATION_CHARACTER_SET_APPLICABILITY
     WHERE
@@ -179,6 +179,7 @@ do
         echo -n "" > "$SP_BUNDLE_PATH/rowsConstraints.tsv"
     fi
 
+
     # process the results and save to the desktop
     FILENAME=$(date "+%Y_%m_%d_%H%M%S_create_${table}_table.php")
     /usr/bin/php "$SP_BUNDLE_PATH/parse.php" "${table}" > $DESTDIR/$FILENAME
@@ -189,5 +190,33 @@ do
 # end loop through tables
 done
 
+# Loop through tables
+for table in "${tables[@]}"
+do
+    # get the table foreign key
+
+    echo "SELECT 
+      TABLE_NAME,COLUMN_NAME,CONSTRAINT_NAME, REFERENCED_TABLE_NAME,REFERENCED_COLUMN_NAME
+    FROM
+      INFORMATION_SCHEMA.KEY_COLUMN_USAGE
+    WHERE
+      REFERENCED_TABLE_SCHEMA = '${SP_SELECTED_DATABASE}' AND TABLE_NAME = '${table}';" > "$SP_QUERY_FILE"
+
+    # execute and save the table structure result
+    execute_sql
+    cp $SP_QUERY_RESULT_FILE "$SP_BUNDLE_PATH/rowsForeignStructure.tsv"
+
+    hasForeignKey=`cat "$SP_BUNDLE_PATH/rowsForeignStructure.tsv" | grep -v "TABLE_NAME" | wc -l | awk '{print $1}'`;
+    if [ $hasForeignKey != 0 ]; then
+        FILENAME=$(date "+%Y_%m_%d_%H%M%S-0_add_fk_to_${table}_table.php")
+        /usr/bin/php "$SP_BUNDLE_PATH/parse.php" "${table}" "foreignkey" > $DESTDIR/$FILENAME
+        echo "<p>Migration Foreing Key saved: <a href=\"SP-REVEAL-FILE://$DESTDIR/$FILENAME\">$FILENAME</a></p>"
+        # clean up
+        clear_temp
+    fi
+    clear_temp
+# end loop through tables
+done
+
 echo "<button onclick=\"window.system.closeHTMLOutputWindow()\">Close</button>"
 exit $SP_BUNDLE_EXIT_SHOW_AS_HTML
diff --git a/ExportToLaravelMigration.spBundle/table.stub b/ExportToLaravelMigration.spBundle/table.stub
@@ -0,0 +1,37 @@
+<?php
+
+use Illuminate\Support\Facades\Schema;
+use Illuminate\Database\Schema\Blueprint;
+use Illuminate\Database\Migrations\Migration;
+
+/**
+ * Migration auto-generated by Sequel Pro Laravel Export (:VERSION:)
+ * @see https://github.com/cviebrock/sequel-pro-laravel-export
+ */
+class DummyClass extends Migration
+{
+    /**
+     * Run the migrations.
+     *
+     * @return void
+     */
+    public function up()
+    {
+        Schema::table('DummyTable', function (Blueprint $table) {
+            // foreign
+        });
+        // extras
+    }
+
+    /**
+     * Reverse the migrations.
+     *
+     * @return void
+     */
+    public function down()
+    {
+        Schema::table('DummyTable', function (Blueprint $table) {
+            // foreignDrop
+        });
+    }
+}
