phorge-phorge/src/applications/phragment/controller/PhragmentPatchController.php

<?php

final class PhragmentPatchController extends PhragmentController {

  private $aid;
  private $bid;

  public function willProcessRequest(array $data) {
    $this->aid = idx($data, "aid", 0);
    $this->bid = idx($data, "bid", 0);
  }

  public function processRequest() {
    $request = $this->getRequest();
    $viewer = $request->getUser();

    // If "aid" is "x", then it means the user wants to generate
    // a patch of an empty file to the version specified by "bid".

    $ids = array($this->aid, $this->bid);
    if ($this->aid === "x") {
      $ids = array($this->bid);
    }

    $versions = id(new PhragmentFragmentVersionQuery())
      ->setViewer($viewer)
      ->withIDs($ids)
      ->execute();

    $version_a = null;
    if ($this->aid !== "x") {
      $version_a = idx($versions, $this->aid, null);
      if ($version_a === null) {
        return new Aphront404Response();
      }
    }

    $version_b = idx($versions, $this->bid, null);
    if ($version_b === null) {
      return new Aphront404Response();
    }

    $file_phids = array();
    if ($version_a !== null) {
      $file_phids[] = $version_a->getFilePHID();
    }
    $file_phids[] = $version_b->getFilePHID();

    $files = id(new PhabricatorFileQuery())
      ->setViewer($viewer)
      ->withPHIDs($file_phids)
      ->execute();
    $files = mpull($files, null, 'getPHID');

    $file_a = null;
    if ($version_a != null) {
      $file_a = idx($files, $version_a->getFilePHID(), null);
    }
    $file_b = idx($files, $version_b->getFilePHID(), null);

    $patch = PhragmentPatchUtil::calculatePatch($file_a, $file_b);

    if ($patch === null) {
      throw new Exception("Unable to compute patch!");
    }

    $a_sequence = 'x';
    if ($version_a !== null) {
      $a_sequence = $version_a->getSequence();
    }

    $name =
      $version_b->getFragment()->getName().'.'.
      $a_sequence.'.'.
      $version_b->getSequence().'.patch';

    $result = PhabricatorFile::buildFromFileDataOrHash(
      $patch,
      array(
        'name' => $name,
        'mime-type' => 'text/plain',
        'ttl' => time() + 60 * 60 * 24,
      ));
    return id(new AphrontRedirectResponse())
      ->setURI($result->getBestURI());
  }

}