Page MenuHomePhabricator
Files F14847168

D7801.id.diff
No OneTemporary
Actions

D7801.id.diff
View Options

Index: scripts/repository/manage_repositories.php
===================================================================
--- scripts/repository/manage_repositories.php
+++ scripts/repository/manage_repositories.php
@@ -22,6 +22,7 @@
new PhabricatorRepositoryManagementListWorkflow(),
new PhabricatorRepositoryManagementDeleteWorkflow(),
new PhabricatorRepositoryManagementMarkImportedWorkflow(),
+ new PhabricatorRepositoryManagementLookupUsersWorkflow(),
new PhabricatorRepositoryManagementImportingWorkflow(),
new PhutilHelpArgumentWorkflow(),
);
Index: src/__phutil_library_map__.php
===================================================================
--- src/__phutil_library_map__.php
+++ src/__phutil_library_map__.php
@@ -554,6 +554,7 @@
'DiffusionRepositoryRef' => 'applications/diffusion/data/DiffusionRepositoryRef.php',
'DiffusionRepositoryTag' => 'applications/diffusion/data/DiffusionRepositoryTag.php',
'DiffusionRequest' => 'applications/diffusion/request/DiffusionRequest.php',
+ 'DiffusionResolveUserQuery' => 'applications/diffusion/query/DiffusionResolveUserQuery.php',
'DiffusionSSHGitReceivePackWorkflow' => 'applications/diffusion/ssh/DiffusionSSHGitReceivePackWorkflow.php',
'DiffusionSSHGitUploadPackWorkflow' => 'applications/diffusion/ssh/DiffusionSSHGitUploadPackWorkflow.php',
'DiffusionSSHGitWorkflow' => 'applications/diffusion/ssh/DiffusionSSHGitWorkflow.php',
@@ -1794,6 +1795,7 @@
'PhabricatorRepositoryManagementEditWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementEditWorkflow.php',
'PhabricatorRepositoryManagementImportingWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementImportingWorkflow.php',
'PhabricatorRepositoryManagementListWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementListWorkflow.php',
+ 'PhabricatorRepositoryManagementLookupUsersWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementLookupUsersWorkflow.php',
'PhabricatorRepositoryManagementMarkImportedWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementMarkImportedWorkflow.php',
'PhabricatorRepositoryManagementPullWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementPullWorkflow.php',
'PhabricatorRepositoryManagementWorkflow' => 'applications/repository/management/PhabricatorRepositoryManagementWorkflow.php',
@@ -2935,6 +2937,7 @@
1 => 'PhabricatorApplicationSearchResultsControllerInterface',
),
'DiffusionRepositoryNewController' => 'DiffusionController',
+ 'DiffusionResolveUserQuery' => 'Phobject',
'DiffusionSSHGitReceivePackWorkflow' => 'DiffusionSSHGitWorkflow',
'DiffusionSSHGitUploadPackWorkflow' => 'DiffusionSSHGitWorkflow',
'DiffusionSSHGitWorkflow' => 'DiffusionSSHWorkflow',
@@ -4362,6 +4365,7 @@
'PhabricatorRepositoryManagementEditWorkflow' => 'PhabricatorRepositoryManagementWorkflow',
'PhabricatorRepositoryManagementImportingWorkflow' => 'PhabricatorRepositoryManagementWorkflow',
'PhabricatorRepositoryManagementListWorkflow' => 'PhabricatorRepositoryManagementWorkflow',
+ 'PhabricatorRepositoryManagementLookupUsersWorkflow' => 'PhabricatorRepositoryManagementWorkflow',
'PhabricatorRepositoryManagementMarkImportedWorkflow' => 'PhabricatorRepositoryManagementWorkflow',
'PhabricatorRepositoryManagementPullWorkflow' => 'PhabricatorRepositoryManagementWorkflow',
'PhabricatorRepositoryManagementWorkflow' => 'PhutilArgumentWorkflow',
Index: src/applications/diffusion/query/DiffusionResolveUserQuery.php
===================================================================
--- /dev/null
+++ src/applications/diffusion/query/DiffusionResolveUserQuery.php
@@ -0,0 +1,130 @@
+<?php
+
+/**
+ * Resolve an author or committer name, like
+ * `"Abraham Lincoln <alincoln@logcab.in>"`, into a valid Phabricator user
+ * account, like `@alincoln`.
+ */
+final class DiffusionResolveUserQuery extends Phobject {
+
+ private $name;
+ private $commit;
+
+ public function withName($name) {
+ $this->name = $name;
+ return $this;
+ }
+
+ public function withCommit($commit) {
+ $this->commit = $commit;
+ return $this;
+ }
+
+ public function execute() {
+ $user_name = $this->name;
+
+ $phid = $this->findUserPHID($this->name);
+ $phid = $this->fireLookupEvent($phid);
+
+ return $phid;
+ }
+
+ private function findUserPHID($user_name) {
+ if (!strlen($user_name)) {
+ return null;
+ }
+
+ $phid = $this->findUserByUserName($user_name);
+ if ($phid) {
+ return $phid;
+ }
+
+ $phid = $this->findUserByEmailAddress($user_name);
+ if ($phid) {
+ return $phid;
+ }
+
+ $phid = $this->findUserByRealName($user_name);
+ if ($phid) {
+ return $phid;
+ }
+
+ // No hits yet, try to parse it as an email address.
+
+ $email = new PhutilEmailAddress($user_name);
+
+ $phid = $this->findUserByEmailAddress($email->getAddress());
+ if ($phid) {
+ return $phid;
+ }
+
+ $display_name = $email->getDisplayName();
+ if ($display_name) {
+ $phid = $this->findUserByUserName($display_name);
+ if ($phid) {
+ return $phid;
+ }
+
+ $phid = $this->findUserByRealName($display_name);
+ if ($phid) {
+ return $phid;
+ }
+ }
+
+ return null;
+ }
+
+
+ /**
+ * Emit an event so installs can do custom lookup of commit authors who may
+ * not be naturally resolvable.
+ */
+ private function fireLookupEvent($guess) {
+
+ $type = PhabricatorEventType::TYPE_DIFFUSION_LOOKUPUSER;
+ $data = array(
+ 'commit' => $this->commit,
+ 'query' => $this->name,
+ 'result' => $guess,
+ );
+
+ $event = new PhabricatorEvent($type, $data);
+ PhutilEventEngine::dispatchEvent($event);
+
+ return $event->getValue('result');
+ }
+
+
+ private function findUserByUserName($user_name) {
+ $by_username = id(new PhabricatorUser())->loadOneWhere(
+ 'userName = %s',
+ $user_name);
+ if ($by_username) {
+ return $by_username->getPHID();
+ }
+ return null;
+ }
+
+
+ private function findUserByRealName($real_name) {
+ // Note, real names are not guaranteed unique, which is why we do it this
+ // way.
+ $by_realname = id(new PhabricatorUser())->loadAllWhere(
+ 'realName = %s',
+ $real_name);
+ if (count($by_realname) == 1) {
+ return reset($by_realname)->getPHID();
+ }
+ return null;
+ }
+
+
+ private function findUserByEmailAddress($email_address) {
+ $by_email = PhabricatorUser::loadOneWithEmailAddress($email_address);
+ if ($by_email) {
+ return $by_email->getPHID();
+ }
+ return null;
+ }
+
+}
Index: src/applications/repository/management/PhabricatorRepositoryManagementLookupUsersWorkflow.php
===================================================================
--- /dev/null
+++ src/applications/repository/management/PhabricatorRepositoryManagementLookupUsersWorkflow.php
@@ -0,0 +1,97 @@
+<?php
+
+final class PhabricatorRepositoryManagementLookupUsersWorkflow
+ extends PhabricatorRepositoryManagementWorkflow {
+
+ public function didConstruct() {
+ $this
+ ->setName('lookup-users')
+ ->setExamples('**lookup-users** __commit__ ...')
+ ->setSynopsis('Resolve user accounts for users attached to __commit__.')
+ ->setArguments(
+ array(
+ array(
+ 'name' => 'commits',
+ 'wildcard' => true,
+ ),
+ ));
+ }
+
+ public function execute(PhutilArgumentParser $args) {
+ $commits = $this->loadCommits($args, 'commits');
+ if (!$commits) {
+ throw new PhutilArgumentUsageException(
+ "Specify one or more commits to resolve users for.");
+ }
+
+ $console = PhutilConsole::getConsole();
+ foreach ($commits as $commit) {
+ $repo = $commit->getRepository();
+ $name = $repo->formatCommitName($commit->getCommitIdentifier());
+
+ $console->writeOut(
+ "%s\n",
+ pht("Examining commit %s...", $name));
+
+ $ref = id(new DiffusionLowLevelCommitQuery())
+ ->setRepository($repo)
+ ->withIdentifier($commit->getCommitIdentifier())
+ ->execute();
+
+ $author = $ref->getAuthor();
+ $console->writeOut(
+ "%s\n",
+ pht('Raw author string: %s', coalesce($author, 'null')));
+
+ if ($author !== null) {
+ $handle = $this->resolveUser($commit, $author);
+ if ($handle) {
+ $console->writeOut(
+ "%s\n",
+ pht('Phabricator user: %s', $handle->getFullName()));
+ } else {
+ $console->writeOut(
+ "%s\n",
+ pht('Unable to resolve a corresponding Phabricator user.'));
+ }
+ }
+
+ $committer = $ref->getCommitter();
+ $console->writeOut(
+ "%s\n",
+ pht('Raw committer string: %s', coalesce($committer, 'null')));
+
+ if ($committer !== null) {
+ $handle = $this->resolveUser($commit, $committer);
+ if ($handle) {
+ $console->writeOut(
+ "%s\n",
+ pht('Phabricator user: %s', $handle->getFullName()));
+ } else {
+ $console->writeOut(
+ "%s\n",
+ pht('Unable to resolve a corresponding Phabricator user.'));
+ }
+ }
+ }
+
+ return 0;
+ }
+
+ private function resolveUser(PhabricatorRepositoryCommit $commit, $name) {
+ $phid = id(new DiffusionResolveUserQuery())
+ ->withCommit($commit)
+ ->withName($name)
+ ->execute();
+
+ if (!$phid) {
+ return null;
+ }
+
+ return id(new PhabricatorHandleQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withPHIDs(array($phid))
+ ->executeOne();
+ }
+
+}
Index: src/applications/repository/management/PhabricatorRepositoryManagementWorkflow.php
===================================================================
--- src/applications/repository/management/PhabricatorRepositoryManagementWorkflow.php
+++ src/applications/repository/management/PhabricatorRepositoryManagementWorkflow.php
@@ -30,5 +30,27 @@
return $repos;
}
+ protected function loadCommits(PhutilArgumentParser $args, $param) {
+ $names = $args->getArg($param);
+ if (!$names) {
+ return null;
+ }
+
+ $query = id(new DiffusionCommitQuery())
+ ->setViewer(PhabricatorUser::getOmnipotentUser())
+ ->withIdentifiers($names);
+
+ $query->execute();
+ $map = $query->getIdentifierMap();
+
+ foreach ($names as $name) {
+ if (empty($map[$name])) {
+ throw new PhutilArgumentUsageException(
+ pht('Commit "%s" does not exist or is ambiguous.', $name));
+ }
+ }
+
+ return $map;
+ }
}
Index: src/applications/repository/worker/commitmessageparser/PhabricatorRepositoryCommitMessageParserWorker.php
===================================================================
--- src/applications/repository/worker/commitmessageparser/PhabricatorRepositoryCommitMessageParserWorker.php
+++ src/applications/repository/worker/commitmessageparser/PhabricatorRepositoryCommitMessageParserWorker.php
@@ -22,7 +22,7 @@
$data->setAuthorName($author);
$data->setCommitDetail(
'authorPHID',
- $this->resolveUserPHID($author));
+ $this->resolveUserPHID($commit, $author));
$data->setCommitMessage($message);
@@ -30,16 +30,13 @@
$data->setCommitDetail('committer', $committer);
$data->setCommitDetail(
'committerPHID',
- $this->resolveUserPHID($committer));
+ $this->resolveUserPHID($commit, $committer));
}
$repository = $this->repository;
- $author_phid = $this->lookupUser(
- $commit,
- $data->getAuthorName(),
- $data->getCommitDetail('authorPHID'));
- $data->setCommitDetail('authorPHID', $author_phid);
+ $author_phid = $data->getCommitDetail('authorPHID');
+ $committer_phid = $data->getCommitDetail('committerPHID');
$user = new PhabricatorUser();
if ($author_phid) {
@@ -87,12 +84,6 @@
'differential.revisionID',
$revision_id);
- $committer_phid = $this->lookupUser(
- $commit,
- $data->getCommitDetail('committer'),
- $data->getCommitDetail('committerPHID'));
- $data->setCommitDetail('committerPHID', $committer_phid);
-
if ($author_phid != $commit->getAuthorPHID()) {
$commit->setAuthorPHID($author_phid);
}
@@ -449,7 +440,7 @@
/**
* Given a set of revisions, returns the revision with the latest
- * updated time. This is ostensibly the most recent revision.
+ * updated time. This is ostensibly the most recent revision.
*/
private function identifyMostRecentRevision(array $revisions) {
assert_instances_of($revisions, 'DifferentialRevision');
@@ -457,98 +448,14 @@
return end($revisions);
}
- /**
- * Emit an event so installs can do custom lookup of commit authors who may
- * not be naturally resolvable.
- */
- private function lookupUser(
+ private function resolveUserPHID(
PhabricatorRepositoryCommit $commit,
- $query,
- $guess) {
-
- $type = PhabricatorEventType::TYPE_DIFFUSION_LOOKUPUSER;
- $data = array(
- 'commit' => $commit,
- 'query' => $query,
- 'result' => $guess,
- );
-
- $event = new PhabricatorEvent($type, $data);
- PhutilEventEngine::dispatchEvent($event);
-
- return $event->getValue('result');
- }
-
- private function resolveUserPHID($user_name) {
- if (!strlen($user_name)) {
- return null;
- }
-
- $phid = $this->findUserByUserName($user_name);
- if ($phid) {
- return $phid;
- }
- $phid = $this->findUserByEmailAddress($user_name);
- if ($phid) {
- return $phid;
- }
- $phid = $this->findUserByRealName($user_name);
- if ($phid) {
- return $phid;
- }
-
- // No hits yet, try to parse it as an email address.
-
- $email = new PhutilEmailAddress($user_name);
-
- $phid = $this->findUserByEmailAddress($email->getAddress());
- if ($phid) {
- return $phid;
- }
+ $user_name) {
- $display_name = $email->getDisplayName();
- if ($display_name) {
- $phid = $this->findUserByUserName($display_name);
- if ($phid) {
- return $phid;
- }
- $phid = $this->findUserByRealName($display_name);
- if ($phid) {
- return $phid;
- }
- }
-
- return null;
- }
-
- private function findUserByUserName($user_name) {
- $by_username = id(new PhabricatorUser())->loadOneWhere(
- 'userName = %s',
- $user_name);
- if ($by_username) {
- return $by_username->getPHID();
- }
- return null;
- }
-
- private function findUserByRealName($real_name) {
- // Note, real names are not guaranteed unique, which is why we do it this
- // way.
- $by_realname = id(new PhabricatorUser())->loadAllWhere(
- 'realName = %s',
- $real_name);
- if (count($by_realname) == 1) {
- return reset($by_realname)->getPHID();
- }
- return null;
- }
-
- private function findUserByEmailAddress($email_address) {
- $by_email = PhabricatorUser::loadOneWithEmailAddress($email_address);
- if ($by_email) {
- return $by_email->getPHID();
- }
- return null;
+ return id(new DiffusionResolveUserQuery())
+ ->withCommit($commit)
+ ->withName($user_name)
+ ->execute();
}
}

File Metadata

Mime Type
text/plain
Expires
Tue, Feb 4, 9:26 AM (40 m, 56 s)
Storage Engine
blob
Storage Format
Encrypted (AES-256-CBC)
Storage Handle
7089301
Default Alt Text
D7801.id.diff (15 KB)

Event Timeline

Phabricator · Built by Phacility