charitable / includes / queries / class-charitable-donor-query.php

class-charitable-donor-query.php in Charitable – Donation & Fundraising Platform (Donation Forms, Recurring Donations & Fundraising Campaigns) 1.7.0.11, at includes/queries/class-charitable-donor-query.php

304 lines 9.0 KB

Raw Download Zip

1	<?php
2	/**
3	* The class responsible for querying donors.
4	*
5	* @package Charitable/Classes/Charitable_Donor_Query
6	* @version 1.0.0
7	* @author David Bisset
8	* @copyright Copyright (c) 2022, WP Charitable LLC
9	* @license http://opensource.org/licenses/gpl-2.0.php GNU Public License
10	*/
11
12	// Exit if accessed directly.
13	if ( ! defined( 'ABSPATH' ) ) {
14	exit;
15	}
16
17	if ( ! class_exists( 'Charitable_Donor_Query' ) ) :
18
19	/**
20	* Charitable_Donor_Query
21	*
22	* @since 1.0.0
23	*/
24	class Charitable_Donor_Query extends Charitable_Query {
25
26	/**
27	* Create new query object.
28	*
29	* @since 1.0.0
30	*
31	* @param array $args Query arguments.
32	*/
33	public function __construct( $args = array() ) {
34	/**
35	* Filter the default arguments for Charitable_Donor_Query.
36	*
37	* @since 1.0.0
38	*
39	* @param array $args The default arguments.
40	*/
41	$defaults = apply_filters(
42	'charitable_donor_query_default_args',
43	array(
44	'output' => 'donors',
45	'status' => array( 'charitable-completed', 'charitable-preapproved' ),
46	'orderby' => 'date',
47	'order' => 'DESC',
48	'number' => 20,
49	'paged' => 1,
50	'fields' => 'all',
51	'campaign' => 0,
52	'distinct_donors' => true,
53	'donor_id' => 0,
54	'include_erased' => 1,
55	'date_query' => array(),
56	'meta_query' => array(),
57	)
58	);
59
60	$this->args = wp_parse_args( $args, $defaults );
61	$this->args['campaign'] = $this->sanitize_campaign();
62	$this->position = 0;
63
64	$this->prepare_query();
65
66	$this->results = $this->get_donors();
67	}
68
69	/**
70	* Return list of donor IDs together with the number of donations they have made.
71	*
72	* @since 1.0.0
73	*
74	* @return object[]
75	*/
76	public function get_donors() {
77	$records = $this->query();
78
79	if ( 'donors' != $this->get( 'output' ) ) {
80	return $records;
81	}
82
83	return array_map( array( $this, 'get_donor_object' ), $records );
84	}
85
86	/**
87	* Returns a Charitable_Donor object for a database record.
88	*
89	* @since 1.5.0
90	*
91	* @param object $record Database record for the donor.
92	* @return Charitable_Donor
93	*/
94	public function get_donor_object( $record ) {
95	$donation_id = isset( $record->donation_id ) ? $record->donation_id : false;
96	$donor = new Charitable_Donor( $record->donor_id, $donation_id );
97
98	/**
99	* Filter the returned object.
100	*
101	* Note that this should always return a Charitable_Donor object.
102	*
103	* @since 1.5.7
104	*
105	* @param Charitable_Donor $donor The instance of `Charitable_Donor`.
106	* @param object $record Database record for the donor.
107	* @param Charitable_Donor_Query $query This query object.
108	*/
109	return apply_filters( 'charitable_donor_query_donor_object', $donor, $record, $this );
110	}
111
112	/**
113	* When retrieving the donor count, get distinct donor IDs.
114	*
115	* @since 1.5.4
116	*
117	* @return string
118	*/
119	public function select_donor_count() {
120	return 'COUNT(DISTINCT d.donor_id)';
121	}
122
123	/**
124	* Set up fields query argument.
125	*
126	* @since 1.0.0
127	*
128	* @return void
129	*/
130	public function setup_fields() {
131	if ( ! $this->get( 'distinct_donors', true ) ) {
132	add_filter( 'charitable_query_fields', array( $this, 'donation_fields' ), 4 );
133	}
134
135	$fields = $this->get( 'fields', 'all' );
136
137	if ( 'all' == $fields ) {
138	add_filter( 'charitable_query_fields', array( $this, 'donation_calc_fields' ), 5 );
139	add_filter( 'charitable_query_fields', array( $this, 'donor_fields' ), 6 );
140	}
141
142	if ( is_array( $fields ) ? in_array( 'amount', $fields ) : 'amount' == $fields ) {
143	add_filter( 'charitable_query_fields', array( $this, 'donation_amount_sum_field' ), 6 );
144	}
145	}
146
147	/**
148	* Set up orderby query argument.
149	*
150	* @since 1.0.0
151	*
152	* @return void
153	*/
154	public function setup_orderby() {
155	$orderby = $this->get( 'orderby', false );
156
157	if ( ! $orderby ) {
158	return;
159	}
160
161	switch ( $orderby ) {
162	case 'date':
163	add_filter( 'charitable_query_orderby', array( $this, 'orderby_date' ) );
164	break;
165
166	case 'donations':
167	add_filter( 'charitable_query_orderby', array( $this, 'orderby_count' ) );
168	break;
169
170	case 'amount':
171	add_filter( 'charitable_query_orderby', array( $this, 'orderby_donation_amount' ) );
172	break;
173
174	case 'name':
175	add_filter( 'charitable_query_orderby', array( $this, 'orderby_name' ) );
176	break;
177	}
178	}
179
180	/**
181	* Order the results by name.
182	*
183	* @since 1.5.0
184	*
185	* @return string
186	*/
187	public function orderby_name() {
188	return 'ORDER BY d.last_name, d.first_name';
189	}
190
191	/**
192	* Set up query grouping.
193	*
194	* @since 1.0.0
195	*
196	* @return void
197	*/
198	public function setup_grouping() {
199	if ( ! $this->get( 'distinct_donors', false ) ) {
200
201	add_filter( 'charitable_query_groupby', array( $this, 'groupby_ID' ) );
202	return;
203	}
204
205	add_filter( 'charitable_query_groupby', array( $this, 'groupby_donor_id' ) );
206	}
207
208	/**
209	* Do not include donors that have had their data erased.
210	*
211	* @since 1.6.0
212	*
213	* @param string $where The WHERE SQL query part.
214	* @return string
215	*/
216	public function where_donor_is_not_erased( $where ) {
217	if ( ! $this->get( 'include_erased' ) ) {
218	$where .= ' AND d.data_erased = "0000-00-00 00:00:00"';
219	}
220
221	return $where;
222	}
223
224	/**
225	* Remove any hooks that have been attached by the class to prevent contaminating other queries.
226	*
227	* @since 1.0.0
228	*
229	* @return void
230	*/
231	public function unhook_callbacks() {
232	remove_action( 'charitable_pre_query', array( $this, 'setup_fields' ) );
233	remove_action( 'charitable_pre_query', array( $this, 'setup_orderby' ) );
234	remove_action( 'charitable_pre_query', array( $this, 'setup_grouping' ) );
235	remove_filter( 'charitable_select_count_fields', array( $this, 'select_donor_count' ) );
236	remove_filter( 'charitable_query_fields', array( $this, 'donation_fields' ), 4 );
237	remove_filter( 'charitable_query_fields', array( $this, 'donation_calc_fields' ), 5 );
238	remove_filter( 'charitable_query_fields', array( $this, 'donor_fields' ), 6 );
239	remove_filter( 'charitable_query_fields', array( $this, 'donation_amount_sum_field' ), 6 );
240	remove_filter( 'charitable_query_join', array( $this, 'join_campaign_donations_table_on_donation' ), 5 );
241	remove_filter( 'charitable_query_join', array( $this, 'join_donors_table' ), 6 );
242	remove_filter( 'charitable_query_join', array( $this, 'join_meta' ), 7 );
243	remove_filter( 'charitable_query_where', array( $this, 'where_status_is_in' ), 5 );
244	remove_filter( 'charitable_query_where', array( $this, 'where_campaign_is_in' ), 6 );
245	remove_filter( 'charitable_query_where', array( $this, 'where_donor_id_is_in' ), 7 );
246	remove_filter( 'charitable_query_where', array( $this, 'where_donor_is_not_erased' ), 8 );
247	remove_filter( 'charitable_query_where', array( $this, 'where_date' ), 9 );
248	remove_filter( 'charitable_query_where', array( $this, 'where_meta' ), 10 );
249	remove_filter( 'charitable_query_groupby', array( $this, 'groupby_donor_id' ) );
250	remove_filter( 'charitable_query_orderby', array( $this, 'orderby_date' ) );
251	remove_filter( 'charitable_query_orderby', array( $this, 'orderby_count' ) );
252	remove_filter( 'charitable_query_orderby', array( $this, 'orderby_donation_amount' ) );
253	remove_filter( 'charitable_query_orderby', array( $this, 'orderby_name' ) );
254	remove_action( 'charitable_post_query', array( $this, 'unhook_callbacks' ) );
255	}
256
257	/**
258	* Set up callbacks for WP_Query filters.
259	*
260	* @since 1.0.0
261	*
262	* @return void
263	*/
264	protected function prepare_query() {
265	add_action( 'charitable_pre_query', array( $this, 'setup_fields' ) );
266	add_action( 'charitable_pre_query', array( $this, 'setup_orderby' ) );
267	add_action( 'charitable_pre_query', array( $this, 'setup_grouping' ) );
268	add_filter( 'charitable_select_count_fields', array( $this, 'select_donor_count' ) );
269	add_filter( 'charitable_query_join', array( $this, 'join_campaign_donations_table_on_donation' ), 5 );
270	add_filter( 'charitable_query_join', array( $this, 'join_donors_table' ), 6 );
271	add_filter( 'charitable_query_join', array( $this, 'join_meta' ), 7 );
272	add_filter( 'charitable_query_where', array( $this, 'where_status_is_in' ), 5 );
273	add_filter( 'charitable_query_where', array( $this, 'where_campaign_is_in' ), 6 );
274	add_filter( 'charitable_query_where', array( $this, 'where_donor_id_is_in' ), 7 );
275	add_filter( 'charitable_query_where', array( $this, 'where_donor_is_not_erased' ), 8 );
276	add_filter( 'charitable_query_where', array( $this, 'where_date' ), 9 );
277	add_filter( 'charitable_query_where', array( $this, 'where_meta' ), 10 );
278	add_action( 'charitable_post_query', array( $this, 'unhook_callbacks' ) );
279	}
280
281	/**
282	* Sanitize the campaign argument.
283	*
284	* @since 1.5.0
285	*
286	* @return int
287	*/
288	protected function sanitize_campaign() {
289	switch ( $this->args['campaign'] ) {
290	case '':
291	case 'all':
292	return 0;
293
294	case 'current':
295	return charitable_get_current_campaign_id();
296
297	default:
298	return $this->args['campaign'];
299	}
300	}
301	}
302
303	endif;
304