classes/SearchEngines.php

   1 <?php
   2 /*
   3  * Laconica - a distributed open-source microblogging tool
   4  * Copyright (C) 2008, Controlez-Vous, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 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 Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 if (!defined('LACONICA')) { exit(1); }
  21
  22 class SearchEngine {
  23     protected $profile;
  24
  25     function __construct($profile) {
  26         $this->profile = $profile;
  27     }
  28
  29     function query($q) {
  30     }
  31
  32     function limit($offset, $count) {
  33         return $this->profile->limit($offset, $count);
  34     }
  35 }
  36
  37 class SphinxSearch extends SearchEngine {
  38     private $sphinx;
  39
  40     function __construct($profile) {
  41         parent::__construct($profile);
  42         $this->sphinx = new SphinxClient;
  43         $this->sphinx->setServer(common_config('sphinx', 'server'), common_config('sphinx', 'port'));
  44     }
  45
  46     function limit($offset, $count) {
  47         $this->sphinx->setLimits($offset, $count);
  48         $this->profile->limit($offset, $count);
  49     }
  50
  51     function query($q) {
  52         $result = $this->sphinx->query($q);
  53         if (!isset($result['matches'])) return false;
  54         $id_set = join(', ', array_keys($result['matches']));
  55         return $this->profile->whereAdd("id in ($id_set)");
  56      }
  57 }
  58
  59 class MySQLSearch extends SearchEngine {
  60     function query($q) {
  61         return $this->profile->whereAdd('MATCH(nickname, fullname, location, bio, homepage) ' .
  62                                                    'against (\''.addslashes($q).'\')');
  63     }
  64 }
  65
  66 class PGSearch extends SearchEngine {
  67     function query($q) {
  68         $this->profile->whereAdd('textsearch @@ plainto_tsquery(\''.addslashes($q).'\')');
  69     }
  70 }
  71