Merge pull request #1 from yournextagency/migration_fix

Updating migration to fix https://github.com/sebastian-lenz/craft-li…
yournextagency · May 15, 2023 · a17d1f7 · a17d1f7
2 parents 67e33b2 + 73909d1
commit a17d1f7
Showing 1 changed file with 66 additions and 39 deletions.
diff --git a/src/migrations/m190417_202153_migrateDataToTable.php b/src/migrations/m190417_202153_migrateDataToTable.php
@@ -12,6 +12,7 @@
 use lenz\linkfield\fields\LinkField;
 use lenz\linkfield\records\LinkRecord;
 use verbb\supertable\fields\SuperTableField;
+use Yii;
 
 /**
  * m190417_202153_migrateDataToTable migration.
@@ -119,7 +120,11 @@ private function updateMatrixField(Matrix $matrixField): void {
    */
   private function updateLinkField(LinkField $field, string $table, string $handlePrefix = ''): void {
     $insertRows = [];
+    $columnsToDrop = [];
     $columnName = ($field->columnPrefix ?: 'field_') . $handlePrefix . $field->handle;
+    if (!$this->db->tableExists($table)) {
+      return;
+    }
     if ($field->columnSuffix) {
       $columnName .= '_' . $field->columnSuffix;
     }
@@ -132,55 +137,77 @@ private function updateLinkField(LinkField $field, string $table, string $handle
       }
     };
 
-    // Make sure the rows actually exist in the elements table.
-    $rows = (new Query())
-      ->select(['t.elementId', 't.siteId', 't.'.$columnName])
-      ->from(['t' => $table])
-      ->innerJoin(['e' => Table::ELEMENTS], '[[t.elementId]] = [[e.id]]')
-      ->all();
+    $yiiTable = Yii::$app->db->schema->getTableSchema($table);
+    if (isset($yiiTable->columns[$columnName])) {
+      // do something
 
-    foreach ($rows as $row) {
-      $payload = Json::decode($row[$columnName]);
-      if (!is_array($payload)) {
-        continue;
-      }
 
-      $type = $payload['type'] ?? null;
-      $value = $payload['value'] ?? '';
-      unset($payload['type']);
-      unset($payload['value']);
+      // Make sure the rows actually exist in the elements table.
+      $rowQuery = (new Query())
+        ->select(['t.elementId', 't.siteId', 't.'.$columnName])
+        ->from(['t' => $table])
+        ->innerJoin(['e' => Table::ELEMENTS], '[[t.elementId]] = [[e.id]]');
+      $rows = $rowQuery->all();
 
-      if ($value && is_numeric($value)) {
-        $doesExist = (new Query())
-          ->select('id')
-          ->where(['id' => $value])
-          ->from('{{%elements}}')
-          ->exists();
+      foreach ($rows as $row) {
 
-        if (!$doesExist) {
-          $value = null;
+        if (substr($row[$columnName], 0, 1) != "{") {
+          continue;
+        }
+
+        $payload = Json::decode((string)$row[$columnName], $associative=true, $depth=512, JSON_THROW_ON_ERROR);
+
+        if (!is_array($payload)) {
+          continue;
+        }
+
+        $type = $payload['type'] ?? null;
+        $value = $payload['value'] ?? '';
+        unset($payload['type']);
+        unset($payload['value']);
+
+        if ($value && is_numeric($value)) {
+          $doesExist = (new Query())
+            ->select('id')
+            ->where(['id' => $value])
+            ->from('{{%elements}}')
+            ->exists();
+
+          if (!$doesExist) {
+            $value = null;
+          }
         }
-      }
 
-      $insertRows[] = [
-        $row['elementId'],                          // elementId
-        $row['siteId'],                             // siteId
-        $field->id,                                 // fieldId
-        is_numeric($value) ? $value : null,         // linkedId
-        is_numeric($value) ? $row['siteId'] : null, // linkedSiteId
-        $type,                                      // type
-        is_numeric($value) ? null : $value,         // linkedUrl
-        Json::encode($payload)                      // payload
-      ];
-
-      if (count($insertRows) > 100) {
-        $writeRows($insertRows);
-        $insertRows = [];
+        $insertRows[] = [
+          $row['elementId'],                          // elementId
+          $row['siteId'],                             // siteId
+          $field->id,                                 // fieldId
+          is_numeric($value) ? $value : null,         // linkedId
+          is_numeric($value) ? $row['siteId'] : null, // linkedSiteId
+          $type,                                      // type
+          is_numeric($value) ? null : $value,         // linkedUrl
+          Json::encode($payload)                      // payload
+        ];
+
+        if (count($insertRows) > 100) {
+          $writeRows($insertRows);
+          $insertRows = [];
+        }
+        if (!in_array($columnName, $columnsToDrop)) {
+          $columnsToDrop[] = $columnName;
+        }
       }
     }
 
     $writeRows($insertRows);
-    $this->dropColumn($table, $columnName);
+
+    foreach ($columnsToDrop as $col) {
+      $yiiTable = Yii::$app->db->schema->getTableSchema($table);
+      if (isset($yiiTable->columns[$columnName])) {
+        $this->dropColumn($table, $col);
+      }
+    }
+    //
   }
 
   /**