give / src / TestData / Commands / FormSeedCommand.php

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

132 lines 3.6 KB

Raw Download Zip

1	<?php
2
3	namespace Give\TestData\Commands;
4
5	use Exception;
6	use Give\TestData\Factories\DonationFormFactory;
7	use Give\TestData\Repositories\DonationFormRepository;
8	use WP_CLI;
9
10	/**
11	* Class FormSeedCommand
12	* @package Give\TestData\Commands
13	*
14	* A WP-CLI command to generate Donation Forms
15	*/
16	class FormSeedCommand
17	{
18
19	/**
20	* @var DonationFormFactory
21	*/
22	private $donationFormFactory;
23	/**
24	* @var DonationFormRepository
25	*/
26	private $donationFormRepository;
27
28	public function __construct(
29	DonationFormFactory $donationFormFactory,
30	DonationFormRepository $donationFormRepository
31	) {
32	$this->donationFormFactory = $donationFormFactory;
33	$this->donationFormRepository = $donationFormRepository;
34	}
35
36	/**
37	* Generate Donation Forms
38	*
39	* ## OPTIONS
40	*
41	* [--count=<count>]
42	* : Number of donations to generate
43	* default: 10
44	*
45	* [--template=<template>]
46	* : Form template
47	* default: random
48	* options:
49	* - sequoia
50	* - legacy
51	* - random
52	*
53	* [--set-goal=<bool>]
54	* : Set donation form goal
55	* default: false
56	*
57	* [--set-terms=<bool>]
58	* : Set donation form terms and conditions
59	* default: false
60	*
61	* [--preview=<preview>]
62	* : Preview generated data
63	* default: false
64	*
65	* [--consistent=<consistent>]
66	* : Generate consistent data
67	* default: false
68	*
69	* ## EXAMPLES
70	*
71	* wp give test-donation-form --count=10 --template=legacy --set-goal=true --set-terms=true
72	*
73	* @when after_wp_load
74	*/
75	public function __invoke($args, $assocArgs)
76	{
77	global $wpdb;
78
79	// Get CLI args
80	$count = WP_CLI\Utils\get_flag_value($assocArgs, 'count', $default = 10);
81	$preview = WP_CLI\Utils\get_flag_value($assocArgs, 'preview', $default = false);
82	$template = WP_CLI\Utils\get_flag_value($assocArgs, 'template', $default = 'random');
83	$setGoal = WP_CLI\Utils\get_flag_value($assocArgs, 'set-goal', $default = false);
84	$setTerms = WP_CLI\Utils\get_flag_value($assocArgs, 'set-terms', $default = false);
85	$consistent = WP_CLI\Utils\get_flag_value($assocArgs, 'consistent', $default = false);
86
87	// Check form template
88	if ( ! $this->donationFormFactory->checkFormTemplate($template)) {
89	WP_CLI::error(
90	WP_CLI::colorize("Unsupported form template: %g{$template}%n")
91	);
92	}
93
94	// Factory config
95	$this->donationFormFactory->setFormTemplate($template);
96	$this->donationFormFactory->setDonationFormGoal($setGoal);
97	$this->donationFormFactory->setTermsAndConditions($setTerms);
98
99	// Generate donation forms
100	$forms = $this->donationFormFactory->consistent($consistent)->make($count);
101
102	if ($preview) {
103	WP_CLI\Utils\format_items(
104	'table',
105	$forms,
106	array_keys($this->donationFormFactory->definition())
107	);
108	} else {
109	$progress = WP_CLI\Utils\make_progress_bar('Generating donation forms', $count);
110
111	// Start DB transaction
112	$wpdb->query('START TRANSACTION');
113
114	try {
115	foreach ($forms as $form) {
116	$this->donationFormRepository->insertDonationForm($form);
117	$progress->tick();
118	}
119
120	$wpdb->query('COMMIT');
121
122	$progress->finish();
123	} catch (Exception $e) {
124	$wpdb->query('ROLLBACK');
125
126	WP_CLI::error($e->getMessage());
127	}
128	}
129	}
130
131	}
132