src/applications/diffusion/identity/DiffusionRepositoryIdentityEngine.php

   1 <?php
   2
   3 final class DiffusionRepositoryIdentityEngine
   4   extends Phobject {
   5
   6   private $viewer;
   7   private $sourcePHID;
   8   private $dryRun;
   9
  10   public function setViewer(PhabricatorUser $viewer) {
  11     $this->viewer = $viewer;
  12     return $this;
  13   }
  14
  15   public function getViewer() {
  16     return $this->viewer;
  17   }
  18
  19   public function setSourcePHID($source_phid) {
  20     $this->sourcePHID = $source_phid;
  21     return $this;
  22   }
  23
  24   public function getSourcePHID() {
  25     if (!$this->sourcePHID) {
  26       throw new PhutilInvalidStateException('setSourcePHID');
  27     }
  28
  29     return $this->sourcePHID;
  30   }
  31
  32   public function setDryRun($dry_run) {
  33     $this->dryRun = $dry_run;
  34     return $this;
  35   }
  36
  37   public function getDryRun() {
  38     return $this->dryRun;
  39   }
  40
  41   public function newResolvedIdentity($raw_identity) {
  42     $identity = $this->loadRawIdentity($raw_identity);
  43
  44     if (!$identity) {
  45       $identity = $this->newIdentity($raw_identity);
  46     }
  47
  48     return $this->updateIdentity($identity);
  49   }
  50
  51   public function newUpdatedIdentity(PhabricatorRepositoryIdentity $identity) {
  52     return $this->updateIdentity($identity);
  53   }
  54
  55   private function loadRawIdentity($raw_identity) {
  56     $viewer = $this->getViewer();
  57
  58     return id(new PhabricatorRepositoryIdentityQuery())
  59       ->setViewer($viewer)
  60       ->withIdentityNames(array($raw_identity))
  61       ->executeOne();
  62   }
  63
  64   private function newIdentity($raw_identity) {
  65     $source_phid = $this->getSourcePHID();
  66
  67     return id(new PhabricatorRepositoryIdentity())
  68       ->setAuthorPHID($source_phid)
  69       ->setIdentityName($raw_identity);
  70   }
  71
  72   private function resolveIdentity(PhabricatorRepositoryIdentity $identity) {
  73     $raw_identity = $identity->getIdentityName();
  74
  75     return id(new DiffusionResolveUserQuery())
  76       ->withName($raw_identity)
  77       ->execute();
  78   }
  79
  80   private function updateIdentity(PhabricatorRepositoryIdentity $identity) {
  81
  82     // If we're updating an identity and it has a manual user PHID associated
  83     // with it but the user is no longer valid, remove the value. This likely
  84     // corresponds to a user that was destroyed.
  85
  86     $assigned_phid = $identity->getManuallySetUserPHID();
  87     $unassigned = DiffusionIdentityUnassignedDatasource::FUNCTION_TOKEN;
  88     if ($assigned_phid && ($assigned_phid !== $unassigned)) {
  89       $viewer = $this->getViewer();
  90       $user = id(new PhabricatorPeopleQuery())
  91         ->setViewer($viewer)
  92         ->withPHIDs(array($assigned_phid))
  93         ->executeOne();
  94       if (!$user) {
  95         $identity->setManuallySetUserPHID(null);
  96       }
  97     }
  98
  99     $resolved_phid = $this->resolveIdentity($identity);
 100
 101     $identity->setAutomaticGuessedUserPHID($resolved_phid);
 102
 103     if ($this->getDryRun()) {
 104       $identity->makeEphemeral();
 105     } else {
 106       $identity->save();
 107     }
 108
 109     return $identity;
 110   }
 111
 112   public function didUpdateEmailAddress($raw_address) {
 113     PhabricatorWorker::scheduleTask(
 114       'PhabricatorRepositoryIdentityChangeWorker',
 115       array(
 116         'emailAddresses' => array($raw_address),
 117       ));
 118   }
 119
 120 }