danielebarchiesi@4: <?php
danielebarchiesi@4: /**
danielebarchiesi@4:  * @file
danielebarchiesi@4:  * Mapper that handles any fields added by the References module (node_reference, user_reference)
danielebarchiesi@4:  *
danielebarchiesi@4:  * Code is effectively a port of the mapper_for_nodereference_field patch to drupal 7, and borrows
danielebarchiesi@4:  * heavily from the nodereference patch
danielebarchiesi@4:  *
danielebarchiesi@4:  */
danielebarchiesi@4: 
danielebarchiesi@4: /**
danielebarchiesi@4:  * Implements hook_feeds_parser_sources_alter().
danielebarchiesi@4:  */
danielebarchiesi@4: function node_reference_feeds_parser_sources_alter(&$sources, $content_type) {
danielebarchiesi@4:   if (!empty($content_type)) {
danielebarchiesi@4:     $fields = field_info_fields();
danielebarchiesi@4:     foreach ($fields as $field) {
danielebarchiesi@4:       if ($field['type'] == 'node_reference' && isset($field['bundles']['node']) && in_array($content_type, $field['bundles']['node'])) {
danielebarchiesi@4:         $sources['parent:node_reference:'. $field['field_name']] = array(
danielebarchiesi@4:           'name' => t('Feed node: Node Reference (nid): @field_name', array('@field_name' => $field['field_name'])),
danielebarchiesi@4:           'description' => t('Node References from the parent feed node.'),
danielebarchiesi@4:           'callback' => 'node_reference_feeds_get_source',
danielebarchiesi@4:         );
danielebarchiesi@4:       }
danielebarchiesi@4:     }
danielebarchiesi@4:   }
danielebarchiesi@4:   return $sources;
danielebarchiesi@4: }
danielebarchiesi@4: 
danielebarchiesi@4: /**
danielebarchiesi@4:  * Callback for mapping parent node references to child node reference values.
danielebarchiesi@4:  *
danielebarchiesi@4:  * @param $source
danielebarchiesi@4:  *   A FeedsSource object.
danielebarchiesi@4:  * @param $result
danielebarchiesi@4:  *   FeedsParserResult object.
danielebarchiesi@4:  * @param $key
danielebarchiesi@4:  *   The key as defined in the _feeds_parser_sources_alter() hook defined above.
danielebarchiesi@4:  * @return array
danielebarchiesi@4:  *   The node ids that the parent node references.
danielebarchiesi@4:  */
danielebarchiesi@4: function node_reference_feeds_get_source(FeedsSource $source, FeedsParserResult $result, $key) {
danielebarchiesi@4:   if ($node = node_load($source->feed_nid)) {
danielebarchiesi@4:     $results = array();
danielebarchiesi@4:     $field = substr($key, 22);
danielebarchiesi@4:     if (!empty($node->{$field}['und'])) {
danielebarchiesi@4:       foreach ($node->{$field}['und'] as $delta => $value) {
danielebarchiesi@4:         $results[] = $value['nid'];
danielebarchiesi@4:       }
danielebarchiesi@4:     }
danielebarchiesi@4:     return $results;
danielebarchiesi@4:   }
danielebarchiesi@4: }
danielebarchiesi@4: 
danielebarchiesi@4: /**
danielebarchiesi@4:  * Implements hook_feeds_processor_targets_alter() for node_reference fields
danielebarchiesi@4:  *
danielebarchiesi@4:  * @see FeedsNodeProcessor::getMappingTargets().
danielebarchiesi@4:  */
danielebarchiesi@4: function node_reference_feeds_processor_targets_alter(&$targets, $entity_type, $bundle_name) {
danielebarchiesi@4:   foreach (field_info_instances($entity_type, $bundle_name) as $name => $instance) {
danielebarchiesi@4:     $info = field_info_field($name);
danielebarchiesi@4:     if ($info['type'] == 'node_reference') {
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':title'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by node title)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by node title.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':nid'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by node ID)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by node ID.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':url'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by Feeds URL)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds URL.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':guid'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by Feeds GUID)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds GUID.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':title:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by node title) -- allow duplicate nodes'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by node title. This target allows duplicate nodes (nodes can appear more than once in a field).', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':nid:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by node ID) -- allow duplicate nodes'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by node ID. This target allows duplicate nodes (nodes can appear more than once in a field).', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':url:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by Feeds URL) -- allow duplicate nodes'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds URL.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':guid:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (Node reference by Feeds GUID) -- allow duplicate nodes'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds GUID.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4:     }
danielebarchiesi@4:   }
danielebarchiesi@4: }
danielebarchiesi@4: /**
danielebarchiesi@4:  * Implements hook_feeds_processor_targets_alter() for user_reference fields
danielebarchiesi@4:  *
danielebarchiesi@4:  * @see FeedsNodeProcessor::getMappingTargets().
danielebarchiesi@4:  *
danielebarchiesi@4:  */
danielebarchiesi@4: function user_reference_feeds_processor_targets_alter(&$targets, $entity_type, $bundle_name) {
danielebarchiesi@4:   foreach (field_info_instances($entity_type, $bundle_name) as $name => $instance) {
danielebarchiesi@4:     $info = field_info_field($name);
danielebarchiesi@4:     if ($info['type'] == 'user_reference') {
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':name'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by user name)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the user matched by user name. ', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':uid'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by user ID)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the user matched by user ID.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':url'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by Feeds URL)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds URL.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':guid'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by Feeds GUID)'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds GUID.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':name:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by user name) -- allow duplicate users'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the user matched by user name. This target allows duplicate users (users can appear more than once in a field).', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':uid:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by user ID) -- allow duplicate users'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the user matched by user ID. This target allows duplicate users (users can appear more than once in a field).', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':url:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by Feeds URL) -- allow duplicate users'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds URL.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4: 
danielebarchiesi@4:       $targets[$name . ':guid:duplicates'] = array(
danielebarchiesi@4:         'name' => $instance['label'] . t(' (User reference by Feeds GUID) -- allow duplicate users'),
danielebarchiesi@4:         'callback' => 'references_feeds_set_target',
danielebarchiesi@4:         'description' => t('The @label field of the node matched by Feeds GUID.', array('@label' => $instance['label'])),
danielebarchiesi@4:         'real_target' => $name);
danielebarchiesi@4:     }
danielebarchiesi@4:   }
danielebarchiesi@4: }
danielebarchiesi@4: 
danielebarchiesi@4: /**
danielebarchiesi@4:  * Callback for mapping both node reference and user_reference fields
danielebarchiesi@4:  *
danielebarchiesi@4:  * Implementation of hook_feeds_set_target().
danielebarchiesi@4:  *
danielebarchiesi@4:  * @param $source
danielebarchiesi@4:  *   A FeedsSource object.
danielebarchiesi@4:  * @param $entity
danielebarchiesi@4:  *   The entity to map to.
danielebarchiesi@4:  * @param $target
danielebarchiesi@4:  *   The target key on $entity to map to.
danielebarchiesi@4:  * @param $value
danielebarchiesi@4:  *   The value to map. Can be an array or a string.
danielebarchiesi@4:  */
danielebarchiesi@4: function references_feeds_set_target($source, $entity, $target, $value) {
danielebarchiesi@4:   if (empty($value)) {
danielebarchiesi@4:     return;
danielebarchiesi@4:   }
danielebarchiesi@4: 
danielebarchiesi@4:   // Handle comma delimited or non-multiple values.
danielebarchiesi@4:   if (!is_array($value)) {
danielebarchiesi@4:     $value = array($value);
danielebarchiesi@4:   }
danielebarchiesi@4: 
danielebarchiesi@4:   // Determine the field we are matching against, and whether duplicates are allowed.
danielebarchiesi@4:   $target_info = explode(':', $target, 3);
danielebarchiesi@4:   if (sizeof($target_info) == 3) {
danielebarchiesi@4:     list($target, $match_key, $duplicates) = $target_info;
danielebarchiesi@4:   }
danielebarchiesi@4:   else {
danielebarchiesi@4:     list($target, $match_key) = $target_info;
danielebarchiesi@4:   }
danielebarchiesi@4: 
danielebarchiesi@4:   // Load field definition.
danielebarchiesi@4:   $info = field_info_field($target);
danielebarchiesi@4:   // Parameters to handle differences between node references and user references
danielebarchiesi@4:   if ($info['type'] == 'user_reference') {
danielebarchiesi@4:     $idname = 'uid';
danielebarchiesi@4:     $typename = 'user';
danielebarchiesi@4:     $validate_function = 'user_reference_potential_references';
danielebarchiesi@4:   }
danielebarchiesi@4:   else {
danielebarchiesi@4:     $idname = 'nid';
danielebarchiesi@4:     $typename = 'node';
danielebarchiesi@4:     $validate_function = 'node_reference_potential_references';
danielebarchiesi@4:   }
danielebarchiesi@4: 
danielebarchiesi@4:   $field = isset($entity->$target) ? $entity->$target : array();
danielebarchiesi@4:   if (!isset($field['und'])) {
danielebarchiesi@4:     $field['und'] = array();
danielebarchiesi@4:   }
danielebarchiesi@4: 
danielebarchiesi@4:   // Match values against nodes and add to field.
danielebarchiesi@4:   foreach ($value as $v) {
danielebarchiesi@4:     // Create options.
danielebarchiesi@4:     $options = array(
danielebarchiesi@4:       'string' => $v,
danielebarchiesi@4:       'match' => 'equals',
danielebarchiesi@4:       'ids' => array(),
danielebarchiesi@4:       'limit' => 1,
danielebarchiesi@4:     );
danielebarchiesi@4: 
danielebarchiesi@4:     switch ($match_key) {
danielebarchiesi@4:       case 'title':
danielebarchiesi@4:       case 'name':
danielebarchiesi@4:         // Validate node title or user name.
danielebarchiesi@4:         if ((is_string($options['string']) && $options['string'] != '') || is_numeric($options['string'])) {
danielebarchiesi@4:           // Lookup potential exact matches for the value (limit to one result).
danielebarchiesi@4:           $matches = $validate_function($info,$options);
danielebarchiesi@4:           // Use the first element of the potential matches.
danielebarchiesi@4:           $options['ids']= key($matches);
danielebarchiesi@4:         }
danielebarchiesi@4: 
danielebarchiesi@4:         // Alert if no match is found.
danielebarchiesi@4:         if (empty($options['ids'])) {
danielebarchiesi@4:           drupal_set_message(t('%title does not match an existing '.$typename, array('%title' => $options['string'])));
danielebarchiesi@4:         }
danielebarchiesi@4:         break;
danielebarchiesi@4: 
danielebarchiesi@4:       case 'nid':
danielebarchiesi@4:       case 'uid':
danielebarchiesi@4:         // Make sure it is a positive integer.
danielebarchiesi@4:         if ((is_int($options['string']) || ctype_digit($options['string'])) && $options['string'] > 0 && $options['string'] !== '') {
danielebarchiesi@4:           // Make sure it is a valid node id or user id for this field.
danielebarchiesi@4:           $matches = $validate_function($info, array($options['string']));
danielebarchiesi@4:           foreach ($matches as $k => $v) {
danielebarchiesi@4:             if ($options['string'] == $k) {
danielebarchiesi@4:               $options['ids'] = $k;
danielebarchiesi@4:             }
danielebarchiesi@4:           }
danielebarchiesi@4:         }
danielebarchiesi@4: 
danielebarchiesi@4:         // Alert if no match is found.
danielebarchiesi@4:         if (empty($options['ids'])) {
danielebarchiesi@4:           drupal_set_message(t('%id is not a valid '.$typename.' id for this field.', array('%id' => $options['string'])));
danielebarchiesi@4:         }
danielebarchiesi@4:         break;
danielebarchiesi@4: 
danielebarchiesi@4:       case 'guid':
danielebarchiesi@4:       case 'url':
danielebarchiesi@4:         // get the value from table feeds-item.
danielebarchiesi@4:         $result = db_query('SELECT f.entity_id FROM {feeds_item} f WHERE f.'.$match_key.' = :v', array( ':v' => $v) );
danielebarchiesi@4:         $options['ids'] = $result->fetchField();
danielebarchiesi@4: 
danielebarchiesi@4:         // Alert if no match is found.
danielebarchiesi@4:         if (empty($options['ids'])) {
danielebarchiesi@4:           drupal_set_message(t('%id is not a valid '.$typename.' id for this field.', array('%id' => $v)));
danielebarchiesi@4:         }
danielebarchiesi@4:         break;
danielebarchiesi@4:     }
danielebarchiesi@4: 
danielebarchiesi@4:     if (!empty($options['ids'])) {
danielebarchiesi@4:       $reference = array($idname => $options['ids']);
danielebarchiesi@4:       if (!empty($duplicates)) {
danielebarchiesi@4:         // Add the reference, ignoring duplicates.
danielebarchiesi@4:         $field['und'][] = $reference;
danielebarchiesi@4:       }
danielebarchiesi@4:       else if (!in_array($reference, $field['und'])) {
danielebarchiesi@4:         // Add the reference only if it doesn't already exist.
danielebarchiesi@4:         $field['und'][] = $reference;
danielebarchiesi@4:       }
danielebarchiesi@4: 
danielebarchiesi@4:       if ($info['cardinality'] == 1) {
danielebarchiesi@4:         break;
danielebarchiesi@4:       }
danielebarchiesi@4:     }
danielebarchiesi@4:   }
danielebarchiesi@4:   $entity->{$target} = $field;
danielebarchiesi@4: }