Diffusion Phabricator (feature/better-testing) Browse

better-testing

PhabricatorBoardLayoutEngine.php
View Options

			<?php

			final class PhabricatorBoardLayoutEngine extends Phobject {

			private $viewer;
			private $boardPHIDs;
			private $objectPHIDs;
			private $boards;
			private $columnMap = array();
			private $objectColumnMap = array();
			private $boardLayout = array();
			private $fetchAllBoards;

			private $remQueue = array();
			private $addQueue = array();

			public function setViewer(PhabricatorUser $viewer) {
			$this->viewer = $viewer;
			return $this;
			}

			public function getViewer() {
			return $this->viewer;
			}

			public function setBoardPHIDs(array $board_phids) {
			$this->boardPHIDs = array_fuse($board_phids);
			return $this;
			}

			public function getBoardPHIDs() {
			return $this->boardPHIDs;
			}

			public function setObjectPHIDs(array $object_phids) {
			$this->objectPHIDs = array_fuse($object_phids);
			return $this;
			}

			public function getObjectPHIDs() {
			return $this->objectPHIDs;
			}

			/**
			* Fetch all boards, even if the board is disabled.
			*/
			public function setFetchAllBoards($fetch_all) {
			$this->fetchAllBoards = $fetch_all;
			return $this;
			}

			public function getFetchAllBoards() {
			return $this->fetchAllBoards;
			}

			public function executeLayout() {
			$viewer = $this->getViewer();

			$boards = $this->loadBoards();
			if (!$boards) {
			return $this;
			}

			$columns = $this->loadColumns($boards);
			$positions = $this->loadPositions($boards);

			foreach ($boards as $board_phid => $board) {
			$board_columns = idx($columns, $board_phid);

			// Don't layout boards with no columns. These boards need to be formally
			// created first.
			if (!$columns) {
			continue;
			}

			$board_positions = idx($positions, $board_phid, array());

			$this->layoutBoard($board, $board_columns, $board_positions);
			}

			return $this;
			}

			public function getColumns($board_phid) {
			$columns = idx($this->boardLayout, $board_phid, array());
			return array_select_keys($this->columnMap, array_keys($columns));
			}

			public function getColumnObjectPositions($board_phid, $column_phid) {
			$columns = idx($this->boardLayout, $board_phid, array());
			return idx($columns, $column_phid, array());
			}


			public function getColumnObjectPHIDs($board_phid, $column_phid) {
			$positions = $this->getColumnObjectPositions($board_phid, $column_phid);
			return mpull($positions, 'getObjectPHID');
			}

			public function getObjectColumns($board_phid, $object_phid) {
			$board_map = idx($this->objectColumnMap, $board_phid, array());

			$column_phids = idx($board_map, $object_phid);
			if (!$column_phids) {
			return array();
			}

			return array_select_keys($this->columnMap, $column_phids);
			}

			public function queueRemovePosition(
			$board_phid,
			$column_phid,
			$object_phid) {

			$board_layout = idx($this->boardLayout, $board_phid, array());
			$positions = idx($board_layout, $column_phid, array());
			$position = idx($positions, $object_phid);

			if ($position) {
			$this->remQueue[] = $position;

			// If this position hasn't been saved yet, get it out of the add queue.
			if (!$position->getID()) {
			foreach ($this->addQueue as $key => $add_position) {
			if ($add_position === $position) {
			unset($this->addQueue[$key]);
			}
			}
			}
			}

			unset($this->boardLayout[$board_phid][$column_phid][$object_phid]);

			return $this;
			}

			public function queueAddPositionBefore(
			$board_phid,
			$column_phid,
			$object_phid,
			$before_phid) {

			return $this->queueAddPositionRelative(
			$board_phid,
			$column_phid,
			$object_phid,
			$before_phid,
			true);
			}

			public function queueAddPositionAfter(
			$board_phid,
			$column_phid,
			$object_phid,
			$after_phid) {

			return $this->queueAddPositionRelative(
			$board_phid,
			$column_phid,
			$object_phid,
			$after_phid,
			false);
			}

			public function queueAddPosition(
			$board_phid,
			$column_phid,
			$object_phid) {
			return $this->queueAddPositionRelative(
			$board_phid,
			$column_phid,
			$object_phid,
			null,
			true);
			}

			private function queueAddPositionRelative(
			$board_phid,
			$column_phid,
			$object_phid,
			$relative_phid,
			$is_before) {

			$board_layout = idx($this->boardLayout, $board_phid, array());
			$positions = idx($board_layout, $column_phid, array());

			// Check if the object is already in the column, and remove it if it is.
			$object_position = idx($positions, $object_phid);
			unset($positions[$object_phid]);

			if (!$object_position) {
			$object_position = id(new PhabricatorProjectColumnPosition())
			->setBoardPHID($board_phid)
			->setColumnPHID($column_phid)
			->setObjectPHID($object_phid);
			}

			$found = false;
			if (!$positions) {
			$object_position->setSequence(0);
			} else {
			foreach ($positions as $position) {
			if (!$found) {
			if ($relative_phid === null) {
			$is_match = true;
			} else {
			$position_phid = $position->getObjectPHID();
			$is_match = ($relative_phid == $position_phid);
			}

			if ($is_match) {
			$found = true;

			$sequence = $position->getSequence();

			if (!$is_before) {
			$sequence++;
			}

			$object_position->setSequence($sequence++);

			if (!$is_before) {
			// If we're inserting after this position, continue the loop so
			// we don't update it.
			continue;
			}
			}
			}

			if ($found) {
			$position->setSequence($sequence++);
			$this->addQueue[] = $position;
			}
			}
			}

			if ($relative_phid && !$found) {
			throw new Exception(
			pht(
			'Unable to find object "%s" in column "%s" on board "%s".',
			$relative_phid,
			$column_phid,
			$board_phid));
			}

			$this->addQueue[] = $object_position;

			$positions[$object_phid] = $object_position;
			$positions = msort($positions, 'getOrderingKey');

			$this->boardLayout[$board_phid][$column_phid] = $positions;

			return $this;
			}

			public function applyPositionUpdates() {
			foreach ($this->remQueue as $position) {
			if ($position->getID()) {
			$position->delete();
			}
			}
			$this->remQueue = array();

			$adds = array();
			$updates = array();

			foreach ($this->addQueue as $position) {
			$id = $position->getID();
			if ($id) {
			$updates[$id] = $position;
			} else {
			$adds[] = $position;
			}
			}
			$this->addQueue = array();

			$table = new PhabricatorProjectColumnPosition();
			$conn_w = $table->establishConnection('w');

			$pairs = array();
			foreach ($updates as $id => $position) {
			// This is ugly because MySQL gets upset with us if it is configured
			// strictly and we attempt inserts which can't work. We'll never actually
			// do these inserts since they'll always collide (triggering the ON
			// DUPLICATE KEY logic), so we just provide dummy values in order to get
			// there.

			$pairs[] = qsprintf(
			$conn_w,
			'(%d, %d, "", "", "")',
			$id,
			$position->getSequence());
			}

			if ($pairs) {
			queryfx(
			$conn_w,
			'INSERT INTO %T (id, sequence, boardPHID, columnPHID, objectPHID)
			VALUES %Q ON DUPLICATE KEY UPDATE sequence = VALUES(sequence)',
			$table->getTableName(),
			implode(', ', $pairs));
			}

			foreach ($adds as $position) {
			$position->save();
			}

			return $this;
			}

			private function loadBoards() {
			$viewer = $this->getViewer();
			$board_phids = $this->getBoardPHIDs();

			$boards = id(new PhabricatorObjectQuery())
			->setViewer($viewer)
			->withPHIDs($board_phids)
			->execute();
			$boards = mpull($boards, null, 'getPHID');

			if (!$this->fetchAllBoards) {
			foreach ($boards as $key => $board) {
			if (!$board->getHasWorkboard()) {
			unset($boards[$key]);
			}
			}
			}

			return $boards;
			}

			private function loadColumns(array $boards) {
			$viewer = $this->getViewer();

			$columns = id(new PhabricatorProjectColumnQuery())
			->setViewer($viewer)
			->withProjectPHIDs(array_keys($boards))
			->execute();
			$columns = msort($columns, 'getOrderingKey');
			$columns = mpull($columns, null, 'getPHID');

			$need_children = array();
			foreach ($boards as $phid => $board) {
			if ($board->getHasMilestones() \|\| $board->getHasSubprojects()) {
			$need_children[] = $phid;
			}
			}

			if ($need_children) {
			$children = id(new PhabricatorProjectQuery())
			->setViewer($viewer)
			->withParentProjectPHIDs($need_children)
			->execute();
			$children = mpull($children, null, 'getPHID');
			$children = mgroup($children, 'getParentProjectPHID');
			} else {
			$children = array();
			}

			$columns = mgroup($columns, 'getProjectPHID');
			foreach ($boards as $board_phid => $board) {
			$board_columns = idx($columns, $board_phid, array());

			// If the project has milestones, create any missing columns.
			if ($board->getHasMilestones() \|\| $board->getHasSubprojects()) {
			$child_projects = idx($children, $board_phid, array());

			if ($board_columns) {
			$next_sequence = last($board_columns)->getSequence() + 1;
			} else {
			$next_sequence = 1;
			}

			$proxy_columns = mpull($board_columns, null, 'getProxyPHID');
			foreach ($child_projects as $child_phid => $child) {
			if (isset($proxy_columns[$child_phid])) {
			continue;
			}

			$new_column = PhabricatorProjectColumn::initializeNewColumn($viewer)
			->attachProject($board)
			->attachProxy($child)
			->setSequence($next_sequence++)
			->setProjectPHID($board_phid)
			->setProxyPHID($child_phid);

			$unguarded = AphrontWriteGuard::beginScopedUnguardedWrites();
			$new_column->save();
			unset($unguarded);

			$board_columns[$new_column->getPHID()] = $new_column;
			}
			}

			$board_columns = msort($board_columns, 'getOrderingKey');

			$columns[$board_phid] = $board_columns;
			}

			foreach ($columns as $board_phid => $board_columns) {
			foreach ($board_columns as $board_column) {
			$column_phid = $board_column->getPHID();
			$this->columnMap[$column_phid] = $board_column;
			}
			}

			return $columns;
			}

			private function loadPositions(array $boards) {
			$viewer = $this->getViewer();

			$object_phids = $this->getObjectPHIDs();
			if (!$object_phids) {
			return array();
			}

			$positions = id(new PhabricatorProjectColumnPositionQuery())
			->setViewer($viewer)
			->withBoardPHIDs(array_keys($boards))
			->withObjectPHIDs($object_phids)
			->execute();
			$positions = msort($positions, 'getOrderingKey');
			$positions = mgroup($positions, 'getBoardPHID');

			return $positions;
			}

			private function layoutBoard(
			$board,
			array $columns,
			array $positions) {

			$viewer = $this->getViewer();

			$board_phid = $board->getPHID();
			$position_groups = mgroup($positions, 'getObjectPHID');

			$layout = array();
			$default_phid = null;
			foreach ($columns as $column) {
			$column_phid = $column->getPHID();
			$layout[$column_phid] = array();

			if ($column->isDefaultColumn()) {
			$default_phid = $column_phid;
			}
			}

			// Find all the columns which are proxies for other objects.
			$proxy_map = array();
			foreach ($columns as $column) {
			$proxy_phid = $column->getProxyPHID();
			if ($proxy_phid) {
			$proxy_map[$proxy_phid] = $column->getPHID();
			}
			}

			$object_phids = $this->getObjectPHIDs();

			// If we have proxies, we need to force cards into the correct proxy
			// columns.
			if ($proxy_map && $object_phids) {
			$edge_query = id(new PhabricatorEdgeQuery())
			->withSourcePHIDs($object_phids)
			->withEdgeTypes(
			array(
			PhabricatorProjectObjectHasProjectEdgeType::EDGECONST,
			));
			$edge_query->execute();

			$project_phids = $edge_query->getDestinationPHIDs();
			$project_phids = array_fuse($project_phids);
			} else {
			$project_phids = array();
			}

			if ($project_phids) {
			$projects = id(new PhabricatorProjectQuery())
			->setViewer($viewer)
			->withPHIDs($project_phids)
			->execute();
			$projects = mpull($projects, null, 'getPHID');
			} else {
			$projects = array();
			}

			// Build a map from every project that any task is tagged with to the
			// ancestor project which has a column on this board, if one exists.
			$ancestor_map = array();
			foreach ($projects as $phid => $project) {
			if (isset($proxy_map[$phid])) {
			$ancestor_map[$phid] = $proxy_map[$phid];
			} else {
			$seen = array($phid);
			foreach ($project->getAncestorProjects() as $ancestor) {
			$ancestor_phid = $ancestor->getPHID();
			$seen[] = $ancestor_phid;
			if (isset($proxy_map[$ancestor_phid])) {
			foreach ($seen as $project_phid) {
			$ancestor_map[$project_phid] = $proxy_map[$ancestor_phid];
			}
			}
			}
			}
			}

			$view_sequence = 1;
			foreach ($object_phids as $object_phid) {
			$positions = idx($position_groups, $object_phid, array());

			// First, check for objects that have corresponding proxy columns. We're
			// going to overwrite normal column positions if a tag belongs to a proxy
			// column, since you can't be in normal columns if you're in proxy
			// columns.
			$proxy_hits = array();
			if ($proxy_map) {
			$object_project_phids = $edge_query->getDestinationPHIDs(
			array(
			$object_phid,
			));

			foreach ($object_project_phids as $project_phid) {
			if (isset($ancestor_map[$project_phid])) {
			$proxy_hits[] = $ancestor_map[$project_phid];
			}
			}
			}

			if ($proxy_hits) {
			// TODO: For now, only one column hit is permissible.
			$proxy_hits = array_slice($proxy_hits, 0, 1);

			$proxy_hits = array_fuse($proxy_hits);

			// Check the object positions: we hope to find a position in each
			// column the object should be part of. We're going to drop any
			// invalid positions and create new positions where positions are
			// missing.
			foreach ($positions as $key => $position) {
			$column_phid = $position->getColumnPHID();
			if (isset($proxy_hits[$column_phid])) {
			// Valid column, mark the position as found.
			unset($proxy_hits[$column_phid]);
			} else {
			// Invalid column, ignore the position.
			unset($positions[$key]);
			}
			}

			// Create new positions for anything we haven't found.
			foreach ($proxy_hits as $proxy_hit) {
			$new_position = id(new PhabricatorProjectColumnPosition())
			->setBoardPHID($board_phid)
			->setColumnPHID($proxy_hit)
			->setObjectPHID($object_phid)
			->setSequence(0)
			->setViewSequence($view_sequence++);

			$this->addQueue[] = $new_position;

			$positions[] = $new_position;
			}
			} else {
			// Ignore any positions in columns which no longer exist. We don't
			// actively destory them because the rest of the code ignores them and
			// there's no real need to destroy the data.
			foreach ($positions as $key => $position) {
			$column_phid = $position->getColumnPHID();
			if (empty($columns[$column_phid])) {
			unset($positions[$key]);
			}
			}

			// If the object has no position, put it on the default column if
			// one exists.
			if (!$positions && $default_phid) {
			$new_position = id(new PhabricatorProjectColumnPosition())
			->setBoardPHID($board_phid)
			->setColumnPHID($default_phid)
			->setObjectPHID($object_phid)
			->setSequence(0)
			->setViewSequence($view_sequence++);

			$this->addQueue[] = $new_position;

			$positions = array(
			$new_position,
			);
			}
			}

			foreach ($positions as $position) {
			$column_phid = $position->getColumnPHID();
			$layout[$column_phid][$object_phid] = $position;
			}
			}

			foreach ($layout as $column_phid => $map) {
			$map = msort($map, 'getOrderingKey');
			$layout[$column_phid] = $map;

			foreach ($map as $object_phid => $position) {
			$this->objectColumnMap[$board_phid][$object_phid][] = $column_phid;
			}
			}

			$this->boardLayout[$board_phid] = $layout;
			}

			}

phab/src/applications/project/engine/PhabricatorBoardLayoutEngine.phpd1842929afaafeature/better-testing

PhabricatorBoardLayoutEngine.phpView Options

phab/src/applications/project/engine/PhabricatorBoardLayoutEngine.php
d1842929afaafeature/better-testing

PhabricatorBoardLayoutEngine.php
View Options