First crack at Twitter-like JSON search results for the API

This commit is contained in:
Zach Copley 2009-03-06 13:33:47 -08:00
parent 87e5badcb6
commit b1f337fe01
4 changed files with 357 additions and 28 deletions

View File

@ -32,6 +32,7 @@ if (!defined('LACONICA')) {
}
require_once INSTALLDIR.'/lib/twitterapi.php';
require_once INSTALLDIR.'/lib/jsonsearchresultslist.php';
/**
* Action handler for Twitter-compatible API search
@ -44,25 +45,48 @@ require_once INSTALLDIR.'/lib/twitterapi.php';
* @see TwitterapiAction
*/
class TwitapisearchAction extends TwitterapiAction
class TwitapisearchjsonAction extends TwitterapiAction
{
var $query;
var $lang;
var $rpp;
var $page;
var $since_id;
var $limit;
var $callback;
var $geocode;
/**
* Initialization.
*
* @param array $args Web and URL arguments
*
* @return boolean false if user doesn't exist
* @return boolean true if nothing goes wrong
*/
function prepare($args)
{
parent::prepare($args);
$qeury = $this->trimmed('query');
$this->query = $this->trimmed('q');
$this->lang = $this->trimmed('lang');
$this->rpp = $this->trimmed('rpp');
if (!$this->rpp) {
$this->rpp = 15;
}
if ($this->rpp > 100) {
$this->rpp = 100;
}
$this->page = $this->trimmed('page');
if (!$this->page) {
$this->page = 1;
}
$this->since_id = $this->trimmed('since_id');
$this->geocode = $this->trimmed('geocode');
return true;
}
@ -78,20 +102,48 @@ class TwitapisearchAction extends TwitterapiAction
function handle($args)
{
parent::handle($args);
$this->showResults($this->limit);
$this->showResults();
}
/**
* Show search results
*
* @param int $limit Number of notices to show
*
* @return void
*/
function showResults($limit)
function showResults()
{
$this->serverError(_('API method under construction.'), $code = 501);
// TODO: Support search operators like from: and to:
$notice = new Notice();
// lcase it for comparison
$q = strtolower($this->query);
$search_engine = $notice->getSearchEngine('identica_notices');
$search_engine->set_sort_mode('chron');
$search_engine->limit(($this->page - 1) * $this->rpp, $this->rpp + 1, true);
$search_engine->query($q);
$cnt = $notice->find();
// TODO: since_id, lang, geocode
$results = new JSONSearchResultsList($notice, $q, $this->rpp, $this->page);
$this->init_document('json');
$results->show();
$this->end_document('json');
}
/**
* This is a read-only action
*
* @return boolean true
*/
function isReadOnly()
{
return true;
}
}

View File

