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