2013-04-04 09:26:00 +00:00
|
|
|
<?php defined("SYSPATH") or die("No direct script access.");
|
|
|
|
/**
|
|
|
|
* Gallery - a web based photo album viewer and editor
|
2013-04-04 09:26:06 +00:00
|
|
|
* Copyright (C) 2000-2013 Bharat Mediratta
|
2013-04-04 09:26:00 +00:00
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or (at
|
|
|
|
* your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful, but
|
|
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
*/
|
|
|
|
class search_Core {
|
2013-04-04 09:26:02 +00:00
|
|
|
/**
|
|
|
|
* Add more terms to the query by wildcarding the stem value of the first
|
|
|
|
* few terms in the query.
|
|
|
|
*/
|
|
|
|
static function add_query_terms($q) {
|
|
|
|
$MAX_TERMS = 5;
|
|
|
|
$terms = explode(" ", $q, $MAX_TERMS);
|
|
|
|
for ($i = 0; $i < min(count($terms), $MAX_TERMS - 1); $i++) {
|
|
|
|
// Don't wildcard quoted or already wildcarded terms
|
|
|
|
if ((substr($terms[$i], 0, 1) != '"') && (substr($terms[$i], -1, 1) != "*")) {
|
|
|
|
$terms[] = rtrim($terms[$i], "s") . "*";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return implode(" ", $terms);
|
|
|
|
}
|
|
|
|
|
2013-04-04 09:26:00 +00:00
|
|
|
static function search($q, $limit, $offset) {
|
2013-04-04 09:26:06 +00:00
|
|
|
return search::search_within_album($q, item::root(), $limit, $offset);
|
|
|
|
}
|
|
|
|
|
|
|
|
static function search_within_album($q, $album, $limit, $offset) {
|
2013-04-04 09:26:00 +00:00
|
|
|
$db = Database::instance();
|
|
|
|
|
2013-04-04 09:26:06 +00:00
|
|
|
$query = self::_build_query_base($q, $album) .
|
2013-04-04 10:25:58 +00:00
|
|
|
" ORDER BY `score` DESC" .
|
|
|
|
" LIMIT $limit OFFSET " . (int)$offset;
|
2013-04-04 09:26:04 +00:00
|
|
|
|
|
|
|
$data = $db->query($query);
|
|
|
|
$count = $db->query("SELECT FOUND_ROWS() as c")->current()->c;
|
|
|
|
|
|
|
|
return array($count, new ORM_Iterator(ORM::factory("item"), $data));
|
|
|
|
}
|
|
|
|
|
2013-04-04 09:26:06 +00:00
|
|
|
private static function _build_query_base($q, $album, $where=array()) {
|
|
|
|
$db = Database::instance();
|
|
|
|
$q = $db->escape($q);
|
|
|
|
|
2013-04-04 09:26:00 +00:00
|
|
|
if (!identity::active_user()->admin) {
|
|
|
|
foreach (identity::group_ids_for_active_user() as $id) {
|
|
|
|
$fields[] = "`view_$id` = TRUE"; // access::ALLOW
|
|
|
|
}
|
2013-04-04 09:26:04 +00:00
|
|
|
$access_sql = " AND (" . join(" OR ", $fields) . ")";
|
2013-04-04 09:26:00 +00:00
|
|
|
} else {
|
|
|
|
$access_sql = "";
|
|
|
|
}
|
|
|
|
|
2013-04-04 09:26:06 +00:00
|
|
|
if ($album->id == item::root()->id) {
|
|
|
|
$album_sql = "";
|
|
|
|
} else {
|
|
|
|
$album_sql =
|
2013-04-04 10:25:58 +00:00
|
|
|
" AND {items}.left_ptr > " . $db->escape($album->left_ptr) .
|
2013-04-04 09:26:06 +00:00
|
|
|
" AND {items}.right_ptr <= " . $db->escape($album->right_ptr);
|
|
|
|
}
|
|
|
|
|
2013-04-04 09:26:04 +00:00
|
|
|
return
|
2013-04-04 09:26:00 +00:00
|
|
|
"SELECT SQL_CALC_FOUND_ROWS {items}.*, " .
|
|
|
|
" MATCH({search_records}.`data`) AGAINST ('$q') AS `score` " .
|
|
|
|
"FROM {items} JOIN {search_records} ON ({items}.`id` = {search_records}.`item_id`) " .
|
2013-04-04 10:25:58 +00:00
|
|
|
"WHERE MATCH({search_records}.`data`) AGAINST ('$q' IN BOOLEAN MODE)" .
|
2013-04-04 09:26:06 +00:00
|
|
|
$album_sql .
|
2013-04-04 09:26:04 +00:00
|
|
|
(empty($where) ? "" : " AND " . join(" AND ", $where)) .
|
2013-04-04 10:25:58 +00:00
|
|
|
$access_sql;
|
2013-04-04 09:26:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @return string An error message suitable for inclusion in the task log
|
|
|
|
*/
|
|
|
|
static function check_index() {
|
|
|
|
list ($remaining) = search::stats();
|
|
|
|
if ($remaining) {
|
|
|
|
site_status::warning(
|
|
|
|
t('Your search index needs to be updated. <a href="%url" class="g-dialog-link">Fix this now</a>',
|
|
|
|
array("url" => html::mark_clean(url::site("admin/maintenance/start/search_task::update_index?csrf=__CSRF__")))),
|
|
|
|
"search_index_out_of_date");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static function update($item) {
|
|
|
|
$data = new ArrayObject();
|
|
|
|
$record = ORM::factory("search_record")->where("item_id", "=", $item->id)->find();
|
|
|
|
if (!$record->loaded()) {
|
|
|
|
$record->item_id = $item->id;
|
|
|
|
}
|
|
|
|
|
|
|
|
$item = $record->item();
|
|
|
|
module::event("item_index_data", $item, $data);
|
|
|
|
$record->data = join(" ", (array)$data);
|
|
|
|
$record->dirty = 0;
|
|
|
|
$record->save();
|
|
|
|
}
|
|
|
|
|
|
|
|
static function stats() {
|
|
|
|
$remaining = db::build()
|
|
|
|
->from("items")
|
|
|
|
->join("search_records", "items.id", "search_records.item_id", "left")
|
|
|
|
->and_open()
|
|
|
|
->where("search_records.item_id", "IS", null)
|
|
|
|
->or_where("search_records.dirty", "=", 1)
|
|
|
|
->close()
|
|
|
|
->count_records();
|
|
|
|
|
|
|
|
$total = ORM::factory("item")->count_all();
|
|
|
|
$percent = round(100 * ($total - $remaining) / $total);
|
|
|
|
|
|
|
|
return array($remaining, $total, $percent);
|
|
|
|
}
|
2013-04-04 09:26:04 +00:00
|
|
|
|
|
|
|
static function get_position($item, $q) {
|
2013-04-04 09:26:06 +00:00
|
|
|
return search::get_position_within_album($item, $q, item::root());
|
|
|
|
}
|
|
|
|
|
|
|
|
static function get_position_within_album($item, $q, $album) {
|
2013-04-04 09:26:04 +00:00
|
|
|
$page_size = module::get_var("gallery", "page_size", 9);
|
2013-04-04 09:26:06 +00:00
|
|
|
$query = self::_build_query_base($q, $album, array("{items}.id = " . $item->id)) .
|
2013-04-04 10:25:58 +00:00
|
|
|
" ORDER BY `score` DESC";
|
2013-04-04 09:26:04 +00:00
|
|
|
$db = Database::instance();
|
|
|
|
|
|
|
|
// Truncate the score by two decimal places as this resolves the issues
|
2013-04-04 09:26:06 +00:00
|
|
|
// that arise due to inexact numeric conversions.
|
|
|
|
$current = $db->query($query)->current();
|
|
|
|
if (!$current) {
|
|
|
|
// We can't find this result in our result set - perhaps we've fallen out of context? Clear
|
|
|
|
// the context and try again.
|
|
|
|
item::clear_display_context_callback();
|
|
|
|
url::redirect(url::current());
|
|
|
|
}
|
|
|
|
$score = $current->score;
|
2013-04-04 09:26:04 +00:00
|
|
|
if (strlen($score) > 7) {
|
|
|
|
$score = substr($score, 0, strlen($score) - 2);
|
|
|
|
}
|
|
|
|
|
2013-04-04 09:26:06 +00:00
|
|
|
// Redo the query but only look for results greater than or equal to our current location
|
|
|
|
// then seek backwards until we find our item.
|
|
|
|
$data = $db->query(self::_build_query_base($q, $album) . " HAVING `score` >= " . $score .
|
2013-04-04 10:25:58 +00:00
|
|
|
" ORDER BY `score` DESC");
|
2013-04-04 09:26:04 +00:00
|
|
|
$data->seek($data->count() - 1);
|
|
|
|
|
|
|
|
while ($data->get("id") != $item->id && $data->prev()->valid()) {
|
|
|
|
}
|
|
|
|
|
|
|
|
return $data->key() + 1;
|
|
|
|
}
|
2013-04-04 09:26:00 +00:00
|
|
|
}
|