phorge-arcanist/src/repository/api/mercurial/ArcanistMercurialAPI.php

<?php

/*
 * Copyright 2012 Facebook, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/**
 * Interfaces with the Mercurial working copies.
 *
 * @group workingcopy
 */
class ArcanistMercurialAPI extends ArcanistRepositoryAPI {

  private $status;
  private $base;
  private $relativeCommit;

  public function getSourceControlSystemName() {
    return 'hg';
  }

  public function getSourceControlBaseRevision() {
    list($stdout) = execx(
      '(cd %s && hg id -ir %s)',
      $this->getPath(),
      $this->getRelativeCommit());
    return $stdout;
  }

  public function getSourceControlPath() {
    return '/';
  }

  public function getBranchName() {
    // TODO: I have nearly no idea how hg local branches work.
    list($stdout) = execx(
      '(cd %s && hg branch)',
      $this->getPath());
    return $stdout;
  }

  public function setRelativeCommit($commit) {
    list($err) = exec_manual(
      '(cd %s && hg id -ir %s)',
      $this->getPath(),
      $commit);

    if ($err) {
      throw new ArcanistUsageException(
        "Commit '{$commit}' is not a valid Mercurial commit identifier.");
    }

    $this->relativeCommit = $commit;
    return $this;
  }

  public function getRelativeCommit() {
    if (empty($this->relativeCommit)) {
      list($stdout) = execx(
        '(cd %s && hg outgoing --branch `hg branch` --style default)',
        $this->getPath());
      $logs = ArcanistMercurialParser::parseMercurialLog($stdout);
      if (!count($logs)) {
        throw new ArcanistUsageException("You have no outgoing changes!");
      }

      $outgoing_revs = ipull($logs, 'rev');

      // This is essentially an implementation of a theoretical `hg merge-base`
      // command.
      $against = 'tip';
      while (true) {
        // NOTE: The "^" and "~" syntaxes were only added in hg 1.9, which is
        // new as of July 2011, so do this in a compatible way. Also, "hg log"
        // and "hg outgoing" don't necessarily show parents (even if given an
        // explicit template consisting of just the parents token) so we need
        // to separately execute "hg parents".

        list($stdout) = execx(
          '(cd %s && hg parents --style default --rev %s)',
          $this->getPath(),
          $against);
        $parents_logs = ArcanistMercurialParser::parseMercurialLog($stdout);

        list($p1, $p2) = array_merge($parents_logs, array(null, null));

        if ($p1 && !in_array($p1['rev'], $outgoing_revs)) {
          $against = $p1['rev'];
          break;
        } else if ($p2 && !in_array($p2['rev'], $outgoing_revs)) {
          $against = $p2['rev'];
          break;
        } else if ($p1) {
          $against = $p1['rev'];
        } else {
          // This is the case where you have a new repository and the entire
          // thing is outgoing; Mercurial literally accepts "--rev null" as
          // meaning "diff against the empty state".
          $against = 'null';
          break;
        }
      }

      $this->relativeCommit = $against;
    }
    return $this->relativeCommit;
  }

  public function getLocalCommitInformation() {
    list($info) = execx(
      '(cd %s && hg log --style default --rev %s..%s --)',
      $this->getPath(),
      $this->getRelativeCommit(),
      $this->getWorkingCopyRevision());
    $logs = ArcanistMercurialParser::parseMercurialLog($info);

    // Get rid of the first log, it's not actually part of the diff. "hg log"
    // is inclusive, while "hg diff" is exclusive.
    array_shift($logs);

    // Expand short hashes (12 characters) to full hashes (40 characters) by
    // issuing a big "hg log" command. Possibly we should do this with parents
    // too, but nothing uses them directly at the moment.
    if ($logs) {
      $cmd = array();
      foreach (ipull($logs, 'rev') as $rev) {
        $cmd[] = csprintf('--rev %s', $rev);
      }

      list($full) = execx(
        '(cd %s && hg log --template %s %C --)',
        $this->getPath(),
        '{node}\\n',
        implode(' ', $cmd));

      $full = explode("\n", trim($full));
      foreach ($logs as $key => $dict) {
        $logs[$key]['rev'] = array_pop($full);
      }
    }

    return $logs;
  }

  public function getBlame($path) {
    list($stdout) = execx(
      '(cd %s && hg annotate -u -v -c --rev %s -- %s)',
      $this->getPath(),
      $this->getRelativeCommit(),
      $path);

    $blame = array();
    foreach (explode("\n", trim($stdout)) as $line) {
      if (!strlen($line)) {
        continue;
      }

      $matches = null;
      $ok = preg_match('/^\s*([^:]+?) [a-f0-9]{12}: (.*)$/', $line, $matches);

      if (!$ok) {
        throw new Exception("Unable to parse Mercurial blame line: {$line}");
      }

      $revision = $matches[2];
      $author = trim($matches[1]);
      $blame[] = array($author, $revision);
    }

    return $blame;
  }

