<?php
/**
 * TPoll Shortcode Handler
 *
 * The HTML is the same for every visitor, so full-page caches can store it.
 * Whether the current visitor has already voted is loaded afterwards via AJAX.
 *
 * @package TPoll
 */

if (!defined('ABSPATH')) {
    exit;
}

class TPoll_Shortcode {

    /**
     * Constructor
     */
    public function __construct() {
        add_shortcode('tpoll', array($this, 'render'));
    }

    /**
     * Text from the settings, or the translated default when the field is empty.
     *
     * @param string $key Setting key.
     */
    public static function text($key) {
        $settings = get_option('tpoll_settings', array());
        if (!empty($settings[$key]) && is_string($settings[$key])) {
            return $settings[$key];
        }

        switch ($key) {
            case 'button_text':
                return __('Vote', '3task-polls');
            case 'voted_text':
                return __('You have already voted.', '3task-polls');
            case 'closed_text':
                return __('This poll is closed.', '3task-polls');
        }
        return '';
    }

    /**
     * Effective display options of a poll: shortcode attribute, then poll setting, then global default.
     *
     * @param TPoll_Poll $poll Poll.
     * @param array      $atts Shortcode attributes.
     * @return array
     */
    public static function display_options($poll, $atts = array()) {
        $global = get_option('tpoll_settings', array());

        $show_results = !empty($atts['show_results']) ? $atts['show_results'] : $poll->get_setting('show_results', $global['show_results'] ?? 'after_vote');
        if (!in_array($show_results, array('after_vote', 'always', 'never'), true)) {
            $show_results = 'after_vote';
        }

        if (isset($atts['show_voters']) && $atts['show_voters'] !== '') {
            $show_voters = ($atts['show_voters'] === 'yes');
        } else {
            $show_voters = (bool) $poll->get_setting('show_voters_count', $global['show_voters_count'] ?? true);
        }

        $theme = !empty($atts['theme']) ? $atts['theme'] : $poll->get_setting('theme', $global['theme'] ?? 'default');
        if (!in_array($theme, array('default', 'minimal', 'rounded', 'square', 'pill'), true)) {
            $theme = 'default';
        }

        $dark_mode = !empty($atts['dark_mode']) ? $atts['dark_mode'] : $poll->get_setting('dark_mode', $global['dark_mode'] ?? 'no');
        if (!in_array($dark_mode, array('auto', 'yes', 'no'), true)) {
            $dark_mode = 'no';
        }

        $bar_color = sanitize_hex_color($poll->get_style('bar_color', '')) ?: sanitize_hex_color($global['results_bar_color'] ?? '');
        $bar_bg = sanitize_hex_color($global['results_bar_bg'] ?? '');
        $animation = ($global['animation'] ?? 'grow') === 'none' ? 'none' : 'grow';

        return compact('show_results', 'show_voters', 'theme', 'dark_mode', 'bar_color', 'bar_bg', 'animation');
    }

    /**
     * Whether results may be shown to the current visitor.
     *
     * @param TPoll_Poll $poll         Poll.
     * @param string     $show_results after_vote, always or never.
     * @param bool       $has_voted    Visitor has voted.
     */
    public static function results_visible($poll, $show_results, $has_voted) {
        if ($show_results === 'never') {
            return false;
        }
        if ($show_results === 'always' || $poll->is_closed()) {
            return true;
        }
        return $has_voted;
    }

