Register
phpBB.com Wiki · Home Projects Help

root / trunk / phpBB / includes / search / search.php

1 5441 naderman
<?php
2 8147 acydburn
/**
3 5441 naderman
*
4 5441 naderman
* @package search
5 5441 naderman
* @version $Id$
6 8147 acydburn
* @copyright (c) 2005 phpBB Group
7 8147 acydburn
* @license http://opensource.org/licenses/gpl-license.php GNU Public License
8 5441 naderman
*
9 5441 naderman
*/
10 5441 naderman
11 5441 naderman
/**
12 8147 acydburn
* @ignore
13 5670 acydburn
*/
14 5670 acydburn
if (!defined('IN_PHPBB'))
15 5670 acydburn
{
16 5670 acydburn
	exit;
17 5670 acydburn
}
18 5670 acydburn
19 5670 acydburn
/**
20 5441 naderman
* search_backend
21 5441 naderman
* optional base class for search plugins providing simple caching based on ACM
22 5441 naderman
* and functions to retrieve ignore_words and synonyms
23 6058 acydburn
* @package search
24 5441 naderman
*/
25 5441 naderman
class search_backend
26 5441 naderman
{
27 8295 davidmj
	const SEARCH_RESULT_NOT_IN_CACHE = 0;
28 8295 davidmj
	const SEARCH_RESULT_IN_CACHE = 1;
29 8295 davidmj
	const SEARCH_RESULT_INCOMPLETE = 2;
30 5441 naderman
31 8295 davidmj
	public $ignore_words = array();
32 8295 davidmj
	public $match_synonym = array();
33 8295 davidmj
	public $replace_synonym = array();
34 8295 davidmj
35 8295 davidmj
	function __construct(&$error)
36 5441 naderman
	{
37 5441 naderman
		// This class cannot be used as a search plugin
38 5441 naderman
		$error = true;
39 5441 naderman
	}
40 5441 naderman
41 5441 naderman
	/**
42 5981 naderman
	* Retrieves a language dependend list of words that should be ignored by the search
43 5441 naderman
	*/
44 8295 davidmj
	public function get_ignore_words()
45 5441 naderman
	{
46 5441 naderman
		if (!sizeof($this->ignore_words))
47 5441 naderman
		{
48 8572 acydburn
			global $user;
49 5441 naderman
50 5981 naderman
			$words = array();
51 5441 naderman
52 8786 acydburn
			if (file_exists("{$user->lang_path}{$user->lang_name}/search_ignore_words." . PHP_EXT))
53 5983 naderman
			{
54 5983 naderman
				// include the file containing ignore words
55 8786 acydburn
				include("{$user->lang_path}{$user->lang_name}/search_ignore_words." . PHP_EXT);
56 5983 naderman
			}
57 5441 naderman
58 5981 naderman
			$this->ignore_words = $words;
59 5981 naderman
			unset($words);
60 5441 naderman
		}
61 5441 naderman
	}
62 5441 naderman
63 5441 naderman
	/**
64 5441 naderman
	* Stores a list of synonyms that should be replaced in $this->match_synonym and $this->replace_synonym and caches them
65 5441 naderman
	*/
66 8295 davidmj
	public function get_synonyms()
67 5441 naderman
	{
68 5441 naderman
		if (!sizeof($this->match_synonym))
69 5441 naderman
		{
70 8572 acydburn
			global $user;
71 5441 naderman
72 5981 naderman
			$synonyms = array();
73 5441 naderman
74 8786 acydburn
			if (file_exists("{$user->lang_path}{$user->lang_name}/search_synonyms." . PHP_EXT))
75 5983 naderman
			{
76 5983 naderman
				// include the file containing synonyms
77 8786 acydburn
				include("{$user->lang_path}{$user->lang_name}/search_synonyms." . PHP_EXT);
78 5983 naderman
			}
79 5441 naderman
80 5981 naderman
			$this->match_synonym = array_keys($synonyms);
81 5981 naderman
			$this->replace_synonym = array_values($synonyms);
82 5441 naderman
83 5981 naderman
			unset($synonyms);
84 5441 naderman
		}
85 5441 naderman
	}
86 5441 naderman
87 5441 naderman
	/**
88 5441 naderman
	* Retrieves cached search results
89 5441 naderman
	*
90 6595 acydburn
	* @param int &$result_count will contain the number of all results for the search (not only for the current page)
91 6595 acydburn
	* @param array &$id_ary is filled with the ids belonging to the requested page that are stored in the cache
92 5441 naderman
	*
93 6007 naderman
	* @return int SEARCH_RESULT_NOT_IN_CACHE or SEARCH_RESULT_IN_CACHE or SEARCH_RESULT_INCOMPLETE
94 5441 naderman
	*/
95 8295 davidmj
	protected function obtain_ids($search_key, &$result_count, &$id_ary, $start, $per_page, $sort_dir)
96 5441 naderman
	{
97 5441 naderman
		global $cache;
98 5441 naderman
99 5441 naderman
		if (!($stored_ids = $cache->get('_search_results_' . $search_key)))
100 5441 naderman
		{
101 5441 naderman
			// no search results cached for this search_key
102 8295 davidmj
			return self::SEARCH_RESULT_NOT_IN_CACHE;
103 5441 naderman
		}
104 5441 naderman
		else
105 5441 naderman
		{
106 5441 naderman
			$result_count = $stored_ids[-1];
107 5441 naderman
			$reverse_ids = ($stored_ids[-2] != $sort_dir) ? true : false;
108 5441 naderman
			$complete = true;
109 5441 naderman
110 5441 naderman
			// change the start to the actual end of the current request if the sort direction differs
111 5441 naderman
			// from the dirction in the cache and reverse the ids later
112 5441 naderman
			if ($reverse_ids)
113 5441 naderman
			{
114 5441 naderman
				$start = $result_count - $start - $per_page;
115 5441 naderman
116 5441 naderman
				// the user requested a page past the last index
117 5441 naderman
				if ($start < 0)
118 5441 naderman
				{
119 8295 davidmj
					return self::SEARCH_RESULT_NOT_IN_CACHE;
120 5441 naderman
				}
121 5441 naderman
			}
122 5441 naderman
123 5441 naderman
			for ($i = $start, $n = $start + $per_page; ($i < $n) && ($i < $result_count); $i++)
124 5441 naderman
			{
125 5441 naderman
				if (!isset($stored_ids[$i]))
126 5441 naderman
				{
127 5441 naderman
					$complete = false;
128 5441 naderman
				}
129 5441 naderman
				else
130 5441 naderman
				{
131 5441 naderman
					$id_ary[] = $stored_ids[$i];
132 5441 naderman
				}
133 5441 naderman
			}
134 5441 naderman
			unset($stored_ids);
135 5441 naderman
136 5441 naderman
			if ($reverse_ids)
137 5441 naderman
			{
138 5441 naderman
				$id_ary = array_reverse($id_ary);
139 5441 naderman
			}
140 5441 naderman
141 5441 naderman
			if (!$complete)
142 5441 naderman
			{
143 8295 davidmj
				return self::SEARCH_RESULT_INCOMPLETE;
144 5441 naderman
			}
145 8295 davidmj
			return self::SEARCH_RESULT_IN_CACHE;
146 5441 naderman
		}
147 5441 naderman
	}
148 5441 naderman
149 5441 naderman
	/**
150 5441 naderman
	* Caches post/topic ids
151 5441 naderman
	*
152 6595 acydburn
	* @param array &$id_ary contains a list of post or topic ids that shall be cached, the first element
153 5441 naderman
	* 	must have the absolute index $start in the result set.
154 5441 naderman
	*/
155 8295 davidmj
	protected function save_ids($search_key, $keywords, $author_ary, $result_count, &$id_ary, $start, $sort_dir)
156 5441 naderman
	{
157 5981 naderman
		global $cache, $config, $db, $user;
158 5441 naderman
159 5441 naderman
		$length = min(sizeof($id_ary), $config['search_block_size']);
160 5441 naderman
161 5482 naderman
		// nothing to cache so exit
162 5482 naderman
		if (!$length)
163 5482 naderman
		{
164 5482 naderman
			return;
165 5482 naderman
		}
166 5482 naderman
167 5441 naderman
		$store_ids = array_slice($id_ary, 0, $length);
168 5441 naderman
169 5441 naderman
		// create a new resultset if there is none for this search_key yet
170 5441 naderman
		// or add the ids to the existing resultset
171 5441 naderman
		if (!($store = $cache->get('_search_results_' . $search_key)))
172 5441 naderman
		{
173 5441 naderman
			// add the current keywords to the recent searches in the cache which are listed on the search page
174 5441 naderman
			if (!empty($keywords) || sizeof($author_ary))
175 5441 naderman
			{
176 5441 naderman
				$sql = 'SELECT search_time
177 6021 acydburn
					FROM ' . SEARCH_RESULTS_TABLE . '
178 5441 naderman
					WHERE search_key = \'' . $db->sql_escape($search_key) . '\'';
179 5441 naderman
				$result = $db->sql_query($sql);
180 5441 naderman
181 5441 naderman
				if (!$db->sql_fetchrow($result))
182 5441 naderman
				{
183 5441 naderman
					$sql_ary = array(
184 5441 naderman
						'search_key'		=> $search_key,
185 5441 naderman
						'search_time'		=> time(),
186 5441 naderman
						'search_keywords'	=> $keywords,
187 5482 naderman
						'search_authors'	=> ' ' . implode(' ', $author_ary) . ' '
188 5441 naderman
					);
189 5441 naderman
190 6021 acydburn
					$sql = 'INSERT INTO ' . SEARCH_RESULTS_TABLE . ' ' . $db->sql_build_array('INSERT', $sql_ary);
191 5441 naderman
					$db->sql_query($sql);
192 5441 naderman
				}
193 5441 naderman
				$db->sql_freeresult($result);
194 5441 naderman
			}
195 6117 acydburn
196 5981 naderman
			$sql = 'UPDATE ' . USERS_TABLE . '
197 5981 naderman
				SET user_last_search = ' . time() . '
198 5981 naderman
				WHERE user_id = ' . $user->data['user_id'];
199 5981 naderman
			$db->sql_query($sql);
200 5441 naderman
201 5441 naderman
			$store = array(-1 => $result_count, -2 => $sort_dir);
202 5441 naderman
			$id_range = range($start, $start + $length - 1);
203 5441 naderman
		}
204 5441 naderman
		else
205 5441 naderman
		{
206 5607 naderman
			// we use one set of results for both sort directions so we have to calculate the indizes
207 5441 naderman
			// for the reversed array and we also have to reverse the ids themselves
208 5441 naderman
			if ($store[-2] != $sort_dir)
209 5441 naderman
			{
210 5441 naderman
				$store_ids = array_reverse($store_ids);
211 5441 naderman
				$id_range = range($store[-1] - $start - $length, $store[-1] - $start - 1);
212 5441 naderman
			}
213 5441 naderman
			else
214 5441 naderman
			{
215 5441 naderman
				$id_range = range($start, $start + $length - 1);
216 5441 naderman
			}
217 5441 naderman
		}
218 5441 naderman
219 5482 naderman
		$store_ids = array_combine($id_range, $store_ids);
220 5482 naderman
221 5441 naderman
		// append the ids
222 5482 naderman
		if (is_array($store_ids))
223 5482 naderman
		{
224 5482 naderman
			$store += $store_ids;
225 5607 naderman
226 5607 naderman
			// if the cache is too big
227 5607 naderman
			if (sizeof($store) - 2 > 20 * $config['search_block_size'])
228 5607 naderman
			{
229 5607 naderman
				// remove everything in front of two blocks in front of the current start index
230 5607 naderman
				for ($i = 0, $n = $id_range[0] - 2 * $config['search_block_size']; $i < $n; $i++)
231 5607 naderman
				{
232 5607 naderman
					if (isset($store[$i]))
233 5607 naderman
					{
234 5607 naderman
						unset($store[$i]);
235 5607 naderman
					}
236 5607 naderman
				}
237 5607 naderman
238 5607 naderman
				// remove everything after two blocks after the current stop index
239 5607 naderman
				end($id_range);
240 5607 naderman
				for ($i = $store[-1] - 1, $n = current($id_range) + 2 * $config['search_block_size']; $i > $n; $i--)
241 5607 naderman
				{
242 5607 naderman
					if (isset($store[$i]))
243 5607 naderman
					{
244 5607 naderman
						unset($store[$i]);
245 5607 naderman
					}
246 5607 naderman
				}
247 5607 naderman
			}
248 5482 naderman
			$cache->put('_search_results_' . $search_key, $store, $config['search_store_results']);
249 5441 naderman
250 8147 acydburn
			$sql = 'UPDATE ' . SEARCH_RESULTS_TABLE . '
251 5482 naderman
				SET search_time = ' . time() . '
252 5482 naderman
				WHERE search_key = \'' . $db->sql_escape($search_key) . '\'';
253 5482 naderman
			$db->sql_query($sql);
254 5482 naderman
		}
255 5482 naderman
256 5441 naderman
		unset($store);
257 5441 naderman
		unset($store_ids);
258 5441 naderman
		unset($id_range);
259 5441 naderman
	}
260 5441 naderman
261 5441 naderman
	/**
262 5441 naderman
	* Removes old entries from the search results table and removes searches with keywords that contain a word in $words.
263 5441 naderman
	*/
264 8295 davidmj
	public function destroy_cache($words, $authors = false)
265 5441 naderman
	{
266 5441 naderman
		global $db, $cache, $config;
267 5441 naderman
268 5482 naderman
		// clear all searches that searched for the specified words
269 5441 naderman
		if (sizeof($words))
270 5441 naderman
		{
271 5441 naderman
			$sql_where = '';
272 5441 naderman
			foreach ($words as $word)
273 5441 naderman
			{
274 7789 acydburn
				$sql_where .= " OR search_keywords " . $db->sql_like_expression($db->any_char . $word . $db->any_char);
275 5441 naderman
			}
276 5441 naderman
277 5441 naderman
			$sql = 'SELECT search_key
278 6021 acydburn
				FROM ' . SEARCH_RESULTS_TABLE . "
279 5441 naderman
				WHERE search_keywords LIKE '%*%' $sql_where";
280 5441 naderman
			$result = $db->sql_query($sql);
281 5441 naderman
282 5441 naderman
			while ($row = $db->sql_fetchrow($result))
283 5441 naderman
			{
284 5441 naderman
				$cache->destroy('_search_results_' . $row['search_key']);
285 5441 naderman
			}
286 6345 acydburn
			$db->sql_freeresult($result);
287 5441 naderman
		}
288 5441 naderman
289 5482 naderman
		// clear all searches that searched for the specified authors
290 5482 naderman
		if (is_array($authors) && sizeof($authors))
291 5482 naderman
		{
292 5482 naderman
			$sql_where = '';
293 5482 naderman
			foreach ($authors as $author)
294 5482 naderman
			{
295 5482 naderman
				$sql_where .= (($sql_where) ? ' OR ' : '') . 'search_authors LIKE \'% ' . (int) $author . ' %\'';
296 5482 naderman
			}
297 5482 naderman
298 5482 naderman
			$sql = 'SELECT search_key
299 6021 acydburn
				FROM ' . SEARCH_RESULTS_TABLE . "
300 5482 naderman
				WHERE $sql_where";
301 5482 naderman
			$result = $db->sql_query($sql);
302 5482 naderman
303 5482 naderman
			while ($row = $db->sql_fetchrow($result))
304 5482 naderman
			{
305 5482 naderman
				$cache->destroy('_search_results_' . $row['search_key']);
306 5482 naderman
			}
307 6345 acydburn
			$db->sql_freeresult($result);
308 5482 naderman
		}
309 5482 naderman
310 5441 naderman
		$sql = 'DELETE
311 6021 acydburn
			FROM ' . SEARCH_RESULTS_TABLE . '
312 5441 naderman
			WHERE search_time < ' . (time() - $config['search_store_results']);
313 5441 naderman
		$db->sql_query($sql);
314 5441 naderman
	}
315 5441 naderman
}
316 5441 naderman
317 5441 naderman
?>