give / src / Campaigns / Migrations / Donations / AddCampaignId.php

AddCampaignId.php in GiveWP – Donation Plugin and Fundraising Platform 4.18.0, at src/Campaigns/Migrations/Donations/AddCampaignId.php

177 lines 4.3 KB

Raw Download Zip

1	<?php
2
3	namespace Give\Campaigns\Migrations\Donations;
4
5	use Give\Donations\ValueObjects\DonationMetaKeys;
6	use Give\Framework\Database\DB;
7	use Give\Framework\Database\Exceptions\DatabaseQueryException;
8	use Give\Framework\Migrations\Contracts\BatchMigration;
9	use Give\Framework\Migrations\Contracts\ReversibleMigration;
10	use Give\Framework\Migrations\Exceptions\DatabaseMigrationException;
11	use Give\Framework\QueryBuilder\QueryBuilder;
12
13	/**
14	* @since 4.0.0
15	*/
16	class AddCampaignId extends BatchMigration implements ReversibleMigration
17	{
18	/**
19	* @inheritDoc
20	*/
21	public static function id(): string
22	{
23	return 'add_campaign_id_to_donations';
24	}
25
26	/**
27	* @inheritDoc
28	*/
29	public static function title(): string
30	{
31	return 'Add campaign id to donations';
32	}
33
34	/**
35	* @inheritdoc
36	*/
37	public static function timestamp(): string
38	{
39	return strtotime('2024-11-22 00:00:00');
40	}
41
42	/**
43	* Base query
44	*
45	* @since 4.0.0
46	*/
47	protected function query(): QueryBuilder
48	{
49	return DB::table('posts')->where('post_type', 'give_payment');
50	}
51
52	/**
53	* @inheritDoc
54	*
55	* @since 4.0.0
56	*
57	* @throws DatabaseMigrationException
58	*/
59	public function runBatch($firstId, $lastId)
60	{
61	$relationships = [];
62
63	try {
64	$data = DB::table('give_campaign_forms')
65	->select('campaign_id', 'form_id')
66	->getAll();
67
68	foreach ($data as $relationship) {
69	$relationships[$relationship->campaign_id][] = $relationship->form_id;
70	}
71
72	$query = $this->query()
73	->select('ID')
74	->attachMeta(
75	'give_donationmeta',
76	'ID',
77	'donation_id',
78	[DonationMetaKeys::FORM_ID(), 'formId']
79	);
80
81	// Migration Runner will pass null for lastId in the last step
82	if (is_null($lastId)) {
83	$query->where('ID', $firstId, '>');
84	} else {
85	$query->whereBetween('ID', $firstId, $lastId);
86	}
87
88	$donations = $query->getAll();
89
90	$donationMeta = [];
91
92	foreach ($donations as $donation) {
93	foreach ($relationships as $campaignId => $formIds) {
94	if (in_array($donation->formId, $formIds)) {
95	$donationMeta[] = [
96	'donation_id' => $donation->ID,
97	'meta_key' => DonationMetaKeys::CAMPAIGN_ID,
98	'meta_value' => $campaignId,
99	];
100
101	break;
102	}
103	}
104	}
105
106	if ( ! empty($donationMeta)) {
107	DB::table('give_donationmeta')
108	->insert($donationMeta, ['%d', '%s', '%d']);
109	}
110	} catch (DatabaseQueryException $exception) {
111	throw new DatabaseMigrationException("An error occurred while adding campaign ID to the donation meta table",
112	0, $exception);
113	}
114	}
115
116	/**
117	* @inheritDoc
118	*/
119	public function getItemsCount(): int
120	{
121	return $this->query()->count();
122	}
123
124	/**
125	* @inheritDoc
126	*/
127	public function getBatchItemsAfter($lastId): ?array
128	{
129	$item = DB::get_row(sprintf(
130	'SELECT MIN(ID) AS first_id, MAX(ID) AS last_id FROM (SELECT ID FROM %1s WHERE ID > %d ORDER BY ID ASC LIMIT %d) as batch',
131	DB::prefix('posts'),
132	$lastId,
133	$this->getBatchSize()
134	));
135
136	if ( ! $item) {
137	return null;
138	}
139
140	return [
141	$item->first_id,
142	$item->last_id,
143	];
144	}
145
146	/**
147	* @inheritDoc
148	*/
149	public function getBatchSize(): int
150	{
151	return 100;
152	}
153
154	/**
155	* @inheritDoc
156	*/
157	public function hasMoreItemsToBatch($lastProcessedId): ?bool
158	{
159	return $this->query()
160	->where('ID', $lastProcessedId, '>')
161	->limit(1)
162	->count();
163	}
164
165	/**
166	* @inheritDoc
167	*
168	* @since 4.6.0
169	*/
170	public function reverse(): void
171	{
172	DB::table('give_donationmeta')
173	->where('meta_key', DonationMetaKeys::CAMPAIGN_ID)
174	->delete();
175	}
176	}
177