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