root/wp-polls/trunk/wp-polls.php

<?php
/*
Plugin Name: WP-Polls
Plugin URI: http://lesterchan.net/portfolio/programming/php/
Description: Adds an AJAX poll system to your WordPress blog. You can easily include a poll into your WordPress's blog post/page. WP-Polls is extremely customizable via templates and css styles and there are tons of options for you to choose to ensure that WP-Polls runs the way you wanted. It now supports multiple selection of answers.
Version: 2.31
Author: Lester 'GaMerZ' Chan
Author URI: http://lesterchan.net
*/


/*  
    Copyright 2008  Lester Chan  (email : lesterchan@gmail.com)

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/


### Load WP-Config File If This File Is Called Directly
if (!function_exists('add_action')) {
    $wp_root = '../../..';
    if (file_exists($wp_root.'/wp-load.php')) {
        require_once($wp_root.'/wp-load.php');
    } else {
        require_once($wp_root.'/wp-config.php');
    }
}


### Use WordPress 2.6 Constants
if (!defined('WP_CONTENT_DIR')) {
    define( 'WP_CONTENT_DIR', ABSPATH.'wp-content');
}
if (!defined('WP_CONTENT_URL')) {
    define('WP_CONTENT_URL', get_option('siteurl').'/wp-content');
}
if (!defined('WP_PLUGIN_DIR')) {
    define('WP_PLUGIN_DIR', WP_CONTENT_DIR.'/plugins');
}
if (!defined('WP_PLUGIN_URL')) {
    define('WP_PLUGIN_URL', WP_CONTENT_URL.'/plugins');
}


### Create Text Domain For Translations
add_action('init', 'polls_textdomain');
function polls_textdomain() {
    if (!function_exists('wp_print_styles')) {
        load_plugin_textdomain('wp-polls', 'wp-content/plugins/wp-polls');
    } else {
        load_plugin_textdomain('wp-polls', false, 'wp-polls');
    }
}


### Polls Table Name
global $wpdb;
$wpdb->pollsq = $wpdb->prefix.'pollsq';
$wpdb->pollsa = $wpdb->prefix.'pollsa';
$wpdb->pollsip     = $wpdb->prefix.'pollsip';


### Function: Poll Administration Menu
add_action('admin_menu', 'poll_menu');
function poll_menu() {
    if (function_exists('add_menu_page')) {
        add_menu_page(__('Polls', 'wp-polls'), __('Polls', 'wp-polls'), 'manage_polls', 'wp-polls/polls-manager.php');
    }
    if (function_exists('add_submenu_page')) {
        add_submenu_page('wp-polls/polls-manager.php', __('Manage Polls', 'wp-polls'), __('Manage Polls', 'wp-polls'), 'manage_polls', 'wp-polls/polls-manager.php');
        add_submenu_page('wp-polls/polls-manager.php', __('Add Poll', 'wp-polls'), __('Add Poll', 'wp-polls'), 'manage_polls', 'wp-polls/polls-add.php');        
        add_submenu_page('wp-polls/polls-manager.php', __('Poll Options', 'wp-polls'), __('Poll Options', 'wp-polls'), 'manage_polls', 'wp-polls/polls-options.php');
        add_submenu_page('wp-polls/polls-manager.php', __('Poll Templates', 'wp-polls'), __('Poll Templates', 'wp-polls'), 'manage_polls', 'wp-polls/polls-templates.php');
        add_submenu_page('wp-polls/polls-manager.php', __('Uninstall WP-Polls', 'wp-polls'), __('Uninstall WP-Polls', 'wp-polls'), 'manage_polls', 'wp-polls/polls-uninstall.php');
    }
}


### Function: Get Poll
function get_poll($temp_poll_id = 0, $display = true) {
    global $wpdb, $polls_loaded;
    // Poll Result Link
    $pollresult_id = intval($_GET['pollresult']);
    // Check Whether Poll Is Disabled
    if(intval(get_option('poll_currentpoll')) == -1) {
        if($display) {
            echo stripslashes(get_option('poll_template_disable'));
            return;
        } else {
            return stripslashes(get_option('poll_template_disable'));
        }        
    // Poll Is Enabled
    } else {
        // Hardcoded Poll ID Is Not Specified
        if(intval($temp_poll_id) == 0) {
            // Random Poll
            if(intval(get_option('poll_currentpoll')) == -2) {
                $random_poll_id = $wpdb->get_var("SELECT pollq_id FROM $wpdb->pollsq WHERE pollq_active = 1 ORDER BY RAND() LIMIT 1");
                $poll_id = intval($random_poll_id);
                if($pollresult_id > 0) {
                    $poll_id = $pollresult_id;
                } elseif(intval($_POST['poll_id']) > 0) {
                    $poll_id = intval($_POST['poll_id']);
                }
            // Current Poll ID Is Not Specified
            } elseif(intval(get_option('poll_currentpoll')) == 0) {
                // Get Lastest Poll ID
                $poll_id = intval(get_option('poll_latestpoll'));
            } else {
                // Get Current Poll ID
                $poll_id = intval(get_option('poll_currentpoll'));
            }
        // Get Hardcoded Poll ID
        } else {
            $poll_id = intval($temp_poll_id);
        }
    }
    
    // Assign All Loaded Poll To $polls_loaded
    if(empty($polls_loaded)) {
        $polls_loaded = array();
    }
    if(!in_array($poll_id, $polls_loaded)) {
        $polls_loaded[] = $poll_id;
    }

    // User Click on View Results Link
    if($pollresult_id == $poll_id) {
        if($display) {
            echo display_pollresult($poll_id);
            return;
        } else {
            return display_pollresult($poll_id);
        }
    // Check Whether User Has Voted
    } else {
        $poll_active = $wpdb->get_var("SELECT pollq_active FROM $wpdb->pollsq WHERE pollq_id = $poll_id");
        $poll_active = intval($poll_active);
        $check_voted = check_voted($poll_id);
        if($poll_active == 0) {
            $poll_close = intval(get_option('poll_close'));
        } else {
            $poll_close = 0;
        }
        if($check_voted > 0 || ($poll_active == 0 && $poll_close == 1)) {
            if($display) {
                echo display_pollresult($poll_id, $check_voted);
                return;
            } else {
                return display_pollresult($poll_id, $check_voted);
            }
        } elseif(!check_allowtovote() || ($poll_active == 0 && $poll_close == 3)) {
            $disable_poll_js = '<script type="text/javascript">poll_disable_voting('.$poll_id.');</script>';
            if($display) {
                echo display_pollvote($poll_id).$disable_poll_js;
                return;
            } else {
                return display_pollvote($poll_id).$disable_poll_js;
            }            
        } elseif($poll_active == 1) {
            if($display) {
                echo display_pollvote($poll_id);
                return;
            } else {
                return display_pollvote($poll_id);
            }
        }
    }    
}


### Function: Displays Polls Header
add_action('wp_head', 'poll_header');
function poll_header() {    
    $poll_ajax_style = get_option('poll_ajax_style');
    $pollbar = get_option('poll_bar');
    wp_register_script('wp-polls', WP_PLUGIN_URL.'/wp-polls/polls-js-packed.js', false, '2.31');
    echo "\n".'<!-- Start Of Script Generated By WP-Polls 2.31 -->'."\n";
    echo '<script type="text/javascript">'."\n";
    echo '/* <![CDATA[ */'."\n";
    echo "\t".'var polls_ajax_url = \''.WP_PLUGIN_URL.'/wp-polls/wp-polls.php'."';\n";
    echo "\t".'var polls_text_wait = \''.js_escape(__('Your last request is still being processed. Please wait a while ...', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_text_valid = \''.js_escape(__('Please choose a valid poll answer.', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_text_multiple = \''.js_escape(__('Maximum number of choices allowed:', 'wp-polls')).'\';'."\n";
    echo "\t".'var poll_show_loading = '.intval($poll_ajax_style['loading']).';'."\n";
    echo "\t".'var poll_show_fading = '.intval($poll_ajax_style['fading']).';'."\n";
    echo '/* ]]> */'."\n";
    echo '</script>'."\n";
    wp_print_scripts(array('sack', 'wp-polls'));
    if(@file_exists(TEMPLATEPATH.'/polls-css.css')) {
        echo '<link rel="stylesheet" href="'.get_stylesheet_directory_uri().'/polls-css.css" type="text/css" media="screen" />'."\n";    
    } else {
        echo '<link rel="stylesheet" href="'.WP_PLUGIN_URL.'/wp-polls/polls-css.css" type="text/css" media="screen" />'."\n";    
    }
    echo '<style type="text/css">'."\n";    
    if($pollbar['style'] == 'use_css') {
        echo '.wp-polls .pollbar {'."\n";
        echo "\t".'margin: 1px;'."\n";
        echo "\t".'font-size: '.($pollbar['height']-2).'px;'."\n";
        echo "\t".'line-height: '.$pollbar['height'].'px;'."\n";
        echo "\t".'height: '.$pollbar['height'].'px;'."\n";
        echo "\t".'background: #'.$pollbar['background'].';'."\n";
        echo "\t".'border: 1px solid #'.$pollbar['border'].';'."\n";
        echo '}'."\n";
    } else {
        echo '.wp-polls .pollbar {'."\n";
        echo "\t".'margin: 1px;'."\n";
        echo "\t".'font-size: '.($pollbar['height']-2).'px;'."\n";
        echo "\t".'line-height: '.$pollbar['height'].'px;'."\n";
        echo "\t".'height: '.$pollbar['height'].'px;'."\n";
        echo "\t".'background-image: url(\''.WP_PLUGIN_URL.'/wp-polls/images/'.$pollbar['style'].'/pollbg.gif\');'."\n";
        echo "\t".'border: 1px solid #'.$pollbar['border'].';'."\n";
        echo '}'."\n";
    }
    echo '</style>'."\n";
    echo '<!-- End Of Script Generated By WP-Polls 2.31 -->'."\n";
}


### Function: Displays Polls Header In WP-Admin
add_action('admin_head', 'poll_header_admin');
function poll_header_admin() {
    wp_register_script('wp-polls-admin', WP_PLUGIN_URL.'/wp-polls/polls-admin-js-packed.js', false, '2.31');
    echo "\n".'<!-- Start Of Script Generated By WP-Polls 2.31 -->'."\n";
    echo '<script type="text/javascript">'."\n";
    echo '/* <![CDATA[ */'."\n";
    echo "\t".'var polls_admin_ajax_url = \''.WP_PLUGIN_URL.'/wp-polls/polls-admin-ajax.php'."';\n";
    echo "\t".'var polls_admin_text_delete_poll = \''.js_escape(__('Delete Poll', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_no_poll_logs = \''.js_escape(__('No poll logs available.', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_delete_all_logs = \''.js_escape(__('Delete All Logs', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_checkbox_delete_all_logs = \''.js_escape(__('Please check the \\\'Yes\\\' checkbox if you want to delete all logs.', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_delete_poll_logs = \''.js_escape(__('Delete Logs For This Poll Only', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_checkbox_delete_poll_logs = \''.js_escape(__('Please check the \\\'Yes\\\' checkbox if you want to delete all logs for this poll ONLY.', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_delete_poll_ans = \''.js_escape(__('Delete Poll Answer', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_open_poll = \''.js_escape(__('Open Poll', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_close_poll = \''.js_escape(__('Close Poll', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_enter_poll_id = \''.js_escape(__('Enter Poll ID', 'wp-polls')).'\';'."\n";
    echo "\t".'var polls_admin_text_enter_poll_id_again = \''.js_escape(__('Error: Poll ID must be numeric', 'wp-polls')).'\n\n'.js_escape(__('Please enter Poll ID again', 'wp-polls')).'\';'."\n";
    echo '/* ]]> */'."\n";
    echo '</script>'."\n";
    wp_print_scripts(array('sack', 'wp-polls-admin'));
    echo '<link rel="stylesheet" href="'.WP_PLUGIN_URL.'/wp-polls/polls-css.css" type="text/css" media="screen" />'."\n";
    echo '<!-- End Of Script Generated By WP-Polls 2.31 -->'."\n";
}


### Function: Displays Polls Footer In WP-Admin
add_action('admin_footer', 'poll_footer_admin');
function poll_footer_admin() {
    // Javascript Code Courtesy Of WP-AddQuicktag (http://bueltge.de/wp-addquicktags-de-plugin/120/)
    echo '<script type="text/javascript">'."\n";
    echo "\t".'if(document.getElementById("ed_toolbar")){'."\n";
    echo "\t\t".'qt_toolbar = document.getElementById("ed_toolbar");'."\n";
    echo "\t\t".'edButtons[edButtons.length] = new edButton("ed_poll","'.__('Poll', 'wp-polls').'", "", "","");'."\n";
    echo "\t\t".'var qt_button = qt_toolbar.lastChild;'."\n";
    echo "\t\t".'while (qt_button.nodeType != 1){'."\n";
    echo "\t\t\t".'qt_button = qt_button.previousSibling;'."\n";
    echo "\t\t".'}'."\n";
    echo "\t\t".'qt_button = qt_button.cloneNode(true);'."\n";
    echo "\t\t".'qt_button.value = "'.__('Poll', 'wp-polls').'";'."\n";
    echo "\t\t".'qt_button.title = "'.__('Insert Poll', 'wp-polls').'";'."\n";
    echo "\t\t".'qt_button.onclick = function () { insertPoll(\'code\', edCanvas);}'."\n";
    echo "\t\t".'qt_button.id = "ed_poll";'."\n";
    echo "\t\t".'qt_toolbar.appendChild(qt_button);'."\n";
    echo "\t".'}'."\n";
    echo '</script>'."\n";
}


### Function: Add Quick Tag For Poll In TinyMCE >= WordPress 2.5
add_action('init', 'poll_tinymce_addbuttons');
function poll_tinymce_addbuttons() {
    if(!current_user_can('edit_posts') && ! current_user_can('edit_pages')) {
        return;
    }
    if(get_user_option('rich_editing') == 'true') {
        add_filter("mce_external_plugins", "poll_tinymce_addplugin");
        add_filter('mce_buttons', 'poll_tinymce_registerbutton');
    }
}
function poll_tinymce_registerbutton($buttons) {
    array_push($buttons, 'separator', 'polls');
    return $buttons;
}
function poll_tinymce_addplugin($plugin_array) {
    $plugin_array['polls'] = WP_PLUGIN_URL.'/wp-polls/tinymce/plugins/polls/editor_plugin.js';
    return $plugin_array;
}


### Function: Check Who Is Allow To Vote
function check_allowtovote() {
    global $user_ID;
    $user_ID = intval($user_ID);
    $allow_to_vote = intval(get_option('poll_allowtovote'));
    switch($allow_to_vote) {
        // Guests Only
        case 0:
            if($user_ID > 0) {
                return false;
            }
            return true;
            break;
        // Registered Users Only
        case 1:
            if($user_ID == 0) {
                return false;
            }
            return true;
            break;
        // Registered Users And Guests
        case 2:
        default:
            return true;
    }
}


### Funcrion: Check Voted By Cookie Or IP
function check_voted($poll_id) {
    $poll_logging_method = intval(get_option('poll_logging_method'));
    switch($poll_logging_method) {
        // Do Not Log
        case 0:
            return 0;
            break;
        // Logged By Cookie
        case 1:
            return check_voted_cookie($poll_id);
            break;
        // Logged By IP
        case 2:
            return check_voted_ip($poll_id);
            break;
        // Logged By Cookie And IP
        case 3:
            $check_voted_cookie = check_voted_cookie($poll_id);
            if(!empty($check_voted_cookie)) {
                return $check_voted_cookie;
            } else {
                return check_voted_ip($poll_id);
            }
            break;
        // Logged By Username
        case 4:
            return check_voted_username($poll_id);
            break;
    }
}


### Function: Check Voted By Cookie
function check_voted_cookie($poll_id) {
    if(!empty($_COOKIE["voted_$poll_id"])) {
        $get_voted_aids = explode(',', $_COOKIE["voted_$poll_id"]);
    } else {
        $get_voted_aids = 0;
    }
    return $get_voted_aids;
}


### Function: Check Voted By IP
function check_voted_ip($poll_id) {
    global $wpdb;
    $log_expiry = intval(get_option('poll_cookielog_expiry'));
    $log_expiry_sql = '';
    if($log_expiry > 0) {
        $log_expiry_sql = 'AND ('.current_time('timestamp').'-(pollip_timestamp+0)) < '.$log_expiry;
    }
    // Check IP From IP Logging Database
    $get_voted_aids = $wpdb->get_col("SELECT pollip_aid FROM $wpdb->pollsip WHERE pollip_qid = $poll_id AND pollip_ip = '".get_ipaddress()."' $log_expiry_sql");
    if($get_voted_aids) {
        return $get_voted_aids;
    } else {
        return 0;
    }
}


### Function: Check Voted By Username
function check_voted_username($poll_id) {
    global $wpdb, $user_ID;
    // Check IP If User Is Guest
    if ($user_ID == 0) {
        return check_voted_ip($poll_id);
    }
    $pollsip_userid = intval($user_ID);
    $log_expiry = intval(get_option('poll_cookielog_expiry'));
    $log_expiry_sql = '';
    if($log_expiry > 0) {
        $log_expiry_sql = 'AND ('.current_time('timestamp').'-(pollip_timestamp+0)) < '.$log_expiry;
    }
    // Check User ID From IP Logging Database
    $get_voted_aids = $wpdb->get_col("SELECT pollip_aid FROM $wpdb->pollsip WHERE pollip_qid = $poll_id AND pollip_userid = $pollsip_userid $log_expiry_sql");
    if($get_voted_aids) {
        return $get_voted_aids;
    } else {
        return 0;
    }
}


### Function: Display Voting Form
function display_pollvote($poll_id, $without_poll_title = false) {
    global $wpdb;
    // Temp Poll Result
    $temp_pollvote = '';
    // Get Poll Question Data
    $poll_question = $wpdb->get_row("SELECT pollq_id, pollq_question, pollq_totalvotes, pollq_timestamp, pollq_expiry, pollq_multiple, pollq_totalvoters FROM $wpdb->pollsq WHERE pollq_id = $poll_id LIMIT 1");
    // Poll Question Variables
    $poll_question_text = stripslashes($poll_question->pollq_question);
    $poll_question_id = intval($poll_question->pollq_id);
    $poll_question_totalvotes = intval($poll_question->pollq_totalvotes);
    $poll_question_totalvoters = intval($poll_question->pollq_totalvoters);
    $poll_start_date = mysql2date(sprintf(__('%s @ %s', 'wp-polls'), get_option('date_format'), get_option('time_format')), gmdate('Y-m-d H:i:s', $poll_question->pollq_timestamp)); 
    $poll_expiry = trim($poll_question->pollq_expiry);
    if(empty($poll_expiry)) {
        $poll_end_date  = __('No Expiry', 'wp-polls');
    } else {
        $poll_end_date  = mysql2date(sprintf(__('%s @ %s', 'wp-polls'), get_option('date_format'), get_option('time_format')), gmdate('Y-m-d H:i:s', $poll_expiry));
    }
    $poll_multiple_ans = intval($poll_question->pollq_multiple);
    $template_question = stripslashes(get_option('poll_template_voteheader'));
    $template_question = str_replace("%POLL_QUESTION%", $poll_question_text, $template_question);
    $template_question = str_replace("%POLL_ID%", $poll_question_id, $template_question);
    $template_question = str_replace("%POLL_TOTALVOTES%", $poll_question_totalvotes, $template_question);
    $template_question = str_replace("%POLL_TOTALVOTERS%", $poll_question_totalvoters, $template_question);
    $template_question = str_replace("%POLL_START_DATE%", $poll_start_date, $template_question);
    $template_question = str_replace("%POLL_END_DATE%", $poll_end_date, $template_question);
    if($poll_multiple_ans > 0) {
        $template_question = str_replace("%POLL_MULTIPLE_ANS_MAX%", $poll_multiple_ans, $template_question);
    } else {
        $template_question = str_replace(