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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

<?php

/**
 * Class MC4WP_Admin_Messages
 *
 * @ignore
 * @since 3.0
 */
class MC4WP_Admin_Messages {


    /**
     * @var array
     */
    protected $bag;

    /**
     * @var bool
     */
    protected $dirty = false;

    /**
     * Add hooks
     */
    public function add_hooks() {
        add_action( 'admin_notices', array( $this, 'show' ) );
        register_shutdown_function( array( $this, 'save' ) );
    }

    private function load() {
        if ( is_null( $this->bag ) ) {
            $this->bag = get_option( 'mc4wp_flash_messages', array() );
        }
    }

    // empty flash bag
    private function reset() {
        $this->bag   = array();
        $this->dirty = true;
    }

    /**
     * Flash a message (shows on next pageload)
     *
     * @param        $message
     * @param string $type
     */
    public function flash( $message, $type = 'success' ) {
        $this->load();
        $this->bag[] = array(
            'text' => $message,
            'type' => $type,
        );
        $this->dirty = true;
    }



    /**
     * Show queued flash messages
     */
    public function show() {
        $this->load();

        foreach ( $this->bag as $message ) {
            echo sprintf( '<div class="notice notice-%s is-dismissible"><p>%s</p></div>', $message['type'], $message['text'] );
        }

        $this->reset();
    }

    /**
     * Save queued messages
     *
     * @hooked `shutdown`
     */
    public function save() {
        if ( $this->dirty ) {
            update_option( 'mc4wp_flash_messages', $this->bag, false );
        }
    }
}