Differential D21079 Diff 50215 src/query/ArcanistMessageRevisionHardpointQuery.php

Changeset View

Standalone View

				<?php

				final class ArcanistMessageRevisionHardpointQuery
				extends ArcanistWorkflowHardpointQuery {

				public function getHardpoints() {
				return array(
				ArcanistWorkingCopyStateRefPro::HARDPOINT_REVISIONREFS,
				);
				}

				protected function canLoadRef(ArcanistRefPro $ref) {
				return ($ref instanceof ArcanistWorkingCopyStateRefPro);
				}

				public function loadHardpoint(array $refs, $hardpoint) {
				yield $this->yieldRequests(
				$refs,
				array(
				ArcanistWorkingCopyStateRefPro::HARDPOINT_COMMITREF,
				));

				$commit_refs = array();
				foreach ($refs as $ref) {
				$commit_refs[] = $ref->getCommitRef();
				}

				yield $this->yieldRequests(
				$commit_refs,
				array(
				ArcanistCommitRefPro::HARDPOINT_MESSAGE,
				));

				$map = array();
				foreach ($refs as $ref_key => $ref) {
				$commit_ref = $ref->getCommitRef();
				$corpus = $commit_ref->getMessage();

				$id = null;
				try {
				$message = ArcanistDifferentialCommitMessage::newFromRawCorpus($corpus);
				$id = $message->getRevisionID();
				} catch (ArcanistUsageException $ex) {
				continue;
				}

				if (!$id) {
				continue;
				}

				$map[$id][$ref_key] = $ref;
				}

				$results = array();
				if ($map) {
				$revisions = (yield $this->yieldConduit(
				'differential.query',
				array(
				'ids' => array_keys($map),
				)));

				foreach ($revisions as $dict) {
				$revision_ref = ArcanistRevisionRefPro::newFromConduit($dict);
				$id = $dict['id'];

				$state_refs = idx($map, $id, array());
				foreach ($state_refs as $ref_key => $state_ref) {
				$results[$ref_key][] = $revision_ref;
				}
				}
				}

				yield $this->yieldMap($results);
				}

				}