Differential D21082 Diff 50222 src/ref/ArcanistWorkingCopyStateRef.php

Changeset View

Standalone View

src/ref/ArcanistWorkingCopyStateRef.php

	<?php			<?php

	final class ArcanistWorkingCopyStateRef			final class ArcanistWorkingCopyStateRef
	extends ArcanistRef {			extends ArcanistRef {

	private $rootDirectory;			const HARDPOINT_COMMITREF = 'commitRef';
				const HARDPOINT_REVISIONREFS = 'revisionRefs';

	public function getRefIdentifier() {			public function getRefDisplayName() {
	// TODO: This could check attached hardpoints and render something more			// TODO: This could check attached hardpoints and render something more
	// insightful.			// insightful.
	return pht('Working Copy State');			return pht('Working Copy State');
	}			}

	public function defineHardpoints() {			protected function newHardpoints() {
				$object_list = new ArcanistObjectListHardpoint();
	return array(			return array(
	'commitRef' => array(			$this->newHardpoint(self::HARDPOINT_COMMITREF),
	'type' => 'ArcanistCommitRef',			$this->newTemplateHardpoint(self::HARDPOINT_REVISIONREFS, $object_list),
	),
	'branchRef' => array(
	'type' => 'ArcanistBranchRef',
	),
	'revisionRefs' => array(
	'type' => 'ArcanistRevisionRef',
	'vector' => true,
	),
	);			);
	}			}

	public function setRootDirectory($root_directory) {			// TODO: This should be "attachCommitRef()".
	$this->rootDirectory = $root_directory;
	return $this;
	}

	public function getRootDirectory() {
	return $this->rootDirectory;
	}

	public function attachBranchRef(ArcanistBranchRef $branch_ref) {
	return $this->attachHardpoint('branchRef', $branch_ref);
	}

	public function getBranchRef() {
	return $this->getHardpoint('branchRef');
	}

	public function setCommitRef(ArcanistCommitRef $commit_ref) {			public function setCommitRef(ArcanistCommitRef $commit_ref) {
	return $this->attachHardpoint('commitRef', $commit_ref);			return $this->attachHardpoint(self::HARDPOINT_COMMITREF, $commit_ref);
	}			}

	public function getCommitRef() {			public function getCommitRef() {
	return $this->getHardpoint('commitRef');			return $this->getHardpoint(self::HARDPOINT_COMMITREF);
	}			}

	public function getRevisionRefs() {			public function getRevisionRefs() {
	return $this->getHardpoint('revisionRefs');			return $this->getHardpoint(self::HARDPOINT_REVISIONREFS);
	}			}

	public function getRevisionRef() {			public function getRevisionRef() {
	if ($this->hasAmbiguousRevisionRefs()) {			if ($this->hasAmbiguousRevisionRefs()) {
	throw new Exception(			throw new Exception(
	pht('State has multiple ambiguous revisions refs.'));			pht('State has multiple ambiguous revisions refs.'));
	}			}

	$refs = $this->getRevisionRefs();			$refs = $this->getRevisionRefs();
	if ($refs) {			if ($refs) {
	return head($refs);			return head($refs);
	}			}

	return null;			return null;
	}			}

	public function hasAmbiguousRevisionRefs() {			public function hasAmbiguousRevisionRefs() {
	return (count($this->getRevisionRefs()) > 1);			return (count($this->getRevisionRefs()) > 1);
	}			}

	protected function canReadHardpoint($hardpoint) {
	switch ($hardpoint) {
	case 'commitRef':
	// If we have a branch ref, we can try to read the commit ref from the
	// branch ref.
	if ($this->hasAttachedHardpoint('branchRef')) {
	if ($this->getBranchRef()->hasAttachedHardpoint('commitRef')) {
	return true;
	}
	}
	break;
	}

	return false;
	}

	protected function readHardpoint($hardpoint) {
	switch ($hardpoint) {
	case 'commitRef':
	return $this->getBranchRef()->getCommitRef();
	}

	return parent::readHardpoint($hardpoint);
	}

	protected function mergeHardpoint($hardpoint, array $src, array $new) {
	if ($hardpoint == 'revisionRefs') {
	$src = mpull($src, null, 'getID');
	$new = mpull($new, null, 'getID');

	foreach ($new as $id => $ref) {
	if (isset($src[$id])) {
	foreach ($ref->getSources() as $source) {
	$src[$id]->addSource($source);
	}
	} else {
	$src[$id] = $ref;
	}
	}

	return array_values($src);
	}

	return parent::mergeHardpoint($hardpoint, $src, $new);
	}

	}			}