From 67b6a3288cec844bbbec9b615259557aa7c4aa95 Mon Sep 17 00:00:00 2001
From: mikeryan <mikeryan@4420.no-reply.drupal.org>
Date: Thu, 12 Apr 2012 18:23:57 +0200
Subject: [PATCH] File collection field handler for migrate module

---
 field_collection.info        |    1 +
 field_collection.migrate.inc |  155 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 156 insertions(+), 0 deletions(-)
 create mode 100644 field_collection.migrate.inc

diff --git a/field_collection.info b/field_collection.info
index 494eb27..a5bc3cf 100644
--- a/field_collection.info
+++ b/field_collection.info
@@ -4,6 +4,7 @@ core = 7.x
 dependencies[] = entity
 files[] = field_collection.test
 files[] = field_collection.info.inc
+files[] = field_collection.migrate.inc
 files[] = views/field_collection_handler_relationship.inc
 configure = admin/structure/field-collections
 package = Fields
diff --git a/field_collection.migrate.inc b/field_collection.migrate.inc
new file mode 100644
index 0000000..aa97461
--- /dev/null
+++ b/field_collection.migrate.inc
@@ -0,0 +1,155 @@
+<?php
+
+/**
+ * @file
+ * Support for the Migrate API.
+ *
+ * Your field collection migration should be run after the host entity migration.
+ * For example, if the collection is attached to nodes being imported by
+ * ArticleMigration, your collection migration class constructor should look like:
+ *
+ * @code
+ * $this->dependencies = array('Article');
+ * ...
+ * $this->destination = new MigrateDestinationFieldCollection($this->machineName,
+ *   array('host_entity_type' => 'node'));
+ * ...
+ * $this->addFieldMapping('host_entity_id', 'source_article_id')
+ *   ->sourceMigration('Article');
+ */
+
+/**
+ * Destination class implementing migration into field_collection.
+ */
+class MigrateDestinationFieldCollection extends MigrateDestinationEntity {
+  /**
+   * The type of entity hosting this collection field (e.g., node).
+   *
+   * @var string
+   */
+  protected $hostEntityType;
+
+  static public function getKeySchema() {
+    return array(
+      'cid' => array(
+        'type' => 'int',
+        'unsigned' => TRUE,
+        'not null' => TRUE,
+        'description' => 'ID of field collection item',
+      ),
+    );
+  }
+
+  /**
+   * Basic initialization.
+   *
+   * @param array $options
+   *  Options applied to collections.
+   */
+  public function __construct($bundle, array $options = array()) {
+    parent::__construct('field_collection_item', $bundle, $options);
+    $this->hostEntityType = $options['host_entity_type'];
+  }
+
+  /**
+   * Returns a list of fields available to be mapped for this collection (bundle).
+   *
+   * @return array
+   *  Keys: machine names of the fields (to be passed to addFieldMapping).
+   *  Values: Human-friendly descriptions of the fields.
+   */
+  public function fields() {
+    $fields = migrate_handler_invoke_all('Entity', 'fields', $this->entityType, $this->bundle);
+    $fields['host_entity_id'] = t('Field collection host ID');
+    return $fields;
+  }
+
+  /**
+   * Import a single term.
+   *
+   * @param $collection
+   *  Collection object to build. Pre-filled with any fields mapped in the migration.
+   * @param $row
+   *  Raw source data object - passed through to prepare/complete handlers.
+   *
+   * @return array
+   *  Array of key fields (item_id only in this case) of the collection that was saved if
+   *  successful. FALSE on failure.
+   */
+  public function import(stdClass $collection, stdClass $row) {
+    if (isset($row->migrate_map_destid1)) {
+      // We're updated an existing entity - start from the previous data.
+      $entity = entity_load_single('field_collection_item', $row->migrate_map_destid1);
+      $updating = TRUE;
+    }
+    else {
+      $entity = entity_create('field_collection_item', array('field_name' => $this->bundle));
+      $updating = FALSE;
+      // The host entity cannot be reset - we only set it on initial insert
+      $host_entity = entity_load_single($this->hostEntityType, $collection->host_entity_id);
+      // Remove an empty value
+      $language = isset($host_entity->language) ? $host_entity->language : LANGUAGE_NONE;
+      if (empty($host_entity->{$this->bundle}[$language][0]['value'])) {
+        unset($host_entity->{$this->bundle});
+      }
+      $entity->setHostEntity($this->hostEntityType, $host_entity);
+    }
+
+    unset($collection->host_entity_id);
+
+    foreach ((array) $collection as $field => $value) {
+      $entity->{$field} = $value;
+    }
+
+    $this->prepare($entity, $row);
+
+    migrate_instrument_start('field_collection_save');
+    $status = entity_save('field_collection_item', $entity);
+    migrate_instrument_stop('field_collection_save');
+
+    if ($status) {
+      $this->complete($entity, $row);
+      if ($updating) {
+        $this->numUpdated++;
+      }
+      else {
+        $this->numCreated++;
+      }
+      return array($entity->item_id);
+    }
+    else {
+      return FALSE;
+    }
+  }
+
+  /**
+   * Delete a migrated collection.
+   *
+   * @param $key
+   *  Array of fields representing the key.
+   */
+  public function rollback(array $key) {
+    $item_id = reset($key);
+
+    $this->prepareRollback($item_id);
+    $field_collection_item = field_collection_item_load($item_id);
+    // If the collection wasn't imported we can't roll it back so check if the
+    // loaded object matches the fieldcollection item class.
+    if ($field_collection_item instanceof  FieldCollectionItemEntity) {
+      $field_collection_item->delete();
+    }
+
+    $this->completeRollback($item_id);
+    return TRUE;
+  }
+}
+
+/**
+ * Implements migrate hook_migrate_api().
+ */
+function field_collection_migrate_api() {
+  $api = array(
+    'api' => 2,
+  );
+  return $api;
+}
-- 
1.7.4.4