  public function getWorkingCopyStatus() {

    if (!isset($this->status)) {
      // A reviewable revision spans multiple local commits in Mercurial, but
      // there is no way to get file change status across multiple commits, so
      // just take the entire diff and parse it to figure out what's changed.

      $diff = $this->getFullMercurialDiff();
      $parser = new ArcanistDiffParser();
      $changes = $parser->parseDiff($diff);

      $status_map = array();

      foreach ($changes as $change) {
        $flags = 0;
        switch ($change->getType()) {
          case ArcanistDiffChangeType::TYPE_ADD:
          case ArcanistDiffChangeType::TYPE_MOVE_HERE:
          case ArcanistDiffChangeType::TYPE_COPY_HERE:
            $flags |= self::FLAG_ADDED;
            break;
          case ArcanistDiffChangeType::TYPE_CHANGE:
          case ArcanistDiffChangeType::TYPE_COPY_AWAY: // Check for changes?
            $flags |= self::FLAG_MODIFIED;
            break;
          case ArcanistDiffChangeType::TYPE_DELETE:
          case ArcanistDiffChangeType::TYPE_MOVE_AWAY:
          case ArcanistDiffChangeType::TYPE_MULTICOPY:
            $flags |= self::FLAG_DELETED;
            break;
        }
        $status_map[$change->getCurrentPath()] = $flags;
      }

      list($stdout) = execx(
        '(cd %s && hg status)',
        $this->getPath());

      $working_status = ArcanistMercurialParser::parseMercurialStatus($stdout);
      foreach ($working_status as $path => $status) {
        if ($status & ArcanistRepositoryAPI::FLAG_UNTRACKED) {
          // If the file is untracked, don't mark it uncommitted.
          continue;
        }
        $status |= self::FLAG_UNCOMMITTED;
        if (!empty($status_map[$path])) {
          $status_map[$path] |= $status;
        } else {
          $status_map[$path] = $status;
        }
      }

      $this->status = $status_map;
    }

    return $this->status;
  }

  private function getDiffOptions() {
    $options = array(
      '--git',
      // NOTE: We can't use "--color never" because that flag is provided
      // by the color extension, which may or may not be enabled. Instead,
      // set the color mode configuration so that color is disabled regardless
      // of whether the extension is present or not.
      '--config color.mode=off',
      '-U'.$this->getDiffLinesOfContext(),
    );
    return implode(' ', $options);
  }

  public function getRawDiffText($path) {
    $options = $this->getDiffOptions();

    list($stdout) = execx(
      '(cd %s && hg diff %C --rev %s --rev %s -- %s)',
      $this->getPath(),
      $options,
      $this->getRelativeCommit(),
      $this->getWorkingCopyRevision(),
      $path);

    return $stdout;
  }

  public function getFullMercurialDiff() {
    $options = $this->getDiffOptions();

    list($stdout) = execx(
      '(cd %s && hg diff %C --rev %s --rev %s --)',
      $this->getPath(),
      $options,
      $this->getRelativeCommit(),
      $this->getWorkingCopyRevision());

    return $stdout;
  }

  public function getOriginalFileData($path) {
    return $this->getFileDataAtRevision($path, $this->getRelativeCommit());
  }

  public function getCurrentFileData($path) {
    return $this->getFileDataAtRevision(
      $path,
      $this->getWorkingCopyRevision());
  }

  private function getFileDataAtRevision($path, $revision) {
    list($err, $stdout) = exec_manual(
      '(cd %s && hg cat --rev %s -- %s)',
      $this->getPath(),
      $revision,
      $path);
    if ($err) {
      // Assume this is "no file at revision", i.e. a deleted or added file.
      return null;
    } else {
      return $stdout;
    }
  }

  private function getWorkingCopyRevision() {
    // In Mercurial, "tip" means the tip of the current branch, not what's in
    // the working copy. The tip may be ahead of the working copy. We need to
    // use "hg summary" to figure out what is actually in the working copy.
    // For instance, "hg up 4 && arc diff" should not show commits 5 and above.

    // Without arguments, "hg id" shows the current working directory's commit,
    // and "--debug" expands it to a 40-character hash.
    list($stdout) = execx(
      '(cd %s && hg --debug id --id)',
      $this->getPath());

    // Even with "--id", "hg id" will print a trailing "+" after the hash
    // if the working copy is dirty (has uncommitted changes). We'll explicitly
    // detect this later by calling getWorkingCopyStatus(); ignore it for now.
    $stdout = trim($stdout);
    return rtrim($stdout, '+');
  }

  public function supportsRelativeLocalCommits() {
    return true;
  }

  public function parseRelativeLocalCommit(array $argv) {
    if (count($argv) == 0) {
      return;
    }
    if (count($argv) != 1) {
      throw new ArcanistUsageException("Specify only one commit.");
    }
    // This does the "hg id" call we need to normalize/validate the revision
    // identifier.
    $this->setRelativeCommit(reset($argv));
  }

  public function getAllLocalChanges() {
    $diff = $this->getFullMercurialDiff();
    $parser = new ArcanistDiffParser();
    return $parser->parseDiff($diff);
  }

  public function supportsLocalBranchMerge() {
    return true;
  }

  public function performLocalBranchMerge($branch, $message) {
    if ($branch) {
      $err = phutil_passthru(
        '(cd %s && hg merge --rev %s && hg commit -m %s)',
        $this->getPath(),
        $branch,
        $message);
    } else {
      $err = phutil_passthru(
        '(cd %s && hg merge && hg commit -m %s)',
        $this->getPath(),
        $message);
    }

    if ($err) {
      throw new ArcanistUsageException("Merge failed!");
    }
  }

  public function getFinalizedRevisionMessage() {
    return "You may now push this commit upstream, as appropriate (e.g. with ".
           "'hg push' or by printing and faxing it).";
  }

}