phorge-phorge/src/applications/differential/storage/DifferentialDraft.php

<?php

final class DifferentialDraft extends DifferentialDAO {

  protected $objectPHID;
  protected $authorPHID;
  protected $draftKey;

  public static function markHasDraft(
    $author_phid,
    $object_phid,
    $draft_key) {
    try {
      id(new DifferentialDraft())
        ->setObjectPHID($object_phid)
        ->setAuthorPHID($author_phid)
        ->setDraftKey($draft_key)
        ->save();
    } catch (AphrontDuplicateKeyQueryException $ex) {
      // no worries
    }
  }

  public static function deleteHasDraft(
    $author_phid,
    $object_phid,
    $draft_key) {
    $draft = id(new DifferentialDraft())->loadOneWhere(
      'objectPHID = %s AND authorPHID = %s AND draftKey = %s',
      $object_phid,
      $author_phid,
      $draft_key);
    if ($draft) {
      $draft->delete();
    }
  }

  public static function deleteAllDrafts(
    $author_phid,
    $object_phid) {

    $drafts = id(new DifferentialDraft())->loadAllWhere(
      'objectPHID = %s AND authorPHID = %s',
      $object_phid,
      $author_phid);
    foreach ($drafts as $draft) {
      $draft->delete();
    }
  }

}