DiffusionLowLevelMercurialBranchesQuery.php
No OneTemporary
Actions

Subscribers

None

File Metadata

	<?php

	/**
	* Execute and parse a low-level Mercurial branches query using `hg branches`.
	*/
	final class DiffusionLowLevelMercurialBranchesQuery
	extends DiffusionLowLevelQuery {

	private $contains;

	public function withContainsCommit($commit) {
	$this->contains = $commit;
	return $this;
	}

	protected function executeQuery() {
	$repository = $this->getRepository();

	$specs = array();
	if ($this->contains !== null) {
	$specs['all'] = hgsprintf(
	'(descendants(%s) and head())',
	$this->contains);
	$specs['open'] = hgsprintf(
	'(descendants(%s) and head() and not closed())',
	$this->contains);
	} else {
	$specs['all'] = hgsprintf('head()');
	$specs['open'] = hgsprintf('head() and not closed()');
	}

	$futures = array();
	foreach ($specs as $key => $spec) {
	$futures[$key] = $repository->getLocalCommandFuture(
	'log --template %s --rev %s',
	'{node}\1{branch}\2',
	$spec);
	}

	$branches = array();
	$open = array();
	foreach (new FutureIterator($futures) as $key => $future) {
	list($stdout) = $future->resolvex();

	$lines = explode("\2", $stdout);
	$lines = array_filter($lines);
	foreach ($lines as $line) {
	list($node, $branch) = explode("\1", $line);
	$id = $node.'/'.$branch;
	if (empty($branches[$id])) {
	$branches[$id] = id(new DiffusionRepositoryRef())
	->setShortName($branch)
	->setCommitIdentifier($node);
	}

	if ($key == 'open') {
	$open[$id] = true;
	}
	}
	}

	foreach ($branches as $id => $branch) {
	$branch->setRawFields(
	array(
	'closed' => (empty($open[$id])),
	));
	}

	return array_values($branches);
	}

	}