]> git.mxchange.org Git - quix0rs-gnu-social.git/blob - lib/search_engines.php
use none instead of unset
[quix0rs-gnu-social.git] / lib / search_engines.php
1 <?php
2 /*
3  * StatusNet - the distributed open-source microblogging tool
4  * Copyright (C) 2008, 2009, StatusNet, 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('STATUSNET') && !defined('LACONICA')) { exit(1); }
21
22 class SearchEngine
23 {
24     protected $target;
25     protected $table;
26
27     function __construct($target, $table)
28     {
29         $this->target = $target;
30         $this->table = $table;
31     }
32
33     function query($q)
34     {
35     }
36
37     function limit($offset, $count, $rss = false)
38     {
39         return $this->target->limit($offset, $count);
40     }
41
42     function set_sort_mode($mode)
43     {
44         switch ($mode) {
45         case 'chron':
46             return $this->target->orderBy('created DESC');
47             break;
48         case 'reverse_chron':
49             return $this->target->orderBy('created ASC');
50             break;
51         case 'nickname_desc':
52             if ($this->table != 'profile') {
53                 throw new Exception(
54                     'nickname_desc sort mode can only be use when searching profile.'
55                 );
56             } else {
57                 return $this->target->orderBy('nickname DESC');
58             }
59             break;
60         case 'nickname_asc':
61             if ($this->table != 'profile') {
62                 throw new Exception(
63                     'nickname_desc sort mode can only be use when searching profile.'
64                 );
65             } else {
66                 return $this->target->orderBy('nickname ASC');
67             }
68             break;
69         default:
70             return $this->target->orderBy('created DESC');
71             break;
72         }
73     }
74 }
75
76 class MySQLSearch extends SearchEngine
77 {
78     function query($q)
79     {
80         if ('profile' === $this->table) {
81             $this->target->whereAdd('MATCH(nickname, fullname, location, bio, homepage) ' .
82                                     'AGAINST (\''.$this->target->escape($q).'\' IN BOOLEAN MODE)');
83             if (strtolower($q) != $q) {
84                 $this->target->whereAdd('MATCH(nickname, fullname, location, bio, homepage) ' .
85                                         'AGAINST (\''.$this->target->escape(strtolower($q)).'\' IN BOOLEAN MODE)', 'OR');
86             }
87             return true;
88         } else if ('notice' === $this->table) {
89
90             // Don't show imported notices
91             $this->target->whereAdd('notice.is_local != ' . Notice::GATEWAY);
92
93             if (strtolower($q) != $q) {
94                 $this->target->whereAdd("( MATCH(content) AGAINST ('" . $this->target->escape($q) .
95                     "' IN BOOLEAN MODE)) OR ( MATCH(content) " .
96                     "AGAINST ('"  . $this->target->escape(strtolower($q)) .
97                     "' IN BOOLEAN MODE))");
98             } else {
99                 $this->target->whereAdd('MATCH(content) ' .
100                                          'AGAINST (\''.$this->target->escape($q).'\' IN BOOLEAN MODE)');
101             }
102
103             return true;
104         } else {
105             throw new ServerException('Unknown table: ' . $this->table);
106         }
107     }
108 }
109
110 class MySQLLikeSearch extends SearchEngine
111 {
112     function query($q)
113     {
114         if ('profile' === $this->table) {
115             $qry = sprintf('(nickname LIKE "%%%1$s%%" OR '.
116                            ' fullname LIKE "%%%1$s%%" OR '.
117                            ' location LIKE "%%%1$s%%" OR '.
118                            ' bio      LIKE "%%%1$s%%" OR '.
119                            ' homepage LIKE "%%%1$s%%")', $this->target->escape($q, true));
120         } else if ('notice' === $this->table) {
121             $qry = sprintf('content LIKE "%%%1$s%%"', $this->target->escape($q, true));
122         } else {
123             throw new ServerException('Unknown table: ' . $this->table);
124         }
125
126         $this->target->whereAdd($qry);
127
128         return true;
129     }
130 }
131
132 class PGSearch extends SearchEngine
133 {
134     function query($q)
135     {
136         if ('profile' === $this->table) {
137             return $this->target->whereAdd('textsearch @@ plainto_tsquery(\''.$this->target->escape($q).'\')');
138         } else if ('notice' === $this->table) {
139
140             // XXX: We need to filter out gateway notices (notice.is_local = -2) --Zach
141
142             return $this->target->whereAdd('to_tsvector(\'english\', content) @@ plainto_tsquery(\''.$this->target->escape($q).'\')');
143         } else {
144             throw new ServerException('Unknown table: ' . $this->table);
145         }
146     }
147 }
148