modules/search.php

   1 <?php
   2 /***************************************************************************
   3  *  Copyright (C) 2003-2007 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         global $page;
  51
  52         $page->assign('formulaire',1);
  53     }
  54
  55     function get_diplomas($school = null)
  56     {
  57         if (is_null($school) && Env::has('school')) {
  58             $school = Env::i('school');
  59         }
  60
  61         if (!is_null($school)) {
  62             $sql = 'SELECT type FROM applis_def WHERE id=' . $school;
  63         } else {
  64             $sql = 'DESCRIBE applis_def type';
  65         }
  66
  67         $res = XDB::query($sql);
  68         $row = $res->fetchOneRow();
  69         if (!is_null($school)) {
  70             $types = $row[0];
  71         } else {
  72             $types = explode('(',$row[1]);
  73             $types = str_replace("'","",substr($types[1],0,-1));
  74         }
  75         global $page;
  76         $page->assign('choix_diplomas', explode(',',$types));
  77     }
  78
  79     function handler_quick(&$page, $action = null, $subaction = null)
  80     {
  81         global $globals;
  82
  83         $res = XDB::query("SELECT  MIN(`diminutif`), MAX(`diminutif`)
  84                              FROM  `groupex`.`asso`
  85                             WHERE  `cat` = 'Promotions'");
  86         list($min, $max) = $res->fetchOneRow();
  87         $page->assign('promo_min', $min);
  88         $page->assign('promo_max', $max);
  89
  90         if (Env::has('quick') || $action == 'geoloc') {
  91             $page->assign('formulaire', 0);
  92
  93             require_once 'userset.inc.php';
  94             $view = new SearchSet(true, $action == 'geoloc' && substr($subaction, -3) == 'swf');
  95             $view->addMod('minifiche', 'Minifiches', true, array('with_score' => true));
  96             if (S::logged() && !Env::i('nonins')) {
  97                 $view->addMod('trombi', 'Trombinoscope', false, array('with_promo' => true, 'with_score' => true));
  98                 $view->addMod('geoloc', 'Planisphère');
  99             }
 100             $view->apply('search', $page, $action, $subaction);
 101
 102             $nb_tot = $view->count();
 103             if ($subaction) {
 104                 return;
 105             }
 106             if (!S::logged() && $nb_tot > $globals->search->public_max) {
 107                 new ThrowError('Votre recherche a généré trop de résultats pour un affichage public.');
 108             } elseif ($nb_tot > $globals->search->private_max) {
 109                 new ThrowError('Recherche trop générale');
 110             } elseif (empty($nb_tot)) {
 111                 new ThrowError('il n\'existe personne correspondant à ces critères dans la base !');
 112             }
 113         } else {
 114             $page->assign('formulaire',1);
 115             $page->addJsLink('ajax.js');
 116         }
 117
 118         require_once dirname(__FILE__) . '/search/search.inc.php';
 119         $page->changeTpl('search/index.tpl');
 120         $page->assign('xorg_title','Polytechnique.org - Annuaire');
 121     }
 122
 123     function handler_advanced(&$page, $action = null, $subaction = null)
 124     {
 125         global $globals;
 126         require_once 'geoloc.inc.php';
 127         require_once dirname(__FILE__) . '/search/search.inc.php';
 128         $page->assign('advanced',1);
 129
 130         if (!Env::has('rechercher') && $action != 'geoloc') {
 131             $this->form_prepare();
 132         } else {
 133             require_once 'userset.inc.php';
 134             $view = new SearchSet(false, $action == 'geoloc' && substr($subaction, -3) == 'swf');
 135             $view->addMod('minifiche', 'Minifiches', true);
 136             $view->addMod('trombi', 'Trombinoscope', false, array('with_promo' => true));
 137             $view->addMod('geoloc', 'Planisphère');
 138             $view->apply('search/adv', $page, $action, $subaction);
 139
 140             if ($subaction) {
 141                 return;
 142             }
 143             $nb_tot = $view->count();
 144             if ($nb_tot > $globals->search->private_max) {
 145                 $this->form_prepare();
 146                 new ThrowError('Recherche trop générale');
 147             }
 148         }
 149
 150         $page->changeTpl('search/index.tpl', $action == 'mini' ? SIMPLE : SKINNED);
 151         $page->addJsLink('ajax.js');
 152         $page->assign('public_directory',0);
 153     }
 154
 155     function handler_autocomplete(&$page, $type = null)
 156     {
 157         // Autocompletion : according to type required, return
 158         // a list of results matching with the number of matches.
 159         // The output format is :
 160         //   result1|nb1
 161         //   result2|nb2
 162         //   ...
 163         header('Content-Type: text/plain; charset="UTF-8"');
 164         $q = preg_replace(array('/\*+$/', // always look for $q*
 165                                 '/([\^\$\[\]])/', // escape special regexp char
 166                                 '/\*/'), // replace joker by regexp joker
 167                            array('',
 168                                  '\\\\\1',
 169                                  '.*'),
 170                            $_REQUEST['q']);
 171         if (!$q) exit();
 172
 173         // try to look in cached results
 174         $cache = XDB::query('SELECT  `result`
 175                                FROM  `search_autocomplete`
 176                               WHERE  `name` = {?} AND
 177                                      `query` = {?} AND
 178                                      `generated` > NOW() - INTERVAL 1 DAY',
 179                              $type, $q);
 180         if ($res = $cache->fetchOneCell()) {
 181             echo $res;
 182             die();
 183         }
 184
 185         // default search
 186         $q = preg_quote($q);
 187         $unique = '`user_id`';
 188         $db = '`auth_user_md5`';
 189         $realid = false;
 190         $beginwith = true;
 191         $field2 = false;
 192         $qsearch = $q;
 193
 194         switch ($type) {
 195           case 'binetTxt':
 196             $db = '`binets_def` INNER JOIN
 197                    `binets_ins` ON(`binets_def`.`id` = `binets_ins`.`binet_id`)';
 198             $field='`binets_def`.`text`';
 199             if (strlen($q) > 2)
 200                 $beginwith = false;
 201             $realid = '`binets_def`.`id`';
 202             break;
 203           case 'city':
 204             $db = '`geoloc_city` INNER JOIN
 205                    `adresses` ON(`geoloc_city`.`id` = `adresses`.`cityid`)';
 206             $unique='`uid`';
 207             $field='`geoloc_city`.`name`';
 208             break;
 209           case 'countryTxt':
 210             $db = '`geoloc_pays` INNER JOIN
 211                    `adresses` ON(`geoloc_pays`.`a2` = `adresses`.`country`)';
 212             $unique='`uid`';
 213             $field = '`geoloc_pays`.`pays`';
 214             $field2 = '`geoloc_pays`.`country`';
 215             $realid='`geoloc_pays`.`a2`';
 216             break;
 217           case 'entreprise':
 218             $db = '`entreprises`';
 219             $field = '`entreprise`';
 220             $unique='`uid`';
 221             break;
 222           case 'firstname':
 223             $field = '`prenom`';
 224             $q = '(^|[ \\-])'.$q;
 225             $beginwith = false;
 226             break;
 227           case 'fonctionTxt':
 228             $db = '`fonctions_def` INNER JOIN
 229                    `entreprises` ON(`entreprises`.`fonction` = `fonctions_def`.`id`)';
 230             $field = '`fonction_fr`';
 231             $unique = '`uid`';
 232             $realid = '`fonctions_def`.`id`';
 233             $qsearch = '(^|[ /\\-])'.$q;
 234             $beginwith = false;
 235             break;
 236           case 'groupexTxt':
 237             $db = '`groupesx_def` INNER JOIN
 238                    `groupesx_ins` ON(`groupesx_def`.`id` = `groupesx_ins`.`gid`)';
 239             $field='`groupesx_def`.`text`';
 240             if (strlen($q) > 2)
 241                 $beginwith = false;
 242             $realid = '`groupesx_def`.`id`';
 243             $unique = '`guid`';
 244             break;
 245           case 'name':
 246             $field = '`nom`';
 247             $field2 = '`nom_usage`';
 248             $qsearch = '(^|[ \\-])'.$q;
 249             $beginwith = false;
 250             break;
 251           case 'nationaliteTxt':
 252             $db = '`geoloc_pays` INNER JOIN
 253                    `auth_user_md5` ON(`geoloc_pays`.`a2` = `auth_user_md5`.`nationalite`)';
 254             $field = 'IF(`geoloc_pays`.`nat`=\'\',
 255                                        `geoloc_pays`.`pays`,
 256                                        `geoloc_pays`.`nat`)';
 257             $realid = '`geoloc_pays`.`a2`';
 258             break;
 259           case 'nickname':
 260             $field = '`profile_nick`';
 261             $db = '`auth_user_quick`';
 262             $qsearch = '(^|[ \\-])'.$q;
 263             $beginwith = false;
 264             break;
 265           case 'poste':
 266             $db = '`entreprises`';
 267             $field = '`poste`';
 268             $unique='`uid`';
 269             break;
 270           case 'schoolTxt':
 271             $db = '`applis_def` INNER JOIN
 272                    `applis_ins` ON(`applis_def`.`id` = `applis_ins`.`aid`)';
 273             $field='`applis_def`.`text`';
 274             $unique = '`uid`';
 275             $realid = '`applis_def`.`id`';
 276             if (strlen($q) > 2)
 277                 $beginwith = false;
 278             break;
 279           case 'secteurTxt':
 280             $db = '`emploi_secteur` INNER JOIN
 281                    `entreprises` ON(`entreprises`.`secteur` = `emploi_secteur`.`id`)';
 282             $field = '`emploi_secteur`.`label`';
 283             $realid = '`emploi_secteur`.`id`';
 284             $unique = '`uid`';
 285             $beginwith = false;
 286             break;
 287           case 'sectionTxt':
 288             $db = '`sections` INNER JOIN
 289                    `auth_user_md5` ON(`auth_user_md5`.`section` = `sections`.`id`)';
 290             $field = '`sections`.`text`';
 291             $realid = '`sections`.`id`';
 292             $beginwith = false;
 293             break;
 294           default: exit();
 295         }
 296
 297         $field_select = $field;
 298         if ($field2) {
 299             $field_select = 'IF('.$field.' REGEXP {?}, '.$field.', '.$field2.')';
 300         }
 301
 302         if ($beginwith) {
 303             $qsearch = '^'.$qsearch;
 304         }
 305         $list = XDB::iterator('SELECT  '.$field_select.' AS field,
 306                                        COUNT(DISTINCT '.$unique.') AS nb
 307                                        '.($realid?(', '.$realid.' AS id'):'').'
 308                                  FROM  '.$db.'
 309                                 WHERE  '.$field.' REGEXP {?}'.
 310                                         ($field2?(' OR '.$field2.' REGEXP {?}'):'').'
 311                              GROUP BY  '.$field_select.'
 312                              ORDER BY  nb DESC
 313                                 LIMIT  11',
 314                                $qsearch, $qsearch, $qsearch, $qsearch);
 315         $nbResults = 0;
 316         $res = "";
 317         while ($result = $list->next()) {
 318             $nbResults++;
 319             if ($nbResults == 11) {
 320                 $res .= '...|1'."\n";
 321             } else {
 322                 $res .= $result['field'].'|';
 323                 $res .= $result['nb'];
 324                 if (isset($result['id'])) {
 325                     $res  .= '|'.$result['id'];
 326                 }
 327                 $res .= "\n";
 328             }
 329         }
 330         XDB::query('REPLACE INTO  `search_autocomplete`
 331                           VALUES  ({?}, {?}, {?}, NOW())',
 332                     $type, $q, $res);
 333         echo $res;
 334         exit();
 335     }
 336
 337     function handler_list(&$page, $type = null, $idVal = null)
 338     {
 339         // Give the list of all values possible of type and builds a select input for it
 340         $field = '`text`';
 341         $id = '`id`';
 342         $where = '';
 343
 344         switch ($type) {
 345           case 'binet':
 346             $db = '`binets_def`';
 347             break;
 348           case 'country':
 349             $db = '`geoloc_pays`';
 350             $field = '`pays`';
 351             $id = '`a2`';
 352             $page->assign('onchange', 'changeCountry(this.value)');
 353             break;
 354           case 'fonction':
 355             $db = '`fonctions_def`';
 356             $field = '`fonction_fr`';
 357             break;
 358           case 'diploma':
 359             header('Content-Type: text/xml; charset="UTF-8"');
 360             $this->get_diplomas();
 361             $page->changeTpl('search/adv.grade.form.tpl', NO_SKIN);
 362             return;
 363           case 'groupex':
 364             $db = '`groupesx_def`';
 365             break;
 366           case 'nationalite':
 367             $db = '`geoloc_pays` INNER JOIN
 368                    `auth_user_md5` ON (`geoloc_pays`.`a2` = `auth_user_md5`.`nationalite`)';
 369             $field = 'IF(`nat`=\'\', `pays`, `nat`)';
 370             $id = '`a2`';
 371             break;
 372           case 'region':
 373             $db = '`geoloc_region`';
 374             $field = '`name`';
 375             $id = '`region`';
 376             if (isset($_REQUEST['country'])) {
 377                 $where .= ' WHERE `a2` = "'.$_REQUEST['country'].'"';
 378             }
 379             break;
 380           case 'school':
 381             $db = '`applis_def`';
 382             $page->assign('onchange', 'changeSchool(this.value)');
 383             break;
 384           case 'section':
 385             $db = '`sections`';
 386             break;
 387           case 'secteur':
 388             $db = '`emploi_secteur`';
 389             $field = '`label`';
 390             break;
 391           default: exit();
 392         }
 393         if (isset($idVal)) {
 394             header('Content-Type: text/plain; charset="UTF-8"');
 395             $result = XDB::query('SELECT '.$field.' AS field FROM '.$db.' WHERE '.$id.' = {?} LIMIT 1',$idVal);
 396             echo $result->fetchOneCell();
 397             exit();
 398         }
 399         header('Content-Type: text/xml; charset="UTF-8"');
 400         $page->changeTpl('include/field.select.tpl', NO_SKIN);
 401         $page->assign('name', $type);
 402         $page->assign('list', XDB::iterator('SELECT  '.$field.' AS field,
 403                                                      '.$id.' AS id
 404                                                FROM  '.$db.$where.'
 405                                            GROUP BY  '.$field.'
 406                                            ORDER BY  '.$field));
 407     }
 408 }
 409
 410 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 411 ?>