@ -0,0 +1,259 @@
<?php
/**
* Laconica, the distributed open-source microblogging tool
*
* widget for displaying a list of notices
*
* PHP version 5
*
* LICENCE: This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* @category Search
* @package Laconica
* @author Zach Copley <zach@controlyourself.ca>
* @copyright 2008-2009 Control Yourself, Inc.
* @license http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
* @link http://laconi.ca/
*/
if (!defined('LACONICA')) {
exit(1);
}
/**
* widget-like class for showing JSON search results
*
* @category Search
* @package Laconica
* @author Zach Copley <zach@controlyourself.ca>
* @license http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
* @link http://laconi.ca/
*
*/
class JSONSearchResultsList
{
protected $notice; // protected attrs invisible to json_encode()
protected $rpp;
// The below attributes are carefully named so the JSON output from
// this obj matches the output from search.twitter.com
var $results;
var $since_id;
var $max_id;
var $refresh_url;
var $results_per_page;
var $completed_in;
var $page;
var $query;
/**
* constructor
*
* @param Notice $notice stream of notices from DB_DataObject
*/
function __construct($notice, $query, $rpp, $page, $since_id = 0)
{
$this->notice = $notice;
$this->query = urlencode($query);
$this->results_per_page = $this->rpp = $rpp;
$this->page = $page;
$this->since_id = $since_id;
$this->results = array();
}
/**
* show the list of search results
*
* @return int count of the search results listed.
*/
function show()
{
$cnt = 0;
$time_start = microtime(true);
while ($this->notice->fetch() && $cnt <= $this->rpp) {
$cnt++;
// XXX: Hmmm. this depends on desc sort order
if (!$this->max_id) {
$this->max_id = (int)$this->notice->id;
}
if ($cnt > $this->rpp) {
break;
}
$item = new ResultItem($this->notice);
array_push($this->results, $item);
}
$time_end = microtime(true);
$this->completed_in = $time_end - $time_start;
// Set other attrs
$this->refresh_url = '?since_id=' . $this->max_id .
'&q=' . $this->query;
// pagination stuff
if ($cnt > $this->rpp) {
$this->next_page = '?page=' . ($this->page + 1) .
'&max_id=' . $this->max_id;
if ($this->rpp != 15) {
$this->next_page .= '&rpp=' . $this->rpp;
}
$this->next_page .= '&q=' . $this->query;
}
if ($this->page > 1) {
$this->previous_page = '?page=' . ($this->page - 1) .
'&max_id=' . $this->max_id;
if ($this->rpp != 15) {
$this->previous_page .= '&rpp=' . $this->rpp;
}
$this->previous_page .= '&q=' . $this->query;
}
print json_encode($this);
return $cnt;
}
}
/**
* widget for displaying a single JSON search result
*
* @category UI
* @package Laconica
* @author Zach Copley <zach@controlyourself.ca>
* @license http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
* @link http://laconi.ca/
* @see JSONSearchResultsList
*/
class ResultItem
{
/** The notice this item is based on. */
protected $notice; // protected attrs invisible to json_encode()
/** The profile associated with the notice. */
protected $profile;
// The below attributes are carefully named so the JSON output from
// this obj matches the output from search.twitter.com
var $text;
var $to_user_id;
var $to_user;
var $from_user;
var $id;
var $from_user_id;
var $iso_language_code;
var $source;
var $profile_image_url;
var $created_at;
/**
* constructor
*
* Also initializes the profile attribute.
*
* @param Notice $notice The notice we'll display
*/
function __construct($notice)
{
$this->notice = $notice;
$this->profile = $notice->getProfile();
$this->buildResult();
}
/**
* Build a search result object
*
* This populates the the result in preparation for JSON encoding.
*
* @return void
*/
function buildResult()
{
$this->text = $this->notice->content;
$replier_profile = null;
if ($this->notice->reply_to) {
$reply = Notice::staticGet(intval($notice->reply_to));
if ($reply) {
$replier_profile = $reply->getProfile();
}
}
$this->to_user_id = ($replier_profile) ?
intval($replier_profile->id) : null;
$this->to_user = ($replier_profile) ?
$replier_profile->nickname : null;
$this->from_user = $this->profile->nickname;
$this->id = $this->notice->id;
$this->from_user_id = $this->profile->id;
$user = User::staticGet('id', $this->profile->id);
$this->iso_language_code = $this->user->language;
$this->source = $this->getSourceLink($this->notice->source);
$avatar = $this->profile->getAvatar(AVATAR_STREAM_SIZE);
$this->profile_image_url = ($avatar) ?
$avatar->displayUrl() : Avatar::defaultImage(AVATAR_STREAM_SIZE);
$this->created_at = date('r', $this->notice->created);
}
/**
* Show the source of the notice
*
* Either the name (and link) of the API client that posted the notice,
* or one of other other channels.
*
* @return string the source of the Notice
*/
function getSourceLink($source)
{
$source_name = _($source);
switch ($source) {
case 'web':
case 'xmpp':
case 'mail':
case 'omb':
case 'api':
break;
default:
$ns = Notice_source::staticGet($source);
if ($ns) {
$source_name = '<a href="' . $ns->url . '">' . $ns->name . '</a>';
}
break;
}
return $source_name;
}
}

View File

@ -354,12 +354,8 @@ class Router
// search
foreach (array('json', 'atom') as $e) {
$m->connect('api/search.'.$e,
array('action' => 'twitapisearch'));
}
$m->connect('api/search.atom', array('action' => 'twitapisearchatom'));
$m->connect('api/search.json', array('action' => 'twitapisearchjson'));
$m->connect('api/trends.json', array('action' => 'twitapitrends'));
// user stuff

View File

@ -24,6 +24,28 @@ class TwitterapiAction extends Action
var $auth_user;
/**
* Initialization.
*
* @param array $args Web and URL arguments
*
* @return boolean false if user doesn't exist
*/
function prepare($args)
{
parent::prepare($args);
return true;
}
/**
* Handle a request
*
* @param array $args Arguments from $_REQUEST
*
* @return void
*/
function handle($args)
{
parent::handle($args);