wc-notice-functions.php — WooCommerce 3.3.0

woocommerce / includes / wc-notice-functions.php

woocommerce / includes Last commit date

wc-notice-functions.php

201 lines

1	<?php
2	/**
3	* WooCommerce Message Functions
4	*
5	* Functions for error/message handling and display.
6	*
7	* @package WooCommerce/Functions
8	* @version 2.1.0
9	*/
10
11	if ( ! defined( 'ABSPATH' ) ) {
12	exit;
13	}
14
15	/**
16	* Get the count of notices added, either for all notices (default) or for one.
17	* particular notice type specified by $notice_type.
18	*
19	* @since 2.1
20	* @param string $notice_type Optional. The name of the notice type - either error, success or notice.
21	* @return int
22	*/
23	function wc_notice_count( $notice_type = '' ) {
24	if ( ! did_action( 'woocommerce_init' ) ) {
25	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.3' );
26	return;
27	}
28
29	$notice_count = 0;
30	$all_notices = WC()->session->get( 'wc_notices', array() );
31
32	if ( isset( $all_notices[ $notice_type ] ) ) {
33
34	$notice_count = count( $all_notices[ $notice_type ] );
35
36	} elseif ( empty( $notice_type ) ) {
37
38	foreach ( $all_notices as $notices ) {
39	$notice_count += count( $notices );
40	}
41	}
42
43	return $notice_count;
44	}
45
46	/**
47	* Check if a notice has already been added.
48	*
49	* @since 2.1
50	* @param string $message The text to display in the notice.
51	* @param string $notice_type Optional. The name of the notice type - either error, success or notice.
52	* @return bool
53	*/
54	function wc_has_notice( $message, $notice_type = 'success' ) {
55	if ( ! did_action( 'woocommerce_init' ) ) {
56	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.3' );
57	return false;
58	}
59
60	$notices = WC()->session->get( 'wc_notices', array() );
61	$notices = isset( $notices[ $notice_type ] ) ? $notices[ $notice_type ] : array();
62	return array_search( $message, $notices, true ) !== false;
63	}
64
65	/**
66	* Add and store a notice.
67	*
68	* @since 2.1
69	* @param string $message The text to display in the notice.
70	* @param string $notice_type Optional. The name of the notice type - either error, success or notice.
71	*/
72	function wc_add_notice( $message, $notice_type = 'success' ) {
73	if ( ! did_action( 'woocommerce_init' ) ) {
74	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.3' );
75	return;
76	}
77
78	$notices = WC()->session->get( 'wc_notices', array() );
79
80	// Backward compatibility.
81	if ( 'success' === $notice_type ) {
82	$message = apply_filters( 'woocommerce_add_message', $message );
83	}
84
85	$notices[ $notice_type ][] = apply_filters( 'woocommerce_add_' . $notice_type, $message );
86
87	WC()->session->set( 'wc_notices', $notices );
88	}
89
90	/**
91	* Set all notices at once.
92	*
93	* @since 2.6.0
94	* @param mixed $notices Array of notices.
95	*/
96	function wc_set_notices( $notices ) {
97	if ( ! did_action( 'woocommerce_init' ) ) {
98	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.6' );
99	return;
100	}
101	WC()->session->set( 'wc_notices', $notices );
102	}
103
104
105	/**
106	* Unset all notices.
107	*
108	* @since 2.1
109	*/
110	function wc_clear_notices() {
111	if ( ! did_action( 'woocommerce_init' ) ) {
112	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.3' );
113	return;
114	}
115	WC()->session->set( 'wc_notices', null );
116	}
117
118	/**
119	* Prints messages and errors which are stored in the session, then clears them.
120	*
121	* @since 2.1
122	*/
123	function wc_print_notices() {
124	if ( ! did_action( 'woocommerce_init' ) ) {
125	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.3' );
126	return;
127	}
128
129	$all_notices = WC()->session->get( 'wc_notices', array() );
130	$notice_types = apply_filters( 'woocommerce_notice_types', array( 'error', 'success', 'notice' ) );
131
132	foreach ( $notice_types as $notice_type ) {
133	if ( wc_notice_count( $notice_type ) > 0 ) {
134	wc_get_template( "notices/{$notice_type}.php", array(
135	'messages' => array_filter( $all_notices[ $notice_type ] ),
136	) );
137	}
138	}
139
140	wc_clear_notices();
141	}
142	add_action( 'woocommerce_shortcode_before_product_cat_loop', 'wc_print_notices', 10 );
143	add_action( 'woocommerce_before_shop_loop', 'wc_print_notices', 10 );
144	add_action( 'woocommerce_before_single_product', 'wc_print_notices', 10 );
145
146	/**
147	* Print a single notice immediately.
148	*
149	* @since 2.1
150	* @param string $message The text to display in the notice.
151	* @param string $notice_type Optional. The singular name of the notice type - either error, success or notice.
152	*/
153	function wc_print_notice( $message, $notice_type = 'success' ) {
154	if ( 'success' === $notice_type ) {
155	$message = apply_filters( 'woocommerce_add_message', $message );
156	}
157
158	wc_get_template( "notices/{$notice_type}.php", array(
159	'messages' => array( apply_filters( 'woocommerce_add_' . $notice_type, $message ) ),
160	) );
161	}
162
163	/**
164	* Returns all queued notices, optionally filtered by a notice type.
165	*
166	* @since 2.1
167	* @param string $notice_type Optional. The singular name of the notice type - either error, success or notice.
168	* @return array\|mixed
169	*/
170	function wc_get_notices( $notice_type = '' ) {
171	if ( ! did_action( 'woocommerce_init' ) ) {
172	wc_doing_it_wrong( __FUNCTION__, __( 'This function should not be called before woocommerce_init.', 'woocommerce' ), '2.3' );
173	return;
174	}
175
176	$all_notices = WC()->session->get( 'wc_notices', array() );
177
178	if ( empty( $notice_type ) ) {
179	$notices = $all_notices;
180	} elseif ( isset( $all_notices[ $notice_type ] ) ) {
181	$notices = $all_notices[ $notice_type ];
182	} else {
183	$notices = array();
184	}
185
186	return $notices;
187	}
188
189	/**
190	* Add notices for WP Errors.
191	*
192	* @param WP_Error $errors Errors.
193	*/
194	function wc_add_wp_error_notices( $errors ) {
195	if ( is_wp_error( $errors ) && $errors->get_error_messages() ) {
196	foreach ( $errors->get_error_messages() as $error ) {
197	wc_add_notice( $error, 'error' );
198	}
199	}
200	}
201