ArcanistBranchWorkflow.php
No OneTemporary
Actions

Size

7 KB

Referenced Files

None

Subscribers

None

ArcanistBranchWorkflow.php
View Options

	<?php

	/**
	* Displays user's git branches
	*
	* @group workflow
	*/
	final class ArcanistBranchWorkflow extends ArcanistBaseWorkflow {

	private $branches;

	public function getWorkflowName() {
	return 'branch';
	}

	public function getCommandSynopses() {
	return phutil_console_format(<<<EOTEXT
	branch [__options__]
	branch __name__ [__start__]
	EOTEXT
	);
	}

	public function getCommandHelp() {
	return phutil_console_format(<<<EOTEXT
	Supports: git
	A wrapper on 'git branch'. It pulls data from Differential and
	displays the revision status next to the branch name.

	By default, branches are sorted chronologically. You can sort them
	by status instead with __--by-status__.

	By default, branches that are "Closed" or "Abandoned" are not
	displayed. You can show them with __--view-all__.

	With __name__, it creates or checks out a branch. If the branch
	__name__ doesn't exist and is in format D123 then the branch of
	revision D123 is checked out.
	EOTEXT
	);
	}

	public function requiresConduit() {
	return true;
	}

	public function requiresRepositoryAPI() {
	return true;
	}

	public function requiresAuthentication() {
	return true;
	}


	public function getArguments() {
	return array(
	'view-all' => array(
	'help' => 'Include closed and abandoned revisions',
	),
	'by-status' => array(
	'help' => 'Sort branches by status instead of time.',
	),
	'*' => 'names',
	);
	}

	public function run() {
	$repository_api = $this->getRepositoryAPI();
	if (!($repository_api instanceof ArcanistGitAPI)) {
	throw new ArcanistUsageException(
	'arc branch is only supported under git.');
	}

	$names = $this->getArgument('names');
	if ($names) {
	if (count($names) > 2) {
	throw new ArcanistUsageException("Specify only one branch.");
	}
	return $this->checkoutBranch($names);
	}

	$branches = $repository_api->getAllBranches();
	if (!$branches) {
	throw new ArcanistUsageException('No branches in this working copy.');
	}

	$branches = $this->loadCommitInfo($branches);

	$revisions = $this->loadRevisions($branches);

	$this->printBranches($branches, $revisions);

	return 0;
	}

	private function checkoutBranch(array $names) {
	$api = $this->getRepositoryAPI();

	list($err, $stdout, $stderr) = $api->execManualLocal(
	'checkout %s',
	reset($names));

	if ($err) {
	$match = null;
	if (preg_match('/^D(\d+)$/', reset($names), $match)) {
	try {
	$diff = $this->getConduit()->callMethodSynchronous(
	'differential.getdiff',
	array(
	'revision_id' => $match[1],
	));

	if ($diff['branch'] != '') {
	$names[0] = $diff['branch'];
	list($err, $stdout, $stderr) = $api->execManualLocal(
	'checkout %s',
	reset($names));
	}
	} catch (ConduitException $ex) {
	}
	}
	}

	if ($err) {
	list($err, $stdout, $stderr) = $api->execManualLocal(
	'checkout -b %Ls',
	$names);
	}

	echo $stdout;
	fprintf(STDERR, $stderr);
	return $err;
	}

	private function loadCommitInfo(array $branches) {
	$repository_api = $this->getRepositoryAPI();

	$futures = array();
	foreach ($branches as $branch) {
	// NOTE: "-s" is an option deep in git's diff argument parser that doesn't
	// seem to have much documentation and has no long form. It suppresses any
	// diff output.
	$futures[$branch['name']] = $repository_api->execFutureLocal(
	'show -s --format=%C %s --',
	'%H%x01%ct%x01%T%x01%s%x01%b',
	$branch['name']);
	}

	$branches = ipull($branches, null, 'name');

	foreach (Futures($futures)->limit(16) as $name => $future) {
	list($info) = $future->resolvex();
	list($hash, $epoch, $tree, $desc, $text) = explode("\1", trim($info), 5);

	$branch = $branches[$name];
	$branch['hash'] = $hash;
	$branch['desc'] = $desc;

	try {
	$text = $desc."\n".$text;
	$message = ArcanistDifferentialCommitMessage::newFromRawCorpus($text);
	$id = $message->getRevisionID();

	$branch += array(
	'epoch' => (int)$epoch,
	'tree' => $tree,
	'revisionID' => $id,
	);
	} catch (ArcanistUsageException $ex) {
	// In case of invalid commit message which fails the parsing,
	// do nothing.
	}

	$branches[$name] = $branch;
	}

	return $branches;
	}

	private function loadRevisions(array $branches) {
	$ids = array();
	$hashes = array();

	foreach ($branches as $branch) {
	if ($branch['revisionID']) {
	$ids[] = $branch['revisionID'];
	}
	$hashes[] = array('gtcm', $branch['hash']);
	$hashes[] = array('gttr', $branch['tree']);
	}

	$calls = array();

	if ($ids) {
	$calls[] = $this->getConduit()->callMethod(
	'differential.query',
	array(
	'ids' => $ids,
	));
	}

	if ($hashes) {
	$calls[] = $this->getConduit()->callMethod(
	'differential.query',
	array(
	'commitHashes' => $hashes,
	));
	}

	$results = array();
	foreach (Futures($calls) as $call) {
	$results[] = $call->resolve();
	}

	return array_mergev($results);
	}

	private function printBranches(array $branches, array $revisions) {
	$revisions = ipull($revisions, null, 'id');

	static $color_map = array(
	'Closed' => 'cyan',
	'Needs Review' => 'magenta',
	'Needs Revision' => 'red',
	'Accepted' => 'green',
	'No Revision' => 'blue',
	'Abandoned' => 'default',
	);

	static $ssort_map = array(
	'Closed' => 1,
	'No Revision' => 2,
	'Needs Review' => 3,
	'Needs Revision' => 4,
	'Accepted' => 5,
	);

	$out = array();
	foreach ($branches as $branch) {
	$revision = idx($revisions, idx($branch, 'revisionID'));

	// If we haven't identified a revision by ID, try to identify it by hash.
	if (!$revision) {
	foreach ($revisions as $rev) {
	$hashes = idx($rev, 'hashes', array());
	foreach ($hashes as $hash) {
	if (($hash[0] == 'gtcm' && $hash[1] == $branch['hash']) \|\|
	($hash[0] == 'gttr' && $hash[1] == $branch['tree'])) {
	$revision = $rev;
	break;
	}
	}
	}
	}

	if ($revision) {
	$desc = 'D'.$revision['id'].': '.$revision['title'];
	$status = $revision['statusName'];
	} else {
	$desc = $branch['desc'];
	$status = 'No Revision';
	}

	if (!$this->getArgument('view-all') && !$branch['current']) {
	if ($status == 'Closed' \|\| $status == 'Abandoned') {
	continue;
	}
	}

	$epoch = $branch['epoch'];

	$color = idx($color_map, $status, 'default');
	$ssort = sprintf('%d%012d', idx($ssort_map, $status, 0), $epoch);

	$out[] = array(
	'name' => $branch['name'],
	'current' => $branch['current'],
	'status' => $status,
	'desc' => $desc,
	'color' => $color,
	'esort' => $epoch,
	'ssort' => $ssort,
	);
	}

	$len_name = max(array_map('strlen', ipull($out, 'name'))) + 2;
	$len_status = max(array_map('strlen', ipull($out, 'status'))) + 2;

	if ($this->getArgument('by-status')) {
	$out = isort($out, 'ssort');
	} else {
	$out = isort($out, 'esort');
	}

	$console = PhutilConsole::getConsole();
	foreach ($out as $line) {
	$color = $line['color'];
	$console->writeOut(
	"%s %s <fg:{$color}>%s</fg> %s\n",
	$line['current'] ? '* ' : ' ',
	str_pad($line['name'], $len_name),
	str_pad($line['status'], $len_status),
	$line['desc']);
	}
	}

	}

File Metadata

Mime Type: text/x-php
Expires: Mon, Mar 24, 00:41 (13 h, 29 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1116193
Default Alt Text: ArcanistBranchWorkflow.php (7 KB)

ArcanistBranchWorkflow.phpNo OneTemporaryActions

ArcanistBranchWorkflow.phpView Options

File Metadata

Event Timeline

ArcanistBranchWorkflow.php
No OneTemporary
Actions

ArcanistBranchWorkflow.php
View Options