You are here

callback_attachments_references_settings.inc in Search API attachments 7

Search API data alteration callback.

File

contrib/search_api_attachments_references/includes/callback_attachments_references_settings.inc
View source
<?php

/**
 * @file
 * Search API data alteration callback.
 */

/**
 * {@inheritdoc}
 */
class SearchApiAttachmentsReferencesAlterSettings extends SearchApiAttachmentsAlterSettings {

  /**
   * {@inheritdoc}
   */
  public function alterItems(array &$items) {

    // Get the reference fields.
    $reference_fields = $this
      ->getReferenceFields();

    // Get the file fields.
    $file_fields = $this
      ->getFileFields();
    foreach ($items as &$item) {
      $item_wrapper = entity_metadata_wrapper($this->index
        ->getEntityType(), $item);

      // Case of entity reference fields.
      foreach ($reference_fields as $reference_field) {
        if (isset($item->{$reference_field['field_name']})) {
          $referenced_entities = $item_wrapper->{$reference_field['field_name']}
            ->value();

          // Manage case of field having no value and continue with next field.
          if (empty($referenced_entities)) {
            continue;
          }

          // Manage case of single value fields by reproducing the structure of
          // multiple values fields.
          if (is_object($referenced_entities)) {
            $referenced_entities = array(
              $referenced_entities,
            );
          }

          // Index the files content.
          if (!empty($file_fields)) {
            foreach ($file_fields as $file_field) {
              foreach ($referenced_entities as $referenced_entity) {

                // The referenced entity has the file field.
                if (isset($referenced_entity->{$file_field['field_name']}) && !empty($referenced_entity->{$file_field['field_name']})) {

                  // Get the files.
                  $files = field_get_items($reference_field['settings']['target_type'], $referenced_entity, $file_field['field_name']);
                  if (!empty($files)) {

                    // Limit to the max number of value per field.
                    if (isset($this->options['number_indexed']) && $this->options['number_indexed'] != '0' && count($files) > $this->options['number_indexed']) {
                      $files = array_slice($files, 0, $this->options['number_indexed']);
                    }
                    foreach ($files as $file) {
                      if ($this
                        ->isFileIndexable($file, $item, $reference_field['field_name'])) {
                        $attachments = 'attachments_' . $file_field['field_name'];
                        if (isset($item->{$attachments})) {
                          $item->{$attachments} .= ' ' . $this
                            ->getFileContent($file);
                        }
                        else {
                          $item->{$attachments} = $this
                            ->getFileContent($file);
                        }
                      }
                    }
                  }
                }
              }
            }
          }
        }
      }
    }
  }

  /**
   * {@inheritdoc}
   */
  public function propertyInfo() {
    $ret = array();
    if ($this->index
      ->getEntityType() == 'file') {
      $ret['attachments_content'] = array(
        'label' => 'File content',
        'description' => 'File content',
        'type' => 'text',
      );
    }
    else {
      $fields = $this
        ->getFileFields();
      foreach ($fields as $name => $field) {
        $ret['attachments_' . $name] = array(
          'label' => 'Attachment content: ' . $name,
          'description' => $name,
          'type' => 'text',
        );
      }
    }
    return $ret;
  }

  /**
   * {@inheritdoc}
   */
  protected function getFileFields() {
    $ret = array();
    foreach (field_info_fields() as $name => $field) {
      if ($field['type'] == 'file') {
        $ret[$name] = $field;
      }
    }
    return $ret;
  }

  /**
   * {@inheritdoc}
   */
  protected function getReferenceFields() {
    $ret = array();
    foreach (field_info_fields() as $name => $field) {
      if ($field['type'] == 'node_reference') {
        $field['settings']['target_type'] = 'node';
        $ret[$name] = $field;
      }
    }
    return $ret;
  }

}