123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185 |
- <?php
- defined('GNUSOCIAL') || die();
- class SearchEngine
- {
- protected $target;
- protected $table;
- public function __construct($target, $table)
- {
- $this->target = $target;
- $this->table = $table;
- }
- public function query($q)
- {
- }
- public function limit($offset, $count, $rss = false)
- {
- return $this->target->limit($offset, $count);
- }
- public function set_sort_mode($mode)
- {
- switch ($mode) {
- case 'chron':
- return $this->target->orderBy('created DESC');
- break;
- case 'reverse_chron':
- return $this->target->orderBy('created ASC');
- break;
- case 'nickname_desc':
- if ($this->table != 'profile') {
- throw new Exception(
- 'nickname_desc sort mode can only be use when searching profile.'
- );
- } else {
- return $this->target->orderBy(sprintf('%1$s.nickname DESC', $this->table));
- }
- break;
- case 'nickname_asc':
- if ($this->table != 'profile') {
- throw new Exception(
- 'nickname_desc sort mode can only be use when searching profile.'
- );
- } else {
- return $this->target->orderBy(sprintf('%1$s.nickname ASC', $this->table));
- }
- break;
- default:
- return $this->target->orderBy('created DESC');
- break;
- }
- }
- }
- class PostgreSQLSearch extends SearchEngine
- {
- public function query($q)
- {
- if ($this->table === 'profile') {
- $cols = implode(" || ' ' || ", array_map(
- function ($col) {
- return sprintf(
- "COALESCE(%s.%s, '')",
- common_database_tablename($this->table),
- $col
- );
- },
- ['nickname', 'fullname', 'location', 'bio', 'homepage']
- ));
- $this->target->whereAdd(sprintf(
- 'to_tsvector(\'english\', %2$s) @@ plainto_tsquery(\'%1$s\')',
- $this->target->escape($q, true),
- $cols
- ));
- return true;
- } elseif ($this->table === 'notice') {
-
- $this->target->whereAdd('notice.is_local <> ' . Notice::GATEWAY);
- $this->target->whereAdd(sprintf(
- 'to_tsvector(\'english\', content) @@ plainto_tsquery(\'%1$s\')',
- $this->target->escape($q, true)
- ));
- return true;
- } else {
- throw new ServerException('Unknown table: ' . $this->table);
- }
- }
- }
- class MySQLSearch extends SearchEngine
- {
- public function query($q)
- {
- if ($this->table === 'profile') {
- $this->target->whereAdd(sprintf(
- 'MATCH (%2$s.nickname, %2$s.fullname, %2$s.location, %2$s.bio, %2$s.homepage) ' .
- 'AGAINST (\'%1$s\' IN BOOLEAN MODE)',
- $this->target->escape($q, true),
- $this->table
- ));
- if (strtolower($q) != $q) {
- $this->target->whereAdd(
- sprintf(
- 'MATCH (%2$s.nickname, %2$s.fullname, %2$s.location, %2$s.bio, %2$s.homepage) ' .
- 'AGAINST (\'%1$s\' IN BOOLEAN MODE)',
- $this->target->escape(strtolower($q), true),
- $this->table
- ),
- 'OR'
- );
- }
- return true;
- } elseif ($this->table === 'notice') {
-
- $this->target->whereAdd('notice.is_local <> ' . Notice::GATEWAY);
- $this->target->whereAdd(sprintf(
- 'MATCH (%2$s.content) AGAINST (\'%1$s\' IN BOOLEAN MODE)',
- $this->target->escape($q, true),
- $this->table
- ));
- if (strtolower($q) != $q) {
- $this->target->whereAdd(
- sprintf(
- 'MATCH (%2$s.content) AGAINST (\'%1$s\' IN BOOLEAN MODE)',
- $this->target->escape(strtolower($q), true),
- $this->table
- ),
- 'OR'
- );
- }
- return true;
- } else {
- throw new ServerException('Unknown table: ' . $this->table);
- }
- }
- }
- class SQLLikeSearch extends SearchEngine
- {
- public function query($q)
- {
- if ($this->table === 'profile') {
- $qry = sprintf(
- '( %2$s.nickname LIKE \'%%%1$s%%\' ' .
- ' OR %2$s.fullname LIKE \'%%%1$s%%\' ' .
- ' OR %2$s.location LIKE \'%%%1$s%%\' ' .
- ' OR %2$s.bio LIKE \'%%%1$s%%\' ' .
- ' OR %2$s.homepage LIKE \'%%%1$s%%\')',
- $this->target->escape($q, true),
- $this->table
- );
- } elseif ($this->table === 'notice') {
- $qry = sprintf('content LIKE \'%%%1$s%%\'', $this->target->escape($q, true));
- } else {
- throw new ServerException('Unknown table: ' . $this->table);
- }
- $this->target->whereAdd($qry);
- return true;
- }
- }
|