forked from GNUsocial/gnu-social
271 lines
7.8 KiB
PHP
271 lines
7.8 KiB
PHP
<?php
|
|
/**
|
|
* Data class to mark notices as bookmarks
|
|
*
|
|
* PHP version 5
|
|
*
|
|
* @category PollPlugin
|
|
* @package StatusNet
|
|
* @author Brion Vibber <brion@status.net>
|
|
* @license http://www.fsf.org/licensing/licenses/agpl.html AGPLv3
|
|
* @link http://status.net/
|
|
*
|
|
* StatusNet - the distributed open-source microblogging tool
|
|
* Copyright (C) 2011, StatusNet, Inc.
|
|
*
|
|
* 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/>.
|
|
*/
|
|
|
|
if (!defined('STATUSNET')) {
|
|
exit(1);
|
|
}
|
|
|
|
/**
|
|
* For storing the poll options and such
|
|
*
|
|
* @category PollPlugin
|
|
* @package StatusNet
|
|
* @author Brion Vibber <brion@status.net>
|
|
* @license http://www.fsf.org/licensing/licenses/agpl.html AGPLv3
|
|
* @link http://status.net/
|
|
*
|
|
* @see DB_DataObject
|
|
*/
|
|
|
|
class Poll extends Managed_DataObject
|
|
{
|
|
public $__table = 'poll'; // table name
|
|
public $id; // char(36) primary key not null -> UUID
|
|
public $uri;
|
|
public $profile_id; // int -> profile.id
|
|
public $question; // text
|
|
public $options; // text; newline(?)-delimited
|
|
public $created; // datetime
|
|
|
|
/**
|
|
* Get an instance by key
|
|
*
|
|
* This is a utility method to get a single instance with a given key value.
|
|
*
|
|
* @param string $k Key to use to lookup (usually 'user_id' for this class)
|
|
* @param mixed $v Value to lookup
|
|
*
|
|
* @return User_greeting_count object found, or null for no hits
|
|
*
|
|
*/
|
|
function staticGet($k, $v=null)
|
|
{
|
|
return Memcached_DataObject::staticGet('Poll', $k, $v);
|
|
}
|
|
|
|
/**
|
|
* Get an instance by compound key
|
|
*
|
|
* This is a utility method to get a single instance with a given set of
|
|
* key-value pairs. Usually used for the primary key for a compound key; thus
|
|
* the name.
|
|
*
|
|
* @param array $kv array of key-value mappings
|
|
*
|
|
* @return Bookmark object found, or null for no hits
|
|
*
|
|
*/
|
|
function pkeyGet($kv)
|
|
{
|
|
return Memcached_DataObject::pkeyGet('Poll', $kv);
|
|
}
|
|
|
|
/**
|
|
* The One True Thingy that must be defined and declared.
|
|
*/
|
|
public static function schemaDef()
|
|
{
|
|
return array(
|
|
'description' => 'Per-notice poll data for Poll plugin',
|
|
'fields' => array(
|
|
'id' => array('type' => 'char', 'length' => 36, 'not null' => true, 'description' => 'UUID'),
|
|
'uri' => array('type' => 'varchar', 'length' => 255, 'not null' => true),
|
|
'profile_id' => array('type' => 'int'),
|
|
'question' => array('type' => 'text'),
|
|
'options' => array('type' => 'text'),
|
|
'created' => array('type' => 'datetime', 'not null' => true),
|
|
),
|
|
'primary key' => array('id'),
|
|
'unique keys' => array(
|
|
'poll_uri_key' => array('uri'),
|
|
),
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Get a bookmark based on a notice
|
|
*
|
|
* @param Notice $notice Notice to check for
|
|
*
|
|
* @return Poll found poll or null
|
|
*/
|
|
function getByNotice($notice)
|
|
{
|
|
return self::staticGet('uri', $notice->uri);
|
|
}
|
|
|
|
function getOptions()
|
|
{
|
|
return explode("\n", $this->options);
|
|
}
|
|
|
|
/**
|
|
* Is this a valid selection index?
|
|
*
|
|
* @param numeric $selection (1-based)
|
|
* @return boolean
|
|
*/
|
|
function isValidSelection($selection)
|
|
{
|
|
if ($selection != intval($selection)) {
|
|
return false;
|
|
}
|
|
if ($selection < 1 || $selection > count($this->getOptions())) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
function getNotice()
|
|
{
|
|
return Notice::staticGet('uri', $this->uri);
|
|
}
|
|
|
|
function bestUrl()
|
|
{
|
|
return $this->getNotice()->bestUrl();
|
|
}
|
|
|
|
/**
|
|
* Get the response of a particular user to this poll, if any.
|
|
*
|
|
* @param Profile $profile
|
|
* @return Poll_response object or null
|
|
*/
|
|
function getResponse(Profile $profile)
|
|
{
|
|
$pr = new Poll_response();
|
|
$pr->poll_id = $this->id;
|
|
$pr->profile_id = $profile->id;
|
|
$pr->find();
|
|
if ($pr->fetch()) {
|
|
return $pr;
|
|
} else {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
function countResponses()
|
|
{
|
|
$pr = new Poll_response();
|
|
$pr->poll_id = $this->id;
|
|
$pr->groupBy('selection');
|
|
$pr->selectAdd('count(profile_id) as votes');
|
|
$pr->find();
|
|
|
|
$raw = array();
|
|
while ($pr->fetch()) {
|
|
// Votes list 1-based
|
|
// Array stores 0-based
|
|
$raw[$pr->selection - 1] = $pr->votes;
|
|
}
|
|
|
|
$counts = array();
|
|
foreach (array_keys($this->getOptions()) as $key) {
|
|
if (isset($raw[$key])) {
|
|
$counts[$key] = $raw[$key];
|
|
} else {
|
|
$counts[$key] = 0;
|
|
}
|
|
}
|
|
return $counts;
|
|
}
|
|
|
|
/**
|
|
* Save a new poll notice
|
|
*
|
|
* @param Profile $profile
|
|
* @param string $question
|
|
* @param array $opts (poll responses)
|
|
*
|
|
* @return Notice saved notice
|
|
*/
|
|
static function saveNew($profile, $question, $opts, $options=null)
|
|
{
|
|
if (empty($options)) {
|
|
$options = array();
|
|
}
|
|
|
|
$p = new Poll();
|
|
|
|
$p->id = UUID::gen();
|
|
$p->profile_id = $profile->id;
|
|
$p->question = $question;
|
|
$p->options = implode("\n", $opts);
|
|
|
|
if (array_key_exists('created', $options)) {
|
|
$p->created = $options['created'];
|
|
} else {
|
|
$p->created = common_sql_now();
|
|
}
|
|
|
|
if (array_key_exists('uri', $options)) {
|
|
$p->uri = $options['uri'];
|
|
} else {
|
|
$p->uri = common_local_url('showpoll',
|
|
array('id' => $p->id));
|
|
}
|
|
|
|
common_log(LOG_DEBUG, "Saving poll: $p->id $p->uri");
|
|
$p->insert();
|
|
|
|
// TRANS: Notice content creating a poll.
|
|
// TRANS: %1$s is the poll question, %2$s is a link to the poll.
|
|
$content = sprintf(_m('Poll: %1$s %2$s'),
|
|
$question,
|
|
$p->uri);
|
|
$link = '<a href="' . htmlspecialchars($p->uri) . '">' . htmlspecialchars($question) . '</a>';
|
|
// TRANS: Rendered version of the notice content creating a poll.
|
|
// TRANS: %s a link to the poll with the question as link description.
|
|
$rendered = sprintf(_m('Poll: %s'), $link);
|
|
|
|
$tags = array('poll');
|
|
$replies = array();
|
|
|
|
$options = array_merge(array('urls' => array(),
|
|
'rendered' => $rendered,
|
|
'tags' => $tags,
|
|
'replies' => $replies,
|
|
'object_type' => PollPlugin::POLL_OBJECT),
|
|
$options);
|
|
|
|
if (!array_key_exists('uri', $options)) {
|
|
$options['uri'] = $p->uri;
|
|
}
|
|
|
|
$saved = Notice::saveNew($profile->id,
|
|
$content,
|
|
array_key_exists('source', $options) ?
|
|
$options['source'] : 'web',
|
|
$options);
|
|
|
|
return $saved;
|
|
}
|
|
}
|