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             $quick = trim(Env::v('quick'));
  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', 'Minifiches', 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');
 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->assign('xorg_title','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
 167         if (!Env::has('rechercher') && $action != 'geoloc') {
 168             $this->form_prepare();
 169         } else {
 170             $textFields = array(
 171                 'country' => array('field' => 'a2', 'table' => 'geoloc_pays', 'text' => 'pays', 'exact' => false),
 172                 'fonction' => array('field' => 'id', 'table' => 'fonctions_def', 'text' => 'fonction_fr', 'exact' => true),
 173                 'secteur' => array('field' => 'id', 'table' => 'emploi_secteur', 'text' => 'label', 'exact' => false),
 174                 'nationalite' => array('field' => 'a2', 'table' => 'geoloc_pays', 'text' => 'nat', 'exact' => 'false'),
 175                 'binet' => array('field' => 'id', 'table' => 'binets_def', 'text' => 'text', 'exact' => false),
 176                 'groupex' => array('field' => 'id', 'table' => 'groupex.asso',
 177                                    'text' => "(a.cat = 'GroupesX' OR a.cat = 'Institutions') AND pub = 'public' AND nom",
 178                                    'exact' => false),
 179                 'section' => array('field' => 'id', 'table' => 'sections', 'text' => 'text', 'exact' => false),
 180                 'school' => array('field' => 'id', 'table' => 'applis_def', 'text' => 'text', 'exact' => false),
 181                 'city' => array('table' => 'geoloc_city', 'text' => 'name', 'exact' => false)
 182             );
 183             foreach ($textFields as $field=>&$query) {
 184                 if (!Env::v($field) && Env::v($field . 'Txt')) {
 185                     $res = XDB::query("SELECT  {$query['field']}
 186                                          FROM  {$query['table']}
 187                                         WHERE  {$query['text']} " . ($query['exact'] ? " = {?}" :
 188                                                                     " LIKE CONCAT('%', {?}, '%')"),
 189                                       Env::v($field . 'Txt'));
 190                     $_REQUEST[$field] = $res->fetchOneCell();
 191                 }
 192             }
 193
 194             require_once 'userset.inc.php';
 195             $view = new SearchSet(false, $action == 'geoloc' && substr($subaction, -3) == 'swf');
 196             $view->addMod('minifiche', 'Minifiches', true);
 197             $view->addMod('trombi', 'Trombinoscope', false, array('with_promo' => true));
 198             $view->addMod('geoloc', 'Planisphère', false, array('with_annu' => 'search/adv'));
 199             $view->apply('search/adv', $page, $action, $subaction);
 200
 201             if ($subaction) {
 202                 return;
 203             }
 204             $nb_tot = $view->count();
 205             if ($nb_tot > $globals->search->private_max) {
 206                 $this->form_prepare();
 207                 new ThrowError('Recherche trop générale');
 208             }
 209         }
 210
 211         $page->changeTpl('search/index.tpl', $action == 'mini' ? SIMPLE : SKINNED);
 212         $page->addJsLink('ajax.js');
 213         $page->assign('public_directory',0);
 214     }
 215
 216     function handler_autocomplete(&$page, $type = null)
 217     {
 218         // Autocompletion : according to type required, return
 219         // a list of results matching with the number of matches.
 220         // The output format is :
 221         //   result1|nb1
 222         //   result2|nb2
 223         //   ...
 224         header('Content-Type: text/plain; charset="UTF-8"');
 225         $q = preg_replace(array('/\*+$/', // always look for $q*
 226                                 '/([\^\$\[\]])/', // escape special regexp char
 227                                 '/\*/'), // replace joker by regexp joker
 228                            array('',
 229                                  '\\\\\1',
 230                                  '.*'),
 231                            $_REQUEST['q']);
 232         if (!$q) exit();
 233
 234         // try to look in cached results
 235         $cache = XDB::query('SELECT  `result`
 236                                FROM  `search_autocomplete`
 237                               WHERE  `name` = {?} AND
 238                                      `query` = {?} AND
 239                                      `generated` > NOW() - INTERVAL 1 DAY',
 240                              $type, $q);
 241         if ($res = $cache->fetchOneCell()) {
 242             echo $res;
 243             die();
 244         }
 245
 246         // default search
 247         $q = preg_quote($q);
 248         $unique = '`user_id`';
 249         $db = '`auth_user_md5`';
 250         $realid = false;
 251         $beginwith = true;
 252         $field2 = false;
 253         $qsearch = $q;
 254
 255         switch ($type) {
 256           case 'binetTxt':
 257             $db = '`binets_def` INNER JOIN
 258                    `binets_ins` ON(`binets_def`.`id` = `binets_ins`.`binet_id`)';
 259             $field='`binets_def`.`text`';
 260             if (strlen($q) > 2)
 261                 $beginwith = false;
 262             $realid = '`binets_def`.`id`';
 263             break;
 264           case 'city':
 265             $db = '`geoloc_city` INNER JOIN
 266                    `adresses` ON(`geoloc_city`.`id` = `adresses`.`cityid`)';
 267             $unique='`uid`';
 268             $field='`geoloc_city`.`name`';
 269             break;
 270           case 'countryTxt':
 271             $db = '`geoloc_pays` INNER JOIN
 272                    `adresses` ON(`geoloc_pays`.`a2` = `adresses`.`country`)';
 273             $unique='`uid`';
 274             $field = '`geoloc_pays`.`pays`';
 275             $field2 = '`geoloc_pays`.`country`';
 276             $realid='`geoloc_pays`.`a2`';
 277             break;
 278           case 'entreprise':
 279             $db = '`entreprises`';
 280             $field = '`entreprise`';
 281             $unique='`uid`';
 282             break;
 283           case 'firstname':
 284             $field = '`prenom`';
 285             $q = '(^|[ \\-])'.$q;
 286             $beginwith = false;
 287             break;
 288           case 'fonctionTxt':
 289             $db = '`fonctions_def` INNER JOIN
 290                    `entreprises` ON(`entreprises`.`fonction` = `fonctions_def`.`id`)';
 291             $field = '`fonction_fr`';
 292             $unique = '`uid`';
 293             $realid = '`fonctions_def`.`id`';
 294             $qsearch = '(^|[ /\\-])'.$q;
 295             $beginwith = false;
 296             break;
 297           case 'groupexTxt':
 298             $db = "groupex.asso AS a INNER JOIN
 299                    groupex.membres AS m ON(a.id = m.asso_id
 300                                            AND (a.cat = 'GroupesX' OR a.cat = 'Institutions')
 301                                            AND a.pub = 'public')";
 302             $field='a.nom';
 303             $field2 = 'a.diminutif';
 304             if (strlen($q) > 2)
 305                 $beginwith = false;
 306             $realid = 'a.id';
 307             $unique = 'm.uid';
 308             break;
 309           case 'name':
 310             $field = '`nom`';
 311             $field2 = '`nom_usage`';
 312             $qsearch = '(^|[ \\-])'.$q;
 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             $qsearch = '(^|[ \\-])'.$q;
 327             $beginwith = false;
 328             break;
 329           case 'poste':
 330             $db = '`entreprises`';
 331             $field = '`poste`';
 332             $unique='`uid`';
 333             break;
 334           case 'schoolTxt':
 335             $db = '`applis_def` INNER JOIN
 336                    `applis_ins` ON(`applis_def`.`id` = `applis_ins`.`aid`)';
 337             $field='`applis_def`.`text`';
 338             $unique = '`uid`';
 339             $realid = '`applis_def`.`id`';
 340             if (strlen($q) > 2)
 341                 $beginwith = false;
 342             break;
 343           case 'secteurTxt':
 344             $db = '`emploi_secteur` INNER JOIN
 345                    `entreprises` ON(`entreprises`.`secteur` = `emploi_secteur`.`id`)';
 346             $field = '`emploi_secteur`.`label`';
 347             $realid = '`emploi_secteur`.`id`';
 348             $unique = '`uid`';
 349             $beginwith = false;
 350             break;
 351           case 'sectionTxt':
 352             $db = '`sections` INNER JOIN
 353                    `auth_user_md5` ON(`auth_user_md5`.`section` = `sections`.`id`)';
 354             $field = '`sections`.`text`';
 355             $realid = '`sections`.`id`';
 356             $beginwith = false;
 357             break;
 358           default: exit();
 359         }
 360
 361         $field_select = $field;
 362         if ($field2) {
 363             $field_select = 'IF('.$field.' REGEXP {?}, '.$field.', '.$field2.')';
 364         }
 365
 366         if ($beginwith) {
 367             $qsearch = '^'.$qsearch;
 368         }
 369         $list = XDB::iterator('SELECT  '.$field_select.' AS field,
 370                                        COUNT(DISTINCT '.$unique.') AS nb
 371                                        '.($realid?(', '.$realid.' AS id'):'').'
 372                                  FROM  '.$db.'
 373                                 WHERE  '.$field.' REGEXP {?}'.
 374                                         ($field2?(' OR '.$field2.' REGEXP {?}'):'').'
 375                              GROUP BY  '.$field_select.'
 376                              ORDER BY  nb DESC
 377                                 LIMIT  11',
 378                                $qsearch, $qsearch, $qsearch, $qsearch);
 379         $nbResults = 0;
 380         $res = "";
 381         while ($result = $list->next()) {
 382             $nbResults++;
 383             if ($nbResults == 11) {
 384                 $res .= '...|1'."\n";
 385             } else {
 386                 $res .= $result['field'].'|';
 387                 $res .= $result['nb'];
 388                 if (isset($result['id'])) {
 389                     $res  .= '|'.$result['id'];
 390                 }
 391                 $res .= "\n";
 392             }
 393         }
 394         XDB::query('REPLACE INTO  `search_autocomplete`
 395                           VALUES  ({?}, {?}, {?}, NOW())',
 396                     $type, $q, $res);
 397         echo $res;
 398         exit();
 399     }
 400
 401     function handler_list(&$page, $type = null, $idVal = null)
 402     {
 403         // Give the list of all values possible of type and builds a select input for it
 404         $field = '`text`';
 405         $id = '`id`';
 406         $where = '';
 407
 408         switch ($type) {
 409           case 'binet':
 410             $db = '`binets_def`';
 411             break;
 412           case 'country':
 413             $db = '`geoloc_pays`';
 414             $field = '`pays`';
 415             $id = '`a2`';
 416             $page->assign('onchange', 'changeCountry(this.value)');
 417             break;
 418           case 'fonction':
 419             $db = '`fonctions_def`';
 420             $field = '`fonction_fr`';
 421             break;
 422           case 'diploma':
 423             header('Content-Type: text/xml; charset="UTF-8"');
 424             $this->get_diplomas();
 425             $page->changeTpl('search/adv.grade.form.tpl', NO_SKIN);
 426             return;
 427           case 'groupex':
 428             $db = 'groupex.asso';
 429             $where = " WHERE (cat = 'GroupesX' OR cat = 'Institutions') AND pub = 'public'";
 430             $field = 'nom';
 431             break;
 432           case 'nationalite':
 433             $db = '`geoloc_pays` INNER JOIN
 434                    `auth_user_md5` ON (`geoloc_pays`.`a2` = `auth_user_md5`.`nationalite`)';
 435             $field = 'IF(`nat`=\'\', `pays`, `nat`)';
 436             $id = '`a2`';
 437             break;
 438           case 'region':
 439             $db = '`geoloc_region`';
 440             $field = '`name`';
 441             $id = '`region`';
 442             if (isset($_REQUEST['country'])) {
 443                 $where .= ' WHERE `a2` = "'.$_REQUEST['country'].'"';
 444             }
 445             break;
 446           case 'school':
 447             $db = '`applis_def`';
 448             $page->assign('onchange', 'changeSchool(this.value)');
 449             break;
 450           case 'section':
 451             $db = '`sections`';
 452             break;
 453           case 'secteur':
 454             $db = '`emploi_secteur`';
 455             $field = '`label`';
 456             break;
 457           default: exit();
 458         }
 459         if (isset($idVal)) {
 460             header('Content-Type: text/plain; charset="UTF-8"');
 461             $result = XDB::query('SELECT '.$field.' AS field FROM '.$db.' WHERE '.$id.' = {?} LIMIT 1',$idVal);
 462             echo $result->fetchOneCell();
 463             exit();
 464         }
 465         header('Content-Type: text/xml; charset="UTF-8"');
 466         $page->changeTpl('include/field.select.tpl', NO_SKIN);
 467         $page->assign('name', $type);
 468         $page->assign('list', XDB::iterator('SELECT  '.$field.' AS field,
 469                                                      '.$id.' AS id
 470                                                FROM  '.$db.$where.'
 471                                            GROUP BY  '.$field.'
 472                                            ORDER BY  '.$field));
 473         $page->assign('with_text_value', true);
 474         $page->assign('onchange', "document.forms.recherche.{$type}Txt.value = this.options[this.selectedIndex].text");
 475     }
 476 }
 477
 478 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 479 ?>