src/applications/repository/management/PhabricatorRepositoryManagementMaintenanceWorkflow.php

   1 <?php
   2
   3 final class PhabricatorRepositoryManagementMaintenanceWorkflow
   4   extends PhabricatorRepositoryManagementWorkflow {
   5
   6   protected function didConstruct() {
   7     $this
   8       ->setName('maintenance')
   9       ->setExamples(
  10         "**maintenance** --start __message__ __repository__ ...\n".
  11         "**maintenance** --stop __repository__")
  12       ->setSynopsis(
  13         pht('Set or clear read-only mode for repository maintenance.'))
  14       ->setArguments(
  15         array(
  16           array(
  17             'name' => 'start',
  18             'param' => 'message',
  19             'help' => pht(
  20               'Put repositories into maintenance mode.'),
  21           ),
  22           array(
  23             'name' => 'stop',
  24             'help' => pht(
  25               'Take repositories out of maintenance mode, returning them '.
  26               'to normal serice.'),
  27           ),
  28           array(
  29             'name' => 'repositories',
  30             'wildcard' => true,
  31           ),
  32         ));
  33   }
  34
  35   public function execute(PhutilArgumentParser $args) {
  36     $viewer = $this->getViewer();
  37
  38     $repositories = $this->loadRepositories($args, 'repositories');
  39     if (!$repositories) {
  40       throw new PhutilArgumentUsageException(
  41         pht('Specify one or more repositories to act on.'));
  42     }
  43
  44     $message = $args->getArg('start');
  45     $is_start = (bool)strlen($message);
  46     $is_stop = $args->getArg('stop');
  47
  48     if (!$is_start && !$is_stop) {
  49       throw new PhutilArgumentUsageException(
  50         pht(
  51           'Use "--start <message>" to put repositories into maintenance '.
  52           'mode, or "--stop" to take them out of maintenance mode.'));
  53     }
  54
  55     if ($is_start && $is_stop) {
  56       throw new PhutilArgumentUsageException(
  57         pht(
  58           'Specify either "--start" or "--stop", but not both.'));
  59     }
  60
  61     $content_source = $this->newContentSource();
  62     $diffusion_phid = id(new PhabricatorDiffusionApplication())->getPHID();
  63
  64     if ($is_start) {
  65       $new_value = $message;
  66     } else {
  67       $new_value = null;
  68     }
  69
  70     foreach ($repositories as $repository) {
  71       $xactions = array();
  72
  73       $xactions[] = $repository->getApplicationTransactionTemplate()
  74         ->setTransactionType(
  75           PhabricatorRepositoryMaintenanceTransaction::TRANSACTIONTYPE)
  76         ->setNewValue($new_value);
  77
  78       $repository->getApplicationTransactionEditor()
  79         ->setActor($viewer)
  80         ->setActingAsPHID($diffusion_phid)
  81         ->setContentSource($content_source)
  82         ->setContinueOnNoEffect(true)
  83         ->setContinueOnMissingFields(true)
  84         ->applyTransactions($repository, $xactions);
  85
  86       if ($is_start) {
  87         echo tsprintf(
  88           "%s\n",
  89           pht(
  90             'Put repository "%s" into maintenance mode.',
  91             $repository->getDisplayName()));
  92       } else {
  93         echo tsprintf(
  94           "%s\n",
  95           pht(
  96             'Took repository "%s" out of maintenance mode.',
  97             $repository->getDisplayName()));
  98       }
  99     }
 100
 101     return 0;
 102   }
 103
 104 }