PhabricatorSubscriptionsExportEngineExtension.php

<?php

final class PhabricatorSubscriptionsExportEngineExtension
  extends PhabricatorExportEngineExtension {

  const EXTENSIONKEY = 'subscriptions';

  public function supportsObject($object) {
    return ($object instanceof PhabricatorSubscribableInterface);
  }

  public function newExportFields() {
    return array(
      id(new PhabricatorPHIDListExportField())
        ->setKey('subscriberPHIDs')
        ->setLabel(pht('Subscriber PHIDs')),
      id(new PhabricatorStringListExportField())
        ->setKey('subscribers')
        ->setLabel(pht('Subscribers')),
    );
  }

  public function newExportData(array $objects) {
    $viewer = $this->getViewer();

    $object_phids = mpull($objects, 'getPHID');

    $projects_query = id(new PhabricatorEdgeQuery())
      ->withSourcePHIDs($object_phids)
      ->withEdgeTypes(
        array(
          PhabricatorObjectHasSubscriberEdgeType::EDGECONST,
        ));
    $projects_query->execute();

    $handles = $viewer->loadHandles($projects_query->getDestinationPHIDs());

    $map = array();
    foreach ($objects as $object) {
      $object_phid = $object->getPHID();

      $project_phids = $projects_query->getDestinationPHIDs(
        array($object_phid),
        array(PhabricatorObjectHasSubscriberEdgeType::EDGECONST));

      $handle_list = $handles->newSublist($project_phids);
      $handle_list = iterator_to_array($handle_list);
      $handle_names = mpull($handle_list, 'getName');
      $handle_names = array_values($handle_names);

      $map[] = array(
        'subscriberPHIDs' => $project_phids,
        'subscribers' => $handle_names,
      );
    }

    return $map;
  }

}