PhabricatorObjectQuery.php
0a62f13464b8stable

PhabricatorObjectQuery.php
View Options

			<?php

			final class PhabricatorObjectQuery
			extends PhabricatorCursorPagedPolicyAwareQuery {

			private $phids = array();
			private $names = array();
			private $types;

			private $namedResults;

			public function withPHIDs(array $phids) {
			$this->phids = $phids;
			return $this;
			}

			public function withNames(array $names) {
			$this->names = $names;
			return $this;
			}

			public function withTypes(array $types) {
			$this->types = $types;
			return $this;
			}

			public function loadPage() {
			if ($this->namedResults === null) {
			$this->namedResults = array();
			}

			$types = PhabricatorPHIDType::getAllTypes();
			if ($this->types) {
			$types = array_select_keys($types, $this->types);
			}

			$names = array_unique($this->names);
			$phids = $this->phids;

			// We allow objects to be named by their PHID in addition to their normal
			// name so that, e.g., CLI tools which accept object names can also accept
			// PHIDs and work as users expect.
			$actually_phids = array();
			if ($names) {
			foreach ($names as $key => $name) {
			if (!strncmp($name, 'PHID-', 5)) {
			$actually_phids[] = $name;
			$phids[] = $name;
			unset($names[$key]);
			}
			}
			}

			$phids = array_unique($phids);

			if ($names) {
			$name_results = $this->loadObjectsByName($types, $names);
			} else {
			$name_results = array();
			}

			if ($phids) {
			$phid_results = $this->loadObjectsByPHID($types, $phids);
			} else {
			$phid_results = array();
			}

			foreach ($actually_phids as $phid) {
			if (isset($phid_results[$phid])) {
			$name_results[$phid] = $phid_results[$phid];
			}
			}

			$this->namedResults += $name_results;

			return $phid_results + mpull($name_results, null, 'getPHID');
			}

			public function getNamedResults() {
			if ($this->namedResults === null) {
			throw new Exception('Call execute() before getNamedResults()!');
			}
			return $this->namedResults;
			}

			private function loadObjectsByName(array $types, array $names) {
			$groups = array();
			foreach ($names as $name) {
			foreach ($types as $type => $type_impl) {
			if (!$type_impl->canLoadNamedObject($name)) {
			continue;
			}
			$groups[$type][] = $name;
			break;
			}
			}

			$results = array();
			foreach ($groups as $type => $group) {
			$results += $types[$type]->loadNamedObjects($this, $group);
			}

			return $results;
			}

			private function loadObjectsByPHID(array $types, array $phids) {
			$results = array();

			$workspace = $this->getObjectsFromWorkspace($phids);

			foreach ($phids as $key => $phid) {
			if (isset($workspace[$phid])) {
			$results[$phid] = $workspace[$phid];
			unset($phids[$key]);
			}
			}

			if (!$phids) {
			return $results;
			}

			$groups = array();
			foreach ($phids as $phid) {
			$type = phid_get_type($phid);
			$groups[$type][] = $phid;
			}

			foreach ($groups as $type => $group) {
			if (isset($types[$type])) {
			$objects = $types[$type]->loadObjects($this, $group);
			$results += mpull($objects, null, 'getPHID');
			}
			}

			return $results;
			}

			protected function didFilterResults(array $filtered) {
			foreach ($this->namedResults as $name => $result) {
			if (isset($filtered[$result->getPHID()])) {
			unset($this->namedResults[$name]);
			}
			}
			}

			/**
			* This query disables policy filtering because it is performed in the
			* subqueries which actually load objects. We don't need to re-filter
			* results, since policies have already been applied.
			*/
			protected function shouldDisablePolicyFiltering() {
			return true;
			}

			public function getQueryApplicationClass() {
			return null;
			}

			}

phab/src/applications/phid/query/PhabricatorObjectQuery.php0a62f13464b8stable

PhabricatorObjectQuery.phpView Options

phab/src/applications/phid/query/PhabricatorObjectQuery.php
0a62f13464b8stable

PhabricatorObjectQuery.php
View Options