PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.php
View Options

	<?php

	final class PhabricatorRepositoryManagementRebuildIdentitiesWorkflow
	extends PhabricatorRepositoryManagementWorkflow {

	protected function didConstruct() {
	$this
	->setName('rebuild-identities')
	->setExamples(
	'rebuild-identities [__options__] __repository__')
	->setSynopsis(pht('Rebuild repository identities from commits.'))
	->setArguments(
	array(
	array(
	'name' => 'repositories',
	'wildcard' => true,
	),
	array(
	'name' => 'all',
	'help' => pht('Rebuild identities across all repositories.'),
	),
	));
	}

	public function execute(PhutilArgumentParser $args) {
	$console = PhutilConsole::getConsole();

	$all = $args->getArg('all');
	$repositories = $args->getArg('repositories');

	if ($all xor empty($repositories)) {
	throw new PhutilArgumentUsageException(
	pht('Specify --all or a list of repositories, but not both.'));
	}

	$query = id(new DiffusionCommitQuery())
	->setViewer(PhabricatorUser::getOmnipotentUser())
	->needCommitData(true);

	if ($repositories) {
	$repos = $this->loadRepositories($args, 'repositories');
	$query->withRepositoryIDs(mpull($repos, 'getID'));
	}

	$iterator = new PhabricatorQueryIterator($query);
	foreach ($iterator as $commit) {
	$needs_update = false;

	$data = $commit->getCommitData();
	$author_name = $data->getAuthorName();

	$author_identity = $this->getIdentityForCommit(
	$commit,
	$author_name);

	$author_phid = $commit->getAuthorIdentityPHID();
	$identity_phid = $author_identity->getPHID();
	if ($author_phid !== $identity_phid) {
	$commit->setAuthorIdentityPHID($identity_phid);
	$data->setCommitDetail('authorIdentityPHID', $identity_phid);
	$needs_update = true;
	}

	$committer_name = $data->getCommitDetail('committer', null);
	$committer_phid = $commit->getCommitterIdentityPHID();
	if (strlen($committer_name)) {
	$committer_identity = $this->getIdentityForCommit(
	$commit,
	$committer_name);
	$identity_phid = $committer_identity->getPHID();
	} else {
	$identity_phid = null;
	}

	if ($committer_phid !== $identity_phid) {
	$commit->setCommitterIdentityPHID($identity_phid);
	$data->setCommitDetail('committerIdentityPHID', $identity_phid);
	$needs_update = true;
	}

	if ($needs_update) {
	$commit->save();
	$data->save();
	echo tsprintf(
	"Rebuilt identities for %s.\n",
	$commit->getDisplayName());
	} else {
	echo tsprintf(
	"No changes for %s.\n",
	$commit->getDisplayName());
	}
	}

	}

	private function getIdentityForCommit(
	PhabricatorRepositoryCommit $commit, $identity_name) {

	static $seen = array();
	$identity_key = PhabricatorHash::digestForIndex($identity_name);
	if (empty($seen[$identity_key])) {
	try {
	$user_phid = id(new DiffusionResolveUserQuery())
	->withCommit($commit)
	->withName($identity_name)
	->execute();

	$identity = id(new PhabricatorRepositoryIdentity())
	->setAuthorPHID($commit->getPHID())
	->setIdentityName($identity_name)
	->setAutomaticGuessedUserPHID($user_phid)
	->save();
	} catch (AphrontDuplicateKeyQueryException $ex) {
	// Somehow this identity already exists?
	$identity = id(new PhabricatorRepositoryIdentityQuery())
	->setViewer(PhabricatorUser::getOmnipotentUser())
	->withIdentityNames(array($identity_name))
	->executeOne();
	}
	$seen[$identity_key] = $identity;
	}

	return $seen[$identity_key];
	}
	}

File Metadata

Mime Type: text/x-php
Expires: Wed, Feb 25, 1:20 AM (22 m, 10 s)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 851790
Default Alt Text: PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.php (3 KB)

PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.php
No OneTemporary
Actions

PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.php
View Options

File Metadata

Event Timeline

PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.phpNo OneTemporaryActions

PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.phpView Options

File Metadata

Event Timeline

PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.php
No OneTemporary
Actions

PhabricatorRepositoryManagementRebuildIdentitiesWorkflow.php
View Options