From fae0500f96781431523b3f809195c02067a34cf7 Mon Sep 17 00:00:00 2001
From: Darren Shelley <darren.shelley@gmail.com>
Date: Wed, 1 Jul 2015 10:30:19 +0100
Subject: [PATCH] Issue #1116408 by mikeryan, marcusx, aidanlis, brockfanning,
 jamsilver, mikeker, rvilar, kostajh, darren shelley: Added Migration Support.

---
 redirect.info        |   1 +
 redirect.migrate.inc | 139 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 140 insertions(+)
 create mode 100644 redirect.migrate.inc

diff --git a/redirect.info b/redirect.info
index c54e5da..8c29a56 100644
--- a/redirect.info
+++ b/redirect.info
@@ -3,6 +3,7 @@ description = Allows users to redirect from old URLs to new URLs.
 core = 7.x
 files[] = redirect.controller.inc
 files[] = redirect.test
+files[] = redirect.migrate.inc
 files[] = views/redirect.views.inc
 ;files[] = views/redirect_handler_field_redirect_type.inc
 files[] = views/redirect_handler_filter_redirect_type.inc
diff --git a/redirect.migrate.inc b/redirect.migrate.inc
new file mode 100644
index 0000000..e657ad3
--- /dev/null
+++ b/redirect.migrate.inc
@@ -0,0 +1,139 @@
+<?php
+
+/**
+ * @file
+ * Migrate support for Redirect module.
+ */
+
+class MigrateRedirectEntityHandler extends MigrateDestinationHandler {
+
+  /**
+   * Constructor.
+   */
+  public function __construct() {
+    $this->registerTypes(array('entity'));
+  }
+
+  /**
+   * Overrides fields().
+   */
+  public function fields() {
+    return array('migrate_redirects' => t('Original path(s) to redirect from.'));
+  }
+
+  /**
+   * Validates a redirect.
+   *
+   * We need to check if a redirect already exists
+   * otherwise if we call redirect_save in complete we get an db
+   * error due to duplicate entries.
+   *
+   * This function is similar to the validate function in the
+   * redirect module but the feedback is handled via the Migrate
+   * saveMessage functionality.
+   */
+  protected function redirectValidate($redirect) {
+    $redirect = (object) $redirect;
+
+    // Check that there there are no redirect loops.
+    $migration = Migration::currentMigration();
+    if (url($redirect->source) == url($redirect->redirect)) {
+      $migration->saveMessage(t('Redirect to self (!redirect) ignored',
+                              array('!redirect' => $redirect->redirect)),
+                  MigrationBase::MESSAGE_INFORMATIONAL);
+      return FALSE;
+    }
+    redirect_hash($redirect);
+    if ($existing = redirect_load_by_hash($redirect->hash)) {
+      if ($redirect->rid != $existing->rid) {
+        $migration->saveMessage(t('The source path is already being redirected.'),
+          MigrationBase::MESSAGE_INFORMATIONAL);
+        return FALSE;
+      }
+    }
+
+    return TRUE;
+  }
+
+  /**
+   * Get redirects from entity or row.
+   */
+  protected function getRedirects($entity, $row) {
+    // If there are multiple redirects defined for the entity, they will be in
+    // $row. If there is just one, it will be in $entity.
+    $migrate_redirects = array();
+    if (isset($row->migrate_redirects) && !empty($row->migrate_redirects)) {
+      $migrate_redirects = $row->migrate_redirects;
+    }
+    else {
+      $migrate_redirects = isset($entity->migrate_redirects) ? $entity->migrate_redirects : NULL;
+    }
+    // If it is not an array already, make it one now.
+    if ($migrate_redirects && !is_array($migrate_redirects)) {
+      $migrate_redirects = array($migrate_redirects);
+    }
+    return $migrate_redirects;
+  }
+
+  /**
+   * Overrides complete().
+   *
+   * @param object $entity
+   *   The Drupal entity.
+   * @param stdClass $row
+   *   The row being migrated.
+   */
+  public function complete($entity, stdClass $row) {
+
+    $migration = Migration::currentMigration();
+    $destination = $migration->getDestination();
+    $entity_type = $destination->getEntityType();
+    $migrate_redirects = $this->getRedirects($entity, $row);
+
+    // We looked up the destination entity_type in the constructor.
+    if (!empty($migrate_redirects) && ($redirect_destination = entity_uri($entity_type, $entity))) {
+      foreach ($migrate_redirects as $path) {
+        $redirect_defaults = array(
+          'status_code' => 301,
+        );
+        if (isset($entity->uid)) {
+          $redirect_defaults['uid'] = $entity->uid;
+        }
+        if (isset($entity->language)) {
+          if ($entity->language == '') {
+            $redirect_defaults['language'] = LANGUAGE_NONE;
+          }
+          else {
+            $redirect_defaults['language'] = $entity->language;
+          }
+        }
+        $redirect = new stdClass();
+        redirect_object_prepare($redirect, $redirect_defaults);
+        $redirect->redirect = $redirect_destination['path'];
+        $parsed = redirect_parse_url($path);
+        $redirect->source = isset($parsed['path']) ? ltrim($parsed['path'], '/') : '';
+        if (!empty($parsed['query'])) {
+          $redirect->source_options['query'] = $parsed['query'];
+        }
+
+        // Only save if the redirect does not already exist.
+        if ($this->redirectValidate($redirect)) {
+          redirect_save($redirect);
+        }
+      }
+    }
+  }
+}
+
+/**
+ * Implements hook_migrate_api().
+ */
+function redirect_migrate_api() {
+  $api = array(
+    'api' => 2,
+    'destination handlers' => array(
+      'MigrateRedirectEntityHandler',
+    ),
+  );
+  return $api;
+}
-- 
1.9.1