give / src / TestData / Commands / DonationSeedCommand.php

DonationSeedCommand.php in GiveWP – Donation Plugin and Fundraising Platform 4.16.8.1, at src/TestData/Commands/DonationSeedCommand.php

145 lines 4.1 KB

Raw Download Zip

1	<?php
2
3	namespace Give\TestData\Commands;
4
5	use Exception;
6	use Give\TestData\Factories\DonationFactory as DonationFactory;
7	use Give\TestData\Repositories\DonationRepository as DonationRepository;
8	use WP_CLI;
9
10	/**
11	* Class DonationSeedCommand
12	* @package Give\TestData\Commands
13	*
14	* A WP-CLI command for seeding donations.
15	*/
16	class DonationSeedCommand
17	{
18	/**
19	* @var DonationFactory
20	*/
21	private $donationFactory;
22	/**
23	* @var DonationRepository
24	*/
25	private $donationRepository;
26
27	/**
28	* @param DonationFactory $donationFactory
29	* @param DonationRepository $donationRepository
30	*/
31	public function __construct(
32	DonationFactory $donationFactory,
33	DonationRepository $donationRepository
34	) {
35	$this->donationFactory = $donationFactory;
36	$this->donationRepository = $donationRepository;
37	}
38
39	/**
40	* Generates Donations
41	*
42	* ## OPTIONS
43	*
44	* [--count=<count>]
45	* : Number of donations to generate
46	* default: 10
47	*
48	* [--status=<status>]
49	* : Donation status
50	* default: publish
51	* options:
52	* - publish
53	* - random
54	* get all available statuses with command:
55	* wp give test-donation-statuses
56	*
57	* [--total-revenue=<amount>]
58	* : Total revenue amount to be generated
59	* default: 0
60	*
61	* [--preview=<preview>]
62	* : Preview generated data
63	* default: false
64	*
65	* [--start-date=<date>]
66	* : Set donation start date. Date format is YYYY-MM-DD
67	* default: false
68	*
69	* [--params=<params>]
70	* : Additional params
71	* default: ''
72	*
73	* [--consistent=<consistent>]
74	* : Generate consistent data
75	* default: false
76	*
77	* ## EXAMPLES
78	*
79	* wp give test-donations --count=50 --status=random --total-revenue=10000 --start-date=2020-11-22 --params=donation_currency=EUR
80	*
81	* @when after_wp_load
82	*/
83	public function __invoke($args, $assocArgs)
84	{
85	global $wpdb;
86	// Get CLI args
87	$count = WP_CLI\Utils\get_flag_value($assocArgs, 'count', $default = 10);
88	$preview = WP_CLI\Utils\get_flag_value($assocArgs, 'preview', $default = false);
89	$status = WP_CLI\Utils\get_flag_value($assocArgs, 'status', $default = 'publish');
90	$totalRevenue = WP_CLI\Utils\get_flag_value($assocArgs, 'total-revenue', $default = 0);
91	$startDate = WP_CLI\Utils\get_flag_value($assocArgs, 'start-date', $default = false);
92	$additional = WP_CLI\Utils\get_flag_value($assocArgs, 'params', $default = '');
93	$consistent = WP_CLI\Utils\get_flag_value($assocArgs, 'consistent', $default = false);
94
95	// Additional params
96	parse_str($additional, $params);
97
98	try {
99	// Factory config
100	$this->donationFactory->setDonationStatus($status);
101
102	if ($totalRevenue) {
103	$this->donationFactory->setDonationAmount(($totalRevenue / $count));
104	}
105
106	if ($startDate) {
107	$this->donationFactory->setDonationStartDate($startDate);
108	}
109
110	// Generate donations
111	$donations = $this->donationFactory->consistent($consistent)->make($count);
112	} catch (Exception $e) {
113	return WP_CLI::error($e->getMessage());
114	}
115
116	if ($preview) {
117	WP_CLI\Utils\format_items(
118	'table',
119	$donations,
120	array_keys($this->donationFactory->definition())
121	);
122	} else {
123	$progress = WP_CLI\Utils\make_progress_bar('Generating donations', $count);
124
125	// Start DB transaction
126	$wpdb->query('START TRANSACTION');
127
128	try {
129	foreach ($donations as $donation) {
130	$this->donationRepository->insertDonation($donation, $params);
131	$progress->tick();
132	}
133
134	$wpdb->query('COMMIT');
135
136	$progress->finish();
137	} catch (Exception $e) {
138	$wpdb->query('ROLLBACK');
139
140	WP_CLI::error($e->getMessage());
141	}
142	}
143	}
144	}
145