ArcanistMercurialCommitGraphQuery.php
View Options

	<?php

	final class ArcanistMercurialCommitGraphQuery
	extends ArcanistCommitGraphQuery {

	private $seen = array();
	private $queryFuture;

	public function execute() {
	$this->beginExecute();
	$this->continueExecute();

	return $this->seen;
	}

	protected function beginExecute() {
	$head_hashes = $this->getHeadHashes();
	$exact_hashes = $this->getExactHashes();

	if (!$head_hashes && !$exact_hashes) {
	throw new Exception(pht('Need head hashes or exact hashes!'));
	}

	$api = $this->getRepositoryAPI();

	$revsets = array();
	if ($head_hashes !== null) {
	$revs = array();
	foreach ($head_hashes as $hash) {
	$revs[] = hgsprintf(
	'ancestors(%s)',
	$hash);
	}
	$revsets[] = $this->joinOrRevsets($revs);
	}

	$tail_hashes = $this->getTailHashes();
	if ($tail_hashes !== null) {
	$revs = array();
	foreach ($tail_hashes as $tail_hash) {
	$revs[] = hgsprintf(
	'descendants(%s)',
	$tail_hash);
	}
	$revsets[] = $this->joinOrRevsets($revs);
	}

	if ($revsets) {
	$revsets = array(
	$this->joinAndRevsets($revs),
	);
	}

	if ($exact_hashes !== null) {
	$revs = array();
	foreach ($exact_hashes as $exact_hash) {
	$revs[] = hgsprintf(
	'%s',
	$exact_hash);
	}
	$revsets[] = array(
	$this->joinOrRevsets($revs),
	);
	}

	$revsets = $this->joinOrRevsets($revs);

	$fields = array(
	'', // Placeholder for "encoding".
	'{node}',
	'{parents}',
	'{date\|rfc822date}',
	'{description\|utf8}',
	);

	$template = implode("\2", $fields)."\1";

	$future = $api->newFuture(
	'log --rev %s --template %s --',
	$revsets,
	$template);
	$future->setResolveOnError(true);
	$future->start();

	$lines = id(new LinesOfALargeExecFuture($future))
	->setDelimiter("\1");
	$lines->rewind();

	$this->queryFuture = $lines;
	}

	protected function continueExecute() {
	$graph = $this->getGraph();
	$lines = $this->queryFuture;
	$limit = $this->getLimit();
	while (true) {
	if (!$lines->valid()) {
	return false;
	}

	$line = $lines->current();
	$lines->next();

	if ($line === "\n") {
	continue;
	}

	$fields = explode("\2", $line);

	if (count($fields) !== 5) {
	throw new Exception(
	pht(
	'Failed to split line "%s" from "git log".',
	$line));
	}

	list($encoding, $hash, $parents, $commit_epoch, $message) = $fields;

	$node = $graph->getNode($hash);
	if (!$node) {
	$node = $graph->newNode($hash);
	}

	$this->seen[$hash] = $node;

	$node
	->setCommitMessage($message)
	->setCommitEpoch((int)strtotime($commit_epoch));

	if (strlen($parents)) {
	$parents = explode(' ', $parents);

	$parent_nodes = array();
	foreach ($parents as $parent) {
	$parent_node = $graph->getNode($parent);
	if (!$parent_node) {
	$parent_node = $graph->newNode($parent);
	}

	$parent_nodes[$parent] = $parent_node;
	$parent_node->addChildNode($node);
	}
	$node->setParentNodes($parent_nodes);
	} else {
	$parents = array();
	}

	if ($limit) {
	if (count($this->seen) >= $limit) {
	break;
	}
	}
	}
	}

	private function joinOrRevsets(array $revsets) {
	return $this->joinRevsets($revsets, false);
	}

	private function joinAndRevsets(array $revsets) {
	return $this->joinRevsets($revsets, true);
	}

	private function joinRevsets(array $revsets, $is_and) {
	if (!$revsets) {
	return array();
	}

	if (count($revsets) === 1) {
	return head($revsets);
	}

	if ($is_and) {
	return '('.implode(' and ', $revsets).')';
	} else {
	return '('.implode(' or ', $revsets).')';
	}
	}

	}

File Metadata

Mime Type: text/x-php
Expires: Mon, Mar 24, 07:56 (2 w, 1 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1241786
Default Alt Text: ArcanistMercurialCommitGraphQuery.php (3 KB)

ArcanistMercurialCommitGraphQuery.php
No OneTemporary
Actions

ArcanistMercurialCommitGraphQuery.php
View Options

File Metadata

Event Timeline

ArcanistMercurialCommitGraphQuery.phpNo OneTemporaryActions

ArcanistMercurialCommitGraphQuery.phpView Options

File Metadata

Event Timeline

ArcanistMercurialCommitGraphQuery.php
No OneTemporary
Actions

ArcanistMercurialCommitGraphQuery.php
View Options