From aecb2c995dfc3f2855c7329432eb7ff26dfa97a8 Mon Sep 17 00:00:00 2001
From: Nicolas Meyer <nicolas@commerceguys.com>
Date: Tue, 30 Aug 2011 15:17:15 +0200
Subject: [PATCH] Issue #1261086 by Haza: Added Implements migrate handler.

---
 entityreference.info                               |    2 +
 entityreference.migrate.inc                        |   26 ------------------
 includes/migrate/entityreference.migrate.class.inc |   28 ++++++++++++++++++++
 3 files changed, 30 insertions(+), 26 deletions(-)
 create mode 100644 includes/migrate/entityreference.migrate.class.inc

diff --git a/entityreference.info b/entityreference.info
index bc6104f..ef24697 100644
--- a/entityreference.info
+++ b/entityreference.info
@@ -3,3 +3,5 @@ description = Provides a field that can reference other entities.
 core = 7.x
 dependencies[] = entity
 
+; Migrate plugins.
+files[] = includes/migrate/entityreference.migrate.class.inc
diff --git a/entityreference.migrate.inc b/entityreference.migrate.inc
index 037fba1..7379e0d 100644
--- a/entityreference.migrate.inc
+++ b/entityreference.migrate.inc
@@ -1,35 +1,9 @@
 <?php
 
 /**
- * @file
- * Support for processing entity reference fields in Migrate.
- */
-
-/**
  * Implement hook_migrate_api().
  */
 function entityreference_migrate_api() {
   return array('api' => 2);
 }
 
-class MigrateEntityReferenceFieldHandler extends MigrateFieldHandler {
-  public function __construct() {
-    $this->registerTypes(array('entityreference'));
-  }
-
-  public function prepare($entity, array $field_info, array $instance, array $values) {
-    $migration = Migration::currentMigration();
-    $arguments = (isset($values['arguments']))? $values['arguments']: array();
-    $language = $this->getFieldLanguage($entity, $field_info, $arguments);
-    // Setup the standard Field API array for saving.
-    $delta = 0;
-    foreach ($values as $value) {
-      $return[$language][$delta]['target_id'] = $value;
-      $delta++;
-    }
-    if (!isset($return)) {
-      $return = NULL;
-    }
-    return $return;
-  }
-}
diff --git a/includes/migrate/entityreference.migrate.class.inc b/includes/migrate/entityreference.migrate.class.inc
new file mode 100644
index 0000000..5cc1265
--- /dev/null
+++ b/includes/migrate/entityreference.migrate.class.inc
@@ -0,0 +1,28 @@
+<?php
+
+/**
+ * @file
+ * Support for processing entity reference fields in Migrate.
+ */
+
+class MigrateEntityReferenceFieldHandler extends MigrateFieldHandler {
+  public function __construct() {
+    $this->registerTypes(array('entityreference'));
+  }
+
+  public function prepare($entity, array $field_info, array $instance, array $values) {
+    $migration = Migration::currentMigration();
+    $arguments = (isset($values['arguments']))? $values['arguments']: array();
+    $language = $this->getFieldLanguage($entity, $field_info, $arguments);
+    // Setup the standard Field API array for saving.
+    $delta = 0;
+    foreach ($values as $value) {
+      $return[$language][$delta]['target_id'] = $value;
+      $delta++;
+    }
+    if (!isset($return)) {
+      $return = NULL;
+    }
+    return $return;
+  }
+}
-- 
1.7.6