    /**
     * Render shortcode
     */
    public function render($atts) {
        $atts = shortcode_atts(array(
            'id' => 0,
            'show_results' => '', // after_vote, always, never (empty = use poll setting)
            'show_voters' => '', // yes, no (empty = use poll setting)
            'theme' => '', // (empty = use poll setting)
            'dark_mode' => '' // auto, yes, no (empty = use poll setting)
        ), $atts, 'tpoll');

        $poll_id = absint($atts['id']);

        if (!$poll_id) {
            return '';
        }

        $poll = TPoll_Poll::get($poll_id);

        if (!$poll) {
            return '';
        }

        if ($poll->status === 'draft' && !current_user_can('edit_tpoll')) {
            return '';
        }

        wp_enqueue_style('tpoll-frontend');
        wp_enqueue_script('tpoll-frontend');

        $opt = self::display_options($poll, $atts);
        $restriction = $poll->get_setting('vote_restriction', 'ip');
        $is_closed = $poll->is_closed();
        $is_upcoming = !$is_closed && $poll->is_upcoming();
        $needs_login = ($restriction === 'user' && !is_user_logged_in());
        $show_form = !$is_closed && !$is_upcoming && $poll->status !== 'draft';
        $show_results_now = self::results_visible($poll, $opt['show_results'], false);

        $classes = array(
            'tpoll-poll',
            'tpoll-type-' . $poll->type,
            'tpoll-theme-' . $opt['theme'],
        );
        if ($opt['dark_mode'] === 'yes') {
            $classes[] = 'tpoll-dark';
        } elseif ($opt['dark_mode'] === 'auto') {
            $classes[] = 'tpoll-dark-auto';
        }
        if ($opt['animation'] === 'none') {
            $classes[] = 'tpoll-no-animation';
        }

        $style = '';
        if ($opt['bar_color']) {
            $style .= '--tpoll-bar:' . $opt['bar_color'] . ';';
        }
        if ($opt['bar_bg']) {
            $style .= '--tpoll-bar-bg:' . $opt['bar_bg'] . ';';
        }

        $question_id = 'tpoll-q-' . $poll_id;

        ob_start();
        ?>
        <div class="<?php echo esc_attr(implode(' ', $classes)); ?>"
             id="tpoll-<?php echo esc_attr((string) $poll_id); ?>"
             data-poll-id="<?php echo esc_attr((string) $poll_id); ?>"
             data-poll-type="<?php echo esc_attr($poll->type); ?>"
             data-show-results="<?php echo esc_attr($opt['show_results']); ?>"
             data-active="<?php echo $show_form ? '1' : '0'; ?>"
             <?php if ($style) : ?>style="<?php echo esc_attr($style); ?>"<?php endif; ?>>

            <?php if (!empty($poll->question)) : ?>
                <div class="tpoll-question" id="<?php echo esc_attr($question_id); ?>">
                    <?php echo esc_html($poll->question); ?>
                </div>
            <?php endif; ?>

            <?php if ($is_closed) : ?>
                <div class="tpoll-closed"><?php echo esc_html(self::text('closed_text')); ?></div>
            <?php elseif ($is_upcoming) : ?>
                <div class="tpoll-closed">
                    <?php
                    if (!empty($poll->start_date)) {
                        printf(
                            /* translators: %s: date and time the poll opens */
                            esc_html__('This poll opens on %s', '3task-polls'),
                            esc_html(mysql2date(get_option('date_format') . ' ' . get_option('time_format'), $poll->start_date))
                        );
                    } else {
                        esc_html_e('This poll is not open yet.', '3task-polls');
                    }
                    ?>
                </div>
            <?php endif; ?>

            <?php if ($show_form && $needs_login) : ?>
                <div class="tpoll-closed tpoll-login-required">
                    <?php esc_html_e('Please log in to vote.', '3task-polls'); ?>
                    <a href="<?php echo esc_url(wp_login_url(get_permalink() ?: home_url('/'))); ?>"><?php esc_html_e('Log in', '3task-polls'); ?></a>
                </div>
            <?php elseif ($show_form) : ?>
                <?php $this->render_poll_form($poll, $question_id, $show_results_now); ?>
                <noscript><p class="tpoll-noscript"><?php esc_html_e('Please enable JavaScript to vote.', '3task-polls'); ?></p></noscript>
            <?php endif; ?>

            <div class="tpoll-voted tpoll-hidden"><?php echo esc_html(self::text('voted_text')); ?></div>

            <div class="tpoll-results-container" aria-live="polite"<?php echo $show_results_now ? '' : ' hidden'; ?>>
                <?php
                if ($show_results_now) {
                    echo self::results_html($poll, TPoll_Vote::get_results($poll_id, $poll), $opt['show_voters']); // phpcs:ignore WordPress.Security.EscapeOutput.OutputNotEscaped -- Escaped inside results_html().
                }
                ?>
            </div>
        </div>
        <?php

        return ob_get_clean();
    }

    /**
     * Render poll form based on type
     */
    private function render_poll_form($poll, $question_id, $show_counts) {
        $answers = $poll->get_answers();

        if ($poll->get_setting('randomize_answers', false)) {
            shuffle($answers);
        }

        ?>
        <form class="tpoll-form" method="post" data-nonce="<?php echo esc_attr(wp_create_nonce('tpoll_vote')); ?>">
            <input type="hidden" name="poll_id" value="<?php echo esc_attr($poll->id); ?>">

            <?php
            switch ($poll->type) {
                case 'single':
                    $this->render_single_choice($poll, $answers, $question_id);
                    break;
                case 'multiple':
                    $this->render_multiple_choice($poll, $answers, $question_id);
                    break;
                case 'rating':
                    $this->render_rating($poll, $answers, $question_id);
                    break;
                case 'voting':
                    $this->render_voting($poll, $answers, $question_id, $show_counts);
                    break;
            }
            ?>

            <?php if (!in_array($poll->type, array('rating', 'voting'), true)) : ?>
                <button type="submit" class="tpoll-submit">
                    <?php echo esc_html(self::text('button_text')); ?>
                </button>
            <?php endif; ?>

            <div class="tpoll-message tpoll-hidden" role="status" aria-live="polite"></div>
        </form>
        <?php
    }

