mediawiki-extensions-AJAXPoll/AJAXPoll.php

382 lines
9.8 KiB
PHP

<?php
/**
* AJAX Poll extension for MediaWiki
* Created by Dariusz Siedlecki, based on the work by Eric David.
* Licensed under the GFDL.
*
* <poll>
* [Option]
* Question
* Answer 1
* Answer 2
* Answer ...
* Answer n
* </poll>
*
* @file
* @ingroup Extensions
* @author Dariusz Siedlecki <datrio@gmail.com>
* @author Jack Phoenix <jack@countervandalism.net>
* @author Thomas Gries
* @maintainer Thomas Gries
* @version 1.600
* @link http://www.mediawiki.org/wiki/Extension:AJAX_Poll Documentation
*/
if( !defined( 'MEDIAWIKI' ) ) {
die( "This is not a valid entry point.\n" );
}
// Extension credits that will show up on Special:Version
$wgExtensionCredits['parserhook'][] = array(
'path' => __FILE__,
'name' => 'AJAX Poll',
'version' => '1.600 20120216',
'author' => array( 'Dariusz Siedlecki', 'Jack Phoenix', 'Thomas Gries', ),
'description' => 'Allows AJAX-based polls with <tt>&lt;poll&gt;</tt> tag',
'url' => 'https://www.mediawiki.org/wiki/Extension:AJAX_Poll',
);
// Internationalization + AJAX function
$dir = dirname( __FILE__ ) . '/';
$wgExtensionMessagesFiles['AJAXPoll'] = $dir . 'AJAXPoll.i18n.php';
$wgAjaxExportList[] = 'submitVote';
$wgHooks['ParserFirstCallInit'][] = 'wfPoll';
/**
* Register <poll> tag with the parser
*
* @param $parser Object: instance of Parser (not necessarily $wgParser)
* @return Boolean: true
*/
function wfPoll( &$parser ) {
$parser->setHook( 'poll', 'renderPoll' );
return true;
}
# The callback function for converting the input text to HTML output
function renderPoll( $input ) {
global $wgParser, $wgUser, $wgOut, $wgTitle, $wgScriptPath;
$wgParser->disableCache();
if ( $wgUser->getName() == '' ) {
$user = wfGetIP();
} else {
$user = $wgUser->getName();
}
// ID of the poll
$ID = strtoupper( md5( $input ) );
$par = new Parser();
$input = $par->parse( $input, $wgTitle, $wgOut->parserOptions() );
$input = trim( strip_tags( $input->getText() ) );
$lines = explode( "\n", trim( $input ) );
// Deprecating AJAX
/*if ( isset( $_POST['ajaxpoll-post-id'] ) && isset( $_POST['ajaxpoll-post-answer'] ) && $_POST['ajaxpoll-post-id'] == $ID ) {
submitVote( $_POST['ajaxpoll-post-id'], intval( $_POST['ajaxpoll-post-answer'] ) );
}*/
$dbw = wfGetDB( DB_MASTER );
$dbw->begin();
/**
* Register poll in the database
*/
$row = $dbw->selectRow(
array( 'poll_info' ),
array( 'COUNT(poll_id) AS count' ),
array( 'poll_id' => $ID ),
__METHOD__
);
if( empty( $row->count ) ) {
$dbw->insert(
'poll_info',
array(
'poll_id' => $ID,
'poll_txt' => $input,
'poll_date' => wfTimestampNow(),
'poll_title' => $wgParser->mTitle->getText()
),
__METHOD__
);
}
$dbw->commit();
// Add CSS
$wgOut->addExtensionStyle( $wgScriptPath . '/extensions/AJAXPoll/AJAXPoll.css' );
switch( $lines[0] ) {
case 'STATS':
$retVal = buildStats( $ID, $user );
break;
default:
$retVal = '<div id="ajaxpoll-container-' . $ID . '">' .
buildHTML( $ID, $user, $lines ) .
'</div>';
break;
}
return $retVal;
}
function buildStats( $ID, $user ) {
$dbw = wfGetDB( DB_MASTER );
$res = $dbw->select(
'poll_vote',
array(
'COUNT(*)',
'COUNT(DISTINCT poll_id)',
'COUNT(DISTINCT poll_user)',
'TIMEDIFF(NOW(), MAX(poll_date))'
),
array(),
__METHOD__
);
$tab = $dbw->fetchRow( $res );
$clock = explode( ':', $tab[3] );
if ( $clock[0] == '00' && $clock[1] == '00' ) {
$x = $clock[2];
$y = 'second';
} elseif( $clock[0] == '00' ) {
$x = $clock[1];
$y = 'minute';
} else {
if ( $clock[0] < 24 ) {
$x = $clock[0];
$y = 'hour';
} else {
$x = floor( $hr / 24 );
$y = 'day';
}
}
$clockago = $x . ' ' . $y . ( $x > 1 ? 's' : '' );
$res = $dbw->select(
'poll_vote',
'COUNT(*)',
array( 'DATE_SUB(CURDATE(), INTERVAL 2 DAY) <= poll_date' ),
__METHOD__
);
$tab2 = $dbw->fetchRow( $res );
return "There are $tab[1] polls and $tab[0] votes given by $tab[2] different people.<br />
The last vote has been given $clockago ago.<br/>
During the last 48 hours, $tab2[0] votes have been given.";
}
function submitVote( $ID, $answer ) {
global $wgUser;
$dbw = wfGetDB( DB_MASTER );
if ( $wgUser->getName() == '' ) {
$user = wfGetIP();
} else {
$user = $wgUser->getName();
}
if ( $wgUser->isAllowed( 'bot' ) ) {
return buildHTML( $ID, $user );
}
$answer = ++$answer;
$q = $dbw->select(
'poll_vote',
'COUNT(*) AS c',
array(
'poll_id' => $ID,
'poll_user' => $user
),
__METHOD__
);
$r = $dbw->fetchRow( $q );
if ( $r['c'] > 0 ) {
$updateQuery = $dbw->update(
'poll_vote',
array(
'poll_answer' => $answer,
'poll_date' => wfTimestampNow()
),
array(
'poll_id' => $ID,
'poll_user' => $user
),
__METHOD__
);
$dbw->commit();
if ( $updateQuery ) {
return buildHTML( $ID, $user, '', 'ajaxpoll-vote-update' );
} else {
return buildHTML( $ID, $user, '', 'ajaxpoll-vote-error' );
}
} else {
$insertQuery = $dbw->insert(
'poll_vote',
array(
'poll_id' => $ID,
'poll_user' => $user,
'poll_ip' => wfGetIP(),
'poll_answer' => $answer,
'poll_date' => wfTimestampNow()
),
__METHOD__
);
$dbw->commit();
if ( $insertQuery ) {
return buildHTML( $ID, $user, '', 'ajaxpoll-vote-add' );
} else {
return buildHTML( $ID, $user, '', 'ajaxpoll-vote-error' );
}
}
}
function buildHTML( $ID, $user, $lines = '', $extra_from_ajax = '' ) {
global $wgTitle, $wgLang, $wgUseAjax;
$dbw = wfGetDB( DB_SLAVE );
$q = $dbw->select(
'poll_info',
array( 'poll_txt', 'poll_date' ),
array( 'poll_id' => $ID ),
__METHOD__
);
$r = $dbw->fetchRow( $q );
if ( empty( $lines ) ) {
$lines = explode( "\n", trim( $r['poll_txt'] ) );
}
$start_date = $r['poll_date'];
$q = $dbw->select(
'poll_vote',
array( 'poll_answer', 'COUNT(*)' ),
array( 'poll_id' => $ID ),
__METHOD__,
array( 'GROUP BY' => 'poll_answer' )
);
$poll_result = array();
while ( $r = $q->fetchRow() ) {
$poll_result[$r[0]] = $r[1];
}
$amountOfVotes = array_sum( $poll_result );
// Did we vote?
$q = $dbw->select(
'poll_vote',
array( 'poll_answer', 'poll_date' ),
array(
'poll_id' => $ID,
'poll_user' => $user
),
__METHOD__
);
if ( $r = $dbw->fetchRow( $q ) ) {
$tmp_date = wfMsg(
'ajaxpoll-your-vote',
$lines[$r[0] - 1],
$wgLang->timeanddate( wfTimestamp( TS_MW, $r[1] ), true /* adjust? */ )
);
}
if ( is_object( $wgTitle ) ) {
if( !empty( $extra_from_ajax ) ) {
$additionalAttributes = ' style="display: block;"';
$message = wfMsg( $extra_from_ajax );
} else {
$additionalAttributes = '';
$message = '';
}
// HTML output has to be on one line thanks to a MediaWiki bug
// @see https://bugzilla.wikimedia.org/show_bug.cgi?id=1319
$ret = '<div id="ajaxpoll-id-' . $ID . '" class="ajaxpoll"><div id="ajaxpoll-ajax-' . $ID . '" class="ajaxpoll-ajax"' .
$additionalAttributes . '>' . $message .
'</div>
<script>
var tmp;
function mover(x){
var sp=$(x).find("span");
tmp=sp.html();
sp.text(sp.attr("title"));
sp.attr("title","");
}
function mout(x){
var sp=$(x).find("span");
sp.attr("title",sp.text());
sp.text(tmp);
}
</script>
<div class="ajaxpoll-question">' . strip_tags( $lines[0] ) . '</div>';
// Different message depending on if the user has already voted or not.
if ( isset( $r[0] ) ) {
$ret .= '<div class="ajaxpoll-misc">' . $tmp_date . '</div>';
} else {
$ret .= '<div class="ajaxpoll-misc">' . wfMsg( 'ajaxpoll-no-vote' ) . '</div>';
}
$ret .= '<form method="post" action="' . $wgTitle->getLocalURL() .
'" id="ajaxpoll-answer-id-' . $ID . '"><input type="hidden" name="ajaxpoll-post-id" value="' . $ID . '" />';
for ( $i = 1; $i < count( $lines ); $i++ ) {
$ans_no = $i - 1;
if ( $amountOfVotes == 0 ) {
$percent = 0;
} else {
$percent = $wgLang->formatNum( round( ( isset( $poll_result[$i + 1] ) ? $poll_result[$i + 1] : 0 ) * 100 / $amountOfVotes, 2 ) );
}
$our = ( isset( $r[0] ) && ( $r[0] - 1 == $i ) );
// If AJAX is enabled, as it is by default in modern MWs, we can
// just use sajax library function here for that AJAX-y feel.
// If not, we'll have to submit the form old-school way...
if ( $wgUseAjax ) {
$submitJS = "sajax_do_call(\"submitVote\",[\"" . $ID . "\",\"" . $i . "\"], $(\"#ajaxpoll-container-" . $ID . "\")[0]);";
} else {
$submitJS = "$(\"#ajaxpoll-answer-id-" . $ID . "\").submit();";
}
// HTML output has to be on one line thanks to a MediaWiki bug
// @see https://bugzilla.wikimedia.org/show_bug.cgi?id=1319
$ret .= "
<div id='ajaxpoll-answer-" . $ans_no . "' class='ajaxpoll-answer'><div class='ajaxpoll-answer-name'><label for='ajaxpoll-answer-radio-" . $ans_no . "' onclick='$(\"#ajaxpoll-ajax-" . $ID . "\").html(\"" . wfMsg( 'ajaxpoll-submitting' ) . "\");$(\"#ajaxpoll-ajax-" . $ID . "\").css(\"display\",\"block\");$(this).addClass(\"ajaxpoll-checkevent\").prop(\"checked\",true); " . $submitJS . "'><input type='radio' id='ajaxpoll-post-answer-" . $ans_no . "' name='ajaxpoll-post-answer' value='" . $i . "'" . ( $our ? 'checked=true ' : '' ) . "/>" . strip_tags( $lines[$i] ) .
"</label></div><div class='ajaxpoll-answer-vote" . ( $our ? ' ajaxpoll-our-vote' : '' ) ."' onmouseover='mover(this)' onmouseout='mout(this);'><span title='" . wfMsg( 'ajaxpoll-percent-votes', sprintf( $percent ) ) . "'>" . ( ( isset( $poll_result ) && !empty( $poll_result[$i + 1] ) ) ? $poll_result[$i + 1] : 0 ) . "</span><div style='width: " . $percent . "%;" . ( $percent == 0 ? ' border:0;' : '' ) . "'></div></div>
</div>
";
}
$ret .= '</form>';
// Display information about the poll (creation date, amount of votes)
$tmp_date = wfMsgExt(
'ajaxpoll-info',
'parsemag', // parse PLURAL
$amountOfVotes, // amount of votes
$wgLang->timeanddate( wfTimestamp( TS_MW, $start_date ), true /* adjust? */ )
);
$ret .= '<div id="ajaxpoll-info">' . $tmp_date . '</div>';
$ret .= '</div>';
} else {
$ret = '';
}
return $ret;
}