1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

<?php

namespace TotalPoll\Shortcodes;
! defined( 'ABSPATH' ) && exit();


/**
 * Poll Shortcode.
 * @package TotalPoll\Shortcode
 * @since   1.0.0
 */
class Poll extends \TotalPollVendors\TotalCore\Shortcodes\Shortcode {

    /**
     * Render shortcode.
     *
     * @return mixed|string
     */
    public function handle() {
        $poll = TotalPoll( 'polls.repository' )->getById( $this->getAttribute( 'id' ) );
        if ( ! $poll ):
            return __( 'Could not load the poll.', 'totalpoll' );
        endif;

        $screen = $this->getAttribute( 'screen', $this->getAttribute( 'fragment' ) );
        if ( $screen ):
            // Override screen when rendering
            add_filter( 'totalpoll/filters/render/screen', function ( $currentScreen, $renderedPoll ) use ( $poll, $screen ) {
                if ( $renderedPoll->getId() || $poll->getId() ) :
                    return $screen;
                endif;

                return $currentScreen;
            }, 10, 2 );

            // Override results visibility
            add_filter(
                'totalpoll/filters/poll/results-hidden',
                function ( $hidden, $renderedPoll ) use ( $poll ) {
                    if ( $renderedPoll->getId() === $poll->getId() ) :
                        return false;
                    endif;

                    return $hidden;
                },
                10, 2
            );

            // Hide buttons
            add_filter( 'totalpoll/filters/form/buttons', function ( $buttons, $renderedPoll ) use ( $poll ) {
                if ( $renderedPoll->getId() === $poll->getId() ) :
                    return [];
                endif;

                return $buttons;
            }, 10, 2 );
        endif;

        return $poll->render();
    }


}