Diffusion Phabricator (feature/better-testing) Browse

better-testing

PhabricatorNotificationBuilder.php
View Options

			<?php

			final class PhabricatorNotificationBuilder extends Phobject {

			private $stories;
			private $parsedStories;
			private $user = null;
			private $showTimestamps = true;

			public function __construct(array $stories) {
			assert_instances_of($stories, 'PhabricatorFeedStory');
			$this->stories = $stories;
			}

			public function setUser($user) {
			$this->user = $user;
			return $this;
			}

			public function setShowTimestamps($show_timestamps) {
			$this->showTimestamps = $show_timestamps;
			return $this;
			}

			public function getShowTimestamps() {
			return $this->showTimestamps;
			}

			private function parseStories() {

			if ($this->parsedStories) {
			return $this->parsedStories;
			}

			$stories = $this->stories;
			$stories = mpull($stories, null, 'getChronologicalKey');

			// Aggregate notifications. Generally, we can aggregate notifications only
			// by object, e.g. "a updated T123" and "b updated T123" can become
			// "a and b updated T123", but we can't combine "a updated T123" and
			// "a updated T234" into "a updated T123 and T234" because there would be
			// nowhere sensible for the notification to link to, and no reasonable way
			// to unambiguously clear it.

			// Build up a map of all the possible aggregations.

			$chronokey_map = array();
			$aggregation_map = array();
			$agg_types = array();
			foreach ($stories as $chronokey => $story) {
			$chronokey_map[$chronokey] = $story->getNotificationAggregations();
			foreach ($chronokey_map[$chronokey] as $key => $type) {
			$agg_types[$key] = $type;
			$aggregation_map[$key]['keys'][$chronokey] = true;
			}
			}

			// Repeatedly select the largest available aggregation until none remain.

			$aggregated_stories = array();
			while ($aggregation_map) {

			// Count the size of each aggregation, removing any which will consume
			// fewer than 2 stories.

			foreach ($aggregation_map as $key => $dict) {
			$size = count($dict['keys']);
			if ($size > 1) {
			$aggregation_map[$key]['size'] = $size;
			} else {
			unset($aggregation_map[$key]);
			}
			}

			// If we're out of aggregations, break out.

			if (!$aggregation_map) {
			break;
			}

			// Select the aggregation we're going to make, and remove it from the
			// map.

			$aggregation_map = isort($aggregation_map, 'size');
			$agg_info = idx(last($aggregation_map), 'keys');
			$agg_key = last_key($aggregation_map);
			unset($aggregation_map[$agg_key]);

			// Select all the stories it aggregates, and remove them from the master
			// list of stories and from all other possible aggregations.

			$sub_stories = array();
			foreach ($agg_info as $chronokey => $ignored) {
			$sub_stories[$chronokey] = $stories[$chronokey];
			unset($stories[$chronokey]);
			foreach ($chronokey_map[$chronokey] as $key => $type) {
			unset($aggregation_map[$key]['keys'][$chronokey]);
			}
			unset($chronokey_map[$chronokey]);
			}

			// Build the aggregate story.

			krsort($sub_stories);
			$story_class = $agg_types[$agg_key];
			$conv = array(head($sub_stories)->getStoryData());

			$new_story = newv($story_class, $conv);
			$new_story->setAggregateStories($sub_stories);
			$aggregated_stories[] = $new_story;
			}

			// Combine the aggregate stories back into the list of stories.

			$stories = array_merge($stories, $aggregated_stories);
			$stories = mpull($stories, null, 'getChronologicalKey');
			krsort($stories);

			$this->parsedStories = $stories;
			return $stories;
			}

			public function buildView() {
			$stories = $this->parseStories();
			$null_view = new AphrontNullView();

			foreach ($stories as $story) {
			try {
			$view = $story->renderView();
			} catch (Exception $ex) {
			// TODO: Render a nice debuggable notice instead?
			continue;
			}

			$view->setShowTimestamp($this->getShowTimestamps());

			$null_view->appendChild($view->renderNotification($this->user));
			}

			return $null_view;
			}

			public function buildDict() {
			$stories = $this->parseStories();
			$dict = array();

			$viewer = $this->user;
			$desktop_key = PhabricatorDesktopNotificationsSetting::SETTINGKEY;
			$desktop_enabled = $viewer->getUserSetting($desktop_key);

			foreach ($stories as $story) {
			if ($story instanceof PhabricatorApplicationTransactionFeedStory) {
			$dict[] = array(
			'desktopReady' => $desktop_enabled,
			'title' => $story->renderText(),
			'body' => $story->renderTextBody(),
			'href' => $story->getURI(),
			'icon' => $story->getImageURI(),
			);
			} else if ($story instanceof PhabricatorNotificationTestFeedStory) {
			$dict[] = array(
			'desktopReady' => $desktop_enabled,
			'title' => pht('Test Notification'),
			'body' => $story->renderText(),
			'href' => null,
			'icon' => PhabricatorUser::getDefaultProfileImageURI(),
			);
			} else {
			$dict[] = array(
			'desktopReady' => false,
			'title' => null,
			'body' => null,
			'href' => null,
			'icon' => null,
			);
			}
			}

			return $dict;
			}
			}

phab/src/applications/notification/builder/PhabricatorNotificationBuilder.php2fc7203dc742feature/better-testing

PhabricatorNotificationBuilder.phpView Options

phab/src/applications/notification/builder/PhabricatorNotificationBuilder.php
2fc7203dc742feature/better-testing

PhabricatorNotificationBuilder.php
View Options