Some fixes and updates.
[platal.git] / modules / search.php
1 <?php
2 /***************************************************************************
3 * Copyright (C) 2003-2008 Polytechnique.org *
4 * http://opensource.polytechnique.org/ *
5 * *
6 * This program is free software; you can redistribute it and/or modify *
7 * it under the terms of the GNU General Public License as published by *
8 * the Free Software Foundation; either version 2 of the License, or *
9 * (at your option) any later version. *
10 * *
11 * This program is distributed in the hope that it will be useful, *
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
14 * GNU General Public License for more details. *
15 * *
16 * You should have received a copy of the GNU General Public License *
17 * along with this program; if not, write to the Free Software *
18 * Foundation, Inc., *
19 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA *
20 ***************************************************************************/
21
22 class SearchModule extends PLModule
23 {
24 function handlers()
25 {
26 return array(
27 'search' => $this->make_hook('quick', AUTH_PUBLIC),
28 'search/adv' => $this->make_hook('advanced', AUTH_COOKIE),
29 'advanced_search.php' => $this->make_hook('redir_advanced', AUTH_PUBLIC),
30 'search/autocomplete' => $this->make_hook('autocomplete', AUTH_COOKIE, 'user', NO_AUTH),
31 'search/list' => $this->make_hook('list', AUTH_COOKIE, 'user', NO_AUTH),
32 );
33 }
34
35 function handler_redir_advanced(&$page, $mode = null)
36 {
37 pl_redirect('search/adv');
38 exit;
39 }
40
41 function form_prepare()
42 {
43 Platal::page()->assign('formulaire',1);
44 }
45
46 function get_diplomas($school = null)
47 {
48 if (is_null($school) && Env::has('school')) {
49 $school = Env::i('school');
50 }
51
52 if ((!is_null($school)) && ($school != '')) {
53 $sql = 'SELECT degreeid
54 FROM profile_education_degree
55 WHERE eduid=' . $school;
56 } else {
57 $sql = 'SELECT id
58 FROM profile_education_degree_enum
59 ORDER BY id';
60 }
61
62 $res = XDB::query($sql);
63 Platal::page()->assign('choix_diplomas', $res->fetchColumn());
64
65 $sql = 'SELECT degree
66 FROM profile_education_degree_enum
67 ORDER BY id';
68 $res = XDB::query($sql);
69 Platal::page()->assign('name_diplomas', $res->fetchColumn());
70 }
71
72 function handler_quick(&$page, $action = null, $subaction = null)
73 {
74 global $globals;
75
76 $res = XDB::query("SELECT MIN(`diminutif`), MAX(`diminutif`)
77 FROM `groupex`.`asso`
78 WHERE `cat` = 'Promotions'");
79 list($min, $max) = $res->fetchOneRow();
80 $page->assign('promo_min', $min);
81 $page->assign('promo_max', $max);
82
83 if (Env::has('quick') || $action == 'geoloc') {
84 $quick = trim(Env::v('quick'));
85 if (S::logged() && !Env::has('page')) {
86 S::logger()->log('search', 'quick=' . $quick);
87 }
88 $list = 'profile|prf|fiche|fic|referent|ref|mentor';
89 if (S::has_perms()) {
90 $list .= '|admin|adm|ax';
91 }
92 if (preg_match('/^(' . $list . '):([-a-z]+(\.[-a-z]+(\.\d{2,4})?)?)$/', replace_accent($quick), $matches)) {
93 $forlife = $matches[2];
94 switch($matches[1]) {
95 case 'admin': case 'adm':
96 $base = 'admin/user/';
97 break;
98 case 'ax':
99 $base = 'profile/ax/';
100 break;
101 case 'profile': case 'prf': case 'fiche': case 'fic':
102 $base = 'profile/';
103 break;
104 case 'referent': case 'ref': case 'mentor':
105 $base = 'referent/';
106 break;
107 }
108
109 require_once 'user.func.inc.php';
110 $login = get_user_forlife($forlife, '_silent_user_callback');
111 if ($login) {
112 pl_redirect($base . $login);
113 }
114 $_REQUEST['quick'] = $forlife;
115 $_GET['quick'] = $forlife;
116 } elseif (strpos($quick, 'doc:') === 0) {
117 $url = 'Docs/Recherche?';
118 $url .= 'action=search&q=' . urlencode(substr($quick, 4));
119 $url .= '&group=' . urlencode('-Equipe,-Main,-PmWiki,-Site,-Review');
120 pl_redirect($url);
121 }
122
123 $page->assign('formulaire', 0);
124
125 require_once 'userset.inc.php';
126 $view = new SearchSet(true, $action == 'geoloc' && substr($subaction, -3) == 'swf');
127 $view->addMod('minifiche', 'Mini-fiches', true, array('with_score' => true));
128 if (S::logged() && !Env::i('nonins')) {
129 $view->addMod('trombi', 'Trombinoscope', false, array('with_promo' => true, 'with_score' => true));
130 $view->addMod('geoloc', 'Planisphère', false, array('with_annu' => 'search/adv'));
131 }
132 $view->apply('search', $page, $action, $subaction);
133
134 $nb_tot = $view->count();
135 $page->assign('search_results_nb', $nb_tot);
136 if ($subaction) {
137 return;
138 }
139 if (!S::logged() && $nb_tot > $globals->search->public_max) {
140 new ThrowError('Votre recherche a généré trop de résultats pour un affichage public.');
141 } elseif ($nb_tot > $globals->search->private_max) {
142 new ThrowError('Recherche trop générale. Une <a href="search/adv">recherche avancée</a> permet de préciser la recherche.');
143 } elseif (empty($nb_tot)) {
144 new ThrowError('Il n\'existe personne correspondant à ces critères dans la base !');
145 }
146 } else {
147 $page->assign('formulaire',1);
148 $page->addJsLink('ajax.js');
149 }
150
151 $this->load('search.inc.php');
152 $page->changeTpl('search/index.tpl');
153 $page->setTitle('Annuaire');
154 }
155
156 function handler_advanced(&$page, $action = null, $subaction = null)
157 {
158 global $globals;
159 require_once 'geoloc.inc.php';
160 $this->load('search.inc.php');
161 $page->assign('advanced',1);
162 $page->addJsLink('jquery.autocomplete.js');
163
164 if (!Env::has('rechercher') && $action != 'geoloc') {
165 $this->form_prepare();
166 } else {
167 $textFields = array(
168 'country' => array('field' => 'a2', 'table' => 'geoloc_pays', 'text' => 'pays', 'exact' => false),
169 'fonction' => array('field' => 'id', 'table' => 'fonctions_def', 'text' => 'fonction_fr', 'exact' => true),
170 'secteur' => array('field' => 'id', 'table' => 'emploi_secteur', 'text' => 'label', 'exact' => false),
171 'nationalite' => array('field' => 'a2', 'table' => 'geoloc_pays', 'text' => 'nat', 'exact' => 'false'),
172 'binet' => array('field' => 'id', 'table' => 'binets_def', 'text' => 'text', 'exact' => false),
173 'networking_type' => array('field' => 'network_type', 'table' => 'profile_networking_enum',
174 'text' => 'name', 'exact' => false),
175 'groupex' => array('field' => 'id', 'table' => 'groupex.asso',
176 'text' => "(a.cat = 'GroupesX' OR a.cat = 'Institutions') AND pub = 'public' AND nom",
177 'exact' => false),
178 'section' => array('field' => 'id', 'table' => 'sections', 'text' => 'text', 'exact' => false),
179 'school' => array('field' => 'id', 'table' => 'profile_education_enum', 'text' => 'name', 'exact' => false),
180 'city' => array('table' => 'geoloc_city', 'text' => 'name', 'exact' => false)
181 );
182 if (!Env::has('page')) {
183 S::logger()->log('search', 'adv=' . var_export($_GET, true));
184 }
185 foreach ($textFields as $field=>&$query) {
186 if (!Env::v($field) && Env::v($field . 'Txt')) {
187 $res = XDB::query("SELECT {$query['field']}
188 FROM {$query['table']}
189 WHERE {$query['text']} " . ($query['exact'] ? " = {?}" :
190 " LIKE CONCAT('%', {?}, '%')"),
191 Env::v($field . 'Txt'));
192 $_REQUEST[$field] = $res->fetchOneCell();
193 }
194 }
195
196 require_once 'userset.inc.php';
197 $view = new SearchSet(false, $action == 'geoloc' && substr($subaction, -3) == 'swf');
198 $view->addMod('minifiche', 'Mini-fiches', true);
199 $view->addMod('trombi', 'Trombinoscope', false, array('with_promo' => true));
200 //$view->addMod('geoloc', 'Planisphère', false, array('with_annu' => 'search/adv'));
201 $view->apply('search/adv', $page, $action, $subaction);
202
203 if ($subaction) {
204 return;
205 }
206 $nb_tot = $view->count();
207 if ($nb_tot > $globals->search->private_max) {
208 $this->form_prepare();
209 new ThrowError('Recherche trop générale.');
210 }
211 }
212
213 $page->changeTpl('search/index.tpl', $action == 'mini' ? SIMPLE : SKINNED);
214 $page->addJsLink('ajax.js');
215 $page->assign('public_directory',0);
216 }
217
218 function handler_autocomplete(&$page, $type = null)
219 {
220 // Autocompletion : according to type required, return
221 // a list of results matching with the number of matches.
222 // The output format is :
223 // result1|nb1
224 // result2|nb2
225 // ...
226 header('Content-Type: text/plain; charset="UTF-8"');
227 $q = preg_replace(array('/\*+$/', // always look for $q*
228 '/([\^\$\[\]])/', // escape special regexp char
229 '/\*/'), // replace joker by regexp joker
230 array('',
231 '\\\\\1',
232 '.*'),
233 $_REQUEST['q']);
234 if (!$q) exit();
235
236 // try to look in cached results
237 $cache = XDB::query('SELECT `result`
238 FROM `search_autocomplete`
239 WHERE `name` = {?} AND
240 `query` = {?} AND
241 `generated` > NOW() - INTERVAL 1 DAY',
242 $type, $q);
243 if ($res = $cache->fetchOneCell()) {
244 echo $res;
245 die();
246 }
247
248 // default search
249 $unique = '`user_id`';
250 $db = '`auth_user_md5`';
251 $realid = false;
252 $beginwith = true;
253 $field2 = false;
254 $qsearch = str_replace(array('%', '_'), '', $q);
255
256 switch ($type) {
257 case 'binetTxt':
258 $db = '`binets_def` INNER JOIN
259 `binets_ins` ON(`binets_def`.`id` = `binets_ins`.`binet_id`)';
260 $field = '`binets_def`.`text`';
261 if (strlen($q) > 2)
262 $beginwith = false;
263 $realid = '`binets_def`.`id`';
264 break;
265 case 'networking_typeTxt':
266 $db = '`profile_networking_enum` INNER JOIN
267 `profile_networking` ON(`profile_networking`.`network_type` = `profile_networking_enum`.`network_type`)';
268 $field = '`profile_networking_enum`.`name`';
269 $unique = 'uid';
270 $realid = '`profile_networking_enum`.`network_type`';
271 break;
272 case 'city':
273 $db = '`geoloc_city` INNER JOIN
274 `adresses` ON(`geoloc_city`.`id` = `adresses`.`cityid`)';
275 $unique='`uid`';
276 $field='`geoloc_city`.`name`';
277 break;
278 case 'countryTxt':
279 $db = '`geoloc_pays` INNER JOIN
280 `adresses` ON(`geoloc_pays`.`a2` = `adresses`.`country`)';
281 $unique = '`uid`';
282 $field = '`geoloc_pays`.`pays`';
283 $field2 = '`geoloc_pays`.`country`';
284 $realid = '`geoloc_pays`.`a2`';
285 break;
286 case 'entreprise':
287 $db = '`entreprises`';
288 $field = '`entreprise`';
289 $unique = '`uid`';
290 break;
291 case 'firstname':
292 $field = '`prenom`';
293 $beginwith = false;
294 break;
295 case 'fonctionTxt':
296 $db = '`fonctions_def` INNER JOIN
297 `entreprises` ON(`entreprises`.`fonction` = `fonctions_def`.`id`)';
298 $field = '`fonction_fr`';
299 $unique = '`uid`';
300 $realid = '`fonctions_def`.`id`';
301 $beginwith = false;
302 break;
303 case 'groupexTxt':
304 $db = "groupex.asso AS a INNER JOIN
305 groupex.membres AS m ON(a.id = m.asso_id
306 AND (a.cat = 'GroupesX' OR a.cat = 'Institutions')
307 AND a.pub = 'public')";
308 $field='a.nom';
309 $field2 = 'a.diminutif';
310 if (strlen($q) > 2)
311 $beginwith = false;
312 $realid = 'a.id';
313 $unique = 'm.uid';
314 break;
315 case 'name':
316 $field = '`nom`';
317 $field2 = '`nom_usage`';
318 $beginwith = false;
319 break;
320 case 'nationaliteTxt':
321 $db = '`geoloc_pays` INNER JOIN
322 `auth_user_md5` ON (`geoloc_pays`.`a2` = `auth_user_md5`.`nationalite` OR
323 `geoloc_pays`.`a2` = `auth_user_md5`.`nationalite2` OR
324 `geoloc_pays`.`a2` = `auth_user_md5`.`nationalite3`)';
325 $field = 'IF(`geoloc_pays`.`nat`=\'\',
326 `geoloc_pays`.`pays`,
327 `geoloc_pays`.`nat`)';
328 $realid = '`geoloc_pays`.`a2`';
329 break;
330 case 'nickname':
331 $field = '`profile_nick`';
332 $db = '`auth_user_quick`';
333 $beginwith = false;
334 break;
335 case 'poste':
336 $db = '`entreprises`';
337 $field = '`poste`';
338 $unique = '`uid`';
339 break;
340 case 'schoolTxt':
341 $db = 'profile_education_enum INNER JOIN
342 profile_education ON (profile_education_enum.id = profile_education.eduid)';
343 $field = 'profile_education_enum.name';
344 $unique = 'uid';
345 $realid = 'profile_education_enum.id';
346 if (strlen($q) > 2)
347 $beginwith = false;
348 break;
349 case 'secteurTxt':
350 $db = '`emploi_secteur` INNER JOIN
351 `entreprises` ON(`entreprises`.`secteur` = `emploi_secteur`.`id`)';
352 $field = '`emploi_secteur`.`label`';
353 $realid = '`emploi_secteur`.`id`';
354 $unique = '`uid`';
355 $beginwith = false;
356 break;
357 case 'sectionTxt':
358 $db = '`sections` INNER JOIN
359 `auth_user_md5` ON(`auth_user_md5`.`section` = `sections`.`id`)';
360 $field = '`sections`.`text`';
361 $realid = '`sections`.`id`';
362 $beginwith = false;
363 break;
364 default: exit();
365 }
366
367 function make_field_test($fields, $beginwith) {
368 $tests = array();
369 $tests[] = $fields . ' LIKE CONCAT({?}, \'%\')';
370 if (!$beginwith) {
371 $tests[] = $fields . ' LIKE CONCAT(\'% \', {?}, \'%\')';
372 $tests[] = $fields . ' LIKE CONCAT(\'%-\', {?}, \'%\')';
373 }
374 return '(' . implode(' OR ', $tests) . ')';
375 }
376 $field_select = $field;
377 $field_t = make_field_test($field, $beginwith);
378 if ($field2) {
379 $field2_t = make_field_test($field2, $beginwith);
380 $field_select = 'IF(' . $field_t . ', ' . $field . ', ' . $field2. ')';
381 }
382 $list = XDB::iterator('SELECT ' . $field_select . ' AS field,
383 COUNT(DISTINCT ' . $unique . ') AS nb
384 ' . ($realid ? (', ' . $realid . ' AS id') : '') . '
385 FROM ' . $db . '
386 WHERE ' . $field_t .
387 ($field2 ? (' OR ' . $field2_t) : '') . '
388 GROUP BY ' . $field_select . '
389 ORDER BY nb DESC
390 LIMIT 11',
391 $qsearch, $qsearch, $qsearch, $qsearch, $qsearch, $qsearch, $qsearch, $qsearch,
392 $qsearch, $qsearch, $qsearch, $qsearch, $qsearch, $qsearch, $qsearch, $qsearch);
393
394 $nbResults = 0;
395 $res = "";
396 while ($result = $list->next()) {
397 $nbResults++;
398 if ($nbResults == 11) {
399 $res .= $q."|-1\n";
400 } else {
401 $res .= $result['field'].'|';
402 $res .= $result['nb'];
403 if (isset($result['id'])) {
404 $res .= '|'.$result['id'];
405 }
406 $res .= "\n";
407 }
408 }
409 XDB::query('REPLACE INTO `search_autocomplete`
410 VALUES ({?}, {?}, {?}, NOW())',
411 $type, $q, $res);
412 echo $res;
413 exit();
414 }
415
416 function handler_list(&$page, $type = null, $idVal = null)
417 {
418 // Give the list of all values possible of type and builds a select input for it
419 $field = '`text`';
420 $id = '`id`';
421 $where = '';
422
423 switch ($type) {
424 case 'binet':
425 $db = '`binets_def`';
426 break;
427 case 'networking_type':
428 $db = '`profile_networking_enum`';
429 $field = '`name`';
430 $id = '`network_type`';
431 break;
432 case 'country':
433 $db = '`geoloc_pays`';
434 $field = '`pays`';
435 $id = '`a2`';
436 $page->assign('onchange', 'changeCountry(this.value)');
437 break;
438 case 'fonction':
439 $db = '`fonctions_def`';
440 $field = '`fonction_fr`';
441 break;
442 case 'diploma':
443 header('Content-Type: text/xml; charset="UTF-8"');
444 $this->get_diplomas();
445 $page->changeTpl('search/adv.grade.form.tpl', NO_SKIN);
446 return;
447 case 'groupex':
448 $db = 'groupex.asso';
449 $where = " WHERE (cat = 'GroupesX' OR cat = 'Institutions') AND pub = 'public'";
450 $field = 'nom';
451 break;
452 case 'nationalite':
453 $db = '`geoloc_pays` INNER JOIN
454 `auth_user_md5` ON (`geoloc_pays`.`a2` = `auth_user_md5`.`nationalite` OR
455 `geoloc_pays`.`a2` = `auth_user_md5`.`nationalite2` OR
456 `geoloc_pays`.`a2` = `auth_user_md5`.`nationalite3`)';
457 $field = 'IF(`nat`=\'\', `pays`, `nat`)';
458 $id = '`a2`';
459 break;
460 case 'region':
461 $db = '`geoloc_region`';
462 $field = '`name`';
463 $id = '`region`';
464 if (isset($_REQUEST['country'])) {
465 $where .= ' WHERE `a2` = "'.$_REQUEST['country'].'"';
466 }
467 break;
468 case 'school':
469 $db = 'profile_education_enum';
470 $field = 'name';
471 $id = 'id';
472 $page->assign('onchange', 'changeSchool(this.value)');
473 break;
474 case 'section':
475 $db = '`sections`';
476 break;
477 case 'secteur':
478 $db = '`emploi_secteur`';
479 $field = '`label`';
480 break;
481 default: exit();
482 }
483 if (isset($idVal)) {
484 header('Content-Type: text/plain; charset="UTF-8"');
485 $result = XDB::query('SELECT '.$field.' AS field FROM '.$db.' WHERE '.$id.' = {?} LIMIT 1',$idVal);
486 echo $result->fetchOneCell();
487 exit();
488 }
489 header('Content-Type: text/xml; charset="UTF-8"');
490 $page->changeTpl('include/field.select.tpl', NO_SKIN);
491 $page->assign('name', $type);
492 $page->assign('list', XDB::iterator('SELECT '.$field.' AS field,
493 '.$id.' AS id
494 FROM '.$db.$where.'
495 GROUP BY '.$field.'
496 ORDER BY '.$field));
497 $page->assign('with_text_value', true);
498 $page->assign('onchange', "document.forms.recherche.{$type}Txt.value = this.options[this.selectedIndex].text");
499 }
500 }
501
502 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
503 ?>