materio_search_api.pages.inc 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402
  1. <?php
  2. /**
  3. * materiobase_search_autocomplete_dbselect()
  4. *
  5. * inspired by taxonomy_autocomplete()
  6. *
  7. * OBSOLETE : this fonction use a direct dbselect request to provide results forautocomplete
  8. *
  9. */
  10. function materio_search_api_autocomplete_dbselect($typed = ''){
  11. // If the request has a '/' in the search text, then the menu system will have
  12. // split it into multiple arguments, recover the intended $tags_typed.
  13. $args = func_get_args();
  14. $typed = implode('/', $args);
  15. /*
  16. TODO riche serach engine + \\ etc gmail like
  17. */
  18. if ($typed != '') {
  19. // Part of the criteria for the query come from the field's own settings.
  20. $vids = array();
  21. $vocabularies = taxonomy_vocabulary_get_names();
  22. foreach ($vocabularies as $voc) {
  23. $vids[] = $voc->vid;
  24. }
  25. $query = db_select('taxonomy_term_data', 't');
  26. $query->addTag('translatable');
  27. $query->addTag('term_access');
  28. // Select rows that match by term name.
  29. $tags_return = $query
  30. ->fields('t', array('tid', 'name'))
  31. ->condition('t.vid', $vids)
  32. ->condition('t.name', '%' . db_like($typed) . '%', 'LIKE')
  33. ->range(0, 10)
  34. ->execute()
  35. ->fetchAllKeyed();
  36. $term_matches = array();
  37. foreach ($tags_return as $tid => $name) {
  38. $n = $name;
  39. // Term names containing commas or quotes must be wrapped in quotes.
  40. // if (strpos($name, ',') !== FALSE || strpos($name, '"') !== FALSE) {
  41. // $n = '"' . str_replace('"', '""', $name) . '"';
  42. // }
  43. $term_matches[$n] = check_plain($name);
  44. }
  45. }
  46. drupal_json_output($term_matches);
  47. }
  48. /**
  49. * materio_search_api_autocomplete_searchapi($typed = '')
  50. *
  51. * GOOD one using searchapi (SOLR)
  52. */
  53. function materio_search_api_autocomplete_searchapi($typed = ''){
  54. // If the request has a '/' in the search text, then the menu system will have
  55. // split it into multiple arguments, recover the intended $tags_typed.
  56. $args = func_get_args();
  57. $typed = implode('/', $args);
  58. // dsm($typed, 'typed');
  59. if ($typed != '') {
  60. // search for patterns like key -another key +lastkey
  61. // and provide auto completion for the last key
  62. preg_match_all('/\s?[\+|-]?[^\s]+/', $typed, $adv_search_q);
  63. preg_match('/^(\+|-)?(.*)$/', trim(array_pop($adv_search_q[0])), $last);
  64. $tosearch = isset($last[2]) ? $last[2] : $typed;
  65. // build the query
  66. global $language;
  67. $index_machine_name = variable_get('autocompletesearchindex_'.$language->language, -1);
  68. $query = search_api_query($index_machine_name);
  69. // $query_filter = $query->createFilter();
  70. // $query_filter->condition('name', $tosearch);
  71. // $query_filter->condition('type', 'article');
  72. // $query->filter($query_filter);
  73. $query->keys($tosearch);
  74. $tags_return = $query->execute();
  75. // dsm($tags_return, '$tags_return');
  76. if($tags_return['result count']){
  77. $term_matches = array();
  78. $index = search_api_index_load($index_machine_name);
  79. $delta = 0;
  80. foreach ($index->loadItems(array_keys($tags_return['results'])) as $item) {
  81. //dsm($item, '$item');
  82. //$term_matches[$item->tid] = check_plain($item->name);
  83. // $term_matches[check_plain($item->name)] = check_plain($item->name);
  84. // TODO: leave tags with nodes
  85. $term_matches[ trim(implode(' ', $adv_search_q[0]).' '.$last[1].$item->name)] = check_plain($item->name);
  86. $delta++;
  87. if($delta > 15)
  88. break;
  89. }
  90. drupal_json_output($term_matches);
  91. }else{
  92. drupal_json_output(array());
  93. }
  94. }else{
  95. return;
  96. }
  97. // dsm($term_matches, 'term_matches');
  98. // return 'debug mode of materio_search_api_autocomplete_searchapi';
  99. }
  100. /**
  101. * materio_search_api_results_search()
  102. *
  103. *
  104. */
  105. function materio_search_api_results_search(){
  106. global $user;
  107. //dsm("materio_search_api_results_search");
  108. $args = func_get_args();
  109. $typed = implode('/', $args);
  110. // dsm($typed, 'typed');
  111. preg_match_all('/\?page=([0-9]+)/', $typed, $pages);
  112. //dsm($pages, '$pages');
  113. if($pages[0]){
  114. $typed = str_replace($pages[0][0], '', $typed);
  115. // $page = $pages[1][0];
  116. }
  117. // else{
  118. // $page = 0;
  119. // }
  120. preg_match_all('/\s?[^\s]+\s?/', $typed, $words);
  121. // dsm($words, "words");
  122. // $match = array('\\', '+', '-', '&', '|', '!', '(', ')', '{', '}', '[', ']', '^', '~', '*', '?', ':', '"', ';', ' ');
  123. // $replace = array('\\\\', '\\+', '\\-', '\\&', '\\|', '\\!', '\\(', '\\)', '\\{', '\\}', '\\[', '\\]', '\\^', '\\~', '\\*', '\\?', '\\:', '\\"', '\\;', '\\ ');
  124. foreach ($words[0] as $word) {
  125. // dsm($word);
  126. $word = trim($word);
  127. // $word = str_replace($match, $replace, $word);
  128. // dsm($word, 'word');
  129. $keys[] = $word;
  130. }
  131. // dsm($keys, 'keys');
  132. global $language;
  133. // if(user_access('use materio search api')){
  134. // }else if(user_access('use materio search api for breves')){
  135. // }
  136. if(user_access('use materio search api')){
  137. // switch index depending on key words type full text or taxonomy term (autocomplete selection)
  138. $searchmode = isset($user->data['materiosearchapi_searchmode']) ? $user->data['materiosearchapi_searchmode'] : "fulltext";
  139. switch($searchmode){
  140. case "fulltext":
  141. default:
  142. $index_machine_name = variable_get('materiaux_breves_'.$language->language, -1);
  143. break;
  144. // case "taxonomy":
  145. // $index_machine_name = variable_get('taxonomysearchindex_'.$language->language, -1);
  146. // break;
  147. }
  148. $index = search_api_index_load($index_machine_name);
  149. }else if(user_access('use materio search api for breves')){
  150. $index_machine_name = variable_get('brevessearchindex_'.$language->language, -1);
  151. // dsm($index_machine_name, '$index_machine_name');
  152. $index = search_api_index_load($index_machine_name);
  153. $could_index_machine_name = variable_get('mainsearchindex_'.$language->language, -1);
  154. $could_index = search_api_index_load($index_machine_name);
  155. }
  156. // $index_machine_name = variable_get('brevessearchindex_'.$language->language, -1);
  157. // dsm($index_machine_name, '$index_machine_name');
  158. $indexed_bundles = $index->options['data_alter_callbacks']['search_api_alter_bundle_filter']['settings']['bundles'];
  159. foreach ($indexed_bundles as $bundle) { $default_bundles[] = $bundle; }
  160. $bundles_filter = isset($user->data['materiosearchapi_bundlesfilter']) ? $user->data['materiosearchapi_bundlesfilter'] : $default_bundles;
  161. // dsm($bundles_filter, 'bundles_filter');
  162. $viewmode = isset($user->data['materiosearchapi_viewmode']) ? $user->data['materiosearchapi_viewmode'] : variable_get('defaultviewmode', 'full');
  163. // dsm($viewmode, 'viewmode');
  164. if ($keys) {
  165. // TODO: cache the results with cache graceful : http://drupal.org/project/cache_graceful
  166. try {
  167. $limit = variable_get($viewmode.'_limite', '10');
  168. $offset = pager_find_page() * $limit; //$page*$limit;//
  169. $query = search_api_query($index_machine_name, array('conjunction'=>'OR', 'parse mode'=>'direct'))
  170. ->keys(implode(' ', $keys))
  171. ->range($offset, $limit);
  172. $filter = $query->createFilter('OR');
  173. foreach ($bundles_filter as $type) {
  174. $filter->condition('type', $type, '=');
  175. }
  176. // dsm($filter, 'filter');
  177. $query->filter($filter);
  178. // $query->setOption('search_api_bypass_access', true);
  179. $query->setOption('search_api_access_account', $user);
  180. $results = $query->execute();
  181. # in case of utilisateur search, run a real search to indicate how much items you could find
  182. if(isset($could_index)){
  183. $could_query = search_api_query($could_index_machine_name, array('conjunction'=>'OR', 'parse mode'=>'direct'))
  184. ->keys(implode(' ', $keys))
  185. ->range($offset, $limit);
  186. // ->filter($filter);
  187. $could_results = $could_query->execute();
  188. // dsm($could_results, 'could_results');
  189. }
  190. }
  191. catch (SearchApiException $e) {
  192. $ret['message'] = t('An error occurred while executing the search. Please try again or contact the site administrator if the problem persists.');
  193. watchdog('materiobasemod', 'An error occurred while executing a search: !msg.', array('!msg' => $e->getMessage()), WATCHDOG_ERROR, l(t('search page'), $_GET['q']));
  194. }
  195. // dsm($results, 'results');
  196. // $accessible_results = array();
  197. // foreach ($results['results'] as $nid => $entity) {
  198. // if(node_access('view', node_load($nid)))
  199. // $accessible_results[$nid] = $entity;
  200. // }
  201. // $results['results'] = $accessible_results;
  202. // $results['result count'] = count($accessible_results);
  203. if(user_access('use materio search api')){
  204. $items = $index->loadItems(array_keys($results['results']));
  205. // dsm($items, 'items');
  206. // $count = $results['result count'];
  207. }else if(user_access('use materio search api for breves')){
  208. $items = array();
  209. $breves = $index->loadItems(array_keys($results['results']));
  210. foreach ($breves as $nid => $breve) {
  211. if(!node_access('view', $breve))
  212. continue;
  213. $items[] = $breve;
  214. $materiaux = field_get_items('node',$breve,'field_materiau_ref');
  215. // dsm($materiaux, 'materiaux');
  216. if($materiaux){
  217. foreach ($materiaux as $value) {
  218. if(!isset($could_results['results'][$value['target_id']]))
  219. continue;
  220. $materiau = node_load($value['target_id']);
  221. if(node_access('view', $materiau))
  222. $items[] = $materiau;
  223. }
  224. }
  225. }
  226. $results['breves count'] = $results['result count'];
  227. $results['result count'] = count($items);
  228. $results['could results'] = $could_results;
  229. }
  230. $ret['results']['#theme'] = 'materio_search_api_results';
  231. $ret['results']['#index'] = $index;//search_api_index_load($index_machine_name);
  232. $ret['results']['#results'] = $results;
  233. $ret['results']['#items'] = $items;
  234. $ret['results']['#view_mode'] = $viewmode;
  235. $ret['results']['#keys'] = $keys;
  236. drupal_set_title('<i class="icon-materio-search"></i>'.check_plain($typed), PASS_THROUGH);
  237. if(isset($results)){
  238. // Load pager.
  239. // if ($results['result count'] > $page->options['per_page']) {
  240. pager_default_initialize($results['result count'], $limit);
  241. $ret['results']['#pager'] = theme('pager');
  242. // }
  243. if (!empty($results['ignored'])) {
  244. drupal_set_message(
  245. t('The following search keys are too short or too common and were therefore ignored: "@list".',
  246. array( '@list' => implode(t('", "'), $results['ignored']) ) ),
  247. 'warning'
  248. );
  249. }
  250. if (!empty($results['warnings'])) {
  251. foreach ($results['warnings'] as $warning) {
  252. drupal_set_message($warning, 'warning');
  253. }
  254. }
  255. }
  256. }
  257. return $ret;
  258. }
  259. /**
  260. * materio_search_api_actuality()
  261. *
  262. */
  263. function materio_search_api_actuality(){
  264. global $user;
  265. if(isset($user->roles[1])){
  266. $date_limit = strtotime('-6 month');
  267. // dsm(date('d m y', $date_limit));
  268. }
  269. $viewmode = isset($user->data['materiosearchapi_viewmode']) ? $user->data['materiosearchapi_viewmode'] : variable_get('defaultviewmode', 'full');
  270. $limit = 10;//variable_get($viewmode.'_limite', '10');
  271. $offset = pager_find_page() * $limit;
  272. // dsm($offset);
  273. $query = new EntityFieldQuery;
  274. $query
  275. ->entityCondition('entity_type', 'node')
  276. ->propertyCondition('status', 1)
  277. ->entityCondition('bundle', array('breve'))
  278. ->propertyOrderBy('created', 'DESC')
  279. ->range($offset,$limit);
  280. if(isset($user->roles[1])){
  281. $query->propertyCondition('created', $date_limit, '>');
  282. }
  283. $result = $query->execute();
  284. // dsm($result, '$result');
  285. $count_query = new EntityFieldQuery;
  286. $count_query
  287. ->entityCondition('entity_type', 'node')
  288. ->propertyCondition('status', 1)
  289. ->entityCondition('bundle', array('breve'));
  290. // dsm($count, 'count');
  291. if(isset($user->roles[1])){
  292. $count_query->propertyCondition('created', $date_limit, '>');
  293. }
  294. $count = $count_query->count()->execute();
  295. pager_default_initialize($count, $limit);
  296. $items = array();
  297. if(isset($result['node'])){
  298. foreach ($result['node'] as $nid => $n) {
  299. $breve = node_load($nid);
  300. if(!node_access('view', $breve))
  301. continue;
  302. $items[] = $breve;
  303. $materiaux = field_get_items('node',$breve,'field_materiau_ref');
  304. // dsm($materiaux, 'materiaux');
  305. if($materiaux){
  306. foreach ($materiaux as $value) {
  307. $materiau = node_load($value['target_id']);
  308. if(node_access('view', $materiau))
  309. $items[] = $materiau;
  310. }
  311. }
  312. }
  313. }
  314. // drupal_set_title(t('Actualities'));
  315. drupal_set_title(t(''));
  316. return theme('materio_search_api_actuality', array(
  317. 'items' => $items,
  318. 'view_mode' => $viewmode,
  319. 'count' => $count,
  320. 'pager' => theme('pager'),
  321. ));
  322. }
  323. function materio_search_api_viewmode_change($vm){
  324. //dsm($vm, 'materio_search_api_viewmode_change');
  325. $rep = _materio_search_api_change_viewmode($vm);
  326. //return 'debug mode for materio_search_api_viewmode_change';
  327. // drupal_json_output($rep);
  328. drupal_goto();
  329. }