phorge-phorge/src/applications/transactions/draft/PhabricatorDraftEngine.php

<?php

abstract class PhabricatorDraftEngine
  extends Phobject {

  private $viewer;
  private $object;
  private $hasVersionedDraft;
  private $versionedDraft;

  final public function setViewer(PhabricatorUser $viewer) {
    $this->viewer = $viewer;
    return $this;
  }

  final public function getViewer() {
    return $this->viewer;
  }

  final public function setObject($object) {
    $this->object = $object;
    return $this;
  }

  final public function getObject() {
    return $this->object;
  }

  final public function setVersionedDraft(
    PhabricatorVersionedDraft $draft = null) {
    $this->hasVersionedDraft = true;
    $this->versionedDraft = $draft;
    return $this;
  }

  final public function getVersionedDraft() {
    if (!$this->hasVersionedDraft) {
      $draft = PhabricatorVersionedDraft::loadDraft(
        $this->getObject()->getPHID(),
        $this->getViewer()->getPHID());
      $this->setVersionedDraft($draft);
    }

    return $this->versionedDraft;
  }

  protected function hasVersionedDraftContent() {
    $draft = $this->getVersionedDraft();
    if (!$draft) {
      return false;
    }

    if ($draft->getProperty('comment')) {
      return true;
    }

    if ($draft->getProperty('actions')) {
      return true;
    }

    return false;
  }

  protected function hasCustomDraftContent() {
    return false;
  }

  final protected function hasAnyDraftContent() {
    if ($this->hasVersionedDraftContent()) {
      return true;
    }

    if ($this->hasCustomDraftContent()) {
      return true;
    }

    return false;
  }

  final public function synchronize() {
    $object_phid = $this->getObject()->getPHID();
    $viewer_phid = $this->getViewer()->getPHID();

    $has_draft = $this->hasAnyDraftContent();

    $draft_type = PhabricatorObjectHasDraftEdgeType::EDGECONST;
    $editor = id(new PhabricatorEdgeEditor());

    if ($has_draft) {
      $editor->addEdge($object_phid, $draft_type, $viewer_phid);
    } else {
      $editor->removeEdge($object_phid, $draft_type, $viewer_phid);
    }

    $editor->save();
  }

  final public static function attachDrafts(
    PhabricatorUser $viewer,
    array $objects) {
    assert_instances_of($objects, 'PhabricatorDraftInterface');

    $viewer_phid = $viewer->getPHID();

    if (!$viewer_phid) {
      // Viewers without a valid PHID can never have drafts.
      foreach ($objects as $object) {
        $object->attachHasDraft($viewer, false);
      }
      return;
    } else {
      $draft_type = PhabricatorObjectHasDraftEdgeType::EDGECONST;

      $edge_query = id(new PhabricatorEdgeQuery())
        ->withSourcePHIDs(mpull($objects, 'getPHID'))
        ->withEdgeTypes(
          array(
            $draft_type,
          ))
        ->withDestinationPHIDs(array($viewer_phid));

      $edge_query->execute();

      foreach ($objects as $object) {
        $has_draft = (bool)$edge_query->getDestinationPHIDs(
          array(
            $object->getPHID(),
          ));

        $object->attachHasDraft($viewer, $has_draft);
      }
    }
  }

}