phorge-phorge/scripts/repository/reparse.php

#!/usr/bin/env php
<?php

$root = dirname(dirname(dirname(__FILE__)));
require_once $root.'/scripts/__init_script__.php';

$args = new PhutilArgumentParser($argv);
$args->setSynopsis(<<<EOHELP
**reparse.php** __what__ __which_parts__ [--trace] [--force]

Rerun the Diffusion parser on specific commits and repositories. Mostly
useful for debugging changes to Diffusion.

e.g. enqueue reparse owners in the TEST repo for all commits:
./reparse.php --all TEST --owners

e.g. do same but exclude before yesterday (local time):
./reparse.php --all TEST --owners --min-date yesterday
./reparse.php --all TEST --owners --min-date "today -1 day"

e.g. do same but exclude before 03/31/2013 (local time):
./reparse.php --all TEST --owners --min-date "03/31/2013"
EOHELP
);

$min_date_usage_examples =
  "Valid examples:\n".
  "  'today', 'today 2pm', '-1 hour', '-2 hours', '-24 hours',\n".
  "  'yesterday', 'today -1 day', 'yesterday 2pm', '2pm -1 day',\n".
  "  'last Monday', 'last Monday 14:00', 'last Monday 2pm',\n".
  "  '31 March 2013', '31 Mar', '03/31', '03/31/2013',\n".
  "See __http://www.php.net/manual/en/datetime.formats.php__ for more.\n";

$args->parseStandardArguments();
$args->parse(
  array(
    // what
    array(
      'name'     => 'revision',
      'wildcard' => true,
    ),
    array(
      'name'     => 'all',
      'param'    => 'callsign or phid',
      'help'     => 'Reparse all commits in the specified repository. This '.
                    'mode queues parsers into the task queue; you must run '.
                    'taskmasters to actually do the parses. Use with '.
                    '__--force-local__ to run the tasks locally instead of '.
                    'with taskmasters.',
    ),
    array(
      'name'     => 'min-date',
      'param'    => 'date',
      'help'     => 'Must be used with __--all__, this will exclude commits '.
                    'which are earlier than __date__.'.
                    "\n".$min_date_usage_examples,
    ),
    // which parts
    array(
      'name'     => 'message',
      'help'     => 'Reparse commit messages.',
    ),
    array(
      'name'     => 'change',
      'help'     => 'Reparse changes.',
    ),
    array(
      'name'     => 'herald',
      'help'     => 'Reevaluate Herald rules (may send huge amounts of email!)',
    ),
    array(
      'name'     => 'owners',
      'help'     => 'Reevaluate related commits for owners packages (may '.
                    'delete existing relationship entries between your '.
                    'package and some old commits!)',
    ),
    array(
      'name'     => 'harbormaster',
      'help'     => 'EXPERIMENTAL. Execute Harbormaster.',
    ),
    // misc options
    array(
      'name'     => 'force',
      'short'    => 'f',
      'help'     => 'Act noninteractively, without prompting.',
    ),
    array(
      'name'     => 'force-local',
      'help'     => 'Only used with __--all__, use this to run the tasks '.
                    'locally instead of deferring them to taskmaster daemons.',
    ),
  ));

$all_from_repo = $args->getArg('all');
$reparse_message = $args->getArg('message');
$reparse_change = $args->getArg('change');
$reparse_herald = $args->getArg('herald');
$reparse_owners = $args->getArg('owners');
$reparse_harbormaster = $args->getArg('harbormaster');
$reparse_what = $args->getArg('revision');
$force = $args->getArg('force');
$force_local = $args->getArg('force-local');
$min_date = $args->getArg('min-date');

if (!$all_from_repo && !$reparse_what) {
  usage('Specify a commit or repository to reparse.');
}

if ($all_from_repo && $reparse_what) {
  $commits = implode(', ', $reparse_what);
  usage(
    "Specify a commit or repository to reparse, not both:\n".
    "All from repo: ".$all_from_repo."\n".
    "Commit(s) to reparse: ".$commits);
}

if (!$reparse_message && !$reparse_change && !$reparse_herald &&
    !$reparse_owners && !$reparse_harbormaster) {
  usage('Specify what information to reparse with --message, --change,  '.
        '--herald, --harbormaster, and/or --owners');
}

$min_timestamp = false;
if ($min_date) {
  $min_timestamp = strtotime($min_date);

  if (!$all_from_repo) {
    usage(
      "You must use --all if you specify --min-date\n".
      "e.g.\n".
      "  ./reparse.php --all TEST --owners --min-date yesterday");
  }

  // previous to PHP 5.1.0 you would compare with -1, instead of false
  if (false === $min_timestamp) {
    usage(
      "Supplied --min-date is not valid\n".
      "Supplied value: '".$min_date."'\n".
      $min_date_usage_examples);
  }
}

