phorge-phorge/src/applications/metamta/management/PhabricatorMailManagementReceiveTestWorkflow.php

<?php

final class PhabricatorMailManagementReceiveTestWorkflow
  extends PhabricatorSearchManagementWorkflow {

  protected function didConstruct() {
    $this
      ->setName('receive-test')
      ->setSynopsis(
        pht(
          'Simulate receiving mail. This is primarily useful if you are '.
          'developing new mail receivers.'))
      ->setExamples(
        "**receive-test** --as alincoln --to D123 < body.txt")
      ->setArguments(
        array(
          array(
            'name'    => 'as',
            'param'   => 'user',
            'help'    => 'Act as the specified user.',
          ),
          array(
            'name'    => 'from',
            'param'   => 'email',
            'help'    => 'Simulate mail delivery "From:" the given user.',
          ),
          array(
            'name'    => 'to',
            'param'   => 'object',
            'help'    => 'Simulate mail delivery "To:" the given object.',
          ),
        ));
  }

  public function execute(PhutilArgumentParser $args) {
    $console = PhutilConsole::getConsole();

    $as = $args->getArg('as');
    if (!$as) {
      throw new PhutilArgumentUsageException(
        pht("Use '--as' to specify the acting user."));
    }

    $user = id(new PhabricatorPeopleQuery())
      ->setViewer(PhabricatorUser::getOmnipotentUser())
      ->withUsernames(array($as))
      ->executeOne();
    if (!$user) {
      throw new PhutilArgumentUsageException(
        pht("No such user '%s' exists.", $as));
    }

    $to = $args->getArg('to');
    if (!$to) {
      throw new PhutilArgumentUsageException(
        "Use '--to' to specify the receiving object or email address.");
    }

    $from = $args->getArg('from');
    if (!$from) {
      $from = $user->loadPrimaryEmail()->getAddress();
    }

    $console->writeErr("%s\n", pht('Reading message body from stdin...'));
    $body = file_get_contents('php://stdin');

    $received = new PhabricatorMetaMTAReceivedMail();
    $header_content = array(
      'Message-ID' => Filesystem::readRandomCharacters(12),
      'From'       => $from,
    );

    if (preg_match('/.+@.+/', $to)) {
      $header_content['to'] = $to;
    } else {
      // We allow the user to use an object name instead of a real address
      // as a convenience. To build the mail, we build a similar message and
      // look for a receiver which will accept it.
      $pseudohash = PhabricatorObjectMailReceiver::computeMailHash('x', 'y');
      $pseudomail = id(new PhabricatorMetaMTAReceivedMail())
        ->setHeaders(
          array(
            'to' => $to.'+1+'.$pseudohash,
          ));

      $receivers = id(new PhutilSymbolLoader())
        ->setAncestorClass('PhabricatorMailReceiver')
        ->loadObjects();

      $receiver = null;
      foreach ($receivers as $possible_receiver) {
        if (!$possible_receiver->isEnabled()) {
          continue;
        }
        if (!$possible_receiver->canAcceptMail($pseudomail)) {
          continue;
        }
        $receiver = $possible_receiver;
        break;
      }

      if (!$receiver) {
        throw new Exception(
          pht("No configured mail receiver can accept mail to '%s'.", $to));
      }

      if (!($receiver instanceof PhabricatorObjectMailReceiver)) {
        $class = get_class($receiver);
        throw new Exception(
          "Receiver '%s' accepts mail to '%s', but is not a ".
          "subclass of PhabricatorObjectMailReceiver.",
          $class,
          $to);
      }

      $object = $receiver->loadMailReceiverObject($to, $user);
      if (!$object) {
        throw new Exception(pht("No such object '%s'!", $to));
      }

      $hash = PhabricatorObjectMailReceiver::computeMailHash(
        $object->getMailKey(),
        $user->getPHID());

      $header_content['to'] = $to.'+'.$user->getID().'+'.$hash.'@test.com';
    }

    $received->setHeaders($header_content);
    $received->setBodies(
      array(
        'text' => $body,
      ));

    $received->save();
    $received->processReceivedMail();

    $console->writeErr(
      "%s\n\n    phabricator/ $ ./bin/mail show-inbound --id %d\n\n",
      pht("Mail received! You can view details by running this command:"),
      $received->getID());
  }

}