Differential D20567 Diff 49060 src/applications/herald/management/HeraldRuleManagementWorkflow.php

Changeset View

Standalone View

src/applications/herald/management/HeraldRuleManagementWorkflow.php

This file was added.

				<?php

				final class HeraldRuleManagementWorkflow
				extends HeraldManagementWorkflow {

				protected function didConstruct() {
				$this
				->setName('rule')
				->setExamples('rule --rule __rule__ --disable')
				->setSynopsis(
				pht(
				'Modify a rule, bypassing policies. This workflow can disable '.
				'problematic personal rules.'))
				->setArguments(
				array(
				array(
				'name' => 'rule',
				'param' => 'rule',
				'help' => pht('Apply changes to this rule.'),
				),
				array(
				'name' => 'disable',
				'help' => pht('Disable the rule.'),
				),
				array(
				'name' => 'enable',
				'help' => pht('Enable the rule.'),
				),
				));
				}

				public function execute(PhutilArgumentParser $args) {
				$viewer = $this->getViewer();

				$rule_name = $args->getArg('rule');
				if (!strlen($rule_name)) {
				throw new PhutilArgumentUsageException(
				pht('Specify an rule for with "--rule".'));
				jmeadorUnsubmitted Not Done Inline Actions Not sure if there's a translation here, but the grammar seems a bit icky. jmeador: Not sure if there's a translation here, but the grammar seems a bit icky.
				amckinleyUnsubmitted Not Done Inline Actions Yeah, "a rule" seems to have been the intent. amckinley: Yeah, "a rule" seems to have been the intent.
				}

				if (preg_match('/^H\d+/', $rule_name)) {
				$rule_id = substr($rule_name, 1);
				} else {
				$rule_id = $rule_name;
				}

				$rule = id(new HeraldRuleQuery())
				->setViewer($viewer)
				->withIDs(array($rule_id))
				->executeOne();
				if (!$rule) {
				throw new PhutilArgumentUsageException(
				pht(
				'Unable to load Herald rule with ID or monogram "%s".',
				$rule_name));
				}

				$is_disable = $args->getArg('disable');
				$is_enable = $args->getArg('enable');

				$xactions = array();

				if ($is_disable && $is_enable) {
				throw new PhutilArgumentUsageException(
				pht(
				'Specify "--enable" or "--disable", but not both.'));
				} else if ($is_disable \|\| $is_enable) {
				$xactions[] = $rule->getApplicationTransactionTemplate()
				->setTransactionType(HeraldRuleDisableTransaction::TRANSACTIONTYPE)
				->setNewValue($is_disable);
				}

				if (!$xactions) {
				throw new PhutilArgumentUsageException(
				pht(
				'Use flags to specify at least one edit to apply to the '.
				'rule (for exmaple, use "--disable" to disable a rule).'));
				jmeadorUnsubmitted Not Done Inline Actions Assuming `exmaple` is the British spelling of `example`. jmeador: Assuming `exmaple` is the British spelling of `example`.
				}

				$herald_phid = id(new PhabricatorHeraldApplication())->getPHID();

				$editor = $rule->getApplicationTransactionEditor()
				->setActor($viewer)
				->setActingAsPHID($herald_phid)
				->setContentSource($this->newContentSource())
				->setContinueOnMissingFields(true)
				->setContinueOnNoEffect(true);

				echo tsprintf(
				"%s\n",
				pht(
				'Applying changes to %s: %s...',
				$rule->getMonogram(),
				$rule->getName()));

				$editor->applyTransactions($rule, $xactions);

				echo tsprintf(
				"%s\n",
				pht('Done.'));


				return 0;
				}

				}