if ($reparse_owners && !$force) {
  echo phutil_console_wrap(
    'You are about to recreate the relationship entries between the commits '.
    'and the packages they touch. This might delete some existing '.
    'relationship entries for some old commits.');

  if (!phutil_console_confirm('Are you ready to continue?')) {
    echo "Cancelled.\n";
    exit(1);
  }
}

$commits = array();
if ($all_from_repo) {
  $repository = id(new PhabricatorRepository())->loadOneWhere(
    'callsign = %s OR phid = %s',
    $all_from_repo,
    $all_from_repo);
  if (!$repository) {
    throw new Exception("Unknown repository {$all_from_repo}!");
  }
  $constraint = '';
  if ($min_timestamp) {
    echo "Excluding entries before UNIX timestamp: ".$min_timestamp."\n";
    $table = new PhabricatorRepositoryCommit();
    $conn_r = $table->establishConnection('r');
    $constraint = qsprintf(
      $conn_r,
      'AND epoch >= %d',
      $min_timestamp);
  }
  $commits = id(new PhabricatorRepositoryCommit())->loadAllWhere(
    'repositoryID = %d %Q',
    $repository->getID(),
    $constraint);
  $callsign = $repository->getCallsign();
  if (!$commits) {
    echo "No commits have been discovered in {$callsign} repository!\n";
    exit;
  }
} else {
  $commits = array();
  foreach ($reparse_what as $identifier) {
    $matches = null;
    if (!preg_match('/r([A-Z]+)([a-z0-9]+)/', $identifier, $matches)) {
      throw new Exception("Can't parse commit identifier!");
    }
    $callsign = $matches[1];
    $commit_identifier = $matches[2];
    $repository = id(new PhabricatorRepository())->loadOneWhere(
      'callsign = %s',
      $callsign);
    if (!$repository) {
      throw new Exception("No repository with callsign '{$callsign}'!");
    }
    $commit = id(new PhabricatorRepositoryCommit())->loadOneWhere(
      'repositoryID = %d AND commitIdentifier = %s',
      $repository->getID(),
      $commit_identifier);
    if (!$commit) {
      throw new Exception(
        "No matching commit '{$commit_identifier}' in repository ".
        "'{$callsign}'. (For git and mercurial repositories, you must specify ".
        "the entire commit hash.)");
    }
    $commits[] = $commit;
  }
}

if ($all_from_repo && !$force_local) {
  echo phutil_console_format(
    '**NOTE**: This script will queue tasks to reparse the data. Once the '.
    'tasks have been queued, you need to run Taskmaster daemons to execute '.
    'them.');
  echo "\n\n";
  echo "QUEUEING TASKS (".number_format(count($commits))." Commits):\n";
}

$progress = new PhutilConsoleProgressBar();
$progress->setTotal(count($commits));

$tasks = array();
foreach ($commits as $commit) {
  $classes = array();
  switch ($repository->getVersionControlSystem()) {
    case PhabricatorRepositoryType::REPOSITORY_TYPE_GIT:
      if ($reparse_message) {
        $classes[] = 'PhabricatorRepositoryGitCommitMessageParserWorker';
      }
      if ($reparse_change) {
        $classes[] = 'PhabricatorRepositoryGitCommitChangeParserWorker';
      }
      break;
    case PhabricatorRepositoryType::REPOSITORY_TYPE_MERCURIAL:
      if ($reparse_message) {
        $classes[] = 'PhabricatorRepositoryMercurialCommitMessageParserWorker';
      }
      if ($reparse_change) {
        $classes[] = 'PhabricatorRepositoryMercurialCommitChangeParserWorker';
      }
      break;
    case PhabricatorRepositoryType::REPOSITORY_TYPE_SVN:
      if ($reparse_message) {
        $classes[] = 'PhabricatorRepositorySvnCommitMessageParserWorker';
      }
      if ($reparse_change) {
        $classes[] = 'PhabricatorRepositorySvnCommitChangeParserWorker';
      }
      break;
  }

  if ($reparse_herald) {
    $classes[] = 'PhabricatorRepositoryCommitHeraldWorker';
  }

  if ($reparse_owners) {
    $classes[] = 'PhabricatorRepositoryCommitOwnersWorker';
  }

  if ($reparse_harbormaster) {
    $classes[] = 'HarbormasterRunnerWorker';
  }

  $spec = array(
    'commitID'  => $commit->getID(),
    'only'      => true,
  );

  if ($all_from_repo && !$force_local) {
    foreach ($classes as $class) {
      PhabricatorWorker::scheduleTask(
        $class,
        $spec,
        PhabricatorWorker::PRIORITY_IMPORT);
    }
  } else {
    foreach ($classes as $class) {
      $worker = newv($class, array($spec));
      $worker->executeTask();
    }
  }

  $progress->update(1);
}

$progress->done();

function usage($message) {
  echo phutil_console_format(
    '**Usage Exception:** '.$message."\n".
    "Use __--help__ to display full help\n");
  exit(1);
}