Differential D18960 Diff 45484 src/infrastructure/export/engine/PhabricatorSubscriptionsExportEngineExtension.php

Changeset View

Standalone View

				<?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;
				}

				}