    /**
     * Render single choice
     */
    private function render_single_choice($poll, $answers, $question_id) {
        ?>
        <div class="tpoll-answers tpoll-single" role="radiogroup" aria-labelledby="<?php echo esc_attr($question_id); ?>">
            <?php foreach ($answers as $answer) : ?>
                <label class="tpoll-answer tpoll-radio">
                    <input type="radio"
                           name="tpoll_answer"
                           value="<?php echo esc_attr($answer->id); ?>"
                           required>
                    <span class="tpoll-checkmark" aria-hidden="true"></span>
                    <span class="tpoll-answer-text">
                        <?php echo esc_html($answer->answer_text); ?>
                    </span>
                </label>
            <?php endforeach; ?>
        </div>
        <?php
    }

    /**
     * Render multiple choice
     */
    private function render_multiple_choice($poll, $answers, $question_id) {
        $max_choices = (int) $poll->get_setting('max_choices', 0);
        ?>
        <div class="tpoll-answers tpoll-multiple" role="group" aria-labelledby="<?php echo esc_attr($question_id); ?>"
             <?php if ($max_choices) : ?>data-max-choices="<?php echo esc_attr((string) $max_choices); ?>"<?php endif; ?>>
            <?php if ($max_choices) : ?>
                <p class="tpoll-hint">
                    <?php
                        /* translators: %d: maximum number of choices allowed */
                        printf(esc_html__('Select up to %d options', '3task-polls'), absint($max_choices));
                    ?>
                </p>
            <?php endif; ?>
            <?php foreach ($answers as $answer) : ?>
                <label class="tpoll-answer tpoll-checkbox">
                    <input type="checkbox"
                           name="tpoll_answers[]"
                           value="<?php echo esc_attr($answer->id); ?>">
                    <span class="tpoll-checkmark" aria-hidden="true"></span>
                    <span class="tpoll-answer-text">
                        <?php echo esc_html($answer->answer_text); ?>
                    </span>
                </label>
            <?php endforeach; ?>
        </div>
        <?php
    }

    /**
     * Render rating (stars). Each star is a button, so it works with the keyboard.
     */
    private function render_rating($poll, $answers, $question_id) {
        $answer = isset($answers[0]) ? $answers[0] : null;
        $max_rating = 5;
        ?>
        <div class="tpoll-rating" data-max="<?php echo esc_attr((string) $max_rating); ?>">
            <input type="hidden" name="tpoll_answer" value="<?php echo $answer ? esc_attr($answer->id) : ''; ?>">
            <input type="hidden" name="tpoll_value" value="0">
            <div class="tpoll-stars" role="group" aria-labelledby="<?php echo esc_attr($question_id); ?>">
                <?php for ($i = 1; $i <= $max_rating; $i++) : ?>
                    <button type="button" class="tpoll-star" data-value="<?php echo esc_attr((string) $i); ?>"
                            aria-label="<?php
                            /* translators: 1: star value, 2: maximum stars */
                            echo esc_attr(sprintf(__('Rate %1$d of %2$d stars', '3task-polls'), $i, $max_rating));
                            ?>">
                        <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="currentColor" width="32" height="32" aria-hidden="true" focusable="false">
                            <path d="M12 2l3.09 6.26L22 9.27l-5 4.87 1.18 6.88L12 17.77l-6.18 3.25L7 14.14 2 9.27l6.91-1.01L12 2z"/>
                        </svg>
                    </button>
                <?php endfor; ?>
            </div>
            <span class="tpoll-rating-text"><?php esc_html_e('Click to rate', '3task-polls'); ?></span>
        </div>
        <?php
    }

