[Summary view] [Print] [Text view]

   1  #!/usr/bin/env php
   2  <?php
   3  
   4  $root = dirname(dirname(dirname(__FILE__)));
   5  require_once $root.'/scripts/__init_script__.php';
   6  
   7  if ($argc < 2) {
   8    throw new Exception(pht('usage: commit-hook <callsign>'));
   9  }
  10  
  11  $engine = new DiffusionCommitHookEngine();
  12  
  13  $repository = id(new PhabricatorRepositoryQuery())
  14    ->setViewer(PhabricatorUser::getOmnipotentUser())
  15    ->withCallsigns(array($argv[1]))
  16    ->needProjectPHIDs(true)
  17    ->executeOne();
  18  
  19  if (!$repository) {
  20    throw new Exception(pht('No such repository "%s"!', $argv[1]));
  21  }
  22  
  23  if (!$repository->isHosted()) {
  24    // This should be redundant, but double check just in case.
  25    throw new Exception(pht('Repository "%s" is not hosted!', $argv[1]));
  26  }
  27  
  28  $engine->setRepository($repository);
  29  
  30  
  31  // Figure out which user is writing the commit.
  32  
  33  if ($repository->isGit() || $repository->isHg()) {
  34    $username = getenv(DiffusionCommitHookEngine::ENV_USER);
  35    if (!strlen($username)) {
  36      throw new Exception(
  37        pht('usage: %s should be defined!', DiffusionCommitHookEngine::ENV_USER));
  38    }
  39  
  40    if ($repository->isHg()) {
  41      // We respond to several different hooks in Mercurial.
  42      $engine->setMercurialHook($argv[2]);
  43    }
  44  
  45  } else if ($repository->isSVN()) {
  46    // NOTE: In Subversion, the entire environment gets wiped so we can't read
  47    // DiffusionCommitHookEngine::ENV_USER. Instead, we've set "--tunnel-user" to
  48    // specify the correct user; read this user out of the commit log.
  49  
  50    if ($argc < 4) {
  51      throw new Exception(pht('usage: commit-hook <callsign> <repo> <txn>'));
  52    }
  53  
  54    $svn_repo = $argv[2];
  55    $svn_txn = $argv[3];
  56    list($username) = execx('svnlook author -t %s %s', $svn_txn, $svn_repo);
  57    $username = rtrim($username, "\n");
  58  
  59    $engine->setSubversionTransactionInfo($svn_txn, $svn_repo);
  60  } else {
  61    throw new Exception(pht('Unknown repository type.'));
  62  }
  63  
  64  $user = id(new PhabricatorPeopleQuery())
  65    ->setViewer(PhabricatorUser::getOmnipotentUser())
  66    ->withUsernames(array($username))
  67    ->executeOne();
  68  
  69  if (!$user) {
  70    throw new Exception(pht('No such user "%s"!', $username));
  71  }
  72  
  73  $engine->setViewer($user);
  74  
  75  
  76  // Read stdin for the hook engine.
  77  
  78  if ($repository->isHg()) {
  79    // Mercurial leaves stdin open, so we can't just read it until EOF.
  80    $stdin = '';
  81  } else {
  82    // Git and Subversion write data into stdin and then close it. Read the
  83    // data.
  84    $stdin = @file_get_contents('php://stdin');
  85    if ($stdin === false) {
  86      throw new Exception(pht('Failed to read stdin!'));
  87    }
  88  }
  89  
  90  $engine->setStdin($stdin);
  91  $engine->setOriginalArgv(array_slice($argv, 2));
  92  
  93  $remote_address = getenv(DiffusionCommitHookEngine::ENV_REMOTE_ADDRESS);
  94  if (strlen($remote_address)) {
  95    $engine->setRemoteAddress($remote_address);
  96  }
  97  
  98  $remote_protocol = getenv(DiffusionCommitHookEngine::ENV_REMOTE_PROTOCOL);
  99  if (strlen($remote_protocol)) {
 100    $engine->setRemoteProtocol($remote_protocol);
 101  }
 102  
 103  try {
 104    $err = $engine->execute();
 105  } catch (DiffusionCommitHookRejectException $ex) {
 106    $console = PhutilConsole::getConsole();
 107  
 108    if (PhabricatorEnv::getEnvConfig('phabricator.serious-business')) {
 109      $preamble = pht('*** PUSH REJECTED BY COMMIT HOOK ***');
 110    } else {
 111      $preamble = pht(<<<EOTXT
 112  +---------------------------------------------------------------+
 113  |      * * * PUSH REJECTED BY EVIL DRAGON BUREAUCRATS * * *     |
 114  +---------------------------------------------------------------+
 115              \
 116               \                    ^    /^
 117                \                  / \  // \
 118                 \   |\___/|      /   \//  .\
 119                  \  /V  V  \__  /    //  | \ \           *----*
 120                    /     /  \/_/    //   |  \  \          \   |
 121                    @___@`    \/_   //    |   \   \         \/\ \
 122                   0/0/|       \/_ //     |    \    \         \  \
 123               0/0/0/0/|        \///      |     \     \       |  |
 124            0/0/0/0/0/_|_ /   (  //       |      \     _\     |  /
 125         0/0/0/0/0/0/`/,_ _ _/  ) ; -.    |    _ _\.-~       /   /
 126                     ,-}        _      *-.|.-~-.           .~    ~
 127    \     \__/        `/\      /                 ~-. _ .-~      /
 128     \____(Oo)           *.   }            {                   /
 129     (    (--)          .----~-.\        \-`                 .~
 130     //__\\\\  \ DENIED!  ///.----..<        \             _ -~
 131    //    \\\\               ///-._ _ _ _ _ _ _{^ - - - - ~
 132  
 133  EOTXT
 134  );
 135    }
 136  
 137    $console->writeErr("%s\n\n", $preamble);
 138    $console->writeErr("%s\n\n", $ex->getMessage());
 139    $err = 1;
 140  }
 141  
 142  exit($err);