Diffusion Phabricator (feature/better-testing) Browse

better-testing

DifferentialRevisionEditor.php
View Options

			<?php

			/**
			* Handle major edit operations to DifferentialRevision -- adding and removing
			* reviewers, diffs, and CCs. Unlike simple edits, these changes trigger
			* complicated email workflows.
			*/
			final class DifferentialRevisionEditor extends PhabricatorEditor {

			protected $revision;

			protected $cc = null;
			protected $reviewers = null;
			protected $diff;
			protected $comments;
			protected $silentUpdate;

			private $auxiliaryFields = array();
			private $contentSource;
			private $isCreate;
			private $aphrontRequestForEventDispatch;


			public function setAphrontRequestForEventDispatch(AphrontRequest $request) {
			$this->aphrontRequestForEventDispatch = $request;
			return $this;
			}

			public function getAphrontRequestForEventDispatch() {
			return $this->aphrontRequestForEventDispatch;
			}

			public function __construct(DifferentialRevision $revision) {
			$this->revision = $revision;
			$this->isCreate = !($revision->getID());
			}

			public static function newRevisionFromConduitWithDiff(
			array $fields,
			DifferentialDiff $diff,
			PhabricatorUser $actor) {

			$revision = new DifferentialRevision();
			$revision->setPHID($revision->generatePHID());
			$revision->setAuthorPHID($actor->getPHID());
			$revision->setStatus(ArcanistDifferentialRevisionStatus::NEEDS_REVIEW);

			$editor = new DifferentialRevisionEditor($revision);
			$editor->setActor($actor);
			$editor->addDiff($diff, null);
			$editor->copyFieldsFromConduit($fields);

			$editor->save();

			return $revision;
			}

			public function copyFieldsFromConduit(array $fields) {

			$actor = $this->getActor();
			$revision = $this->revision;
			$revision->loadRelationships();

			$all_fields = DifferentialFieldSelector::newSelector()
			->getFieldSpecifications();

			$aux_fields = array();
			foreach ($all_fields as $aux_field) {
			$aux_field->setRevision($revision);
			$aux_field->setDiff($this->diff);
			$aux_field->setUser($actor);
			if ($aux_field->shouldAppearOnCommitMessage()) {
			$aux_fields[$aux_field->getCommitMessageKey()] = $aux_field;
			}
			}

			foreach ($fields as $field => $value) {
			if (empty($aux_fields[$field])) {
			throw new Exception(
			"Parsed commit message contains unrecognized field '{$field}'.");
			}
			$aux_fields[$field]->setValueFromParsedCommitMessage($value);
			}

			foreach ($aux_fields as $aux_field) {
			$aux_field->validateField();
			}

			$this->setAuxiliaryFields($all_fields);
			}

			public function setAuxiliaryFields(array $auxiliary_fields) {
			assert_instances_of($auxiliary_fields, 'DifferentialFieldSpecification');
			$this->auxiliaryFields = $auxiliary_fields;
			return $this;
			}

			public function getRevision() {
			return $this->revision;
			}

			public function setReviewers(array $reviewers) {
			$this->reviewers = $reviewers;
			return $this;
			}

			public function setCCPHIDs(array $cc) {
			$this->cc = $cc;
			return $this;
			}

			public function setContentSource(PhabricatorContentSource $content_source) {
			$this->contentSource = $content_source;
			return $this;
			}

			public function addDiff(DifferentialDiff $diff, $comments) {
			if ($diff->getRevisionID() &&
			$diff->getRevisionID() != $this->getRevision()->getID()) {
			$diff_id = (int)$diff->getID();
			$targ_id = (int)$this->getRevision()->getID();
			$real_id = (int)$diff->getRevisionID();
			throw new Exception(
			"Can not attach diff #{$diff_id} to Revision D{$targ_id}, it is ".
			"already attached to D{$real_id}.");
			}
			$this->diff = $diff;
			$this->comments = $comments;

			$repository = id(new DifferentialRepositoryLookup())
			->setViewer($this->getActor())
			->setDiff($diff)
			->lookupRepository();

			if ($repository) {
			$this->getRevision()->setRepositoryPHID($repository->getPHID());
			}

			return $this;
			}

			protected function getDiff() {
			return $this->diff;
			}

			protected function getComments() {
			return $this->comments;
			}

			protected function getActorPHID() {
			return $this->getActor()->getPHID();
			}

			public function isNewRevision() {
			return !$this->getRevision()->getID();
			}

			/**
			* A silent update does not trigger Herald rules or send emails. This is used
			* for auto-amends at commit time.
			*/
			public function setSilentUpdate($silent) {
			$this->silentUpdate = $silent;
			return $this;
			}

			public function save() {
			$revision = $this->getRevision();

			$is_new = $this->isNewRevision();
			if ($is_new) {
			$this->initializeNewRevision($revision);
			}

			$revision->loadRelationships();

			$this->willWriteRevision();

			if ($this->reviewers === null) {
			$this->reviewers = $revision->getReviewers();
			}

			if ($this->cc === null) {
			$this->cc = $revision->getCCPHIDs();
			}

			if ($is_new) {
			$content_blocks = array();
			foreach ($this->auxiliaryFields as $field) {
			if ($field->shouldExtractMentions()) {
			$content_blocks[] = $field->renderValueForCommitMessage(false);
			}
			}
			$phids = PhabricatorMarkupEngine::extractPHIDsFromMentions(
			$content_blocks);
			$this->cc = array_unique(array_merge($this->cc, $phids));
			}

			$diff = $this->getDiff();
			if ($diff) {
			$revision->setLineCount($diff->getLineCount());
			}

			// Save the revision, to generate its ID and PHID if it is new. We need
			// the ID/PHID in order to record them in Herald transcripts, but don't
			// want to hold a transaction open while running Herald because it is
			// potentially somewhat slow. The downside is that we may end up with a
			// saved revision/diff pair without appropriate CCs. We could be better
			// about this -- for example:
			//
			// - Herald can't affect reviewers, so we could compute them before
			// opening the transaction and then save them in the transaction.
			// - Herald doesn't really need PHIDs to compute its effects, we could
			// run it before saving these objects and then hand over the PHIDs later.
			//
			// But this should address the problem of orphaned revisions, which is
			// currently the only problem we experience in practice.

			$revision->openTransaction();

			if ($diff) {
			$revision->setBranchName($diff->getBranch());
			$revision->setArcanistProjectPHID($diff->getArcanistProjectPHID());
			}

			$revision->save();

			if ($diff) {
			$diff->setRevisionID($revision->getID());
			$diff->save();
			}

			$revision->saveTransaction();


			// We're going to build up three dictionaries: $add, $rem, and $stable. The
			// $add dictionary has added reviewers/CCs. The $rem dictionary has
			// reviewers/CCs who have been removed, and the $stable array is
			// reviewers/CCs who haven't changed. We're going to send new reviewers/CCs
			// a different ("welcome") email than we send stable reviewers/CCs.

			$old = array(
			'rev' => array_fill_keys($revision->getReviewers(), true),
			'ccs' => array_fill_keys($revision->getCCPHIDs(), true),
			);

			$xscript_header = null;
			$xscript_uri = null;

			$new = array(
			'rev' => array_fill_keys($this->reviewers, true),
			'ccs' => array_fill_keys($this->cc, true),
			);

			$rem_ccs = array();
			$xscript_phid = null;
			if ($diff) {
			$adapter = HeraldDifferentialRevisionAdapter::newLegacyAdapter(
			$revision,
			$diff);
			$adapter->setExplicitCCs($new['ccs']);
			$adapter->setExplicitReviewers($new['rev']);
			$adapter->setForbiddenCCs($revision->loadUnsubscribedPHIDs());

			$xscript = HeraldEngine::loadAndApplyRules($adapter);
			$xscript_uri = '/herald/transcript/'.$xscript->getID().'/';
			$xscript_phid = $xscript->getPHID();
			$xscript_header = $xscript->getXHeraldRulesHeader();

			$xscript_header = HeraldTranscript::saveXHeraldRulesHeader(
			$revision->getPHID(),
			$xscript_header);

			$sub = array(
			'rev' => array(),
			'ccs' => $adapter->getCCsAddedByHerald(),
			);
			$rem_ccs = $adapter->getCCsRemovedByHerald();
			} else {
			$sub = array(
			'rev' => array(),
			'ccs' => array(),
			);
			}

			// Remove any CCs which are prevented by Herald rules.
			$sub['ccs'] = array_diff_key($sub['ccs'], $rem_ccs);
			$new['ccs'] = array_diff_key($new['ccs'], $rem_ccs);

			$add = array();
			$rem = array();
			$stable = array();
			foreach (array('rev', 'ccs') as $key) {
			$add[$key] = array();
			if ($new[$key] !== null) {
			$add[$key] += array_diff_key($new[$key], $old[$key]);
			}
			$add[$key] += array_diff_key($sub[$key], $old[$key]);

			$combined = $sub[$key];
			if ($new[$key] !== null) {
			$combined += $new[$key];
			}
			$rem[$key] = array_diff_key($old[$key], $combined);

			$stable[$key] = array_diff_key($old[$key], $add[$key] + $rem[$key]);
			}

			self::updateReviewers(
			$revision,
			$this->getActor(),
			array_keys($add['rev']),
			array_keys($rem['rev']),
			$this->getActorPHID());

			// We want to attribute new CCs to a "reasonPHID", representing the reason
			// they were added. This is either a user (if some user explicitly CCs
			// them, or uses "Add CCs...") or a Herald transcript PHID, indicating that
			// they were added by a Herald rule.

			if ($add['ccs'] \|\| $rem['ccs']) {
			$reasons = array();
			foreach ($add['ccs'] as $phid => $ignored) {
			if (empty($new['ccs'][$phid])) {
			$reasons[$phid] = $xscript_phid;
			} else {
			$reasons[$phid] = $this->getActorPHID();
			}
			}
			foreach ($rem['ccs'] as $phid => $ignored) {
			if (empty($new['ccs'][$phid])) {
			$reasons[$phid] = $this->getActorPHID();
			} else {
			$reasons[$phid] = $xscript_phid;
			}
			}
			} else {
			$reasons = $this->getActorPHID();
			}

			self::alterCCs(
			$revision,
			$this->cc,
			array_keys($rem['ccs']),
			array_keys($add['ccs']),
			$reasons);

			$this->updateAuxiliaryFields();

			// Add the author and users included from Herald rules to the relevant set
			// of users so they get a copy of the email.
			if (!$this->silentUpdate) {
			if ($is_new) {
			$add['rev'][$this->getActorPHID()] = true;
			if ($diff) {
			$add['rev'] += $adapter->getEmailPHIDsAddedByHerald();
			}
			} else {
			$stable['rev'][$this->getActorPHID()] = true;
			if ($diff) {
			$stable['rev'] += $adapter->getEmailPHIDsAddedByHerald();
			}
			}
			}

			$mail = array();

			$phids = array($this->getActorPHID());

			$handles = id(new PhabricatorHandleQuery())
			->setViewer($this->getActor())
			->withPHIDs($phids)
			->execute();
			$actor_handle = $handles[$this->getActorPHID()];

			$changesets = null;
			$comment = null;
			if ($diff) {
			$changesets = $diff->loadChangesets();
			// TODO: This should probably be in DifferentialFeedbackEditor?
			if (!$is_new) {
			$comment = $this->createComment();
			}
			if ($comment) {
			$mail[] = id(new DifferentialNewDiffMail(
			$revision,
			$actor_handle,
			$changesets))
			->setActor($this->getActor())
			->setIsFirstMailAboutRevision($is_new)
			->setIsFirstMailToRecipients($is_new)
			->setComments($this->getComments())
			->setToPHIDs(array_keys($stable['rev']))
			->setCCPHIDs(array_keys($stable['ccs']));
			}

			// Save the changes we made above.

			$diff->setDescription(preg_replace('/\n.*/s', '', $this->getComments()));
			$diff->save();

			$this->updateAffectedPathTable($revision, $diff, $changesets);
			$this->updateRevisionHashTable($revision, $diff);

			// An updated diff should require review, as long as it's not closed
			// or accepted. The "accepted" status is "sticky" to encourage courtesy
			// re-diffs after someone accepts with minor changes/suggestions.

			$status = $revision->getStatus();
			if ($status != ArcanistDifferentialRevisionStatus::CLOSED &&
			$status != ArcanistDifferentialRevisionStatus::ACCEPTED) {
			$revision->setStatus(ArcanistDifferentialRevisionStatus::NEEDS_REVIEW);
			}

			} else {
			$diff = $revision->loadActiveDiff();
			if ($diff) {
			$changesets = $diff->loadChangesets();
			} else {
			$changesets = array();
			}
			}

			$revision->save();

			$this->didWriteRevision();

			$event_data = array(
			'revision_id' => $revision->getID(),
			'revision_phid' => $revision->getPHID(),
			'revision_name' => $revision->getTitle(),
			'revision_author_phid' => $revision->getAuthorPHID(),
			'action' => $is_new
			? DifferentialAction::ACTION_CREATE
			: DifferentialAction::ACTION_UPDATE,
			'feedback_content' => $is_new
			? phutil_utf8_shorten($revision->getSummary(), 140)
			: $this->getComments(),
			'actor_phid' => $revision->getAuthorPHID(),
			);

			$mailed_phids = array();
			if (!$this->silentUpdate) {
			$revision->loadRelationships();

			if ($add['rev']) {
			$message = id(new DifferentialNewDiffMail(
			$revision,
			$actor_handle,
			$changesets))
			->setActor($this->getActor())
			->setIsFirstMailAboutRevision($is_new)
			->setIsFirstMailToRecipients(true)
			->setToPHIDs(array_keys($add['rev']));

			if ($is_new) {
			// The first time we send an email about a revision, put the CCs in
			// the "CC:" field of the same "Review Requested" email that reviewers
			// get, so you don't get two initial emails if you're on a list that
			// is CC'd.
			$message->setCCPHIDs(array_keys($add['ccs']));
			}

			$mail[] = $message;
			}

			// If we added CCs, we want to send them an email, but only if they were
			// not already a reviewer and were not added as one (in these cases, they
			// got a "NewDiff" mail, either in the past or just a moment ago). You can
			// still get two emails, but only if a revision is updated and you are
			// added as a reviewer at the same time a list you are on is added as a
			// CC, which is rare and reasonable.

			$implied_ccs = self::getImpliedCCs($revision);
			$implied_ccs = array_fill_keys($implied_ccs, true);
			$add['ccs'] = array_diff_key($add['ccs'], $implied_ccs);

			if (!$is_new && $add['ccs']) {
			$mail[] = id(new DifferentialCCWelcomeMail(
			$revision,
			$actor_handle,
			$changesets))
			->setActor($this->getActor())
			->setIsFirstMailToRecipients(true)
			->setToPHIDs(array_keys($add['ccs']));
			}

			foreach ($mail as $message) {
			$message->setHeraldTranscriptURI($xscript_uri);
			$message->setXHeraldRulesHeader($xscript_header);
			$message->send();

			$mailed_phids[] = $message->getRawMail()->buildRecipientList();
			}
			$mailed_phids = array_mergev($mailed_phids);
			}

			id(new PhabricatorFeedStoryPublisher())
			->setStoryType('PhabricatorFeedStoryDifferential')
			->setStoryData($event_data)
			->setStoryTime(time())
			->setStoryAuthorPHID($revision->getAuthorPHID())
			->setRelatedPHIDs(
			array(
			$revision->getPHID(),
			$revision->getAuthorPHID(),
			))
			->setPrimaryObjectPHID($revision->getPHID())
			->setSubscribedPHIDs(
			array_merge(
			array($revision->getAuthorPHID()),
			$revision->getReviewers(),
			$revision->getCCPHIDs()))
			->setMailRecipientPHIDs($mailed_phids)
			->publish();

			id(new PhabricatorSearchIndexer())
			->indexDocumentByPHID($revision->getPHID());
			}

			public static function addCCAndUpdateRevision(
			$revision,
			$phid,
			PhabricatorUser $actor) {

			self::addCC($revision, $phid, $actor->getPHID());

			$type = PhabricatorEdgeConfig::TYPE_OBJECT_HAS_UNSUBSCRIBER;
			id(new PhabricatorEdgeEditor())
			->setActor($actor)
			->removeEdge($revision->getPHID(), $type, $phid)
			->save();
			}

			public static function removeCCAndUpdateRevision(
			$revision,
			$phid,
			PhabricatorUser $actor) {

			self::removeCC($revision, $phid, $actor->getPHID());

			$type = PhabricatorEdgeConfig::TYPE_OBJECT_HAS_UNSUBSCRIBER;
			id(new PhabricatorEdgeEditor())
			->setActor($actor)
			->addEdge($revision->getPHID(), $type, $phid)
			->save();
			}

			public static function addCC(
			DifferentialRevision $revision,
			$phid,
			$reason) {
			return self::alterCCs(
			$revision,
			$revision->getCCPHIDs(),
			$rem = array(),
			$add = array($phid),
			$reason);
			}

			public static function removeCC(
			DifferentialRevision $revision,
			$phid,
			$reason) {
			return self::alterCCs(
			$revision,
			$revision->getCCPHIDs(),
			$rem = array($phid),
			$add = array(),
			$reason);
			}

			protected static function alterCCs(
			DifferentialRevision $revision,
			array $stable_phids,
			array $rem_phids,
			array $add_phids,
			$reason_phid) {

			$dont_add = self::getImpliedCCs($revision);
			$add_phids = array_diff($add_phids, $dont_add);

			return self::alterRelationships(
			$revision,
			$stable_phids,
			$rem_phids,
			$add_phids,
			$reason_phid,
			DifferentialRevision::RELATION_SUBSCRIBED);
			}

			private static function getImpliedCCs(DifferentialRevision $revision) {
			return array_merge(
			$revision->getReviewers(),
			array($revision->getAuthorPHID()));
			}

			public static function updateReviewers(
			DifferentialRevision $revision,
			PhabricatorUser $actor,
			array $add_phids,
			array $remove_phids) {

			$reviewers = $revision->getReviewers();

			// This is here until the new way proves stable enough
			// See https://secure.phabricator.com/T1279
			self::alterReviewers(
			$revision,
			$reviewers,
			$remove_phids,
			$add_phids,
			$actor->getPHID());

			$editor = id(new PhabricatorEdgeEditor())
			->setActor($actor);

			$options = array(
			'data' => array(
			'status' => DifferentialReviewerStatus::STATUS_ADDED
			)
			);

			$reviewer_phids_map = array_fill_keys($reviewers, true);

			foreach ($add_phids as $phid) {

			// Adding an already existing edge again would have cause memory loss
			// That is, the previous state for that reviewer would be lost
			if (isset($reviewer_phids_map[$phid])) {
			continue;
			}

			$editor->addEdge(
			$revision->getPHID(),
			PhabricatorEdgeConfig::TYPE_DREV_HAS_REVIEWER,
			$phid,
			$options);
			}

			foreach ($remove_phids as $phid) {
			$editor->removeEdge(
			$revision->getPHID(),
			PhabricatorEdgeConfig::TYPE_DREV_HAS_REVIEWER,
			$phid);
			}

			$editor->save();
			}

			public static function updateReviewerStatus(
			DifferentialRevision $revision,
			PhabricatorUser $actor,
			$reviewer_phid,
			$status) {

			$reviewers = $revision->getReviewers();
			if (!in_array($reviewer_phid, $reviewers)) {
			// This is here until the new way proves stable enough
			// See https://secure.phabricator.com/T1279
			self::alterReviewers(
			$revision,
			$reviewers,
			array(),
			array($reviewer_phid),
			$actor->getPHID());
			}

			$options = array(
			'data' => array(
			'status' => $status
			)
			);

			$active_diff = $revision->loadActiveDiff();
			if ($active_diff) {
			$options['data']['diff'] = $active_diff->getID();
			}

			id(new PhabricatorEdgeEditor())
			->setActor($actor)
			->addEdge(
			$revision->getPHID(),
			PhabricatorEdgeConfig::TYPE_DREV_HAS_REVIEWER,
			$reviewer_phid,
			$options)
			->save();
			}

			/**
			* @deprecated
			*/
			private static function alterReviewers(
			DifferentialRevision $revision,
			array $stable_phids,
			array $rem_phids,
			array $add_phids,
			$reason_phid) {

			return self::alterRelationships(
			$revision,
			$stable_phids,
			$rem_phids,
			$add_phids,
			$reason_phid,
			DifferentialRevision::RELATION_REVIEWER);
			}

			private static function alterRelationships(
			DifferentialRevision $revision,
			array $stable_phids,
			array $rem_phids,
			array $add_phids,
			$reason_phid,
			$relation_type) {

			$rem_map = array_fill_keys($rem_phids, true);
			$add_map = array_fill_keys($add_phids, true);

			$seq_map = array_values($stable_phids);
			$seq_map = array_flip($seq_map);
			foreach ($rem_map as $phid => $ignored) {
			if (!isset($seq_map[$phid])) {
			$seq_map[$phid] = count($seq_map);
			}
			}
			foreach ($add_map as $phid => $ignored) {
			if (!isset($seq_map[$phid])) {
			$seq_map[$phid] = count($seq_map);
			}
			}

			$raw = $revision->getRawRelations($relation_type);
			$raw = ipull($raw, null, 'objectPHID');

			$sequence = count($seq_map);
			foreach ($raw as $phid => $ignored) {
			if (isset($seq_map[$phid])) {
			$raw[$phid]['sequence'] = $seq_map[$phid];
			} else {
			$raw[$phid]['sequence'] = $sequence++;
			}
			}
			$raw = isort($raw, 'sequence');

			foreach ($raw as $phid => $ignored) {
			if (isset($rem_map[$phid])) {
			unset($raw[$phid]);
			}
			}

			foreach ($add_phids as $add) {
			$reason = is_array($reason_phid)
			? idx($reason_phid, $add)
			: $reason_phid;

			$raw[$add] = array(
			'objectPHID' => $add,
			'sequence' => idx($seq_map, $add, $sequence++),
			'reasonPHID' => $reason,
			);
			}

			$conn_w = $revision->establishConnection('w');

			$sql = array();
			foreach ($raw as $relation) {
			$sql[] = qsprintf(
			$conn_w,
			'(%d, %s, %s, %d, %s)',
			$revision->getID(),
			$relation_type,
			$relation['objectPHID'],
			$relation['sequence'],
			$relation['reasonPHID']);
			}

			$conn_w->openTransaction();
			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE revisionID = %d AND relation = %s',
			DifferentialRevision::RELATIONSHIP_TABLE,
			$revision->getID(),
			$relation_type);
			if ($sql) {
			queryfx(
			$conn_w,
			'INSERT INTO %T
			(revisionID, relation, objectPHID, sequence, reasonPHID)
			VALUES %Q',
			DifferentialRevision::RELATIONSHIP_TABLE,
			implode(', ', $sql));
			}
			$conn_w->saveTransaction();

			$revision->loadRelationships();
			}


			private function createComment() {
			$revision_id = $this->revision->getID();
			$comment = id(new DifferentialComment())
			->setAuthorPHID($this->getActorPHID())
			->setRevisionID($revision_id)
			->setContent($this->getComments())
			->setAction(DifferentialAction::ACTION_UPDATE)
			->setMetadata(
			array(
			DifferentialComment::METADATA_DIFF_ID => $this->getDiff()->getID(),
			));

			if ($this->contentSource) {
			$comment->setContentSource($this->contentSource);
			}

			$comment->save();

			return $comment;
			}

			private function updateAuxiliaryFields() {
			$aux_map = array();
			foreach ($this->auxiliaryFields as $aux_field) {
			$key = $aux_field->getStorageKey();
			if ($key !== null) {
			$val = $aux_field->getValueForStorage();
			$aux_map[$key] = $val;
			}
			}

			if (!$aux_map) {
			return;
			}

			$revision = $this->revision;

			$fields = id(new DifferentialAuxiliaryField())->loadAllWhere(
			'revisionPHID = %s AND name IN (%Ls)',
			$revision->getPHID(),
			array_keys($aux_map));
			$fields = mpull($fields, null, 'getName');

			foreach ($aux_map as $key => $val) {
			$obj = idx($fields, $key);
			if (!strlen($val)) {
			// If the new value is empty, just delete the old row if one exists and
			// don't add a new row if it doesn't.
			if ($obj) {
			$obj->delete();
			}
			} else {
			if (!$obj) {
			$obj = new DifferentialAuxiliaryField();
			$obj->setRevisionPHID($revision->getPHID());
			$obj->setName($key);
			}

			if ($obj->getValue() !== $val) {
			$obj->setValue($val);
			$obj->save();
			}
			}
			}
			}

			private function willWriteRevision() {
			foreach ($this->auxiliaryFields as $aux_field) {
			$aux_field->willWriteRevision($this);
			}

			$this->dispatchEvent(
			PhabricatorEventType::TYPE_DIFFERENTIAL_WILLEDITREVISION);
			}

			private function didWriteRevision() {
			foreach ($this->auxiliaryFields as $aux_field) {
			$aux_field->didWriteRevision($this);
			}

			$this->dispatchEvent(
			PhabricatorEventType::TYPE_DIFFERENTIAL_DIDEDITREVISION);
			}

			private function dispatchEvent($type) {
			$event = new PhabricatorEvent(
			$type,
			array(
			'revision' => $this->revision,
			'new' => $this->isCreate,
			));

			$event->setUser($this->getActor());

			$request = $this->getAphrontRequestForEventDispatch();
			if ($request) {
			$event->setAphrontRequest($request);
			}

			PhutilEventEngine::dispatchEvent($event);
			}

			/**
			* Update the table which links Differential revisions to paths they affect,
			* so Diffusion can efficiently find pending revisions for a given file.
			*/
			private function updateAffectedPathTable(
			DifferentialRevision $revision,
			DifferentialDiff $diff,
			array $changesets) {
			assert_instances_of($changesets, 'DifferentialChangeset');

			$project = $diff->loadArcanistProject();
			if (!$project) {
			// Probably an old revision from before projects.
			return;
			}

			$repository = $project->loadRepository();
			if (!$repository) {
			// Probably no project <-> repository link, or the repository where the
			// project lives is untracked.
			return;
			}

			$path_prefix = null;

			$local_root = $diff->getSourceControlPath();
			if ($local_root) {
			// We're in a working copy which supports subdirectory checkouts (e.g.,
			// SVN) so we need to figure out what prefix we should add to each path
			// (e.g., trunk/projects/example/) to get the absolute path from the
			// root of the repository. DVCS systems like Git and Mercurial are not
			// affected.

			// Normalize both paths and check if the repository root is a prefix of
			// the local root. If so, throw it away. Note that this correctly handles
			// the case where the remote path is "/".
			$local_root = id(new PhutilURI($local_root))->getPath();
			$local_root = rtrim($local_root, '/');

			$repo_root = id(new PhutilURI($repository->getRemoteURI()))->getPath();
			$repo_root = rtrim($repo_root, '/');

			if (!strncmp($repo_root, $local_root, strlen($repo_root))) {
			$path_prefix = substr($local_root, strlen($repo_root));
			}
			}

			$paths = array();
			foreach ($changesets as $changeset) {
			$paths[] = $path_prefix.'/'.$changeset->getFilename();
			}

			// Mark this as also touching all parent paths, so you can see all pending
			// changes to any file within a directory.
			$all_paths = array();
			foreach ($paths as $local) {
			foreach (DiffusionPathIDQuery::expandPathToRoot($local) as $path) {
			$all_paths[$path] = true;
			}
			}
			$all_paths = array_keys($all_paths);

			$path_ids =
			PhabricatorRepositoryCommitChangeParserWorker::lookupOrCreatePaths(
			$all_paths);

			$table = new DifferentialAffectedPath();
			$conn_w = $table->establishConnection('w');

			$sql = array();
			foreach ($path_ids as $path_id) {
			$sql[] = qsprintf(
			$conn_w,
			'(%d, %d, %d, %d)',
			$repository->getID(),
			$path_id,
			time(),
			$revision->getID());
			}

			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE revisionID = %d',
			$table->getTableName(),
			$revision->getID());
			foreach (array_chunk($sql, 256) as $chunk) {
			queryfx(
			$conn_w,
			'INSERT INTO %T (repositoryID, pathID, epoch, revisionID) VALUES %Q',
			$table->getTableName(),
			implode(', ', $chunk));
			}
			}


			/**
			* Update the table connecting revisions to DVCS local hashes, so we can
			* identify revisions by commit/tree hashes.
			*/
			private function updateRevisionHashTable(
			DifferentialRevision $revision,
			DifferentialDiff $diff) {

			$vcs = $diff->getSourceControlSystem();
			if ($vcs == DifferentialRevisionControlSystem::SVN) {
			// Subversion has no local commit or tree hash information, so we don't
			// have to do anything.
			return;
			}

			$property = id(new DifferentialDiffProperty())->loadOneWhere(
			'diffID = %d AND name = %s',
			$diff->getID(),
			'local:commits');
			if (!$property) {
			return;
			}

			$hashes = array();

			$data = $property->getData();
			switch ($vcs) {
			case DifferentialRevisionControlSystem::GIT:
			foreach ($data as $commit) {
			$hashes[] = array(
			ArcanistDifferentialRevisionHash::HASH_GIT_COMMIT,
			$commit['commit'],
			);
			$hashes[] = array(
			ArcanistDifferentialRevisionHash::HASH_GIT_TREE,
			$commit['tree'],
			);
			}
			break;
			case DifferentialRevisionControlSystem::MERCURIAL:
			foreach ($data as $commit) {
			$hashes[] = array(
			ArcanistDifferentialRevisionHash::HASH_MERCURIAL_COMMIT,
			$commit['rev'],
			);
			}
			break;
			}

			$conn_w = $revision->establishConnection('w');

			$sql = array();
			foreach ($hashes as $info) {
			list($type, $hash) = $info;
			$sql[] = qsprintf(
			$conn_w,
			'(%d, %s, %s)',
			$revision->getID(),
			$type,
			$hash);
			}

			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE revisionID = %d',
			ArcanistDifferentialRevisionHash::TABLE_NAME,
			$revision->getID());

			if ($sql) {
			queryfx(
			$conn_w,
			'INSERT INTO %T (revisionID, type, hash) VALUES %Q',
			ArcanistDifferentialRevisionHash::TABLE_NAME,
			implode(', ', $sql));
			}
			}

			private function initializeNewRevision(DifferentialRevision $revision) {
			// These fields aren't nullable; set them to sensible defaults if they
			// haven't been configured. We're just doing this so we can generate an
			// ID for the revision if we don't have one already.
			$revision->setLineCount(0);
			if ($revision->getStatus() === null) {
			$revision->setStatus(ArcanistDifferentialRevisionStatus::NEEDS_REVIEW);
			}
			if ($revision->getTitle() === null) {
			$revision->setTitle('Untitled Revision');
			}
			if ($revision->getAuthorPHID() === null) {
			$revision->setAuthorPHID($this->getActorPHID());
			}
			if ($revision->getSummary() === null) {
			$revision->setSummary('');
			}
			if ($revision->getTestPlan() === null) {
			$revision->setTestPlan('');
			}
			}

			}

phab/src/applications/differential/editor/DifferentialRevisionEditor.php874a9b7fe3f8feature/better-testing

DifferentialRevisionEditor.phpView Options

phab/src/applications/differential/editor/DifferentialRevisionEditor.php
874a9b7fe3f8feature/better-testing

DifferentialRevisionEditor.php
View Options