Differential D16566 Diff 39875 src/workflow/ArcanistLeaseWorkflow.php

Changeset View

Standalone View

src/workflow/ArcanistLeaseWorkflow.php

This file was added.

				<?php

				final class ArcanistLeaseWorkflow extends ArcanistWorkflow {

				public function getWorkflowName() {
				return 'lease';
				}

				public function getCommandSynopses() {
				return phutil_console_format(<<<EOTEXT
				lease [__options__]
				EOTEXT
				);
				}

				public function getCommandHelp() {
				return phutil_console_format(<<<EOTEXT

				Lease resources from Drydock.

				When given no arguments, this workflow will show you any blueprints
				which you have requested authorization to use, and your authorization
				state for each one.

				When given a blueprint ID, this workflow will attempt to create a
				lease using the specified blueprint. If you are not yet authorized
				to use the blueprint, you will be prompted to request authorization.

				Lease attributes should be supplied as a path to a file containing
				JSON, or you can specify - and pipe JSON to the workflow over
				stdin. See documentation for the conduit method 'drydock.createlease'
				for more information about what attributes you can supply.

				Successfully created leases may not be immediately active after the
				workflow exits. You can instruct this workflow to block, waiting
				until the lease is active, by supplying the --wait argument.
				EOTEXT
				);
				}

				public function getArguments() {
				return array(
				'blueprint' => array(
				'help' => pht('ID of the blueprint to use when acquiring a lease.'),
				'param' => 'id',
				),
				'wait' => array(
				'help' => pht('Block until the lease is active.'),
				),
				'*' => 'attributes',
				);
				}

				public function requiresAuthentication() {
				return true;
				}

				public function run() {
				if (!$this->getPassedArguments()) {
				$this->listAuthorizations();
				return 0;
				}

				$attributes = $this->getArgument('attributes');
				if (count($attributes)) {
				$file = head($attributes);
				if ($file === '-') {
				$json = file_get_contents('php://stdin');
				} else {
				$json = Filesystem::readFile($file);
				}
				$attributes = phutil_json_decode($json);
				}

				$blueprint_id = (int)$this->getArgument('blueprint');
				if (!$blueprint_id) {
				throw new ArcanistUsageException('You must provide a blueprint ID.');
				}
				$blueprint = $this->getConduit()->callMethodSynchronous(
				'drydock.blueprint.search',
				array(
				'constraints' => array(
				'ids' => array($blueprint_id),
				),
				));
				$blueprint = $blueprint['data'][0];
				try {
				$lease = $this->getConduit()->callMethodSynchronous(
				'drydock.createlease',
				array(
				'blueprintPHID' => $blueprint['phid'],
				'leaseAttributes' => $attributes,
				));
				} catch (ConduitClientException $e) {
				if ($e->getErrorCode() === 'ERR_NOT_AUTHORIZED') {
				if (phutil_console_confirm(pht(
				'You are not authorized to use this blueprint, request '.
				'authorization?'))) {
				$this->requestAuthorization($blueprint['phid']);
				return 1;
				}
				}
				throw $e;
				}

				echo phutil_console_format(
				"%s\n %s __%s__\n\n",
				pht('Lease queued for activation.'),
				pht('Lease URI:'),
				rtrim($this->getConfigFromAnySource('phabricator.uri'), '/').'/'.
				'drydock/lease/'.$lease['id'].'/');

				if ($this->getArgument('wait')) {
				$this->waitUntilActive($lease['phid']);
				}

				return 0;
				}

				private function waitUntilActive($lease_phid) {
				$lease = $this->getConduit()->callMethodSynchronous(
				'drydock.lease.search',
				array(
				'constraints' => array(
				'phids' => array($lease_phid),
				),
				));
				$lease_status = idxv($lease, array('data', 0, 'fields', 'status', 'value'));
				switch ($lease_status) {
				case 'pending':
				case 'acquired':
				sleep(1);
				$this->waitUntilActive($lease_phid);
				break;
				case 'active':
				return;
				case 'released':
				case 'broken':
				case 'destroyed':
				throw new Exception(pht('Lease failed to activate.'));
				}
				}

				private function requestAuthorization($blueprint_phid) {
				$console = PhutilConsole::getConsole();
				$authorization = $this->getConduit()->callMethodSynchronous(
				'drydock.requestauthorization',
				array(
				'blueprintPHID' => $blueprint_phid,
				));
				echo phutil_console_format(
				"%s\n %s __%s__\n\n",
				pht('Authorization request issued.'),
				pht('Request URI:'),
				rtrim($this->getConfigFromAnySource('phabricator.uri'), '/').'/'.
				'drydock/authorization/'.$authorization['id'].'/');
				}

				private function listAuthorizations() {
				$console = PhutilConsole::getConsole();
				$console->writeOut("<bg:blue> %s </bg>\n", pht('BLUEPRINT AUTHORIZATIONS'));
				$conduit = $this->getConduit();
				$authorizations = $conduit->callMethodSynchronous(
				'drydock.authorization.search',
				array(
				'constraints' => array(
				'objectPHIDs' => array($this->getUserPHID()),
				),
				));
				$authorizations = idx($authorizations, 'data');

				if (!$authorizations) {
				$console->writeOut("\t%s\n", pht('No authorizations found.'));
				}

				$authorization_fields = ipull($authorizations, 'fields');
				$blueprint_phids = ipull($authorization_fields, 'blueprintPHID');

				$blueprints = $conduit->callMethodSynchronous(
				'drydock.blueprint.search',
				array(
				'constraints' => array(
				'phids' => $blueprint_phids,
				),
				));
				$blueprints = idx($blueprints, 'data');
				$blueprints = ipull($blueprints, null, 'phid');

				$table = (new PhutilConsoleTable())
				->addColumn('id', array('title' => 'ID'))
				->addColumn('name', array('title' => 'Name'))
				->addColumn('type', array('title' => 'Type'))
				->addColumn('authorization', array('title' => 'Authorization'))
				->setBorders(true);

				foreach ($authorizations as $authorization) {
				$blueprint_phid = idxv($authorization, array('fields', 'blueprintPHID'));
				$blueprint = idx($blueprints, $blueprint_phid);
				$table->addRow(array(
				'id' => idx($blueprint, 'id'),
				'name' => idxv($blueprint, array('fields', 'name')),
				'type' => idxv($blueprint, array('fields', 'type')),
				'authorization' => idxv($authorization, array(
				'fields',
				'blueprintAuthorizationState',
				'name')),
				Lint: Array Separator Inline Actions Multi-lined arrays should have trailing commas. Lint: Array Separator: Multi-lined arrays should have trailing commas.
				));
				}

				$table->draw();
				}

				}