materio-base-legacy/entity_translation_search_api.module
2012-10-30 10:39:29 +01:00

62 lines
2.2 KiB
Plaintext

<?php
define('ENTITY_TRANSLATION_SEARCH_API_NO_TRANSLATION', 'none');
/**
* Implements hook_search_api_search_api_item_type_info_alter().
*/
function entity_translation_search_api_search_api_item_type_info_alter(array &$infos) {
foreach ($infos as $type => $info) {
if (entity_get_info($type)) {
$infos[$type]['datasource controller'] = 'EntityTranslationDataSourceController';
}
}
}
/**
* Implements hook_form_alter().
*/
function entity_translation_search_api_form_alter(&$form, &$form_state, $form_id) {
if (in_array($form_id, array(
'search_api_admin_add_index',
'search_api_admin_index_edit',
))) {
if ($form_id == 'search_api_admin_index_edit') {
$index = $form_state['index'];
$default = isset($index->options['entity_translation_language']) ? $index->options['entity_translation_language'] : ENTITY_TRANSLATION_SEARCH_API_NO_TRANSLATION;
}
else {
$default = ENTITY_TRANSLATION_SEARCH_API_NO_TRANSLATION;
}
$languages = language_list('enabled');
$languages = $languages[1];
$names = array(ENTITY_TRANSLATION_SEARCH_API_NO_TRANSLATION => t('No translation'));
foreach ($languages as $langcode => $item) {
$name = t($item->name);
$names[$langcode] = $name . ($item->native != $name ? ' (' . $item->native . ')' : '');
}
$form['options']['entity_translation_language'] = array(
'#type' => (count($names) <= 5 ? 'radios' : 'select'),
'#title' => t('Entity translation language'),
'#description' => t('To which language the entities should be translated before indexing.'),
'#default_value' => $default,
'#options' => $names,
);
}
}
/**
* Implements hook_search_api_index_items_alter().
* For compatibility with language control data alteration.
*/
function entity_translation_search_api_search_api_index_items_alter(array &$items, SearchApiIndex $index) {
$target_language = $index->options['entity_translation_language'];
if ($target_language != ENTITY_TRANSLATION_SEARCH_API_NO_TRANSLATION) {
foreach ($items as $id => $item) {
if (isset($item->translations->data[$target_language])) {
$item->language = $target_language;
}
}
}
}