<?php

/**
 * @file
 * Base integration with the Migrate API class.
 */

/**
 * Custom extended MigrateFieldHandler class.
 */
class MigrateCountryFieldHandler extends MigrateFieldHandler {
  public function __construct() {
    $this->registerTypes(array('country'));
  }

  public function prepare($entity, array $field_info, array $instance, array $values) {
    $arguments = (isset($values['arguments'])) ? $values['arguments'] : array();
    $language = $this->getFieldLanguage($entity, $field_info, $arguments);
    $delta = 0;

    foreach ($values as $value) {
      $return[$language][$delta]['iso2'] = $value;
      $delta++;
    }

    return isset($return) ? $return : NULL;
  }
}