materio_search_api.pages.inc 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479
  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. // dsm($adv_search_q, "adv_search_q");
  64. preg_match('/^(\+|-)?(.*)$/', trim(array_pop($adv_search_q[0])), $last);
  65. // dsm($last, "last");
  66. $tosearch = isset($last[2]) ? $last[2] : $typed;
  67. // build the query
  68. global $language;
  69. $index_machine_name = variable_get('autocompletesearchindex_'.$language->language, -1);
  70. $query = search_api_query($index_machine_name);
  71. // $query_filter = $query->createFilter();
  72. // $query_filter->condition('name', $tosearch);
  73. // $query_filter->condition('type', 'article');
  74. // $query->filter($query_filter);
  75. $query->keys($tosearch);
  76. $tags_return = $query->execute();
  77. // dsm($tags_return, '$tags_return');
  78. if($tags_return['result count']){
  79. $term_matches = array();
  80. $index = search_api_index_load($index_machine_name);
  81. $delta = 0;
  82. foreach ($index->loadItems(array_keys($tags_return['results'])) as $item) {
  83. //dsm($item, '$item');
  84. //$term_matches[$item->tid] = check_plain($item->name);
  85. // $term_matches[check_plain($item->name)] = check_plain($item->name);
  86. // TODO: leave tags with nodes
  87. $term_matches[trim(implode(' ', $adv_search_q[0]).' "'.$last[1].$item->name).'"'] = check_plain($item->name);
  88. $delta++;
  89. if($delta > 15)
  90. break;
  91. }
  92. drupal_json_output($term_matches);
  93. }else{
  94. drupal_json_output(array());
  95. }
  96. }else{
  97. return;
  98. }
  99. // dsm($term_matches, 'term_matches');
  100. // return 'debug mode of materio_search_api_autocomplete_searchapi';
  101. }
  102. /**
  103. * materio_search_api_results_search()
  104. *
  105. *
  106. */
  107. function materio_search_api_results_search(){
  108. //dsm("materio_search_api_results_search");
  109. // retreive typed words separated by slashes as a sentence
  110. $keys = func_get_args();
  111. // dsm($args, 'args');
  112. $typed = implode('/', $keys);
  113. // remove query page params
  114. preg_match_all('/\?page=([0-9]+)/', $typed, $pages);
  115. //dsm($pages, '$pages');
  116. if($pages[0]){
  117. $typed = str_replace($pages[0][0], '', $typed);
  118. }
  119. // dsm($typed, 'typed');
  120. if ($typed) {
  121. global $language;
  122. global $user;
  123. # retrieve viewmode and then use it to define the query range
  124. $viewmode = isset($user->data['materiosearchapi_viewmode'])
  125. ? $user->data['materiosearchapi_viewmode']
  126. : variable_get('defaultviewmode', 'full');
  127. $limit = variable_get($viewmode.'_limite', '10');
  128. $offset = pager_find_page() * $limit; //$page*$limit;//
  129. # Normal search
  130. if(user_access('use materio search api')){
  131. // dsm('normal search');
  132. $results = msa_solrquery_materiauxbreves($typed, $language, $user, $offset, $limit);
  133. }
  134. # only breves search (+ related materials)
  135. else if(user_access('use materio search api for breves')){
  136. // dsm('limited search');
  137. $results = msa_solrquery_breves($typed, $language, $user, $offset, $limit);
  138. }
  139. // for all case
  140. $ret['results']['#results'] = $results;
  141. $ret['results']['#items'] = $results['items'];
  142. $ret['results']['#index'] = $results['index'];
  143. $ret['results']['#theme'] = 'materio_search_api_results';
  144. $ret['results']['#keys'] = $typed;
  145. $ret['results']['#view_mode'] = $viewmode;
  146. // page title
  147. $page_title = $typed;
  148. drupal_set_title('<i class="icon-materio-search"></i>'.check_plain($page_title), PASS_THROUGH);
  149. // render results
  150. if(isset($results)){
  151. // Load pager.
  152. // if ($results['result count'] > $page->options['per_page']) {
  153. pager_default_initialize($results['result count'], $limit);
  154. $ret['results']['#pager'] = theme('pager');
  155. // }
  156. if (!empty($results['ignored'])) {
  157. drupal_set_message(
  158. t('The following search keys are too short or too common and were therefore ignored: "@list".',
  159. array( '@list' => implode(t('", "'), $results['ignored']) ) ),
  160. 'warning'
  161. );
  162. }
  163. if (!empty($results['warnings'])) {
  164. foreach ($results['warnings'] as $warning) {
  165. drupal_set_message($warning, 'warning');
  166. }
  167. }
  168. }
  169. }
  170. // dsm($ret, 'ret');
  171. return $ret;
  172. }
  173. function msa_solrquery_materiauxbreves($typed, $language, $user, $offset, $limit){
  174. // dsm($offset, 'offset');
  175. // dsm($limit, 'limit');
  176. // -- communs --//
  177. # define default bundle option (materiaux, breves)
  178. $default_bundles = array();
  179. if(isset($taxo_index->options['data_alter_callbacks']['search_api_alter_bundle_filter']['settings']['bundles'])){
  180. $indexed_bundles = $taxo_index->options['data_alter_callbacks']['search_api_alter_bundle_filter']['settings']['bundles'];
  181. foreach ($indexed_bundles as $bundle) { $default_bundles[] = $bundle; }
  182. }
  183. # choose solr query bundle option
  184. $bundles_filter = isset($user->data['materiosearchapi_bundlesfilter'])
  185. ? $user->data['materiosearchapi_bundlesfilter']
  186. : $default_bundles;
  187. # choose solr query options
  188. $query_options = array('conjunction'=>'OR', 'parse mode'=>'direct');
  189. // -- -- taxo search -- -- //
  190. # define search api solr index
  191. $taxo_index_machine_name = variable_get('taxonomysearchindex_'.$language->language, -1);
  192. $taxo_index = search_api_index_load($taxo_index_machine_name);
  193. #create the solr query for taxonomy search
  194. $taxo_query = search_api_query($taxo_index_machine_name, $query_options)
  195. ->keys($typed);
  196. # apply bundle options to solr query if usefull
  197. if(count($bundles_filter)){
  198. $filter = $taxo_query->createFilter('OR');
  199. foreach ($bundles_filter as $type)
  200. $filter->condition('type', $type, '=');
  201. $taxo_query->filter($filter);
  202. }
  203. // $query->setOption('search_api_bypass_access', true);
  204. # add user access solr query option
  205. $taxo_query->setOption('search_api_access_account', $user);
  206. #execute first time to get the all items, to be able to filter the full text research
  207. $taxo_total_results = $taxo_query->execute();
  208. // dsm($taxo_total_results, "taxo total results");
  209. #add range to retriev the real current results
  210. $taxo_query->range($offset, $limit);
  211. # execute solr query and record results
  212. $taxo_results = $taxo_query->execute();
  213. // dsm($taxo_results, "taxo results");
  214. $taxo_items = $taxo_index->loadItems(array_keys($taxo_results['results']));
  215. // dsm($taxo_items, 'taxo items');
  216. $taxo_results['items'] = $taxo_items;
  217. $taxo_results['index'] = $taxo_index;//search_api_index_load($index_machine_name);
  218. // -- -- full text search -- -- //
  219. #define search api solr index
  220. $fulltxt_index_machine_name = variable_get('fulltextsearchindex_'.$language->language, -1);
  221. $fulltxt_index = search_api_index_load($fulltxt_index_machine_name);
  222. #then calculate the good offset and limit for the complementary search
  223. $fulltxt_offset = $offset - $taxo_results['result count'];
  224. #create the solr query for taxonomy search
  225. $fulltxt_query = search_api_query($fulltxt_index_machine_name, $query_options)
  226. ->keys($typed)
  227. ->range(($fulltxt_offset > 0 ? $fulltxt_offset : 0), $limit); // change offset to match with the first query (taxonomy)
  228. # apply bundle options to solr query if usefull
  229. if(count($bundles_filter)){
  230. $filter = $fulltxt_query->createFilter('OR');
  231. foreach ($bundles_filter as $type)
  232. $filter->condition('type', $type, '=');
  233. $fulltxt_query->filter($filter);
  234. }
  235. # filter to remove precedent search query
  236. # !!!!!! THIS WILL NOT WORK, IT DOESN'T KNOW THE ITEMS FROM PAGES BEFORE THE CURRENT ONE ...
  237. $nid_filter = $fulltxt_query->createFilter('AND');
  238. foreach ($taxo_total_results['results'] as $nid => $item)
  239. $nid_filter->condition('nid', $nid, '<>');
  240. $fulltxt_query->filter($nid_filter);
  241. # add user access solr query option
  242. $fulltxt_query->setOption('search_api_access_account', $user);
  243. # execute solr query and record results
  244. $fulltxt_results = $fulltxt_query->execute();
  245. // dsm($fulltxt_results, "fulltxt_results");
  246. # add the full text result count to the taxo result to have the total of items
  247. $taxo_results['result count'] += $fulltxt_results['result count'];
  248. # if we are at the end of the first search results list
  249. if($fulltxt_offset >= 0){
  250. $fulltxt_items = $fulltxt_index->loadItems(array_keys($fulltxt_results['results']));
  251. // dsm($fulltxt_items, 'full txt items');
  252. $taxo_results['items'] += $fulltxt_items;
  253. }
  254. # TODO: cache the results with cache graceful : http://drupal.org/project/cache_graceful
  255. return $taxo_results;
  256. }
  257. function msa_solrquery_breves($typed, $language, $user, $offset, $limit){
  258. $index_machine_name = variable_get('brevessearchindex_'.$language->language, -1);
  259. // dsm($index_machine_name, '$index_machine_name');
  260. $index = search_api_index_load($index_machine_name);
  261. # choose solr query options
  262. $query_options = array('conjunction'=>'OR', 'parse mode'=>'direct');
  263. #create the solr query
  264. $query = search_api_query($index_machine_name, $query_options)
  265. ->keys($typed)
  266. ->range($offset, $limit);
  267. // $query->setOption('search_api_bypass_access', true);
  268. # add user access solr query option
  269. $query->setOption('search_api_access_account', $user);
  270. # execute solr query and record results
  271. $results = $query->execute();
  272. // dsm($results, 'results');
  273. $results['index'] = $index;//search_api_index_load($index_machine_name);
  274. #could
  275. $could_index_machine_name = variable_get('fulltextsearchindex_'.$language->language, -1);
  276. $could_index = search_api_index_load($could_index_machine_name);
  277. # in case of free user search, run a real search to indicate how much items you could find
  278. $could_query = search_api_query($could_index_machine_name, array('conjunction'=>'OR', 'parse mode'=>'direct'))
  279. ->keys($typed);
  280. $could_results = $could_query->execute();
  281. // dsm($could_results, 'could_results');
  282. $results['could results'] = $could_results;
  283. # add items : breve + materials wich are in the could result
  284. $items = array();
  285. $breves = $index->loadItems(array_keys($results['results']));
  286. foreach ($breves as $nid => $breve) {
  287. if(!node_access('view', $breve))
  288. continue;
  289. $items[] = $breve;
  290. $materiaux = field_get_items('node',$breve,'field_materiau_ref');
  291. // dsm($materiaux, 'materiaux');
  292. if($materiaux){
  293. foreach ($materiaux as $value) {
  294. if(!isset($could_results['results'][$value['target_id']]))
  295. continue;
  296. $materiau = node_load($value['target_id']);
  297. if(node_access('view', $materiau))
  298. $items[] = $materiau;
  299. }
  300. }
  301. }
  302. $results['items'] = $items;
  303. $results['breves count'] = $results['result count'];
  304. $results['result count'] = count($items);
  305. // dsm($results, "results");
  306. # TODO: cache the results with cache graceful : http://drupal.org/project/cache_graceful
  307. return $results;
  308. }
  309. /**
  310. * materio_search_api_actuality()
  311. *
  312. */
  313. function materio_search_api_actuality(){
  314. global $user;
  315. if(isset($user->roles[1])){
  316. $date_limit = strtotime('-6 month');
  317. // dsm(date('d m y', $date_limit));
  318. }
  319. $viewmode = isset($user->data['materiosearchapi_viewmode']) ? $user->data['materiosearchapi_viewmode'] : variable_get('defaultviewmode', 'full');
  320. $limit = 10;//variable_get($viewmode.'_limite', '10');
  321. $offset = pager_find_page() * $limit;
  322. // dsm($offset);
  323. $query = new EntityFieldQuery;
  324. $query
  325. ->entityCondition('entity_type', 'node')
  326. ->propertyCondition('status', 1)
  327. ->entityCondition('bundle', array('breve'))
  328. ->propertyOrderBy('created', 'DESC')
  329. ->range($offset,$limit);
  330. if(isset($user->roles[1])){
  331. $query->propertyCondition('created', $date_limit, '>');
  332. }
  333. $result = $query->execute();
  334. // dsm($result, '$result');
  335. $count_query = new EntityFieldQuery;
  336. $count_query
  337. ->entityCondition('entity_type', 'node')
  338. ->propertyCondition('status', 1)
  339. ->entityCondition('bundle', array('breve'));
  340. // dsm($count, 'count');
  341. if(isset($user->roles[1])){
  342. $count_query->propertyCondition('created', $date_limit, '>');
  343. }
  344. $count = $count_query->count()->execute();
  345. pager_default_initialize($count, $limit);
  346. $items = array();
  347. if(isset($result['node'])){
  348. foreach ($result['node'] as $nid => $n) {
  349. $breve = node_load($nid);
  350. if(!node_access('view', $breve))
  351. continue;
  352. $items[] = $breve;
  353. $materiaux = field_get_items('node',$breve,'field_materiau_ref');
  354. // dsm($materiaux, 'materiaux');
  355. if($materiaux){
  356. foreach ($materiaux as $value) {
  357. $materiau = node_load($value['target_id']);
  358. if(node_access('view', $materiau))
  359. $items[] = $materiau;
  360. }
  361. }
  362. }
  363. }
  364. // drupal_set_title(t('Actualities'));
  365. drupal_set_title(t(''));
  366. return theme('materio_search_api_actuality', array(
  367. 'items' => $items,
  368. 'view_mode' => $viewmode,
  369. 'count' => $count,
  370. 'pager' => theme('pager'),
  371. ));
  372. }
  373. function materio_search_api_viewmode_change($vm){
  374. //dsm($vm, 'materio_search_api_viewmode_change');
  375. $rep = _materio_search_api_change_viewmode($vm);
  376. //return 'debug mode for materio_search_api_viewmode_change';
  377. // drupal_json_output($rep);
  378. drupal_goto();
  379. }