    /**
     * Render voting (thumbs up/down)
     */
    private function render_voting($poll, $answers, $question_id, $show_counts) {
        $up = '';
        $down = '';

        if ($show_counts) {
            $results = TPoll_Vote::get_results($poll->id, $poll);
            if ($results) {
                $up = (string) $results['up'];
                $down = (string) $results['down'];
            }
        }

        $answer = isset($answers[0]) ? $answers[0] : null;
        ?>
        <div class="tpoll-voting" role="group" aria-labelledby="<?php echo esc_attr($question_id); ?>">
            <input type="hidden" name="tpoll_answer" value="<?php echo $answer ? esc_attr($answer->id) : ''; ?>">
            <input type="hidden" name="tpoll_value" value="0">
            <button type="button" class="tpoll-vote-btn tpoll-vote-up" data-value="1" aria-label="<?php esc_attr_e('Thumbs up', '3task-polls'); ?>">
                <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="currentColor" width="24" height="24" aria-hidden="true" focusable="false">
                    <path d="M2 20h2c.55 0 1-.45 1-1v-9c0-.55-.45-1-1-1H2v11zm19.83-7.12c.11-.25.17-.52.17-.8V11c0-1.1-.9-2-2-2h-5.5l.92-4.65c.05-.22.02-.46-.08-.66-.23-.45-.52-.86-.88-1.22L14 2 7.59 8.41C7.21 8.79 7 9.3 7 9.83v7.84C7 18.95 8.05 20 9.34 20h8.11c.7 0 1.36-.37 1.72-.97l2.66-6.15z"/>
                </svg>
                <span class="tpoll-count tpoll-count-up"><?php echo esc_html($up); ?></span>
            </button>
            <button type="button" class="tpoll-vote-btn tpoll-vote-down" data-value="-1" aria-label="<?php esc_attr_e('Thumbs down', '3task-polls'); ?>">
                <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="currentColor" width="24" height="24" aria-hidden="true" focusable="false">
                    <path d="M22 4h-2c-.55 0-1 .45-1 1v9c0 .55.45 1 1 1h2V4zM2.17 11.12c-.11.25-.17.52-.17.8V13c0 1.1.9 2 2 2h5.5l-.92 4.65c-.05.22-.02.46.08.66.23.45.52.86.88 1.22L10 22l6.41-6.41c.38-.38.59-.89.59-1.42V6.34C17 5.05 15.95 4 14.66 4h-8.1c-.71 0-1.36.37-1.72.97l-2.67 6.15z"/>
                </svg>
                <span class="tpoll-count tpoll-count-down"><?php echo esc_html($down); ?></span>
            </button>
        </div>
        <?php
    }

    /**
     * Results HTML, shared by the shortcode and the AJAX responses.
     *
     * @param TPoll_Poll|null $poll        Poll.
     * @param array|null      $results     Result of TPoll_Vote::get_results().
     * @param bool       $show_voters Show the voter count.
     * @return string
     */
    public static function results_html($poll, $results, $show_voters) {
        if (!$poll || !$results) {
            return '';
        }

        ob_start();
        ?>
        <div class="tpoll-results">
            <?php if ($poll->type === 'voting') : ?>
                <div class="tpoll-result-item tpoll-result-voting">
                    <span class="tpoll-result-up">
                        <?php
                        /* translators: %d: number of thumbs up */
                        printf(esc_html__('Thumbs up: %d', '3task-polls'), absint($results['up']));
                        ?>
                    </span>
                    <span class="tpoll-result-down">
                        <?php
                        /* translators: %d: number of thumbs down */
                        printf(esc_html__('Thumbs down: %d', '3task-polls'), absint($results['down']));
                        ?>
                    </span>
                </div>
            <?php elseif ($poll->type === 'rating') : ?>
                <div class="tpoll-average-rating">
                    <?php
                    printf(
                        /* translators: 1: average rating, 2: number of ratings */
                        esc_html(_n('Average rating: %1$s / 5 (%2$d rating)', 'Average rating: %1$s / 5 (%2$d ratings)', (int) $results['total_votes'], '3task-polls')),
                        esc_html(number_format_i18n((float) $results['average_rating'], 1)),
                        absint($results['total_votes'])
                    );
                    ?>
                </div>
            <?php else : ?>
                <?php foreach ($results['answers'] as $answer) : ?>
                    <div class="tpoll-result-item">
                        <div class="tpoll-result-text">
                            <span class="tpoll-result-answer"><?php echo esc_html($answer['text']); ?></span>
                            <span class="tpoll-result-percent"><?php echo esc_html(number_format_i18n((float) $answer['percentage'], 1)); ?>&nbsp;%</span>
                        </div>
                        <div class="tpoll-result-bar-wrap" aria-hidden="true">
                            <div class="tpoll-result-bar" style="width: <?php echo esc_attr((string) min(100, (float) $answer['percentage'])); ?>%"></div>
                        </div>
                        <div class="tpoll-result-votes">
                            <?php
                            printf(
                                /* translators: %d: number of votes */
                                esc_html(_n('%d vote', '%d votes', $answer['votes'], '3task-polls')),
                                absint($answer['votes'])
                            );
                            ?>
                        </div>
                    </div>
                <?php endforeach; ?>
                <?php if ($poll->type === 'multiple') : ?>
                    <p class="tpoll-result-note"><?php esc_html_e('Percentages show the share of voters who picked each answer. Voters could pick more than one.', '3task-polls'); ?></p>
                <?php endif; ?>
            <?php endif; ?>

            <?php if ($show_voters) : ?>
                <div class="tpoll-total-voters">
                    <?php
                        printf(
                            /* translators: %d: number of voters */
                            esc_html(_n('%d voter', '%d voters', $results['total_voters'], '3task-polls')),
                            absint($results['total_voters'])
                        );
                    ?>
                </div>
            <?php endif; ?>
        </div>
        <?php
        return ob_get_clean();
    }
}