src/applications/conduit/ssh/ConduitSSHWorkflow.php

   1 <?php
   2
   3 final class ConduitSSHWorkflow extends PhabricatorSSHWorkflow {
   4
   5   protected function didConstruct() {
   6     $this->setName('conduit');
   7     $this->setArguments(
   8       array(
   9         array(
  10           'name'      => 'method',
  11           'wildcard'  => true,
  12         ),
  13       ));
  14   }
  15
  16   public function execute(PhutilArgumentParser $args) {
  17     $time_start = microtime(true);
  18
  19     $methodv = $args->getArg('method');
  20     if (!$methodv) {
  21       throw new Exception(pht('No Conduit method provided.'));
  22     } else if (count($methodv) > 1) {
  23       throw new Exception(pht('Too many Conduit methods provided.'));
  24     }
  25
  26     $method = head($methodv);
  27
  28     $json = $this->readAllInput();
  29     $raw_params = null;
  30     try {
  31       $raw_params = phutil_json_decode($json);
  32     } catch (PhutilJSONParserException $ex) {
  33       throw new PhutilProxyException(
  34         pht('Invalid JSON input.'),
  35         $ex);
  36     }
  37
  38     $params = idx($raw_params, 'params', '[]');
  39     $params = phutil_json_decode($params);
  40     $metadata = idx($params, '__conduit__', array());
  41     unset($params['__conduit__']);
  42
  43     $call = null;
  44     $error_code = null;
  45     $error_info = null;
  46
  47     try {
  48       $call = new ConduitCall($method, $params);
  49       $call->setUser($this->getSSHUser());
  50
  51       $result = $call->execute();
  52     } catch (ConduitException $ex) {
  53       $result = null;
  54       $error_code = $ex->getMessage();
  55       if ($ex->getErrorDescription()) {
  56         $error_info = $ex->getErrorDescription();
  57       } else if ($call) {
  58         $error_info = $call->getErrorDescription($error_code);
  59       }
  60     }
  61
  62     $response = id(new ConduitAPIResponse())
  63       ->setResult($result)
  64       ->setErrorCode($error_code)
  65       ->setErrorInfo($error_info);
  66
  67     $json_out = json_encode($response->toDictionary());
  68     $json_out = $json_out."\n";
  69
  70     $this->getIOChannel()->write($json_out);
  71
  72     // NOTE: Flush here so we can get an accurate result for the duration,
  73     // if the response is large and the receiver is slow to read it.
  74     $this->getIOChannel()->flush();
  75
  76     $connection_id = idx($metadata, 'connectionID');
  77     $log = id(new PhabricatorConduitMethodCallLog())
  78       ->setCallerPHID($this->getSSHUser()->getPHID())
  79       ->setConnectionID($connection_id)
  80       ->setMethod($method)
  81       ->setError((string)$error_code)
  82       ->setDuration(phutil_microseconds_since($time_start))
  83       ->save();
  84   }
  85 }