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