'Search API',
'description' => 'Create and configure search engines.',
'page callback' => 'search_api_admin_overview',
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
);
$items[$pre . '/overview'] = array(
'title' => 'Overview',
'type' => MENU_DEFAULT_LOCAL_TASK,
'weight' => -10,
);
$items[$pre . '/add_server'] = array(
'title' => 'Add server',
'description' => 'Create a new search server.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_add_server'),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'weight' => -1,
'type' => MENU_LOCAL_ACTION,
);
$items[$pre . '/add_index'] = array(
'title' => 'Add index',
'description' => 'Create a new search index.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_add_index'),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'type' => MENU_LOCAL_ACTION,
);
$items[$pre . '/server/%search_api_server'] = array(
'title' => 'View server',
'title callback' => 'search_api_admin_item_title',
'title arguments' => array(5),
'description' => 'View server details.',
'page callback' => 'search_api_admin_server_view',
'page arguments' => array(5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
);
$items[$pre . '/server/%search_api_server/view'] = array(
'title' => 'View',
'weight' => -10,
'type' => MENU_DEFAULT_LOCAL_TASK,
);
$items[$pre . '/server/%search_api_server/edit'] = array(
'title' => 'Edit',
'description' => 'Edit server details.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_server_edit', 5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'weight' => -1,
'type' => MENU_LOCAL_TASK,
);
$items[$pre . '/server/%search_api_server/delete'] = array(
'title' => 'Delete',
'title callback' => 'search_api_title_delete_page',
'title arguments' => array(5),
'description' => 'Delete server.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_confirm', 'server', 'delete', 5),
'access callback' => 'search_api_access_delete_page',
'access arguments' => array(5),
'file' => 'search_api.admin.inc',
'type' => MENU_LOCAL_TASK,
);
$items[$pre . '/index/%search_api_index'] = array(
'title' => 'View index',
'title callback' => 'search_api_admin_item_title',
'title arguments' => array(5),
'description' => 'View index details.',
'page callback' => 'search_api_admin_index_view',
'page arguments' => array(5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
);
$items[$pre . '/index/%search_api_index/view'] = array(
'title' => 'View',
'type' => MENU_DEFAULT_LOCAL_TASK,
'weight' => -10,
);
$items[$pre . '/index/%search_api_index/status'] = array(
'title' => 'Status',
'description' => 'Display and work on index status.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_index_status_form', 5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'weight' => -8,
'type' => MENU_LOCAL_TASK,
'context' => MENU_CONTEXT_INLINE | MENU_CONTEXT_PAGE,
);
$items[$pre . '/index/%search_api_index/edit'] = array(
'title' => 'Settings',
'description' => 'Edit index settings.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_index_edit', 5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'weight' => -6,
'type' => MENU_LOCAL_TASK,
'context' => MENU_CONTEXT_INLINE | MENU_CONTEXT_PAGE,
);
$items[$pre . '/index/%search_api_index/fields'] = array(
'title' => 'Fields',
'description' => 'Select indexed fields.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_index_fields', 5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'weight' => -4,
'type' => MENU_LOCAL_TASK,
'context' => MENU_CONTEXT_INLINE | MENU_CONTEXT_PAGE,
);
$items[$pre . '/index/%search_api_index/workflow'] = array(
'title' => 'Workflow',
'description' => 'Edit index workflow.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_index_workflow', 5),
'access arguments' => array('administer search_api'),
'file' => 'search_api.admin.inc',
'weight' => -2,
'type' => MENU_LOCAL_TASK,
'context' => MENU_CONTEXT_INLINE | MENU_CONTEXT_PAGE,
);
$items[$pre . '/index/%search_api_index/delete'] = array(
'title' => 'Delete',
'title callback' => 'search_api_title_delete_page',
'title arguments' => array(5),
'description' => 'Delete index.',
'page callback' => 'drupal_get_form',
'page arguments' => array('search_api_admin_confirm', 'index', 'delete', 5),
'access callback' => 'search_api_access_delete_page',
'access arguments' => array(5),
'file' => 'search_api.admin.inc',
'type' => MENU_LOCAL_TASK,
);
return $items;
}
/**
* Implements hook_theme().
*/
function search_api_theme() {
$themes['search_api_server'] = array(
'variables' => array(
'id' => NULL,
'name' => '',
'machine_name' => '',
'description' => NULL,
'enabled' => NULL,
'class_name' => NULL,
'class_description' => NULL,
'options' => array(),
'status' => ENTITY_CUSTOM,
),
'file' => 'search_api.admin.inc',
);
$themes['search_api_index'] = array(
'variables' => array(
'id' => NULL,
'name' => '',
'machine_name' => '',
'description' => NULL,
'item_type' => NULL,
'enabled' => NULL,
'server' => NULL,
'options' => array(),
'fields' => array(),
'indexed_items' => 0,
'total_items' => 0,
'status' => ENTITY_CUSTOM,
'read_only' => 0,
),
'file' => 'search_api.admin.inc',
);
$themes['search_api_admin_item_order'] = array(
'render element' => 'element',
'file' => 'search_api.admin.inc',
);
$themes['search_api_admin_fields_table'] = array(
'render element' => 'element',
'file' => 'search_api.admin.inc',
);
return $themes;
}
/**
* Implements hook_permission().
*/
function search_api_permission() {
return array(
'administer search_api' => array(
'title' => t('Administer Search API'),
'description' => t('Create and configure Search API servers and indexes.'),
),
);
}
/**
* Implements hook_cron().
*
* Will index $options['cron-limit'] items for each enabled index.
*/
function search_api_cron() {
$queue = DrupalQueue::get('search_api_indexing_queue');
foreach (search_api_index_load_multiple(FALSE, array('enabled' => TRUE, 'read_only' => 0)) as $index) {
$limit = isset($index->options['cron_limit'])
? $index->options['cron_limit']
: SEARCH_API_DEFAULT_CRON_LIMIT;
if ($limit) {
try {
$task = array('index' => $index->machine_name);
$ids = search_api_get_items_to_index($index, -1);
if (!$ids) {
continue;
}
$batches = $limit > 0 ? array_chunk($ids, $limit, TRUE) : array($ids);
foreach ($batches as $batch) {
$task['items'] = $batch;
$queue->createItem($task);
}
// Mark items as queued so they won't be inserted into the queue again
// on the next cron run.
search_api_track_item_queued($index, $ids);
}
catch (SearchApiException $e) {
watchdog_exception('search_api', $e);
}
}
}
}
/**
* Implements hook_cron_queue_info().
*
* Defines a queue for saved searches that should be checked for new items.
*/
function search_api_cron_queue_info() {
return array(
'search_api_indexing_queue' => array(
'worker callback' => '_search_api_indexing_queue_process',
'time' => variable_get('search_api_index_worker_callback_runtime', 15),
),
);
}
/**
* Implements hook_entity_info().
*/
function search_api_entity_info() {
$info['search_api_server'] = array(
'label' => t('Search server'),
'controller class' => 'EntityAPIControllerExportable',
'metadata controller class' => FALSE,
'entity class' => 'SearchApiServer',
'base table' => 'search_api_server',
'uri callback' => 'search_api_server_url',
'module' => 'search_api',
'exportable' => TRUE,
'entity keys' => array(
'id' => 'id',
'label' => 'name',
'name' => 'machine_name',
),
);
$info['search_api_index'] = array(
'label' => t('Search index'),
'controller class' => 'EntityAPIControllerExportable',
'metadata controller class' => FALSE,
'entity class' => 'SearchApiIndex',
'base table' => 'search_api_index',
'uri callback' => 'search_api_index_url',
'module' => 'search_api',
'exportable' => TRUE,
'entity keys' => array(
'id' => 'id',
'label' => 'name',
'name' => 'machine_name',
),
);
return $info;
}
/**
* Implements hook_entity_property_info().
*/
function search_api_entity_property_info() {
$info['search_api_server']['properties'] = array(
'id' => array(
'label' => t('ID'),
'type' => 'integer',
'description' => t('The primary identifier for a server.'),
'schema field' => 'id',
'validation callback' => 'entity_metadata_validate_integer_positive',
),
'name' => array(
'label' => t('Name'),
'type' => 'text',
'description' => t('The displayed name for a server.'),
'schema field' => 'name',
'required' => TRUE,
),
'machine_name' => array(
'label' => t('Machine name'),
'type' => 'token',
'description' => t('The internally used machine name for a server.'),
'schema field' => 'machine_name',
'required' => TRUE,
),
'description' => array(
'label' => t('Description'),
'type' => 'text',
'description' => t('The displayed description for a server.'),
'schema field' => 'description',
'sanitize' => 'filter_xss',
),
'class' => array(
'label' => t('Service class'),
'type' => 'text',
'description' => t('The ID of the service class to use for this server.'),
'schema field' => 'class',
'required' => TRUE,
),
'enabled' => array(
'label' => t('Enabled'),
'type' => 'boolean',
'description' => t('A flag indicating whether the server is enabled.'),
'schema field' => 'enabled',
),
);
$info['search_api_index']['properties'] = array(
'id' => array(
'label' => t('ID'),
'type' => 'integer',
'description' => t('An integer identifying the index.'),
'schema field' => 'id',
'validation callback' => 'entity_metadata_validate_integer_positive',
),
'name' => array(
'label' => t('Name'),
'type' => 'text',
'description' => t('A name to be displayed for the index.'),
'schema field' => 'name',
'required' => TRUE,
),
'machine_name' => array(
'label' => t('Machine name'),
'type' => 'token',
'description' => t('The internally used machine name for an index.'),
'schema field' => 'machine_name',
'required' => TRUE,
),
'description' => array(
'label' => t('Description'),
'type' => 'text',
'description' => t("A string describing the index' use to users."),
'schema field' => 'description',
'sanitize' => 'filter_xss',
),
'server' => array(
'label' => t('Server ID'),
'type' => 'token',
'description' => t('The machine name of the search_api_server with which data should be indexed.'),
'schema field' => 'server',
),
'server_entity' => array(
'label' => t('Server'),
'type' => 'search_api_server',
'description' => t('The search_api_server with which data should be indexed.'),
'getter callback' => 'search_api_index_get_server',
),
'item_type' => array(
'label' => t('Item type'),
'type' => 'token',
'description' => t('The type of items stored in this index.'),
'schema field' => 'item_type',
'required' => TRUE,
),
'enabled' => array(
'label' => t('Enabled'),
'type' => 'boolean',
'description' => t('A flag indicating whether the index is enabled.'),
'schema field' => 'enabled',
),
'read_only' => array(
'label' => t('Read only'),
'type' => 'boolean',
'description' => t('A flag indicating whether the index is read-only.'),
'schema field' => 'read_only',
),
);
return $info;
}
/**
* Implements hook_search_api_server_insert().
*
* Calls the postCreate() method for the server.
*/
function search_api_search_api_server_insert(SearchApiServer $server) {
$server->postCreate();
}
/**
* Implements hook_search_api_server_update().
*
* Calls the server's postUpdate() method and marks all of this server's indexes
* for reindexing, if necessary.
*/
function search_api_search_api_server_update(SearchApiServer $server) {
if ($server->postUpdate()) {
foreach (search_api_index_load_multiple(FALSE, array('server' => $server->machine_name)) as $index) {
$index->reindex();
}
}
if ($server->enabled != $server->original->enabled) {
if ($server->enabled) {
// Were there any changes in the server's indexes while it was disabled?
$tasks = variable_get('search_api_tasks', array());
if (isset($tasks[$server->machine_name])) {
foreach ($tasks[$server->machine_name] as $index_id => $index_tasks) {
$index = search_api_index_load($index_id);
foreach ($index_tasks as $task) {
switch ($task) {
case 'add':
$server->addIndex($index);
break;
case 'clear':
$server->deleteItems('all', $index);
break;
case 'clear all':
// Would normally be used with a fake index ID of "", since it doesn't matter.
$server->deleteItems('all');
break;
case 'fields':
if ($server->fieldsUpdated($index)) {
_search_api_index_reindex($index);
}
break;
case 'remove':
$server->removeIndex($index ? $index : $index_id);
break;
default:
if (substr($task, 0, 7) == 'delete-') {
$id = substr($task, 7);
$server->deleteItems(array($id), $index);
}
else {
watchdog('search_api', t('Unknown task "@task" for server "@name".', array('@task' => $task, '@name' => $server->machine_name)), NULL, WATCHDOG_WARNING);
}
}
}
}
unset($tasks[$server->machine_name]);
variable_set('search_api_tasks', $tasks);
}
}
else {
foreach (search_api_index_load_multiple(FALSE, array('server' => $server->machine_name, 'enabled' => 1)) as $index) {
$index->update(array('enabled' => 0));
}
}
}
}
/**
* Implements hook_search_api_server_delete().
*
* Calls the preDelete() method for the server.
*/
function search_api_search_api_server_delete(SearchApiServer $server) {
$server->preDelete();
// Only react on real delete, not revert.
if (!$server->hasStatus(ENTITY_IN_CODE)) {
foreach (search_api_index_load_multiple(FALSE, array('server' => $server->machine_name)) as $index) {
$index->update(array('server' => NULL, 'enabled' => FALSE));
}
}
$tasks = variable_get('search_api_tasks', array());
unset($tasks[$server->machine_name]);
variable_set('search_api_tasks', $tasks);
}
/**
* Implements hook_search_api_index_insert().
*
* Adds the index to its server (if any) and starts tracking indexed items (if
* the index is enabled).
*/
function search_api_search_api_index_insert(SearchApiIndex $index) {
$index->postCreate();
}
/**
* Implements hook_search_api_index_update().
*/
function search_api_search_api_index_update(SearchApiIndex $index) {
// If the server was changed, we have to call the appropriate service class
// hook methods.
if ($index->server != $index->original->server) {
// Server changed - inform old and new ones.
if ($index->original->server) {
$old_server = search_api_server_load($index->original->server);
// The server might have changed because the old one was deleted:
if ($old_server) {
if ($old_server->enabled) {
$old_server->removeIndex($index);
}
else {
$tasks = variable_get('search_api_tasks', array());
// When we add or remove an index, we can ignore all other tasks.
$tasks[$old_server->machine_name][$index->machine_name] = array('remove');
variable_set('search_api_tasks', $tasks);
}
}
}
if ($index->server) {
$new_server = $index->server(TRUE);
// If the server is enabled, we call addIndex(); otherwise, we save the task.
if ($new_server->enabled) {
$new_server->addIndex($index);
}
else {
$tasks = variable_get('search_api_tasks', array());
// When we add or remove an index, we can ignore all other tasks.
$tasks[$new_server->machine_name][$index->machine_name] = array('add');
variable_set('search_api_tasks', $tasks);
unset($new_server);
}
}
// We also have to re-index all content
_search_api_index_reindex($index);
}
// If the fields were changed, call the appropriate service class hook method
// and re-index the content, if necessary.
$old_fields = $index->original->options + array('fields' => array());
$old_fields = $old_fields['fields'];
$new_fields = $index->options + array('fields' => array());
$new_fields = $new_fields['fields'];
if ($old_fields != $new_fields) {
if ($index->server && $index->server()->fieldsUpdated($index)) {
_search_api_index_reindex($index);
}
}
// If the index's enabled or read-only status is being changed, queue or
// dequeue items for indexing.
if (!$index->read_only && $index->enabled != $index->original->enabled) {
if ($index->enabled) {
$index->queueItems();
}
else {
$index->dequeueItems();
}
}
elseif ($index->read_only != $index->original->read_only) {
if ($index->read_only) {
$index->dequeueItems();
}
else {
$index->queueItems();
}
}
// If the cron batch size changed, empty the cron queue for this index.
$old_cron = $index->original->options + array('cron_limit' => NULL);
$old_cron = $old_cron['cron_limit'];
$new_cron = $index->options + array('cron_limit' => NULL);
$new_cron = $new_cron['cron_limit'];
if ($old_cron !== $new_cron) {
_search_api_empty_cron_queue($index, TRUE);
}
}
/**
* Implements hook_search_api_index_delete().
*
* Removes all data for indexes not available any more.
*/
function search_api_search_api_index_delete(SearchApiIndex $index) {
$index->postDelete();
}
/**
* Implements hook_features_export_alter().
*
* Adds dependency information for exported servers.
*/
function search_api_features_export_alter(&$export, $module_name) {
if (isset($export['features']['search_api_server'])) {
// Get a list of the modules that provide storage engines.
$hook = 'search_api_service_info';
$classes = array();
foreach (module_implements('search_api_service_info') as $module) {
$function = $module . '_' . $hook;
$engines = $function();
foreach ($engines as $service => $specs) {
$classes[$service] = $module;
}
}
// Check all of the exported server specifications.
foreach ($export['features']['search_api_server'] as $server_name) {
// Load the server's object.
$server = search_api_server_load($server_name);
$module = $classes[$server->class];
// Ensure that the module responsible for the server object is listed as
// a dependency.
if (!isset($export['dependencies'][$module])) {
$export['dependencies'][$module] = $module;
}
}
// Ensure the dependencies list is still sorted alphabetically.
ksort($export['dependencies']);
}
}
/**
* Implements hook_entity_insert().
*
* Marks the new item as to-index for all indexes on entities of the specified
* type.
*
* @param $entity
* The new entity.
* @param $type
* The entity's type.
*/
function search_api_entity_insert($entity, $type) {
// When inserting a new search index, the new index was already inserted into
// the tracking table. This would lead to a duplicate-key issue, if we would
// continue.
// We also only react on entity operations for types with property
// information, as we don't provide search integration for the others.
if ($type == 'search_api_index' || !entity_get_property_info($type)) {
return;
}
list($id) = entity_extract_ids($type, $entity);
if (isset($id)) {
search_api_track_item_insert($type, array($id));
}
}
/**
* Implements hook_entity_update().
*
* Marks the item as changed for all indexes on entities of the specified type.
*
* @param $entity
* The updated entity.
* @param $type
* The entity's type.
*/
function search_api_entity_update($entity, $type) {
// We only react on entity operations for types with property information, as
// we don't provide search integration for the others.
if (!entity_get_property_info($type)) {
return;
}
list($id) = entity_extract_ids($type, $entity);
if (isset($id)) {
search_api_track_item_change($type, array($id));
}
}
/**
* Implements hook_entity_delete().
*
* Removes the item from the tracking table and deletes it from all indexes.
*
* @param $entity
* The updated entity.
* @param $type
* The entity's type.
*/
function search_api_entity_delete($entity, $type) {
// We only react on entity operations for types with property information, as
// we don't provide search integration for the others.
if (!entity_get_property_info($type)) {
return;
}
list($id) = entity_extract_ids($type, $entity);
if (isset($id)) {
search_api_track_item_delete($type, array($id));
}
}
/**
* Implements hook_search_api_item_type_info().
*
* Adds item types for all entity types with property information.
*/
function search_api_search_api_item_type_info() {
$types = array();
foreach (entity_get_property_info() as $type => $property_info) {
if ($info = entity_get_info($type)) {
$types[$type] = array(
'name' => $info['label'],
'datasource controller' => 'SearchApiEntityDataSourceController',
);
}
}
return $types;
}
/**
* Implements hook_modules_enabled().
*/
function search_api_modules_enabled(array $modules) {
// New modules might offer additional entity types, invalidating the cached
// item type information.
drupal_static_reset('search_api_get_item_type_info');
}
/**
* Implements hook_modules_disabled().
*/
function search_api_modules_disabled(array $modules) {
// The disabled modules might have offered entity types, which are now
// invalid. Therefore, clear the cached item type informaiton.
drupal_static_reset('search_api_get_item_type_info');
}
/**
* Implements hook_search_api_alter_callback_info().
*/
function search_api_search_api_alter_callback_info() {
$callbacks['search_api_alter_bundle_filter'] = array(
'name' => t('Bundle filter'),
'description' => t('Exclude items from indexing based on their bundle (content type, vocabulary, …).'),
'class' => 'SearchApiAlterBundleFilter',
// Filters should be executed first.
'weight' => -10,
);
$callbacks['search_api_alter_add_url'] = array(
'name' => t('URL field'),
'description' => t("Adds the item's URL to the indexed data."),
'class' => 'SearchApiAlterAddUrl',
);
$callbacks['search_api_alter_add_aggregation'] = array(
'name' => t('Aggregated fields'),
'description' => t('Gives you the ability to define additional fields, containing data from one or more other fields.'),
'class' => 'SearchApiAlterAddAggregation',
);
$callbacks['search_api_alter_add_viewed_entity'] = array(
'name' => t('Complete entity view'),
'description' => t('Adds an additional field containing the whole HTML content of the entity when viewed.'),
'class' => 'SearchApiAlterAddViewedEntity',
);
$callbacks['search_api_alter_add_hierarchy'] = array(
'name' => t('Index hierarchy'),
'description' => t('Allows to index hierarchical fields along with all their ancestors.'),
'class' => 'SearchApiAlterAddHierarchy',
);
$callbacks['search_api_alter_language_control'] = array(
'name' => t('Language control'),
'description' => t('Lets you determine the language of items in the index.'),
'class' => 'SearchApiAlterLanguageControl',
);
$callbacks['search_api_alter_node_access'] = array(
'name' => t('Node access'),
'description' => t('Add node access information to the index.'),
'class' => 'SearchApiAlterNodeAccess',
);
$callbacks['search_api_alter_node_status'] = array(
'name' => t('Exclude unpublished nodes'),
'description' => t('Exclude unpublished nodes from the index.'),
'class' => 'SearchApiAlterNodeStatus',
);
return $callbacks;
}
/**
* Implements hook_search_api_processor_info().
*/
function search_api_search_api_processor_info() {
$processors['search_api_case_ignore'] = array(
'name' => t('Ignore case'),
'description' => t('This processor will make searches case-insensitive for all fulltext fields (and, optionally, also for filters on string fields).'),
'class' => 'SearchApiIgnoreCase',
);
$processors['search_api_html_filter'] = array(
'name' => t('HTML filter'),
'description' => t('Strips HTML tags from fulltext fields and decodes HTML entities. ' .
'Use this processor when indexing HTML data, e.g., node bodies for certain text formats.
' .
'The processor also allows to boost (or ignore) the contents of specific elements.'),
'class' => 'SearchApiHtmlFilter',
'weight' => 10,
);
$processors['search_api_tokenizer'] = array(
'name' => t('Tokenizer'),
'description' => t('Tokenizes fulltext data by stripping whitespace. ' .
'This processor allows you to specify which characters make up words and which characters should be ignored, using regular expression syntax. ' .
'Otherwise it is up to the search server implementation to decide how to split indexed fulltext data.'),
'class' => 'SearchApiTokenizer',
'weight' => 20,
);
$processors['search_api_stopwords'] = array(
'name' => t('Stopwords'),
'description' => t('This processor prevents certain words from being indexed and removes them from search terms. ' .
'For best results, it should only be executed after tokenizing.'),
'class' => 'SearchApiStopWords',
'weight' => 30,
);
return $processors;
}
/**
* Inserts new unindexed items for all indexes on the specified type.
*
* @param $type
* The item type of the new items.
* @param array $item_id
* The IDs of the new items.
*/
function search_api_track_item_insert($type, array $item_ids) {
$conditions = array(
'enabled' => 1,
'item_type' => $type,
'read_only' => 0,
);
$indexes = search_api_index_load_multiple(FALSE, $conditions);
if (!$indexes) {
return;
}
search_api_get_datasource_controller($type)->trackItemInsert($item_ids, $indexes);
foreach ($indexes as $index) {
if (!empty($index->options['index_directly'])) {
$indexed = search_api_index_specific_items_delayed($index, $item_ids);
}
}
}
/**
* Mark the items with the specified IDs as "dirty", i.e., as needing to be reindexed.
*
* For indexes for which items should be indexed immediately, the items are
* indexed directly, instead.
*
* @param $type
* The type of items, specific to the data source.
* @param array $item_ids
* The IDs of the items to be marked dirty.
*/
function search_api_track_item_change($type, array $item_ids) {
$conditions = array(
'enabled' => 1,
'item_type' => $type,
'read_only' => 0,
);
$indexes = search_api_index_load_multiple(FALSE, $conditions);
if (!$indexes) {
return;
}
search_api_get_datasource_controller($type)->trackItemChange($item_ids, $indexes);
foreach ($indexes as $index) {
if (!empty($index->options['index_directly'])) {
// For indexes with the index_directly option set, queue the items to be
// indexed at the end of the request.
try {
search_api_index_specific_items_delayed($index, $item_ids);
}
catch (SearchApiException $e) {
watchdog_exception('search_api', $e);
}
}
}
}
/**
* Marks items as queued for indexing for the specified index.
*
* @param SearchApiIndex $index
* The index on which items were queued.
* @param array $item_ids
* The ids of the queued items.
*/
function search_api_track_item_queued(SearchApiIndex $index, array $item_ids) {
$index->datasource()->trackItemQueued($item_ids, $index);
}
/**
* Marks items as successfully indexed for the specified index.
*
* @param SearchApiIndex $index
* The index on which items were indexed.
* @param array $item_ids
* The ids of the indexed items.
*/
function search_api_track_item_indexed(SearchApiIndex $index, array $item_ids) {
$index->datasource()->trackItemIndexed($item_ids, $index);
}
/**
* Removes items from all indexes.
*
* @param $type
* The type of the items.
* @param array $item_ids
* The IDs of the deleted items.
*/
function search_api_track_item_delete($type, array $item_ids) {
// First, delete the item from the tracking table.
$conditions = array(
'enabled' => 1,
'item_type' => $type,
'read_only' => 0,
);
$indexes = search_api_index_load_multiple(FALSE, $conditions);
if ($indexes) {
search_api_get_datasource_controller($type)->trackItemDelete($item_ids, $indexes);
}
// Then, delete it from all servers. Servers of disabled indexes have to be
// considered, too!
unset($conditions['enabled']);
foreach (search_api_index_load_multiple(FALSE, $conditions) as $index) {
if ($index->server) {
$server = $index->server();
if ($server->enabled) {
$server->deleteItems($item_ids, $index);
}
else {
$tasks = variable_get('search_api_tasks', array());
foreach ($item_ids as $id) {
$tasks[$server->machine_name][$index->machine_name][] = 'delete-' . $id;
}
variable_set('search_api_tasks', $tasks);
}
}
}
}
/**
* Indexes items for the specified index. Only items marked as changed are
* indexed, in their order of change (if known).
*
* @param SearchApiIndex $index
* The index on which items should be indexed.
* @param $limit
* The number of items which should be indexed at most. -1 means no limit.
*
* @throws SearchApiException
* If any error occurs during indexing.
*
* @return
* Number of successfully indexed items.
*/
function search_api_index_items(SearchApiIndex $index, $limit = -1) {
// Don't try to index read-only indexes.
if ($index->read_only) {
return 0;
}
$queue = DrupalQueue::get('search_api_indexing_queue');
$queue->createQueue();
$indexed = 0;
$unlimited = $limit < 0;
$release_items = array();
while (($unlimited || $indexed < $limit) && ($item = $queue->claimItem(30))) {
if ($item->data['index'] === $index->machine_name) {
$indexed += _search_api_indexing_queue_process($item->data);
$queue->deleteItem($item);
}
else {
$release_items[] = $item;
}
}
foreach ($release_items as $item) {
$queue->releaseItem($item);
}
if ($unlimited || $indexed < $limit) {
$ids = search_api_get_items_to_index($index, $unlimited ? -1 : $limit - $indexed);
if ($ids) {
$indexed += count(search_api_index_specific_items($index, $ids));
}
}
return $indexed;
}
/**
* Indexes the specified items on the given index.
*
* Items which were successfully indexed are marked as such afterwards.
*
* @param SearchApiIndex $index
* The index on which items should be indexed.
* @param array $ids
* The IDs of the items which should be indexed.
*
* @throws SearchApiException
* If any error occurs during indexing.
*
* @return
* The IDs of all successfully indexed items.
*/
function search_api_index_specific_items(SearchApiIndex $index, array $ids) {
$items = $index->loadItems($ids);
// Clone items because data alterations may alter them.
$cloned_items = array();
foreach ($items as $id => $item) {
$cloned_items[$id] = clone $item;
}
$indexed = $items ? $index->index($cloned_items) : array();
if ($indexed) {
search_api_track_item_indexed($index, $indexed);
// If some items could not be indexed, we don't want to try re-indexing
// them right away, so we mark them as "freshly" changed. Sadly, there is
// no better way than to mark them as indexed first...
if (count($indexed) < count($ids)) {
// Believe it or not but this is actually quite faster than the equivalent
// $diff = array_diff($ids, $indexed);
$diff = array_keys(array_diff_key(array_flip($ids), array_flip($indexed)));
$index->datasource()->trackItemIndexed($diff, $index);
$index->datasource()->trackItemChange($diff, array($index));
}
}
return $indexed;
}
/**
* Queues items for indexing at the end of the page request.
*
* @param SearchApiIndex $index
* The index on which items should be indexed.
* @param array $ids
* The IDs of the items which should be indexed.
*
* @return array
* The current contents of the queue, as a reference.
*
* @see search_api_index_specific_items()
* @see _search_api_index_queued_items()
*/
function &search_api_index_specific_items_delayed(SearchApiIndex $index = NULL, array $ids = array()) {
// We cannot use drupal_static() here because the static cache is reset during
// batch processing, which breaks batch handling.
static $queue = array();
static $registered = FALSE;
// Only register the shutdown function once.
if (empty($registered)) {
drupal_register_shutdown_function('_search_api_index_queued_items');
$registered = TRUE;
}
// Allow for empty call to just retrieve the queue.
if ($index && $ids) {
$index_id = $index->machine_name;
$queue += array($index_id => array());
$queue[$index_id] += drupal_map_assoc($ids);
}
return $queue;
}
/**
* Returns a list of items that need to be indexed for the specified index.
*
* @param SearchApiIndex $index
* The index for which items should be retrieved.
* @param $limit
* The maximum number of items to retrieve. -1 means no limit.
*
* @return array
* An array of IDs of items that need to be indexed.
*/
function search_api_get_items_to_index(SearchApiIndex $index, $limit = -1) {
if ($limit == 0) {
return array();
}
return $index->datasource()->getChangedItems($index, $limit);
}
/**
* Creates a search query on a specified search index.
*
* @param $id
* The ID or machine name of the index to execute the search on.
* @param $options
* An associative array of options. The following are recognized:
* - filters: Either a SearchApiQueryFilterInterface object or an array of
* filters used to filter the search.
* - sort: An array of sort directives of the form $field => $order, where
* $order is either 'ASC' or 'DESC'.
* - offset: The position of the first returned search results relative to the
* whole result in the index.
* - limit: The maximum number of search results to return. -1 means no limit.
* - 'query class': The query class to use. Must be a subtype of
* SearchApiQueryInterface.
* - conjunction: The type of conjunction to use for this query - either
* 'AND' or 'OR'. 'AND' by default.
* - 'parse mode': The mode with which to parse the $keys variable, if it
* is set and not already an array. See SearchApiQuery::parseModes() for
* parse modes recognized by the SearchApiQuery class.
* Subclasses might define additional modes.
*
* @return SearchApiQueryInterface
* An object for searching on the specified index.
*/
function search_api_query($id, array $options = array()) {
$index = search_api_index_load($id);
if (!$index) {
throw new SearchApiException(t('Unknown index with ID @id.', array('@id' => $id)));
}
return $index->query($options);
}
/**
* Static store for the searches executed on the current page. Can either be
* used to store an executed search, or to retrieve a previously stored
* search.
*
* @param $search_id
* For pages displaying multiple searches, an optional ID identifying the
* search in questions. When storing a search, this is filled automatically,
* unless it is manually set.
* @param SearchApiQuery $query
* When storing an executed search, the query that was executed. NULL
* otherwise.
* @param array $results
* When storing an executed search, the returned results as specified by
* SearchApiQueryInterface::execute(). An empty array, otherwise.
*
* @return array
* If a search with the specified ID was executed, an array containing
* ($query, $results) as used in this function's parameters. If $search_id is
* NULL, an array of all executed searches will be returned, keyed by ID.
*/
function search_api_current_search($search_id = NULL, SearchApiQuery $query = NULL, array $results = array()) {
$searches = &drupal_static(__FUNCTION__, array());
if (isset($query)) {
if (!isset($search_id)) {
$search_id = $query->getOption('search id');
}
$base = $search_id;
$i = 0;
while (isset($searches[$search_id])) {
$search_id = $base . '-' . ++$i;
}
$searches[$search_id] = array($query, $results);
}
if (isset($search_id)) {
return isset($searches[$search_id]) ? $searches[$search_id] : NULL;
}
return $searches;
}
/**
* Returns all field types recognized by the Search API framework.
*
* @return array
* An associative array with all recognized types as keys, mapped to their
* translated display names.
*
* @see search_api_default_field_types()
* @see search_api_get_data_type_info()
*/
function search_api_field_types() {
$types = search_api_default_field_types();
foreach (search_api_get_data_type_info() as $id => $type) {
$types[$id] = $type['name'];
}
return $types;
}
/**
* Returns the default field types recognized by the Search API framework.
*
* @return array
* An associative array with the default types as keys, mapped to their
* translated display names.
*/
function search_api_default_field_types() {
return array(
'text' => t('Fulltext'),
'string' => t('String'),
'integer' => t('Integer'),
'decimal' => t('Decimal'),
'date' => t('Date'),
'duration' => t('Duration'),
'boolean' => t('Boolean'),
'uri' => t('URI'),
);
}
/**
* Returns either all custom field type definitions, or a specific one.
*
* @param $type
* If specified, the type whose definition should be returned.
*
* @return array
* If $type was not given, an array containing all custom data types, in the
* format specified by hook_search_api_data_type_info().
* Otherwise, the definition for the given type, or NULL if it is unknown.
*
* @see hook_search_api_data_type_info()
*/
function search_api_get_data_type_info($type = NULL) {
$types = &drupal_static(__FUNCTION__);
if (!isset($types)) {
$default_types = search_api_default_field_types();
$types = module_invoke_all('search_api_data_type_info');
$types = $types ? $types : array();
foreach ($types as &$type_info) {
if (!isset($type_info['fallback']) || !isset($default_types[$type_info['fallback']])) {
$type_info['fallback'] = 'string';
}
}
drupal_alter('search_api_data_type_info', $types);
}
if (isset($type)) {
return isset($types[$type]) ? $types[$type] : NULL;
}
return $types;
}
/**
* Returns either a list of all available service infos, or a specific one.
*
* @see hook_search_api_service_info()
*
* @param $id
* The ID of the service info to retrieve.
*
* @return array
* If $id was not specified, an array of all available service classes.
* Otherwise, either the service info with the specified id (if it exists),
* or NULL.
*/
function search_api_get_service_info($id = NULL) {
$services = &drupal_static(__FUNCTION__);
if (!isset($services)) {
$services = module_invoke_all('search_api_service_info');
// Allow other modules to alter definitions
drupal_alter('search_api_service_info', $services);
}
if (isset($id)) {
return isset($services[$id]) ? $services[$id] : NULL;
}
return $services;
}
/**
* Returns information for either all item types, or a specific one.
*
* @param $type
* If set, the item type whose information should be returned.
*
* @return
* If $type is given, either an array containing the information of that item
* type, or NULL if it is unknown. Otherwise, an array keyed by type IDs
* containing the information for all item types. Item type information is
* formatted as specified by hook_search_api_item_type_info(), and has all
* optional fields filled with the defaults.
*
* @see hook_search_api_item_type_info()
*/
function search_api_get_item_type_info($type = NULL) {
$types = &drupal_static(__FUNCTION__);
if (!isset($types)) {
$types = module_invoke_all('search_api_item_type_info');
drupal_alter('search_api_item_type_info', $types);
}
if (isset($type)) {
return isset($types[$type]) ? $types[$type] : NULL;
}
return $types;
}
/**
* Get a data source controller object for the specified type.
*
* @param $type
* The type whose data source controller should be returned.
*
* @return SearchApiDataSourceControllerInterface
* The type's data source controller.
*
* @throws SearchApiException
* If the type is unknown or specifies an invalid data source controller.
*/
function search_api_get_datasource_controller($type) {
$datasources = &drupal_static(__FUNCTION__, array());
if (empty($datasources[$type])) {
$info = search_api_get_item_type_info($type);
if (isset($info['datasource controller']) && class_exists($info['datasource controller'])) {
$datasources[$type] = new $info['datasource controller']($type);
}
if (empty($datasources[$type]) || !($datasources[$type] instanceof SearchApiDataSourceControllerInterface)) {
unset($datasources[$type]);
throw new SearchApiException(t('Unknown or invalid item type @type.', array('@type' => $type)));
}
}
return $datasources[$type];
}
/**
* Returns a list of all available data alter callbacks.
*
* @see hook_search_api_alter_callback_info()
*
* @return array
* An array of all available data alter callbacks, keyed by function name.
*/
function search_api_get_alter_callbacks() {
$callbacks = &drupal_static(__FUNCTION__);
if (!isset($callbacks)) {
$callbacks = module_invoke_all('search_api_alter_callback_info');
// Initialization of optional entries with default values
foreach ($callbacks as $id => $callback) {
$callbacks[$id] += array('enabled' => TRUE, 'weight' => 0);
}
}
return $callbacks;
}
/**
* Returns a list of all available pre- and post-processors.
*
* @see hook_search_api_processor_info()
*
* @return array
* An array of all available processors, keyed by id.
*/
function search_api_get_processors() {
$processors = &drupal_static(__FUNCTION__);
if (!isset($processors)) {
$processors = module_invoke_all('search_api_processor_info');
// Initialization of optional entries with default values
foreach ($processors as $id => $processor) {
$processors[$id] += array('enabled pre' => TRUE, 'enabled post' => TRUE, 'weight' => 0);
}
}
return $processors;
}
/**
* Implements hook_search_api_query_alter().
*
* Adds node access to the query, if enabled.
*
* @param SearchApiQueryInterface $query
* The SearchApiQueryInterface object representing the search query.
*/
function search_api_search_api_query_alter(SearchApiQueryInterface $query) {
$index = $query->getIndex();
// Only add node access if the necessary fields are indexed in the index, and
// unless disabled explicitly by the query.
$fields = $index->options['fields'];
if (!empty($fields['search_api_access_node']) && !empty($fields['status']) && !empty($fields['author']) && !$query->getOption('search_api_bypass_access')) {
$account = $query->getOption('search_api_access_account', $GLOBALS['user']);
if (is_numeric($account)) {
$account = user_load($account);
}
if (is_object($account)) {
try {
_search_api_query_add_node_access($account, $query);
}
catch (SearchApiException $e) {
watchdog_exception('search_api', $e);
}
}
else {
watchdog('search_api', 'An illegal user UID was given for node access: @uid.', array('@uid' => $query->getOption('search_api_access_account', $GLOBALS['user'])), WATCHDOG_WARNING);
}
}
}
/**
* Build a node access subquery.
*
* @param $account
* The user object, who searches.
*
* @return SearchApiQueryFilter
*/
function _search_api_query_add_node_access($account, SearchApiQueryInterface $query) {
if (!user_access('access content', $account)) {
// Simple hack for returning no results.
$query->condition('status', 0);
$query->condition('status', 1);
watchdog('search_api', 'User @name tried to execute a search, but cannot access content.', array('@name' => theme('username', array('account' => $account))), WATCHDOG_NOTICE);
return;
}
// Only filter for user which don't have full node access.
if (!user_access('bypass node access', $account)) {
// Filter by node "published" status.
if (user_access('view own unpublished content')) {
$filter = $query->createFilter('OR');
$filter->condition('status', NODE_PUBLISHED);
$filter->condition('author', $account->uid);
$query->filter($filter);
}
else {
$query->condition('status', NODE_PUBLISHED);
}
// Filter by node access grants.
$filter = $query->createFilter('OR');
$grants = node_access_grants('view', $account);
foreach ($grants as $realm => $gids) {
foreach ($gids as $gid) {
$filter->condition('search_api_access_node', "node_access_$realm:$gid");
}
}
$filter->condition('search_api_access_node', 'node_access__all');
$query->filter($filter);
}
}
/**
* Utility function for determining whether a field of the given type contains
* text data.
*
* @param $type
* A string containing the type to check.
* @param array $allowed
* Optionally, an array of allowed types.
*
* @return
* TRUE if $type is either one of the specified types, or a list of such
* values. FALSE otherwise.
*/
function search_api_is_text_type($type, array $allowed = array('text')) {
return array_search(search_api_extract_inner_type($type), $allowed) !== FALSE;
}
/**
* Utility function for determining whether a field of the given type contains
* a list of any kind.
*
* @param $type
* A string containing the type to check.
*
* @return
* TRUE iff $type is a list type ("list<*>").
*/
function search_api_is_list_type($type) {
return substr($type, 0, 5) == 'list<';
}
/**
* Utility function for determining the nesting level of a list type.
*
* @param $type
* A string containing the type to check.
*
* @return
* The nesting level of the type. 0 for singular types, 1 for lists of
* singular types, etc.
*/
function search_api_list_nesting_level($type) {
$level = 0;
while (search_api_is_list_type($type)) {
$type = substr($type, 5, -1);
++$level;
}
return $level;
}
/**
* Utility function for nesting a type to the same level as another type.
* I.e., after $t = search_api_nest_type($type, $nested_type);
is
* executed, the following statements will always be true:
* @code
* search_api_list_nesting_level($t) == search_api_list_nesting_level($nested_type);
* search_api_extract_inner_type($t) == search_api_extract_inner_type($type);
* @endcode
*
* @param $type
* The type to wrap.
* @param $nested_type
* Another type, determining the nesting level.
*
* @return
* A list version of $type, as specified above.
*/
function search_api_nest_type($type, $nested_type) {
while (search_api_is_list_type($nested_type)) {
$nested_type = substr($nested_type, 5, -1);
$type = "list<$type>";
}
return $type;
}
/**
* Utility function for extracting the contained primitive type of a list type.
*
* @param $type
* A string containing the list type to process.
*
* @return
* A string containing the primitive type contained within the list, e.g.
* "text" for "list" (or for "list>"). If $type is no list
* type, it is returned unchanged.
*/
function search_api_extract_inner_type($type) {
while (search_api_is_list_type($type)) {
$type = substr($type, 5, -1);
}
return $type;
}
/**
* Utility function for extracting specific fields from an EntityMetadataWrapper
* object.
*
* @param EntityMetadataWrapper $wrapper
* The wrapper from which to extract fields.
* @param array $fields
* The fields to extract, as stored in an index. I.e., the array keys are
* field names, the values are arrays with the keys "name", "type", "boost"
* and "indexed" (although only "type" is used by this function).
* @param array $value_options
* An array of options that should be passed to the
* EntityMetadataWrapper::value() method (see there).
*
* @return
* The $fields array with additional "value" and "original_type" keys set.
*/
function search_api_extract_fields(EntityMetadataWrapper $wrapper, array $fields, array $value_options = array()) {
// If $wrapper is a list of entities, we have to aggregate their field values.
$wrapper_info = $wrapper->info();
if (search_api_is_list_type($wrapper_info['type'])) {
foreach ($fields as $field => &$info) {
$info['value'] = array();
$info['original_type'] = $info['type'];
}
unset($info);
try {
foreach ($wrapper as $i => $w) {
$nested_fields = search_api_extract_fields($w, $fields, $value_options);
foreach ($nested_fields as $field => $info) {
if (isset($info['value'])) {
$fields[$field]['value'][] = $info['value'];
}
if (isset($info['original_type'])) {
$fields[$field]['original_type'] = $info['original_type'];
}
}
}
}
catch (EntityMetadataWrapperException $e) {
// Catch exceptions caused by not set list values.
}
return $fields;
}
$nested = array();
$entity_infos = entity_get_info();
foreach ($fields as $field => &$info) {
$pos = strpos($field, ':');
if ($pos === FALSE) {
// Set "defaults" in case an error occurs later.
$info['value'] = NULL;
$info['original_type'] = $info['type'];
try {
$info['value'] = $wrapper->$field->value($value_options);
// For fulltext fields with options, also include the option labels.
if (search_api_is_text_type($info['type']) && $wrapper->$field->optionsList('view')) {
_search_api_add_option_values($info['value'], $wrapper->$field->optionsList('view'));
}
$property_info = $wrapper->$field->info();
$info['original_type'] = $property_info['type'];
// For entities, we extract the entity ID instead of the whole object.
// @todo Use 'identifier' => TRUE instead of always loading the object.
$t = search_api_extract_inner_type($property_info['type']);
if (isset($entity_infos[$t])) {
// If no object is set, set this field to NULL.
$info['value'] = $info['value'] ? _search_api_extract_entity_value($wrapper->$field, search_api_is_text_type($info['type'])) : NULL;
}
}
catch (EntityMetadataWrapperException $e) {
// This might happen for entity-typed properties that are NULL, e.g.,
// for comments without parent.
}
}
else {
list($prefix, $key) = explode(':', $field, 2);
$nested[$prefix][$key] = $info;
}
}
unset($info);
foreach ($nested as $prefix => $nested_fields) {
if (isset($wrapper->$prefix)) {
$nested_fields = search_api_extract_fields($wrapper->$prefix, $nested_fields, $value_options);
foreach ($nested_fields as $field => $info) {
$fields["$prefix:$field"] = $info;
}
}
else {
foreach ($nested_fields as $field => &$info) {
$info['value'] = NULL;
$info['original_type'] = $info['type'];
}
}
}
return $fields;
}
/**
* Helper method for adding additional text data to fields with an option list.
*/
function _search_api_add_option_values(&$value, array $options) {
if (is_array($value)) {
foreach ($value as &$v) {
_search_api_add_option_values($v, $options);
}
return;
}
if (is_scalar($value) && isset($options[$value])) {
$value .= ' ' . $options[$value];
}
}
/**
* Helper method for extracting the ID (and possibly label) of an entity-valued field.
*/
function _search_api_extract_entity_value(EntityMetadataWrapper $wrapper, $fulltext = FALSE) {
$v = $wrapper->value();
if (is_array($v)) {
$ret = array();
foreach ($wrapper as $item) {
$values = _search_api_extract_entity_value($item, $fulltext);
if ($values) {
$ret[] = $values;
}
}
return $ret;
}
if ($v) {
$ret = $wrapper->getIdentifier();
if ($fulltext && ($label = $wrapper->label())) {
$ret .= ' ' . $label;
}
return $ret;
}
return NULL;
}
/**
* Load the search server with the specified id.
*
* @param $id
* The search server's id.
* @param $reset
* Whether to reset the internal cache.
*
* @return SearchApiServer
* An object representing the server with the specified id.
*/
function search_api_server_load($id, $reset = FALSE) {
$ret = search_api_server_load_multiple(array($id), array(), $reset);
return $ret ? reset($ret) : FALSE;
}
/**
* Load multiple servers at once, determined by IDs or machine names, or by
* other conditions.
*
* @see entity_load()
*
* @param $ids
* An array of server IDs or machine names, or FALSE to load all servers.
* @param $conditions
* An array of conditions on the {search_api_server} table in the form
* 'field' => $value.
* @param $reset
* Whether to reset the internal entity_load cache.
*
* @return array
* An array of server objects keyed by machine name.
*/
function search_api_server_load_multiple($ids = array(), $conditions = array(), $reset = FALSE) {
return entity_load_multiple_by_name('search_api_server', $ids, $conditions, $reset);
}
/**
* Entity uri callback.
*/
function search_api_server_url(SearchApiServer $server) {
return array(
'path' => 'admin/config/search/search_api/server/' . $server->machine_name,
'options' => array(),
);
}
/**
* Title callback for determining which title should be displayed for the
* "delete" local task.
*
* @param Entity $entity
* The server or index for which the menu link is displayed.
*
* @return string
* A translated version of either "Delete" or "Revert".
*/
function search_api_title_delete_page(Entity $entity) {
return $entity->hasStatus(ENTITY_OVERRIDDEN) ? t('Revert') : t('Delete');
}
/**
* Access callback for determining if a server's or index' "delete" page should
* be accessible.
*
* @param Entity $entity
* The server or index for which the access to the delete page is checked.
*
* @return
* TRUE if the delete page can be accessed by the user, FALSE otherwise.
*/
function search_api_access_delete_page(Entity $entity) {
return user_access('administer search_api') && $entity->hasStatus(ENTITY_CUSTOM);
}
/**
* Inserts a new search server into the database.
*
* @param array $values
* An array containing the values to be inserted.
*
* @return
* The newly inserted server's id, or FALSE on error.
*/
function search_api_server_insert(array $values) {
$server = entity_create('search_api_server', $values);
$server->is_new = TRUE;
$server->save();
return $server->id;
}
/**
* Changes a server's settings.
*
* @param $id
* The ID or machine name of the server whose values should be changed.
* @param array $fields
* The new field values to set. The enabled field can't be set this way, use
* search_api_server_enable() and search_api_server_disable() instead.
*
* @return
* 1 if fields were changed, 0 if the fields already had the desired values.
* FALSE on failure.
*/
function search_api_server_edit($id, array $fields) {
$server = search_api_server_load($id, TRUE);
$ret = $server->update($fields);
return $ret ? 1 : $ret;
}
/**
* Enables a search server. Will also check for remembered tasks for this server
* and execute them.
*
* @param $id
* The ID or machine name of the server to enable.
*
* @return
* 1 on success, 0 or FALSE on failure.
*/
function search_api_server_enable($id) {
$server = search_api_server_load($id, TRUE);
$ret = $server->update(array('enabled' => 1));
return $ret ? 1 : $ret;
}
/**
* Disables a search server, along with all associated indexes.
*
* @param $id
* The ID or machine name of the server to disable.
*
* @return
* 1 on success, 0 or FALSE on failure.
*/
function search_api_server_disable($id) {
$server = search_api_server_load($id, TRUE);
$ret = $server->update(array('enabled' => 0));
return $ret ? 1 : $ret;
}
/**
* Deletes a search server and disables all associated indexes.
*
* @param $id
* The ID or machine name of the server to delete.
*
* @return
* 1 on success, 0 or FALSE on failure.
*/
function search_api_server_delete($id) {
$server = search_api_server_load($id, TRUE);
$server->delete();
return 1;
}
/**
* Loads the Search API index with the specified id.
*
* @param $id
* The index' id.
* @param $reset
* Whether to reset the internal cache.
*
* @return SearchApiIndex
* A completely loaded index object, or NULL if no such index exists.
*/
function search_api_index_load($id, $reset = FALSE) {
$ret = search_api_index_load_multiple(array($id), array(), $reset);
return $ret ? reset($ret) : FALSE;
}
/**
* Load multiple indexes at once, determined by IDs or machine names, or by
* other conditions.
*
* @see entity_load()
*
* @param $ids
* An array of index IDs or machine names, or FALSE to load all indexes.
* @param $conditions
* An array of conditions on the {search_api_index} table in the form
* 'field' => $value.
* @param $reset
* Whether to reset the internal entity_load cache.
*
* @return array
* An array of index objects keyed by machine name.
*/
function search_api_index_load_multiple($ids = array(), $conditions = array(), $reset = FALSE) {
// This line is a workaround for a weird PDO bug in PHP 5.2.
// See http://drupal.org/node/889286.
new SearchApiIndex();
return entity_load_multiple_by_name('search_api_index', $ids, $conditions, $reset);
}
/**
* Determines a search index' indexing status.
*
* @param SearchApiIndex $index
* The index whose indexing status should be determined.
*
* @return array
* An associative array containing two keys (in this order):
* - indexed: The number of items already indexed in their latest version.
* - total: The total number of items that have to be indexed for this index.
*/
function search_api_index_status(SearchApiIndex $index) {
return $index->datasource()->getIndexStatus($index);
}
/**
* Entity uri callback.
*/
function search_api_index_url(SearchApiIndex $index) {
return array(
'path' => 'admin/config/search/search_api/index/' . $index->machine_name,
'options' => array(),
);
}
/**
* Property callback.
*
* @return SearchApiServer
* The server this index currently resides on, or NULL if the index
* is currently unassigned.
*/
function search_api_index_get_server(SearchApiIndex $index) {
return $index->server();
}
/**
* Inserts a new search index into the database.
*
* @param array $values
* An array containing the values to be inserted.
*
* @return
* The newly inserted index' id, or FALSE on error.
*/
function search_api_index_insert(array $values) {
$index = entity_create('search_api_index', $values);
$index->is_new = TRUE;
$index->save();
return $index->id;
}
/**
* Changes an index' settings.
*
* @param $id
* The edited index' id.
* @param array $fields
* The new field values to set.
*
* @return
* 1 if fields were changed, 0 if the fields already had the desired values.
* FALSE on failure.
*/
function search_api_index_edit($id, array $fields) {
$index = search_api_index_load($id, TRUE);
$ret = $index->update($fields);
return $ret ? 1 : $ret;
}
/**
* Changes an index' indexed field settings.
*
* @param $id
* The ID or machine name of the index whose fields should be changed.
* @param array $fields
* The new indexed field settings.
*
* @return
* 1 if the field settings were changed, 0 if they already had the desired
* values. FALSE on failure.
*/
function search_api_index_edit_fields($id, array $fields) {
$index = search_api_index_load($id, TRUE);
$options = $index->options;
$options['fields'] = $fields;
$ret = $index->update(array('options' => $options));
return $ret ? 1 : $ret;
}
/**
* Enables a search index.
*
* @param $id
* The ID or machine name of the index to enable.
*
* @throws SearchApiException
* If the index' server isn't enabled.
*
* @return
* 1 on success, 0 or FALSE on failure.
*/
function search_api_index_enable($id) {
$index = search_api_index_load($id, TRUE);
$ret = $index->update(array('enabled' => 1));
return $ret ? 1 : $ret;
}
/**
* Disables a search index.
*
* @param $id
* The ID or machine name of the index to disable.
*
* @return
* 1 on success, 0 or FALSE on failure.
*/
function search_api_index_disable($id) {
$index = search_api_index_load($id, TRUE);
$ret = $index->update(array('enabled' => 0));
return $ret ? 1 : $ret;
}
/**
* Schedules a search index for re-indexing.
*
* @param $id
* The ID or machine name of the index to re-index.
*
* @return
* TRUE on success, FALSE on failure.
*/
function search_api_index_reindex($id) {
$index = search_api_index_load($id);
return $index->reindex();
}
/**
* Helper method for marking all items on an index as needing re-indexing.
*
* @param SearchApiIndex $index
* The index whose items should be re-indexed.
*/
function _search_api_index_reindex(SearchApiIndex $index) {
$index->datasource()->trackItemChange(FALSE, array($index), TRUE);
_search_api_empty_cron_queue($index);
}
/**
* Helper method for removing all of an index's jobs from the cron queue.
*
* @param SearchApiIndex $index
* The index whose jobs should be removed.
* @param $mark_changed
* If TRUE, mark all items in the queue as "changed" again. Defaults to FALSE.
*/
function _search_api_empty_cron_queue(SearchApiIndex $index, $mark_changed = FALSE) {
$index_id = $index->machine_name;
$queue = DrupalQueue::get('search_api_indexing_queue');
$queue->createQueue();
$ids = array();
$release_items = array();
while ($item = $queue->claimItem()) {
if ($item->data['index'] === $index_id) {
$queue->deleteItem($item);
if ($mark_changed) {
$ids = array_merge($ids, $item->data['items']);
}
}
else {
$release_items[] = $item;
}
}
foreach ($release_items as $item) {
$queue->releaseItem($item);
}
if ($ids) {
$index->datasource()->trackItemChange($ids, array($index), TRUE);
}
}
/**
* Clears a search index and schedules all of its items for re-indexing.
*
* @param $id
* The ID or machine name of the index to clear.
*
* @return
* TRUE on success, FALSE on failure.
*/
function search_api_index_clear($id) {
$index = search_api_index_load($id);
return $index->clear();
}
/**
* Deletes a search index.
*
* @param $id
* The ID or machine name of the index to delete.
*
* @return
* TRUE on success, FALSE on failure.
*/
function search_api_index_delete($id) {
$index = search_api_index_load($id);
if (!$index) {
return FALSE;
}
$index->delete();
return TRUE;
}
/**
* Options list callback for search indexes.
*
* @return array
* An array of search index machine names mapped to their human-readable
* names.
*/
function search_api_index_options_list() {
$ret = array(
NULL => '- ' . t('All') . ' -',
);
foreach (search_api_index_load_multiple(FALSE) as $id => $index) {
$ret[$id] = $index->name;
}
return $ret;
}
/**
* Cron queue worker callback for indexing some items.
*
* @param array $task
* An associative array containing:
* - index: The ID of the index on which items should be indexed.
* - items: The items that should be indexed.
*
* @return
* The number of successfully indexed items.
*/
function _search_api_indexing_queue_process(array $task) {
$index = search_api_index_load($task['index']);
try {
if ($index && $index->enabled && !$index->read_only && $task['items']) {
$indexed = search_api_index_specific_items($index, $task['items']);
$num = count($indexed);
// If some items couldn't be indexed, mark them as dirty again.
if ($num < count($task['items'])) {
// Believe it or not but this is actually quite faster than the equivalent
// $diff = array_diff($task['items'], $indexed);
$diff = array_keys(array_diff_key(array_flip($task['items']), array_flip($indexed)));
// Mark the items as dirty again.
$index->datasource()->trackItemChange($diff, array($index), TRUE);
}
if ($num) {
watchdog('search_api', t('Indexed @num items for index @name', array('@num' => $num, '@name' => $index->name)), NULL, WATCHDOG_INFO);
}
return $num;
}
}
catch (SearchApiException $e) {
watchdog_exception('search_api', $e);
}
}
/**
* Shutdown function which indexes all queued items, if any.
*/
function _search_api_index_queued_items() {
$queue = &search_api_index_specific_items_delayed();
try {
if ($queue) {
$indexes = search_api_index_load_multiple(array_keys($queue));
foreach ($indexes as $index_id => $index) {
search_api_index_specific_items($index, $queue[$index_id]);
}
}
// Reset the queue so we don't index the items twice by accident.
$queue = array();
}
catch (SearchApiException $e) {
watchdog_exception('search_api', $e);
}
}
/**
* Helper function to be used as a "property info alter" callback.
*
* If a wrapped entity is passed to this function, all its available properties
* and fields, regardless of bundle, are added to the wrapper.
*/
function _search_api_wrapper_add_all_properties(EntityMetadataWrapper $wrapper, array $property_info) {
if ($properties = entity_get_all_property_info($wrapper->type())) {
$property_info['properties'] = $properties;
}
return $property_info;
}
/**
* Helper function for converting data to a custom type.
*/
function _search_api_convert_custom_type($callback, $value, $original_type, $type, $nesting_level) {
if ($nesting_level == 0) {
return call_user_func($callback, $value, $original_type, $type);
}
if (!is_array($value)) {
return NULL;
}
--$nesting_level;
$values = array();
foreach ($value as $v) {
$v = _search_api_convert_custom_type($callback, $v, $original_type, $type, $nesting_level);
if (isset($v) && !(is_array($v) && !$v)) {
$values[] = $v;
}
}
return $values;
}
/**
* Create and set a batch for indexing items.
*
* @param SearchApiIndex $index
* The index for which items should be indexed.
* @param $batch_size
* Number of items to index per batch.
* @param $limit
* Maximum number of items to index.
* @param $remaining
* Remaining items to index.
* @param $drush
* Boolean specifying whether this was called from drush or not.
*/
function _search_api_batch_indexing_create(SearchApiIndex $index, $batch_size, $limit, $remaining, $drush = FALSE) {
if ($limit !== 0 && $batch_size !== 0) {
$t = !empty($drush) ? 'dt' : 't';
if ($limit < 0 || $limit > $remaining) {
$limit = $remaining;
}
if ($batch_size < 0) {
$batch_size = $remaining;
}
$batch = array(
'title' => $t('Indexing items'),
'operations' => array(
array('_search_api_batch_indexing_callback', array($index, $batch_size, $limit, $drush)),
),
'progress_message' => $t('Completed about @percentage% of the indexing operation.'),
'finished' => '_search_api_batch_indexing_finished',
'file' => drupal_get_path('module', 'search_api') . '/search_api.module',
);
batch_set($batch);
return TRUE;
}
return FALSE;
}
/**
* Batch API callback for the indexing functionality.
*
* @param SearchApiIndex $index
* The index for which items should be indexed.
* @param integer $batch_size
* Number of items to index per batch.
* @param integer $limit
* Maximum number of items to index.
* @param boolean $drush
* Boolean specifying whether this was called from drush or not.
* @param array $context
* The batch context.
*/
function _search_api_batch_indexing_callback(SearchApiIndex $index, $batch_size, $limit, $drush = FALSE, array &$context) {
// Persistent data among batch runs.
if (!isset($context['sandbox']['limit'])) {
$context['sandbox']['limit'] = $limit;
$context['sandbox']['batch_size'] = $batch_size;
$context['sandbox']['progress'] = 0;
}
// Persistent data for results.
if (!isset($context['results']['indexed'])) {
$context['results']['indexed'] = 0;
$context['results']['not indexed'] = 0;
$context['results']['drush'] = $drush;
}
// Number of items to index for this run.
$to_index = min($context['sandbox']['limit'] - $context['sandbox']['progress'], $context['sandbox']['batch_size']);
// Index the items.
$indexed = search_api_index_items($index, $to_index);
$context['results']['indexed'] += $indexed;
// Display progress message.
if ($indexed > 0) {
$format_plural = $context['results']['drush'] === TRUE ? '_search_api_drush_format_plural' : 'format_plural';
$context['message'] = $format_plural($context['results']['indexed'], 'Successfully indexed 1 item.', 'Successfully indexed @count items.');
}
// Some items couldn't be indexed.
if ($indexed !== $to_index) {
$context['results']['not indexed'] += $to_index - $indexed;
}
$context['sandbox']['progress'] += $to_index;
// Everything has been indexed.
if ($indexed === 0 || $context['sandbox']['progress'] >= $context['sandbox']['limit']) {
$context['finished'] = 1;
}
else {
$context['finished'] = $context['sandbox']['progress'] / $context['sandbox']['limit'];
}
}
/**
* Batch API finishing callback for the indexing functionality.
*
* @param boolean $success
* Result of the batch operation.
* @param array $results
* Results.
* @param array $operations
* Remaining batch operation to process.
*/
function _search_api_batch_indexing_finished($success, $results, $operations) {
// Check if called from drush.
if (!empty($results['drush'])) {
$drupal_set_message = 'drush_log';
$format_plural = '_search_api_drush_format_plural';
$t = 'dt';
$success_message = 'success';
}
else {
$drupal_set_message = 'drupal_set_message';
$format_plural = 'format_plural';
$t = 't';
$success_message = 'status';
}
// Display result messages.
if ($success) {
if (!empty($results['indexed'])) {
$drupal_set_message($format_plural($results['indexed'], 'Successfully indexed 1 item.', 'Successfully indexed @count items.'), $success_message);
if (!empty($results['not indexed'])) {
$drupal_set_message($format_plural($results['not indexed'], '1 item could not be indexed. Check the logs for details.', '@count items could not be indexed. Check the logs for details.'), 'warning');
}
}
else {
$drupal_set_message($t("Couldn't index items. Check the logs for details."), 'error');
}
}
else {
$drupal_set_message($t("An error occurred while trying to index items. Check the logs for details."), 'error');
}
}