* @author Jack Phoenix * @author Thomas Gries * @maintainer Thomas Gries * @link http://www.mediawiki.org/wiki/Extension:AJAX_Poll Documentation */ class AJAXPoll { /** * Register tag with the parser * * @param $parser Object: instance of Parser (not necessarily $wgParser) * @return Boolean: true */ static function onParserInit( $parser ) { global $wgOut; $parser->setHook( 'poll', array( __CLASS__, 'AJAXPollRender' ) ); $wgOut->addModules( 'ext.ajaxpoll' ); return true; } # The callback function for converting the input text to HTML output static function AJAXPollRender( $input, $params = array(), Parser $parser ) { global $wgUser, $wgOut, $wgTitle, $wgScriptPath, $wgUseAjax; $parser->disableCache(); $parser->addTrackingCategory( 'ajaxpoll-tracking-category' ); 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 ) ); // compatibility for non-ajax requests - just in case if ( !$wgUseAjax ) { $responseId = "ajaxpoll-post-id"; $responseAnswer = "ajaxpoll-post-answer-$id"; if ( isset( $_POST[$responseId] ) && isset( $_POST[$responseAnswer] ) && $_POST[$responseId] == $id ) { AJAXPoll::submitVote( $id, intval( $_POST[$responseAnswer] ) ); } } $dbw = wfGetDB( DB_MASTER ); $dbw->begin( __METHOD__ ); /** * Register poll in the database */ $row = $dbw->selectRow( array( 'ajaxpoll_info' ), array( 'COUNT(poll_id) AS count' ), array( 'poll_id' => $id ), __METHOD__ ); if( empty( $row->count ) ) { $dbw->insert( 'ajaxpoll_info', array( 'poll_id' => $id, 'poll_txt' => $input, 'poll_date' => wfTimestampNow(), ), __METHOD__ ); } $dbw->commit( __METHOD__ ); switch( $lines[0] ) { case 'STATS': $retVal = AJAXPoll::buildStats( $id, $user ); break; default: $retVal = '
' . AJAXPoll::buildHTML( $id, $user, $lines ) . '
'; break; } return $retVal; } private static function buildStats( $id, $user ) { $dbr = wfGetDB( DB_SLAVE ); $res = $dbr->select( 'ajaxpoll_vote', array( 'COUNT(*)', 'COUNT(DISTINCT poll_id)', 'COUNT(DISTINCT poll_user)', 'TIMEDIFF(NOW(), MAX(poll_date))' ), array(), __METHOD__ ); $tab = $dbr->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 = $dbr->select( 'ajaxpoll_vote', 'COUNT(*)', array( 'DATE_SUB(CURDATE(), INTERVAL 2 DAY) <= poll_date' ), __METHOD__ ); $tab2 = $dbr->fetchRow( $res ); return "There are $tab[1] polls and $tab[0] votes given by $tab[2] different people.
The last vote has been given $clockago ago.
During the last 48 hours, $tab2[0] votes have been given."; } public static function submitVote( $id, $answer ) { global $wgUser,$wgOut; // echo "id: $id ans $answer
"; $dbw = wfGetDB( DB_MASTER ); if ( $wgUser->getName() == '' ) { $user = wfGetIP(); } else { $user = $wgUser->getName(); } if ( !$wgUser->isAllowed( 'ajaxpoll-vote' ) || $wgUser->isAllowed( 'bot' ) ) { return AJAXPoll::buildHTML( $id, $user ); } if ( $answer != 0 ) { $answer = ++$answer; $q = $dbw->select( 'ajaxpoll_vote', 'COUNT(*) AS count', array( 'poll_id' => $id, 'poll_user' => $user ), __METHOD__ ); $row = $dbw->fetchRow( $q ); if ( $row['count'] > 0 ) { $updateQuery = $dbw->update( 'ajaxpoll_vote', array( 'poll_answer' => $answer, 'poll_date' => wfTimestampNow() ), array( 'poll_id' => $id, 'poll_user' => $user, ), __METHOD__ ); $dbw->commit(); $pollContainerText = ( $updateQuery ) ? 'ajaxpoll-vote-update' : 'ajaxpoll-vote-error'; } else { $insertQuery = $dbw->insert( 'ajaxpoll_vote', array( 'poll_id' => $id, 'poll_user' => $user, 'poll_ip' => wfGetIP(), 'poll_answer' => $answer, 'poll_date' => wfTimestampNow() ), __METHOD__ ); $dbw->commit(); $pollContainerText = ( $insertQuery ) ? 'ajaxpoll-vote-add' : 'ajaxpoll-vote-error'; } } else { // revoking a vote $deleteQuery = $dbw->delete( 'ajaxpoll_vote', array( 'poll_id' => $id, 'poll_user' => $user, ), __METHOD__ ); $dbw->commit(); $pollContainerText = ( $deleteQuery ) ? 'ajaxpoll-vote-revoked' : 'ajaxpoll-vote-error'; } return AJAXPoll::buildHTML( $id, $user, '', $pollContainerText ); } private static function buildHTML( $id, $user, $lines = '', $extra_from_ajax = '' ) { global $wgTitle, $wgUser, $wgLang, $wgUseAjax; $dbr = wfGetDB( DB_SLAVE ); $q = $dbr->select( 'ajaxpoll_info', array( 'poll_txt', 'poll_date' ), array( 'poll_id' => $id ), __METHOD__ ); $row = $dbr->fetchRow( $q ); if ( empty( $lines ) ) { $lines = explode( "\n", trim( $row['poll_txt'] ) ); } $start_date = $row['poll_date']; $q = $dbr->select( 'ajaxpoll_vote', array( 'poll_answer', 'COUNT(*)' ), array( 'poll_id' => $id ), __METHOD__, array( 'GROUP BY' => 'poll_answer' ) ); $poll_result = array(); while ( $row = $q->fetchRow() ) { $poll_result[$row[0]] = $row[1]; } $amountOfVotes = array_sum( $poll_result ); // Did we vote? $q = $dbr->select( 'ajaxpoll_vote', array( 'poll_answer', 'poll_date' ), array( 'poll_id' => $id, 'poll_user' => $user ), __METHOD__ ); if ( $row = $dbr->fetchRow( $q ) ) { $ourLastVoteDate = wfMsg( 'ajaxpoll-your-vote', $lines[$row[0] - 1], $wgLang->timeanddate( wfTimestamp( TS_MW, $row[1] ), true /* adjust? */ ) ); } if ( is_object( $wgTitle ) ) { if( !empty( $extra_from_ajax ) ) { $attributes = ' style="display:block;"'; $ajaxMessage = wfMsg( $extra_from_ajax ); } else { $attributes = ' style="display:none;"'; $ajaxMessage = ''; } // HTML output has to be on one line thanks to a MediaWiki bug // @see https://bugzilla.wikimedia.org/show_bug.cgi?id=1319 $ret = '
' . $ajaxMessage . '
' . strip_tags( $lines[0] ) . '
'; // Different message depending on if the user has already voted or not, or is entitled to vote $canRevoke = false; if ( $wgUser->isAllowed( 'ajaxpoll-vote' ) ) { if ( isset( $row[0] ) ) { $message = $ourLastVoteDate; $canRevoke = true; $lines[] = wfMsg( 'ajaxpoll-revoke-vote' ); } else { $message = wfMsg( 'ajaxpoll-no-vote' ); } } else { $message = wfMsg( 'ajaxpoll-vote-permission' ); } $ret .= '
' . $message . '
'; $ret .= '
'; for ( $i = 1; $i < count( $lines ); $i++ ) { $vote = !( $canRevoke && ( $i == count( $lines ) - 1 ) ); // answers are counted from 1 ... n // last answer is pseudo-answer for "I want to revoke vote" // and becomes answer number 0 $answer = ( $vote ) ? $i : 0; $xid = $id . "-" . $answer; 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( $row[0] ) && ( $row[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 ( $wgUser->isAllowed( 'ajaxpoll-vote' ) ) { // HTML output has to be on one line thanks to a MediaWiki bug // @see https://bugzilla.wikimedia.org/show_bug.cgi?id=1319 if ( $vote ) { $ret .= "
" . ( ( isset( $poll_result ) && !empty( $poll_result[$i + 1] ) ) ? $poll_result[$i + 1] : 0 ) . "
"; } else { $ret .= "
"; } } else { $ret .= "
" . ( ( isset( $poll_result ) && !empty( $poll_result[$i + 1] ) ) ? $poll_result[$i + 1] : 0 ) . "
"; } } $ret .= '
'; // Display information about the poll (creation date, amount of votes) $pollSummary = wfMsgExt( 'ajaxpoll-info', 'parsemag', // parse PLURAL $amountOfVotes, // amount of votes $wgLang->timeanddate( wfTimestamp( TS_MW, $start_date ), true /* adjust? */ ) ); $ret .= '
' . $pollSummary . '
poll-id ' . $id . '
'; } else { $ret = ''; } return $ret; } public static function onLoadExtensionSchemaUpdates( $updater = null ) { if ( $updater === null ) { // no < 1.17 support } else { // >= 1.17 support if ( $updater->tableExists( 'poll_info' ) ) { # poll_info.poll_title field was dropped in AJAXPoll version 1.72 $updater->dropExtensionField( 'poll_info', 'poll_title', dirname( __FILE__ ) . '/patches/drop-field--poll_info-poll_title.sql' ); $updater->addExtensionTable( 'ajaxpoll_info', dirname( __FILE__ ) . '/patches/rename-table--poll_info.sql' ); } else { $updater->addExtensionTable( 'ajaxpoll_info', dirname( __FILE__ ) . '/patches/create-table--ajaxpoll_info.sql' ); } if ( $updater->tableExists( 'poll_vote' ) ) { $updater->addExtensionTable( 'ajaxpoll_vote', dirname( __FILE__ ) . '/patches/rename-table--poll_vote.sql' ); } else { $updater->addExtensionTable( 'ajaxpoll_vote', dirname( __FILE__ ) . '/patches/create-table--ajaxpoll_vote.sql' ); } } return true; } } /* class AJAXPoll */