Newer
Older
// $Id$
/**
* @file
* Enables your site to capture votes on different topics in the form of multiple
* choice questions.
*/
Dries Buytaert
committed
/**
* Implementation of hook_help().
*/
Gábor Hojtsy
committed
function poll_help($path, $arg) {
switch ($path) {
$output = '<p>' . t('The poll module can be used to create simple polls for site users. A poll is a simple, multiple choice questionnaire which displays the cumulative results of the answers to the poll. Having polls on the site is a good way to receive feedback from community members.') . '</p>';
$output .= '<p>' . t('When creating a poll, enter the question being posed, as well as the potential choices (and beginning vote counts for each choice). The status and duration (length of time the poll remains active for new votes) can also be specified. Use the <a href="@poll">poll</a> menu item to view all current polls. To vote in or view the results of a specific poll, click on the poll itself.', array('@poll' => url('poll'))) . '</p>';
$output .= '<p>' . t('For more information, see the online handbook entry for <a href="@poll">Poll module</a>.', array('@poll' => 'http://drupal.org/handbook/modules/poll/')) . '</p>';
Dries Buytaert
committed
return $output;
Dries Buytaert
committed
/**
* Implementation of hook_init().
*/
function poll_init() {
drupal_add_css(drupal_get_path('module', 'poll') . '/poll.css');
Dries Buytaert
committed
}
Dries Buytaert
committed
/**
* Implementation of hook_theme().
Dries Buytaert
committed
*/
function poll_theme() {
return array(
Dries Buytaert
committed
'poll_vote' => array(
'template' => 'poll-vote',
Dries Buytaert
committed
'arguments' => array('form' => NULL),
),
'poll_choices' => array(
'arguments' => array('form' => NULL),
),
Dries Buytaert
committed
'poll_results' => array(
'template' => 'poll-results',
Dries Buytaert
committed
'arguments' => array('raw_title' => NULL, 'results' => NULL, 'votes' => NULL, 'raw_links' => NULL, 'block' => NULL, 'nid' => NULL, 'vote' => NULL),
Dries Buytaert
committed
),
'poll_bar' => array(
'template' => 'poll-bar',
Dries Buytaert
committed
'arguments' => array('title' => NULL, 'votes' => NULL, 'total_votes' => NULL, 'vote' => NULL, 'block' => NULL),
Dries Buytaert
committed
),
);
}
Dries Buytaert
committed
/**
* Implementation of hook_perm().
*/
function poll_perm() {
$perms = node_list_permissions('poll');
$perms += array(
Angie Byron
committed
'vote on polls' => array(
'title' => t('Vote on polls'),
'description' => t('Cast votes on polls.'),
),
'cancel own vote' => array(
'title' => t('Cancel own vote'),
'description' => t('Retract and optionally change own votes.'),
),
'inspect all votes' => array(
'title' => t('Inspect all votes'),
'description' => t('View voting results.'),
),
);
return $perms;
Dries Buytaert
committed
}
Dries Buytaert
committed
/**
* Implementation of hook_access().
*/
function poll_access($op, $node, $account) {
switch ($op) {
case 'create':
Gábor Hojtsy
committed
return user_access('create poll content', $account);
case 'update':
Gábor Hojtsy
committed
return user_access('edit any poll content', $account) || (user_access('edit own poll content', $account) && ($node->uid == $account->uid));
Gábor Hojtsy
committed
case 'delete':
return user_access('delete any poll content', $account) || (user_access('delete own poll content', $account) && ($node->uid == $account->uid));
Dries Buytaert
committed
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/**
* Implementation of hook_menu().
*/
function poll_menu() {
$items['poll'] = array(
'title' => 'Polls',
'page callback' => 'poll_page',
'access arguments' => array('access content'),
'type' => MENU_SUGGESTED_ITEM,
);
$items['node/%node/votes'] = array(
'title' => 'Votes',
'page callback' => 'poll_votes',
'page arguments' => array(1),
'access callback' => '_poll_menu_access',
'access arguments' => array(1, 'inspect all votes', FALSE),
'weight' => 3,
'type' => MENU_LOCAL_TASK,
);
$items['node/%node/results'] = array(
'title' => 'Results',
'page callback' => 'poll_results',
'page arguments' => array(1),
'access callback' => '_poll_menu_access',
'access arguments' => array(1, 'access content', TRUE),
'weight' => 3,
'type' => MENU_LOCAL_TASK,
);
Dries Buytaert
committed
return $items;
}
/**
* Callback function to see if a node is acceptable for poll menu items.
*/
function _poll_menu_access($node, $perm, $inspect_allowvotes) {
return user_access($perm) && ($node->type == 'poll') && ($node->allowvotes || !$inspect_allowvotes);
}
Dries Buytaert
committed
/**
Dries Buytaert
committed
* Implementation of hook_block_list().
*/
function poll_block_list() {
if (user_access('access content')) {
$blocks['recent']['info'] = t('Most recent poll');
return $blocks;
}
}
/**
* Implementation of hook_block_view().
Dries Buytaert
committed
*
* Generates a block containing the latest poll.
*/
Dries Buytaert
committed
function poll_block_view($delta = '') {
Dries Buytaert
committed
// Retrieve the latest poll.
Dries Buytaert
committed
$select = db_select('node', 'n');
$select->join('poll', 'p', 'p.nid = n.nid');
$select->fields('n', array('nid'))
->condition('n.status', 1)
->condition('p.active', 1)
->orderBy('n.created', 'DESC')
->range(0, 1)
->addTag('node_access');
$record = $select->execute()->fetchObject();
Dries Buytaert
committed
if ($record) {
$poll = node_load($record->nid);
if ($poll->nid) {
$poll = poll_view($poll, TRUE, FALSE, TRUE);
Dries Buytaert
committed
$block['subject'] = t('Poll');
$block['content'] = $poll->content;
Dries Buytaert
committed
return $block;
Dries Buytaert
committed
/**
* Implementation of hook_cron().
*
* Closes polls that have exceeded their allowed runtime.
*/
Dries Buytaert
committed
$nids = db_query('SELECT p.nid FROM {poll} p INNER JOIN {node} n ON p.nid = n.nid WHERE (n.created + p.runtime) < :request_time AND p.active = :active AND p.runtime <> :runtime', array(':request_time' => REQUEST_TIME, ':active' => 1, ':runtime' => 0))->fetchCol();
if (!empty($nids)) {
db_update('poll')
->fields(array('active' => 0))
->condition('nid', $nids, 'IN')
->execute();
Dries Buytaert
committed
/**
Dries Buytaert
committed
* Implementation of hook_node_info().
Dries Buytaert
committed
*/
Dries Buytaert
committed
function poll_node_info() {
return array(
'poll' => array(
'name' => t('Poll'),
'base' => 'poll',
'description' => t('A <em>poll</em> is a question with a set of possible responses. A <em>poll</em>, once created, automatically provides a simple running count of the number of votes received for each response.'),
Dries Buytaert
committed
'title_label' => t('Question'),
'has_body' => FALSE,
)
);
Dries Buytaert
committed
}
Dries Buytaert
committed
/**
* Implementation of hook_form().
*/
function poll_form($node, $form_state) {
global $user;
$admin = user_access('administer nodes') || user_access('edit any poll content') || (user_access('edit own poll content') && $user->uid == $node->uid);
Neil Drumm
committed
$type = node_get_types('type', $node);
$form = array(
'#cache' => TRUE,
);
$form['title'] = array(
'#type' => 'textfield',
'#title' => check_plain($type->title_label),
'#required' => TRUE,
'#default_value' => $node->title,
'#weight' => -5,
);
Dries Buytaert
committed
if (isset($form_state['choice_count'])) {
$choice_count = $form_state['choice_count'];
}
else {
$choice_count = max(2, empty($node->choice) ? 2 : count($node->choice));
// Add a wrapper for the choices and more button.
$form['choice_wrapper'] = array(
'#tree' => FALSE,
Gábor Hojtsy
committed
'#weight' => -4,
Angie Byron
committed
'#prefix' => '<div class="clearfix" id="poll-choice-wrapper">',
'#suffix' => '</div>',
);
// Container for just the poll choices.
$form['choice_wrapper']['choice'] = array(
'#prefix' => '<div id="poll-choices">',
'#suffix' => '</div>',
'#theme' => 'poll_choices',
);
// Add the current choices to the form.
Dries Buytaert
committed
$delta = 0;
$weight = 0;
if (isset($node->choice)) {
$delta = count($node->choice);
$weight = -$delta;
foreach ($node->choice as $chid => $choice) {
$key = 'chid:' . $chid;
Dries Buytaert
committed
$form['choice_wrapper']['choice'][$key] = _poll_choice_form($key, $choice['chid'], $choice['chtext'], $choice['chvotes'], $choice['weight'], $choice_count);
$weight = ($choice['weight'] > $weight) ? $choice['weight'] : $weight;
}
}
Dries Buytaert
committed
// Add initial or additional choices.
$existing_delta = $delta;
for ($delta; $delta < $choice_count; $delta++) {
$key = 'new:' . ($delta - $existing_delta);
Dries Buytaert
committed
$form['choice_wrapper']['choice'][$key] = _poll_choice_form($key, NULL, '', 0, $weight, $choice_count);
}
// We name our button 'poll_more' to avoid conflicts with other modules using
// AHAH-enabled buttons with the id 'more'.
$form['choice_wrapper']['poll_more'] = array(
'#type' => 'submit',
'#value' => t('More choices'),
'#description' => t("If the amount of boxes above isn't enough, click here to add more choices."),
'#weight' => 1,
'#submit' => array('poll_more_choices_submit'), // If no javascript action.
'#ahah' => array(
Angie Byron
committed
'callback' => 'poll_choice_js',
'wrapper' => 'poll-choices',
'method' => 'replace',
'effect' => 'fade',
),
);
Dries Buytaert
committed
$duration = array(0 => t('Unlimited')) + drupal_map_assoc(array(86400, 172800, 345600, 604800, 1209600, 2419200, 4838400, 9676800, 31536000), "format_interval");
$active = array(0 => t('Closed'), 1 => t('Active'));
$form['settings'] = array(
'#type' => 'fieldset',
'#collapsible' => TRUE,
'#title' => t('Poll settings'),
'#weight' => -3,
'#access' => $admin,
);
Dries Buytaert
committed
$form['settings']['active'] = array(
'#type' => 'radios',
'#title' => t('Poll status'),
'#default_value' => isset($node->active) ? $node->active : 1,
'#options' => $active,
'#description' => t('When a poll is closed, visitors can no longer vote for it.'),
'#access' => $admin,
);
$form['settings']['runtime'] = array(
'#type' => 'select',
'#title' => t('Poll duration'),
Steven Wittens
committed
'#default_value' => isset($node->runtime) ? $node->runtime : 0,
Dries Buytaert
committed
'#options' => $duration,
'#description' => t('After this period, the poll will be closed automatically.'),
);
/**
* Submit handler to add more choices to a poll form. This handler is used when
* javascript is not available. It makes changes to the form state and the
* entire form is rebuilt during the page reload.
*/
function poll_more_choices_submit($form, &$form_state) {
// Set the form to rebuild and run submit handlers.
node_form_submit_build_node($form, $form_state);
// Make the changes we want to the form state.
if ($form_state['values']['poll_more']) {
Angie Byron
committed
$n = $_GET['q'] == 'system/ahah' ? 1 : 5;
$form_state['choice_count'] = count($form_state['values']['choice']) + $n;
}
}
Dries Buytaert
committed
function _poll_choice_form($key, $chid = NULL, $value = '', $votes = 0, $weight = 0, $size = 10) {
$admin = user_access('administer nodes');
$form = array(
'#tree' => TRUE,
);
// We'll manually set the #parents property of these fields so that
// their values appear in the $form_state['values']['choice'] array.
Dries Buytaert
committed
$form['chid'] = array(
'#type' => 'value',
'#value' => $chid,
'#parents' => array('choice', $key, 'chid'),
);
$form['chtext'] = array(
'#type' => 'textfield',
'#default_value' => $value,
Dries Buytaert
committed
'#parents' => array('choice', $key, 'chtext'),
);
Dries Buytaert
committed
$form['chvotes'] = array(
'#type' => 'textfield',
'#default_value' => $votes,
'#size' => 5,
'#maxlength' => 7,
'#parents' => array('choice', $key, 'chvotes'),
'#access' => user_access('administer nodes'),
);
$form['weight'] = array(
'#type' => 'weight',
'#default_value' => $weight,
'#delta' => $size,
'#parents' => array('choice', $key, 'weight'),
);
return $form;
}
/**
Angie Byron
committed
* Menu callback for AHAH additions. Render the new poll choices.
*/
Angie Byron
committed
function poll_choice_js($form, $form_state) {
$choice_form = $form['choice_wrapper']['choice'];
Angie Byron
committed
// Prevent duplicate wrappers.
unset($choice_form['#prefix'], $choice_form['#suffix']);
$output = theme('status_messages') . drupal_render($choice_form);
drupal_json(array('status' => TRUE, 'data' => $output));
}
Dries Buytaert
committed
/**
* Implementation of hook_submit().
*/
function poll_node_form_submit(&$form, &$form_state) {
// Renumber fields
$form_state['values']['choice'] = array_values($form_state['values']['choice']);
$form_state['values']['teaser'] = poll_teaser((object)$form_state['values']);
Dries Buytaert
committed
* Implementation of hook_validate().
Dries Buytaert
committed
function poll_validate($node, $form) {
Dries Buytaert
committed
if (isset($node->title)) {
// Check for at least two options and validate amount of votes:
$realchoices = 0;
// Renumber fields
$node->choice = array_values($node->choice);
foreach ($node->choice as $i => $choice) {
if ($choice['chtext'] != '') {
$realchoices++;
}
if (isset($choice['chvotes']) && $choice['chvotes'] < 0) {
Dries Buytaert
committed
form_set_error("choice][$i][chvotes", t('Negative values are not allowed.'));
}
}
Dries Buytaert
committed
if ($realchoices < 2) {
form_set_error("choice][$realchoices][chtext", t('You must fill in at least two choices.'));
}
}
Dries Buytaert
committed
}
/**
* Implementation of hook_node_prepare_translation().
*/
function poll_node_prepare_translation($node) {
if ($node->type == 'poll') {
$node->choice = $node->translation_source->choice;
}
}
Dries Buytaert
committed
/**
* Implementation of hook_load().
*/
function poll_load($nodes) {
global $user;
foreach ($nodes as $node) {
Dries Buytaert
committed
$poll = db_query("SELECT runtime, active FROM {poll} WHERE nid = :nid", array(':nid' => $node->nid))->fetchObject();
// Load the appropriate choices into the $poll object.
$poll->choice = db_query("SELECT chid, chtext, chvotes, weight FROM {poll_choice} WHERE nid = :nid ORDER BY weight", array(':nid' => $node->nid))->fetchAllAssoc('chid', PDO::FETCH_ASSOC);
// Determine whether or not this user is allowed to vote.
$poll->allowvotes = FALSE;
if (user_access('vote on polls') && $poll->active) {
if ($user->uid) {
Dries Buytaert
committed
$result = db_query('SELECT chid FROM {poll_vote} WHERE nid = :nid AND uid = :uid', array(':nid' => $node->nid, ':uid' => $user->uid))->fetchObject();
}
else {
Dries Buytaert
committed
$result = db_query("SELECT chid FROM {poll_vote} WHERE nid = :nid AND hostname = :hostname", array(':nid' => $node->nid, ':hostname' => ip_address()))->fetchObject();
}
if ($result) {
$poll->vote = $result->chid;
}
else {
$poll->vote = -1;
$poll->allowvotes = TRUE;
}
foreach ($poll as $key => $value) {
$nodes[$node->nid]->$key = $value;
}
}
Dries Buytaert
committed
/**
Dries Buytaert
committed
* Implementation of hook_insert().
Dries Buytaert
committed
*/
Dries Buytaert
committed
function poll_insert($node) {
if (!user_access('administer nodes')) {
// Make sure all votes are 0 initially
foreach ($node->choice as $i => $choice) {
$node->choice[$i]['chvotes'] = 0;
}
$node->active = 1;
}
Dries Buytaert
committed
db_insert('poll')
->fields(array(
'nid' => $node->nid,
'runtime' => $node->runtime,
'active' => $node->active,
))
->execute();
Dries Buytaert
committed
foreach ($node->choice as $choice) {
if ($choice['chtext'] != '') {
Dries Buytaert
committed
db_insert('poll_choice')
->fields(array(
'nid' => $node->nid,
'chtext' => $choice['chtext'],
'chvotes' => $choice['chvotes'],
'weight' => $choice['weight'],
))
->execute();
Dries Buytaert
committed
}
}
}
/**
* Implementation of hook_update().
*/
function poll_update($node) {
Gábor Hojtsy
committed
// Update poll settings.
Dries Buytaert
committed
db_update('poll')
->fields(array(
'runtime' => $node->runtime,
'active' => $node->active,
))
->condition('nid', $node->nid)
->execute();
Dries Buytaert
committed
Dries Buytaert
committed
// Poll choices with empty titles signifies removal. We remove all votes to
// the removed options, so people who voted on them can vote again.
foreach ($node->choice as $key => $choice) {
if (!empty($choice['chtext'])) {
Dries Buytaert
committed
db_merge('poll_choice')
->key(array('chid' => $choice['chid']))
->fields(array(
'nid' => $node->nid,
'chtext' => $choice['chtext'],
'chvotes' => (int) $choice['chvotes'],
'weight' => $choice['weight'],
))
->updateExcept('nid')
->execute();
Gábor Hojtsy
committed
}
else {
Dries Buytaert
committed
db_delete('poll_vote')
->condition('nid', $node->nid)
->condition('chid', $key)
->execute();
Dries Buytaert
committed
}
}
}
/**
* Implementation of hook_delete().
*/
function poll_delete($node) {
Dries Buytaert
committed
db_delete('poll')
->condition('nid', $node->nid)
->execute();
db_delete('poll_choice')
->condition('nid', $node->nid)
->execute();
db_delete('poll_vote')
->condition('nid', $node->nid)
->execute();
Dries Buytaert
committed
}
/**
* Implementation of hook_view().
*
* @param $block
* An extra parameter that adapts the hook to display a block-ready
* rendering of the poll.
*/
function poll_view($node, $teaser = FALSE, $block = FALSE) {
Dries Buytaert
committed
global $user;
$output = '';
// Special display for side-block
if ($block) {
// No 'read more' link
$node->readmore = FALSE;
Dries Buytaert
committed
$node->teaser = '';
Dries Buytaert
committed
$links = module_invoke_all('link', 'node', $node, 1);
$links[] = array('title' => t('Older polls'), 'href' => 'poll', 'attributes' => array('title' => t('View the list of polls on this site.')));
if ($node->allowvotes && $block) {
$links[] = array('title' => t('Results'), 'href' => 'node/' . $node->nid . '/results', 'attributes' => array('title' => t('View the current poll results.')));
Dries Buytaert
committed
}
$node->links = $links;
}
if (!empty($node->allowvotes) && ($block || empty($node->show_results))) {
Dries Buytaert
committed
$node->content['poll_view_voting'] = drupal_get_form('poll_view_voting', $node, $block);
Dries Buytaert
committed
}
else {
Dries Buytaert
committed
$node->content['poll_view_results'] = array('#markup' => poll_view_results($node, $teaser, $block));
Dries Buytaert
committed
}
return $node;
Dries Buytaert
committed
/**
Dries Buytaert
committed
* Creates a simple teaser that lists all the choices.
Dries Buytaert
committed
*
* This is primarily used for RSS.
Dries Buytaert
committed
*/
$teaser = NULL;
if ($choice['chtext'] != '') {
$teaser .= '* ' . check_plain($choice['chtext']) . "\n";
}
}
return $teaser;
}
Dries Buytaert
committed
* Generates the voting form for a poll.
Dries Buytaert
committed
*
* @ingroup forms
* @see poll_vote()
* @see phptemplate_preprocess_poll_vote()
Dries Buytaert
committed
function poll_view_voting(&$form_state, $node, $block) {
$list = array();
foreach ($node->choice as $i => $choice) {
$list[$i] = check_plain($choice['chtext']);
$form['choice'] = array(
'#type' => 'radios',
'#default_value' => -1,
'#options' => $list,
);
Dries Buytaert
committed
Dries Buytaert
committed
$form['vote'] = array(
Dries Buytaert
committed
'#type' => 'submit',
Dries Buytaert
committed
'#value' => t('Vote'),
'#submit' => array('poll_vote'),
);
// Store the node so we can get to it in submit functions.
$form['#node'] = $node;
$form['#block'] = $block;
// Set form caching because we could have multiple of these forms on
// the same page, and we want to ensure the right one gets picked.
$form['#cache'] = TRUE;
// Provide a more cleanly named voting form theme.
$form['#theme'] = 'poll_vote';
Dries Buytaert
committed
return $form;
/**
* Validation function for processing votes
*/
function poll_view_voting_validate($form, &$form_state) {
if ($form_state['values']['choice'] == -1) {
form_set_error( 'choice', t('Your vote could not be recorded because you did not select any of the choices.'));
}
}
Dries Buytaert
committed
/**
* Submit handler for processing a vote
*/
function poll_vote($form, &$form_state) {
$node = $form['#node'];
$choice = $form_state['values']['choice'];
global $user;
Dries Buytaert
committed
db_insert('poll_vote')
->fields(array(
'nid' => $node->nid,
'chid' => $choice,
'uid' => $user->uid,
'hostname' => $user->uid ? ip_address() : '',
))
->execute();
Dries Buytaert
committed
// Add one to the votes.
Dries Buytaert
committed
db_update('poll_choice')
->expression('chvotes', 'chvotes + 1')
->condition('chid', $choice)
->execute();
Dries Buytaert
committed
cache_clear_all();
drupal_set_message(t('Your vote was recorded.'));
// Return the user to whatever page they voted from.
}
/**
* Themes the voting form for a poll.
Dries Buytaert
committed
*
* Inputs: $form
Dries Buytaert
committed
function template_preprocess_poll_vote(&$variables) {
$form = $variables['form'];
$variables['choice'] = drupal_render($form['choice']);
$variables['title'] = check_plain($form['#node']->title);
$variables['vote'] = drupal_render($form['vote']);
$variables['rest'] = drupal_render_children($form);
Dries Buytaert
committed
$variables['block'] = $form['#block'];
// If this is a block, allow a different tpl.php to be used.
if ($variables['block']) {
$variables['template_files'][] = 'poll-vote-block';
}
Dries Buytaert
committed
/**
* Generates a graphical representation of the results of a poll.
*/
function poll_view_results($node, $teaser, $block) {
Steven Wittens
committed
$total_votes = 0;
$max_votes = 0;
if (isset($choice['chvotes'])) {
$total_votes += $choice['chvotes'];
$max_votes = max($max_votes, $choice['chvotes']);
}
Steven Wittens
committed
$poll_results = '';
if (!empty($choice['chtext'])) {
Gábor Hojtsy
committed
$chvotes = isset($choice['chvotes']) ? $choice['chvotes'] : NULL;
$poll_results .= theme('poll_bar', $choice['chtext'], $chvotes, $total_votes, isset($node->vote) && $node->vote == $i, $block);
Dries Buytaert
committed
}
Dries Buytaert
committed
return theme('poll_results', $node->title, $poll_results, $total_votes, isset($node->links) ? $node->links : array(), $block, $node->nid, isset($node->vote) ? $node->vote : NULL);
/**
* Theme the admin poll form for choices.
Gábor Hojtsy
committed
*
* @ingroup themeable
*/
function theme_poll_choices($form) {
Dries Buytaert
committed
drupal_add_tabledrag('poll-choice-table', 'order', 'sibling', 'poll-weight');
Gábor Hojtsy
committed
Dries Buytaert
committed
$delta = 0;
$rows = array();
$headers = array(
Dries Buytaert
committed
'',
t('Choice'),
t('Vote count'),
Dries Buytaert
committed
t('Weight'),
);
foreach (element_children($form) as $key) {
Dries Buytaert
committed
$delta++;
// Set special classes for drag and drop updating.
$form[$key]['weight']['#attributes']['class'] = 'poll-weight';
// Build the table row.
$row = array(
'data' => array(
Dries Buytaert
committed
array('class' => 'choice-flag'),
drupal_render($form[$key]['chtext']),
drupal_render($form[$key]['chvotes']),
drupal_render($form[$key]['weight']),
),
Dries Buytaert
committed
'class' => 'draggable',
);
Dries Buytaert
committed
// Add any additional classes set on the row.
$row['class'] .= isset($form[$key]['#attributes']['class']) ? ' ' . $form[$key]['#attributes']['class'] : '';
Dries Buytaert
committed
$rows[] = $row;
}
Dries Buytaert
committed
$output = theme('table', $headers, $rows, array('id' => 'poll-choice-table'));
$output .= drupal_render_children($form);
return $output;
}
Dries Buytaert
committed
* Preprocess the poll_results theme hook.
*
* Inputs: $raw_title, $results, $votes, $raw_links, $block, $nid, $vote. The
* $raw_* inputs to this are naturally unsafe; often safe versions are
* made to simply overwrite the raw version, but in this case it seems likely
* that the title and the links may be overridden by the theme layer, so they
* are left in with a different name for that purpose.
*
* @see poll-results.tpl.php
* @see poll-results-block.tpl.php
* @see theme_poll_results()
Dries Buytaert
committed
function template_preprocess_poll_results(&$variables) {
$variables['links'] = theme('links', $variables['raw_links']);
if (isset($variables['vote']) && $variables['vote'] > -1 && user_access('cancel own vote')) {
$variables['cancel_form'] = drupal_get_form('poll_cancel_form', $variables['nid']);
Dries Buytaert
committed
$variables['title'] = check_plain($variables['raw_title']);
// If this is a block, allow a different tpl.php to be used.
if ($variables['block']) {
$variables['template_files'][] = 'poll-results-block';
Dries Buytaert
committed
/**
Dries Buytaert
committed
* Preprocess the poll_bar theme hook.
*
* Inputs: $title, $votes, $total_votes, $voted, $block
*
* @see poll-bar.tpl.php
* @see poll-bar-block.tpl.php
* @see theme_poll_bar()
Dries Buytaert
committed
*/
Dries Buytaert
committed
function template_preprocess_poll_bar(&$variables) {
if ($variables['block']) {
$variables['template_files'][] = 'poll-bar-block';
Dries Buytaert
committed
}
Dries Buytaert
committed
$variables['title'] = check_plain($variables['title']);
$variables['percentage'] = round($variables['votes'] * 100 / max($variables['total_votes'], 1));
Dries Buytaert
committed
}
Dries Buytaert
committed
* Builds the cancel form for a poll.
*
* @ingroup forms
* @see poll_cancel()
Dries Buytaert
committed
function poll_cancel_form(&$form_state, $nid) {
// Store the nid so we can get to it in submit functions.
$form['#nid'] = $nid;
Dries Buytaert
committed
$form['submit'] = array(
Dries Buytaert
committed
'#type' => 'submit',
Dries Buytaert
committed
'#value' => t('Cancel your vote'),
'#submit' => array('poll_cancel')
);
Dries Buytaert
committed
$form['#cache'] = TRUE;
Dries Buytaert
committed
Dries Buytaert
committed
return $form;
}
Dries Buytaert
committed
/**
Dries Buytaert
committed
* Submit callback for poll_cancel_form
Dries Buytaert
committed
*/
Dries Buytaert
committed
function poll_cancel($form, &$form_state) {
Dries Buytaert
committed
global $user;
Dries Buytaert
committed
$node = node_load($form['#nid']);
Dries Buytaert
committed
Dries Buytaert
committed
db_delete('poll_vote')
->condition('nid', $node->nid)
->condition($user->uid ? 'uid' : 'hostname', $user->uid ? $user->uid : ip_address())
->execute();
Dries Buytaert
committed
// Subtract from the votes.
Dries Buytaert
committed
db_update('poll_choice')
->expression('chvotes', 'chvotes - 1')
->condition('chid', $node->vote)
->execute();
Gerhard Killesreiter
committed
/**
Angie Byron
committed
* Implementation of hook_user_cancel().
Gerhard Killesreiter
committed
*/
function poll_user_cancel($edit, $account, $method) {
Angie Byron
committed
switch ($method) {
case 'user_cancel_reassign':
Dries Buytaert
committed
db_update('poll_vote')
->fields(array('uid' => 0))
->condition('uid', $account->uid)
->execute();
Angie Byron
committed
break;
case 'user_cancel_delete':
Dries Buytaert
committed
db_delete('poll_vote')
->condition('uid', $account->uid)
->execute();
Angie Byron
committed
break;
}
Gerhard Killesreiter
committed
}