Page MenuHomePhabricator
Files F15464158

D12783.id30729.diff
No OneTemporary
Actions

D12783.id30729.diff
View Options

diff --git a/src/__phutil_library_map__.php b/src/__phutil_library_map__.php
--- a/src/__phutil_library_map__.php
+++ b/src/__phutil_library_map__.php
@@ -696,6 +696,8 @@
'DrydockBlueprintCoreCustomField' => 'applications/drydock/customfield/DrydockBlueprintCoreCustomField.php',
'DrydockBlueprintCreateController' => 'applications/drydock/controller/DrydockBlueprintCreateController.php',
'DrydockBlueprintCustomField' => 'applications/drydock/customfield/DrydockBlueprintCustomField.php',
+ 'DrydockBlueprintCustomFieldBlueprints' => 'applications/drydock/customfield/DrydockBlueprintCustomFieldBlueprints.php',
+ 'DrydockBlueprintDatasource' => 'applications/drydock/typeahead/DrydockBlueprintDatasource.php',
'DrydockBlueprintEditController' => 'applications/drydock/controller/DrydockBlueprintEditController.php',
'DrydockBlueprintEditor' => 'applications/drydock/editor/DrydockBlueprintEditor.php',
'DrydockBlueprintImplementation' => 'applications/drydock/blueprint/DrydockBlueprintImplementation.php',
@@ -877,6 +879,7 @@
'HarbormasterDAO' => 'applications/harbormaster/storage/HarbormasterDAO.php',
'HarbormasterHTTPRequestBuildStepImplementation' => 'applications/harbormaster/step/HarbormasterHTTPRequestBuildStepImplementation.php',
'HarbormasterLeaseHostBuildStepImplementation' => 'applications/harbormaster/step/HarbormasterLeaseHostBuildStepImplementation.php',
+ 'HarbormasterLeaseWorkingCopyBuildStepImplementation' => 'applications/harbormaster/step/HarbormasterLeaseWorkingCopyBuildStepImplementation.php',
'HarbormasterManagePlansCapability' => 'applications/harbormaster/capability/HarbormasterManagePlansCapability.php',
'HarbormasterManagementBuildWorkflow' => 'applications/harbormaster/management/HarbormasterManagementBuildWorkflow.php',
'HarbormasterManagementUpdateWorkflow' => 'applications/harbormaster/management/HarbormasterManagementUpdateWorkflow.php',
@@ -3962,6 +3965,8 @@
),
'DrydockBlueprintCreateController' => 'DrydockBlueprintController',
'DrydockBlueprintCustomField' => 'PhabricatorCustomField',
+ 'DrydockBlueprintCustomFieldBlueprints' => 'PhabricatorStandardCustomFieldPHIDs',
+ 'DrydockBlueprintDatasource' => 'PhabricatorTypeaheadDatasource',
'DrydockBlueprintEditController' => 'DrydockBlueprintController',
'DrydockBlueprintEditor' => 'PhabricatorApplicationTransactionEditor',
'DrydockBlueprintListController' => 'DrydockBlueprintController',
@@ -4194,6 +4199,7 @@
'HarbormasterDAO' => 'PhabricatorLiskDAO',
'HarbormasterHTTPRequestBuildStepImplementation' => 'HarbormasterBuildStepImplementation',
'HarbormasterLeaseHostBuildStepImplementation' => 'HarbormasterBuildStepImplementation',
+ 'HarbormasterLeaseWorkingCopyBuildStepImplementation' => 'HarbormasterBuildStepImplementation',
'HarbormasterManagePlansCapability' => 'PhabricatorPolicyCapability',
'HarbormasterManagementBuildWorkflow' => 'HarbormasterManagementWorkflow',
'HarbormasterManagementUpdateWorkflow' => 'HarbormasterManagementWorkflow',
diff --git a/src/applications/drydock/blueprint/DrydockBlueprintImplementation.php b/src/applications/drydock/blueprint/DrydockBlueprintImplementation.php
--- a/src/applications/drydock/blueprint/DrydockBlueprintImplementation.php
+++ b/src/applications/drydock/blueprint/DrydockBlueprintImplementation.php
@@ -10,6 +10,7 @@
private $activeResource;
private $activeLease;
private $instance;
+ private $scopes = array();
abstract public function getType();
abstract public function getInterface(
@@ -599,10 +600,11 @@
DrydockResource $resource = null,
DrydockLease $lease = null) {
- if (($this->activeResource !== null) ||
- ($this->activeLease !== null)) {
- throw new Exception('There is already an active resource or lease!');
- }
+ $scope = array(
+ 'resource' => $resource,
+ 'lease' => $lease,
+ );
+ array_push($this->scopes, $scope);
$this->activeResource = $resource;
$this->activeLease = $lease;
@@ -611,8 +613,20 @@
}
public function popActiveScope() {
- $this->activeResource = null;
- $this->activeLease = null;
+ if (count($this->scopes) === 0) {
+ throw new Exception('Unable to pop active scope; no scopes active');
+ }
+
+ array_pop($this->scopes);
+
+ if (count($this->scopes) === 0) {
+ $this->activeResource = null;
+ $this->activeLease = null;
+ } else {
+ $current = last($this->scopes);
+ $this->activeResource = $current['resource'];
+ $this->activeLease = $current['lease'];
+ }
}
}
diff --git a/src/applications/drydock/blueprint/DrydockWorkingCopyBlueprintImplementation.php b/src/applications/drydock/blueprint/DrydockWorkingCopyBlueprintImplementation.php
--- a/src/applications/drydock/blueprint/DrydockWorkingCopyBlueprintImplementation.php
+++ b/src/applications/drydock/blueprint/DrydockWorkingCopyBlueprintImplementation.php
@@ -15,14 +15,127 @@
return pht('Allows Drydock to check out working copies of repositories.');
}
+ private function resolveRelatedObjectsForLease(DrydockLease $lease) {
+ if ($lease->getAttribute('resolved.target') !== null) {
+ return;
+ }
+
+ // TODO: Support more ways of specifying working copy leases.
+ $buildable_phid = $lease->getAttribute('buildablePHID');
+
+ if ($buildable_phid) {
+ $buildable = id(new HarbormasterBuildableQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withPHIDs(array($buildable_phid))
+ ->needContainerObjects(true)
+ ->executeOne();
+ if ($buildable === null) {
+ throw new Exception(pht(
+ 'No buildable found with PHID %s',
+ $buildable_phid));
+ }
+ $buildable_object = $buildable->getBuildableObject();
+ $container_object = $buildable->getContainerObject();
+
+ if ($buildable_object instanceof PhabricatorRepositoryCommit &&
+ $container_object instanceof PhabricatorRepository) {
+ $lease->setAttribute('resolved.target', 'commit');
+ $lease->setAttribute(
+ 'resolved.commitIdentifier',
+ $buildable_object->getCommitIdentifier());
+ $lease->setAttribute(
+ 'resolved.repositoryPHID',
+ $container_object->getPHID());
+ $lease->save();
+
+ $this->log(pht(
+ 'Resolved working copy target as "commit"'));
+ $this->log(pht(
+ 'Resolved working copy commit identifier as "%s"',
+ $lease->getAttribute('resolved.commitIdentifier')));
+ $this->log(pht(
+ 'Resolved working copy repository PHID as "%s"',
+ $lease->getAttribute('resolved.repositoryPHID')));
+ } else if ($buildable_object instanceof DifferentialDiff &&
+ $container_object instanceof DifferentialRevision) {
+ $lease->setAttribute('resolved.target', 'diff');
+ $lease->setAttribute(
+ 'resolved.diffID',
+ $buildable_object->getID());
+ $lease->setAttribute(
+ 'resolved.revisionID',
+ $container_object->getID());
+ $lease->setAttribute(
+ 'resolved.repositoryPHID',
+ $container_object->getRepository()->getPHID());
+ $lease->setAttribute(
+ 'resolved.baseRevision',
+ $buildable_object->getSourceControlBaseRevision());
+ $lease->save();
+
+ $this->log(pht(
+ 'Resolved working copy target as "diff"'));
+ $this->log(pht(
+ 'Resolved working copy diff ID as "%d"',
+ $lease->getAttribute('resolved.diffID')));
+ $this->log(pht(
+ 'Resolved working copy revision ID as "%d"',
+ $lease->getAttribute('resolved.revisionID')));
+ $this->log(pht(
+ 'Resolved working copy repository PHID as "%s"',
+ $lease->getAttribute('resolved.repositoryPHID')));
+ $this->log(pht(
+ 'Resolved working copy base revision as "%s"',
+ $lease->getAttribute('resolved.baseRevision')));
+ }
+ }
+ }
+
protected function canAllocateLease(
DrydockResource $resource,
DrydockLease $lease) {
+ $platform_match =
+ $lease->getAttribute('platform') === $resource->getAttribute('platform');
+ $custom_match = DrydockCustomAttributes::hasRequirements(
+ $lease->getAttributes(),
+ $this->getDetail('attributes'));
+
+ $resource_repo = $resource->getAttribute('repositoryPHID');
+ $this->resolveRelatedObjectsForLease($lease);
+ $lease_repo = $lease->getAttribute('resolved.repositoryPHID');
+
+ $host_lease_id = $lease->getAttribute('hostLeaseID');
+ if ($host_lease_id !== null) {
+ $host_lease = id(new DrydockLeaseQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withIDs(array($host_lease_id))
+ ->executeOne();
+ if ($host_lease === null) {
+ throw new Exception(pht(
+ 'No lease found with ID %d',
+ $host_lease_id));
+ }
+ if ($host_lease->getResource()->getBlueprint()->getPHID()
+ !== $this->getDetail('host-blueprint')) {
+ $this->log(pht(
+ 'This blueprint can not allocate a resource on the required host.'));
+ return false;
+ }
+ }
- $resource_repo = $resource->getAttribute('repositoryID');
- $lease_repo = $lease->getAttribute('repositoryID');
+ $can_allocate = $platform_match && $custom_match &&
+ $resource_repo && $lease_repo &&
+ ($resource_repo == $lease_repo);
- return ($resource_repo && $lease_repo && ($resource_repo == $lease_repo));
+ if ($can_allocate) {
+ $this->log(pht(
+ 'This blueprint can allocate a resource for the specified lease.'));
+ } else {
+ $this->log(pht(
+ 'This blueprint can not allocate a resource for the specified lease.'));
+ }
+
+ return $can_allocate;
}
protected function shouldAllocateLease(
@@ -30,7 +143,7 @@
DrydockResource $resource,
DrydockLease $lease) {
- return $context->getCurrentResourceLeaseCount() === 0;
+ return true;
}
protected function executeInitializePendingResource(
@@ -41,20 +154,24 @@
DrydockResource $resource,
DrydockLease $lease) {
- $repository_id = $lease->getAttribute('repositoryID');
- if (!$repository_id) {
+ $this->resolveRelatedObjectsForLease($lease);
+
+ $target = $lease->getAttribute('resolved.target');
+ if (!$target) {
throw new Exception(
- "Lease is missing required 'repositoryID' attribute.");
+ 'Unable to resolve working copy target for lease.');
}
+ $repository_phid = $lease->getAttribute('resolved.repositoryPHID');
+
$repository = id(new PhabricatorRepositoryQuery())
->setViewer(PhabricatorUser::getOmnipotentUser())
- ->withIDs(array($repository_id))
+ ->withPHIDs(array($repository_phid))
->executeOne();
if (!$repository) {
throw new Exception(
- "Repository '{$repository_id}' does not exist!");
+ "Repository with PHID '{$repository_phid}' does not exist!");
}
switch ($repository->getVersionControlSystem()) {
@@ -64,39 +181,185 @@
throw new Exception('Unsupported VCS!');
}
- // TODO: Policy stuff here too.
- $host_lease = id(new DrydockLease())
+ // When allocating the resource, we always need to get a new lease
+ // that is owned by the resource. If the lease has specified an
+ // existing lease, we can a new lease against the same resource as
+ // that lease.
+
+ $host_lease = null;
+ $host_lease_id = $lease->getAttribute('hostLeaseID');
+ $resource_id = null;
+ if ($host_lease_id !== null) {
+ $host_lease = id(new DrydockLeaseQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withIDs(array($host_lease_id))
+ ->executeOne();
+ if ($host_lease === null) {
+ throw new Exception(pht(
+ 'No lease found with ID %d',
+ $host_lease_id));
+ }
+ $resource_id = $host_lease->getResource()->getID();
+ }
+
+ if ($resource_id) {
+ $this->log(pht(
+ 'Limiting lease acquisition to resource ID %d.',
+ $resource_id));
+ }
+
+ $this->log('Acquiring new host lease for working copy...');
+
+ $resource_lease = id(new DrydockLease())
->setResourceType('host')
+ ->setAttributes(
+ array(
+ 'resourceID' => $resource_id,
+ 'platform' => $lease->getAttribute('platform'),
+ ))
->waitUntilActive();
- $path = $host_lease->getAttribute('path').$repository->getCallsign();
+ $this->log(pht(
+ 'Lease %d acquired for working copy resource.',
+ $resource_lease->getID()));
- $this->log(
- pht('Cloning %s into %s....', $repository->getCallsign(), $path));
+ if ($lease->getAttribute('platform') === 'windows') {
+ $cmd = $resource_lease->getInterface(
+ 'command-'.PhutilCommandString::MODE_WINDOWSCMD);
+ } else {
+ $cmd = $resource_lease->getInterface(
+ 'command-'.PhutilCommandString::MODE_BASH);
+ }
- $cmd = $host_lease->getInterface('command');
+ $this->log(pht(
+ 'Cloning repository to %s...',
+ $resource_lease->getAttribute('path')));
+
+ // For some reason %P doesn't work here (it gets translated
+ // into xxxxx which gets passed through as xxxxx to the underlying
+ // command, rather than just being masked in the output).
$cmd->execx(
- 'git clone --origin origin %P %s',
- $repository->getRemoteURIEnvelope(),
- $path);
+ 'git clone --bare %s .',
+ $repository->getRemoteURIEnvelope()->openEnvelope());
- $this->log(pht('Complete.'));
+ $this->log('Cloned repository cache.');
$resource
->setName('Working Copy ('.$repository->getCallsign().')')
->setStatus(DrydockResourceStatus::STATUS_OPEN)
- ->setAttribute('lease.host', $host_lease->getID())
- ->setAttribute('path', $path)
- ->setAttribute('repositoryID', $repository->getID())
+ ->setAttribute('host.lease', $resource_lease->getID())
+ ->setAttribute('host.resource', $resource_lease->getResource()->getID())
+ ->setAttribute('path', $resource_lease->getAttribute('path'))
+ ->setAttribute('repositoryPHID', $repository->getPHID())
+ ->setAttribute('platform', $resource_lease->getAttribute('platform'))
->save();
-
return $resource;
}
protected function executeAcquireLease(
DrydockResource $resource,
DrydockLease $lease) {
- return;
+
+ $resource_lease = id(new DrydockLeaseQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withIDs(array($resource->getAttribute('host.lease')))
+ ->executeOne();
+ if ($resource_lease === null) {
+ throw new Exception(pht(
+ 'No resource found with ID %d',
+ $resource->getAttribute('host.lease')));
+ }
+
+ // We must lock the resource while we perform the cache update,
+ // because otherwise we'll end up running multiple read-write
+ // VCS operations in the same directory at the same time.
+ $lock = PhabricatorGlobalLock::newLock(
+ 'drydock-working-copy-cache-update-'.$resource_lease->getID());
+ $lock->lock(1000000);
+ try {
+ if ($lease->getAttribute('platform') === 'windows') {
+ $cmd = $resource_lease->getInterface(
+ 'command-'.PhutilCommandString::MODE_WINDOWSCMD);
+ } else {
+ $cmd = $resource_lease->getInterface(
+ 'command-'.PhutilCommandString::MODE_BASH);
+ }
+
+ $this->log(pht(
+ 'Fetching latest commits for repository at "%s"',
+ $resource->getAttribute('path')));
+ $cmd->exec('git fetch --all');
+ $this->log(pht('Fetched latest commits.'));
+
+ $lock->unlock();
+ } catch (Exception $ex) {
+ $lock->unlock();
+ throw $ex;
+ }
+
+ $host_lease = null;
+ $host_lease_id = $lease->getAttribute('hostLeaseID');
+ if ($host_lease_id !== null) {
+ $host_lease = id(new DrydockLeaseQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withIDs(array($host_lease_id))
+ ->executeOne();
+ if ($host_lease === null) {
+ throw new Exception(pht(
+ 'No lease found with ID %d',
+ $host_lease_id));
+ }
+ $this->log(pht(
+ 'Using existing host lease %d for working copy resource.',
+ $host_lease_id));
+ } else {
+ $this->log('Acquiring new host lease for working copy...');
+ $host_lease = id(new DrydockLease())
+ ->setResourceType('host')
+ ->setAttributes(
+ array(
+ 'resourceID' => $resource->getAttribute('host.resource'),
+ 'platform' => $lease->getAttribute('platform'),
+ ))
+ ->waitUntilActive();
+ $host_lease_id = $host_lease->getID();
+ $this->log(pht(
+ 'Lease %d acquired for working copy resource.',
+ $host_lease_id));
+ }
+
+ $lease->setAttribute('host.lease', $host_lease_id);
+
+ if ($lease->getAttribute('platform') === 'windows') {
+ $cmd = $lease->getInterface(
+ 'command-'.PhutilCommandString::MODE_WINDOWSCMD);
+ } else {
+ $cmd = $lease->getInterface(
+ 'command-'.PhutilCommandString::MODE_BASH);
+ }
+
+ $this->log(pht(
+ 'Cloning from cache path "%s" to lease path "%s"',
+ $resource->getAttribute('path'),
+ $host_lease->getAttribute('path')));
+ $cmd->execx(
+ 'git clone %s .',
+ $resource->getAttribute('path'));
+ $this->log(pht('Cloned from cache'));
+
+ if ($lease->getAttribute('resolved.target') === 'commit') {
+ $this->log(pht(
+ 'Checking out target commit "%s"',
+ $lease->getAttribute('resolved.commitIdentifier')));
+ $cmd->execx(
+ 'git checkout -f %s',
+ $lease->getAttribute('resolved.commitIdentifier'));
+ $this->log(pht('Checked out commit'));
+ } else {
+ throw new Exception(pht(
+ 'Target type %s not yet supported.',
+ $lease->getAttribute('resolved.target')));
+ }
}
public function getType() {
@@ -108,19 +371,37 @@
DrydockLease $lease,
$type) {
- switch ($type) {
- case 'command':
- return $this
- ->loadLease($resource->getAttribute('lease.host'))
- ->getInterface($type);
- }
-
- throw new Exception("No interface of type '{$type}'.");
+ return $this
+ ->loadLease($lease->getAttribute('host.lease'))
+ ->getInterface($type);
}
protected function executeReleaseLease(
DrydockResource $resource,
- DrydockLease $lease) {}
+ DrydockLease $lease) {
+
+ if (!$lease->getAttribute('hostLeaseID')) {
+ $this->log(pht(
+ 'Releasing host lease %d',
+ $lease->getAttribute('host.lease')));
+ try {
+ $host_lease = $this->loadLease($lease->getAttribute('host.lease'));
+
+ $host_resource = $host_lease->getResource();
+ $host_blueprint = $host_resource->getBlueprint();
+ $host_blueprint->releaseLease($host_resource, $host_lease);
+
+ $this->log(pht(
+ 'Released host lease %d',
+ $lease->getAttribute('host.lease')));
+ } catch (Exception $ex) {
+ $this->log(pht(
+ 'Unable to release host lease %d: "%s"',
+ $lease->getAttribute('host.lease'),
+ (string)$ex));
+ }
+ }
+ }
protected function shouldCloseUnleasedResource(
DrydockAllocationContext $context,
@@ -130,7 +411,57 @@
}
protected function executeCloseResource(DrydockResource $resource) {
- // TODO: Remove leased directory
+ $this->log(pht(
+ 'Releasing resource host lease %d',
+ $resource->getAttribute('host.lease')));
+ try {
+ $host_lease = $this->loadLease($resource->getAttribute('host.lease'));
+
+ $host_resource = $host_lease->getResource();
+ $host_blueprint = $host_resource->getBlueprint();
+ $host_blueprint->releaseLease($host_resource, $host_lease);
+
+ $this->log(pht(
+ 'Released resource host lease %d',
+ $resource->getAttribute('host.lease')));
+ } catch (Exception $ex) {
+ $this->log(pht(
+ 'Unable to release resource host lease %d: "%s"',
+ $resource->getAttribute('host.lease'),
+ (string)$ex));
+ }
}
+ public function getFieldSpecifications() {
+ return array(
+ 'host-config' => array(
+ 'name' => pht('Host Configuration'),
+ 'type' => 'header',
+ ),
+ 'host-blueprint' => array(
+ 'name' => pht('Host Blueprint'),
+ 'type' => 'blueprints',
+ 'required' => true,
+ 'limit' => 1,
+ 'blueprint-type' => 'host',
+ 'caption' => pht(
+ 'The blueprint which provides hosts that this '.
+ 'blueprint will operate on.'),
+ ),
+ 'attr-header' => array(
+ 'name' => pht('Working Copy Attributes'),
+ 'type' => 'header',
+ ),
+ 'attributes' => array(
+ 'name' => pht('Working Copy Attributes'),
+ 'type' => 'textarea',
+ 'caption' => pht(
+ 'A newline separated list of working copy attributes. '.
+ 'Each attribute should be specified in a key=value format.'),
+ 'monospace' => true,
+ ),
+ ) + parent::getFieldSpecifications();
+ }
+
+
}
diff --git a/src/applications/drydock/customfield/DrydockBlueprintCustomFieldBlueprints.php b/src/applications/drydock/customfield/DrydockBlueprintCustomFieldBlueprints.php
new file mode 100644
--- /dev/null
+++ b/src/applications/drydock/customfield/DrydockBlueprintCustomFieldBlueprints.php
@@ -0,0 +1,50 @@
+<?php
+
+final class DrydockBlueprintCustomFieldBlueprints
+ extends PhabricatorStandardCustomFieldPHIDs {
+
+ public function getFieldType() {
+ return 'blueprints';
+ }
+
+ public function renderEditControl(array $handles) {
+ $value = $this->getFieldValue();
+
+ $control = id(new AphrontFormTokenizerControl())
+ ->setUser($this->getViewer())
+ ->setLabel($this->getFieldName())
+ ->setName($this->getFieldKey())
+ ->setDatasource(id(new DrydockBlueprintDatasource())
+ ->setParameters(array(
+ 'type' => $this->getFieldConfigValue('blueprint-type'),
+ )))
+ ->setCaption($this->getCaption())
+ ->setValue(nonempty($value, array()));
+
+ $limit = $this->getFieldConfigValue('limit');
+ if ($limit) {
+ $control->setLimit($limit);
+ }
+
+ return $control;
+ }
+
+ public function appendToApplicationSearchForm(
+ PhabricatorApplicationSearchEngine $engine,
+ AphrontFormView $form,
+ $value,
+ array $handles) {
+
+ $control = id(new AphrontFormTokenizerControl())
+ ->setLabel($this->getFieldName())
+ ->setName($this->getFieldKey())
+ ->setDatasource(id(new DrydockBlueprintDatasource())
+ ->setParameters(array(
+ 'type' => $this->getFieldConfigValue('blueprint-type'),
+ )))
+ ->setValue(nonempty($value, array()));
+
+ $form->appendControl($control);
+ }
+
+}
diff --git a/src/applications/drydock/phid/DrydockBlueprintPHIDType.php b/src/applications/drydock/phid/DrydockBlueprintPHIDType.php
--- a/src/applications/drydock/phid/DrydockBlueprintPHIDType.php
+++ b/src/applications/drydock/phid/DrydockBlueprintPHIDType.php
@@ -29,6 +29,7 @@
$blueprint = $objects[$phid];
$id = $blueprint->getID();
+ $handle->setName($blueprint->getBlueprintName());
$handle->setURI("/drydock/blueprint/{$id}/");
}
}
diff --git a/src/applications/drydock/typeahead/DrydockBlueprintDatasource.php b/src/applications/drydock/typeahead/DrydockBlueprintDatasource.php
new file mode 100644
--- /dev/null
+++ b/src/applications/drydock/typeahead/DrydockBlueprintDatasource.php
@@ -0,0 +1,52 @@
+<?php
+
+final class DrydockBlueprintDatasource
+ extends PhabricatorTypeaheadDatasource {
+
+ public function isBrowsable() {
+ return true;
+ }
+
+ public function getBrowseTitle() {
+ return pht('Browse Blueprints');
+ }
+
+ public function getPlaceholderText() {
+ return pht('Type blueprint names...');
+ }
+
+ public function getDatasourceApplicationClass() {
+ return 'PhabricatorDrydockApplication';
+ }
+
+ public function loadResults() {
+ $viewer = $this->getViewer();
+
+ $blueprint_type = $this->getParameter('type');
+
+ $blueprints = id(new DrydockBlueprintQuery())
+ ->setViewer($viewer)
+ ->execute();
+ $blueprints = mpull($blueprints, null, 'getPHID');
+
+ if (count($blueprints) === 0) {
+ return array();
+ }
+
+ $results = array();
+ foreach ($blueprints as $phid => $blueprint) {
+ if ($blueprint_type !== null &&
+ $blueprint->getImplementation()->getType() !== $blueprint_type) {
+ continue;
+ }
+
+ $results[] = id(new PhabricatorTypeaheadResult())
+ ->setName($blueprint->getBlueprintName())
+ ->setURI('/')
+ ->setPHID($phid);
+ }
+
+ return $results;
+ }
+
+}
diff --git a/src/applications/drydock/worker/DrydockAllocatorWorker.php b/src/applications/drydock/worker/DrydockAllocatorWorker.php
--- a/src/applications/drydock/worker/DrydockAllocatorWorker.php
+++ b/src/applications/drydock/worker/DrydockAllocatorWorker.php
@@ -104,6 +104,12 @@
continue;
}
+ if ($lease->getAttribute('resourceID') !== null &&
+ $candidate->getID() !== $lease->getAttribute('resourceID')) {
+ unset($pool[$key]);
+ continue;
+ }
+
$blueprint = $blueprints[$candidate->getBlueprintPHID()];
$implementation = $blueprint->getImplementation();
@@ -144,6 +150,12 @@
continue;
}
+ if ($lease->getAttribute('resourceID') !== null &&
+ $candidate->getID() !== $lease->getAttribute('resourceID')) {
+ unset($pool[$key]);
+ continue;
+ }
+
$blueprint = $blueprints[$candidate->getBlueprintPHID()];
$implementation = $blueprint->getImplementation();
diff --git a/src/applications/harbormaster/step/HarbormasterLeaseWorkingCopyBuildStepImplementation.php b/src/applications/harbormaster/step/HarbormasterLeaseWorkingCopyBuildStepImplementation.php
new file mode 100644
--- /dev/null
+++ b/src/applications/harbormaster/step/HarbormasterLeaseWorkingCopyBuildStepImplementation.php
@@ -0,0 +1,84 @@
+<?php
+
+final class HarbormasterLeaseWorkingCopyBuildStepImplementation
+ extends HarbormasterBuildStepImplementation {
+
+ public function getName() {
+ return pht('Lease Working Copy');
+ }
+
+ public function getGenericDescription() {
+ return pht(
+ 'Obtain a lease on a Drydock working copy of the '.
+ 'current buildable for performing builds.');
+ }
+
+ public function execute(
+ HarbormasterBuild $build,
+ HarbormasterBuildTarget $build_target) {
+
+ $settings = $this->getSettings();
+
+ $custom_attributes = DrydockCustomAttributes::parse(
+ $settings['attributes']);
+
+ // Create the lease.
+ $lease = id(new DrydockLease())
+ ->setResourceType('working-copy')
+ ->setAttributes(
+ array(
+ 'platform' => $settings['platform'],
+ 'buildablePHID' => $build->getBuildablePHID(),
+ ) + $custom_attributes)
+ ->queueForActivation();
+
+ // Create the associated artifact.
+ $artifact = $build->createArtifact(
+ $build_target,
+ $settings['name'],
+ HarbormasterBuildArtifact::TYPE_HOST);
+ $artifact->setArtifactData(array(
+ 'drydock-lease' => $lease->getID(),
+ ));
+ $artifact->save();
+
+ // Wait until the lease is fulfilled.
+ // TODO: This will throw an exception if the lease can't be fulfilled;
+ // we should treat that as build failure not build error.
+ $lease->waitUntilActive();
+ }
+
+ public function getArtifactOutputs() {
+ return array(
+ array(
+ 'name' => pht('Leased Working Copy'),
+ 'key' => $this->getSetting('name'),
+ 'type' => HarbormasterBuildArtifact::TYPE_HOST,
+ ),
+ );
+ }
+
+ public function getFieldSpecifications() {
+ return array(
+ 'name' => array(
+ 'name' => pht('Artifact Name'),
+ 'type' => 'text',
+ 'required' => true,
+ ),
+ 'platform' => array(
+ 'name' => pht('Host Platform'),
+ 'type' => 'text',
+ 'required' => true,
+ ),
+ 'attributes' => array(
+ 'name' => pht('Required Attributes'),
+ 'type' => 'textarea',
+ 'caption' => pht(
+ 'A newline separated list of required working copy attributes. '.
+ 'Each attribute should be specified in a key=value format.'),
+ 'monospace' => true,
+ ),
+ );
+ }
+
+}

File Metadata

Mime Type
text/plain
Expires
Thu, Apr 3, 6:50 AM (11 h, 23 m ago)
Storage Engine
blob
Storage Format
Encrypted (AES-256-CBC)
Storage Handle
7730069
Default Alt Text
D12783.id30729.diff (28 KB)

Event Timeline

Phabricator · Built by Phacility