Diffusion Phabricator (feature/better-testing) Browse

better-testing

PhabricatorOwnersPackage.php
View Options

			<?php

			final class PhabricatorOwnersPackage
			extends PhabricatorOwnersDAO
			implements
			PhabricatorPolicyInterface,
			PhabricatorApplicationTransactionInterface,
			PhabricatorCustomFieldInterface,
			PhabricatorDestructibleInterface,
			PhabricatorConduitResultInterface,
			PhabricatorFulltextInterface,
			PhabricatorNgramsInterface {

			protected $name;
			protected $auditingEnabled;
			protected $autoReview;
			protected $description;
			protected $primaryOwnerPHID;
			protected $mailKey;
			protected $status;
			protected $viewPolicy;
			protected $editPolicy;
			protected $dominion;

			private $paths = self::ATTACHABLE;
			private $owners = self::ATTACHABLE;
			private $customFields = self::ATTACHABLE;
			private $pathRepositoryMap = array();

			const STATUS_ACTIVE = 'active';
			const STATUS_ARCHIVED = 'archived';

			const AUTOREVIEW_NONE = 'none';
			const AUTOREVIEW_SUBSCRIBE = 'subscribe';
			const AUTOREVIEW_REVIEW = 'review';
			const AUTOREVIEW_BLOCK = 'block';

			const DOMINION_STRONG = 'strong';
			const DOMINION_WEAK = 'weak';

			public static function initializeNewPackage(PhabricatorUser $actor) {
			$app = id(new PhabricatorApplicationQuery())
			->setViewer($actor)
			->withClasses(array('PhabricatorOwnersApplication'))
			->executeOne();

			$view_policy = $app->getPolicy(
			PhabricatorOwnersDefaultViewCapability::CAPABILITY);
			$edit_policy = $app->getPolicy(
			PhabricatorOwnersDefaultEditCapability::CAPABILITY);

			return id(new PhabricatorOwnersPackage())
			->setAuditingEnabled(0)
			->setAutoReview(self::AUTOREVIEW_NONE)
			->setDominion(self::DOMINION_STRONG)
			->setViewPolicy($view_policy)
			->setEditPolicy($edit_policy)
			->attachPaths(array())
			->setStatus(self::STATUS_ACTIVE)
			->attachOwners(array())
			->setDescription('');
			}

			public static function getStatusNameMap() {
			return array(
			self::STATUS_ACTIVE => pht('Active'),
			self::STATUS_ARCHIVED => pht('Archived'),
			);
			}

			public static function getAutoreviewOptionsMap() {
			return array(
			self::AUTOREVIEW_NONE => array(
			'name' => pht('No Autoreview'),
			),
			self::AUTOREVIEW_SUBSCRIBE => array(
			'name' => pht('Subscribe to Changes'),
			),
			self::AUTOREVIEW_REVIEW => array(
			'name' => pht('Review Changes'),
			),
			self::AUTOREVIEW_BLOCK => array(
			'name' => pht('Review Changes (Blocking)'),
			),
			);
			}

			public static function getDominionOptionsMap() {
			return array(
			self::DOMINION_STRONG => array(
			'name' => pht('Strong (Control All Paths)'),
			'short' => pht('Strong'),
			),
			self::DOMINION_WEAK => array(
			'name' => pht('Weak (Control Unowned Paths)'),
			'short' => pht('Weak'),
			),
			);
			}

			protected function getConfiguration() {
			return array(
			// This information is better available from the history table.
			self::CONFIG_TIMESTAMPS => false,
			self::CONFIG_AUX_PHID => true,
			self::CONFIG_COLUMN_SCHEMA => array(
			'name' => 'sort',
			'description' => 'text',
			'primaryOwnerPHID' => 'phid?',
			'auditingEnabled' => 'bool',
			'mailKey' => 'bytes20',
			'status' => 'text32',
			'autoReview' => 'text32',
			'dominion' => 'text32',
			),
			) + parent::getConfiguration();
			}

			public function generatePHID() {
			return PhabricatorPHID::generateNewPHID(
			PhabricatorOwnersPackagePHIDType::TYPECONST);
			}

			public function save() {
			if (!$this->getMailKey()) {
			$this->setMailKey(Filesystem::readRandomCharacters(20));
			}

			return parent::save();
			}

			public function isArchived() {
			return ($this->getStatus() == self::STATUS_ARCHIVED);
			}

			public function setName($name) {
			$this->name = $name;
			return $this;
			}

			public function loadOwners() {
			if (!$this->getID()) {
			return array();
			}
			return id(new PhabricatorOwnersOwner())->loadAllWhere(
			'packageID = %d',
			$this->getID());
			}

			public function loadPaths() {
			if (!$this->getID()) {
			return array();
			}
			return id(new PhabricatorOwnersPath())->loadAllWhere(
			'packageID = %d',
			$this->getID());
			}

			public static function loadAffectedPackages(
			PhabricatorRepository $repository,
			array $paths) {

			if (!$paths) {
			return array();
			}

			return self::loadPackagesForPaths($repository, $paths);
			}

			public static function loadOwningPackages($repository, $path) {
			if (empty($path)) {
			return array();
			}

			return self::loadPackagesForPaths($repository, array($path), 1);
			}

			private static function loadPackagesForPaths(
			PhabricatorRepository $repository,
			array $paths,
			$limit = 0) {

			$fragments = array();
			foreach ($paths as $path) {
			foreach (self::splitPath($path) as $fragment) {
			$fragments[$fragment][$path] = true;
			}
			}

			$package = new PhabricatorOwnersPackage();
			$path = new PhabricatorOwnersPath();
			$conn = $package->establishConnection('r');

			$repository_clause = qsprintf(
			$conn,
			'AND p.repositoryPHID = %s',
			$repository->getPHID());

			// NOTE: The list of $paths may be very large if we're coming from
			// the OwnersWorker and processing, e.g., an SVN commit which created a new
			// branch. Break it apart so that it will fit within 'max_allowed_packet',
			// and then merge results in PHP.

			$rows = array();
			foreach (array_chunk(array_keys($fragments), 128) as $chunk) {
			$rows[] = queryfx_all(
			$conn,
			'SELECT pkg.id, pkg.dominion, p.excluded, p.path
			FROM %T pkg JOIN %T p ON p.packageID = pkg.id
			WHERE p.path IN (%Ls) AND pkg.status IN (%Ls) %Q',
			$package->getTableName(),
			$path->getTableName(),
			$chunk,
			array(
			self::STATUS_ACTIVE,
			),
			$repository_clause);
			}
			$rows = array_mergev($rows);

			$ids = self::findLongestPathsPerPackage($rows, $fragments);

			if (!$ids) {
			return array();
			}

			arsort($ids);
			if ($limit) {
			$ids = array_slice($ids, 0, $limit, $preserve_keys = true);
			}
			$ids = array_keys($ids);

			$packages = $package->loadAllWhere('id in (%Ld)', $ids);
			$packages = array_select_keys($packages, $ids);

			return $packages;
			}

			public static function loadPackagesForRepository($repository) {
			$package = new PhabricatorOwnersPackage();
			$ids = ipull(
			queryfx_all(
			$package->establishConnection('r'),
			'SELECT DISTINCT packageID FROM %T WHERE repositoryPHID = %s',
			id(new PhabricatorOwnersPath())->getTableName(),
			$repository->getPHID()),
			'packageID');

			return $package->loadAllWhere('id in (%Ld)', $ids);
			}

			public static function findLongestPathsPerPackage(array $rows, array $paths) {

			// Build a map from each path to all the package paths which match it.
			$path_hits = array();
			$weak = array();
			foreach ($rows as $row) {
			$id = $row['id'];
			$path = $row['path'];
			$length = strlen($path);
			$excluded = $row['excluded'];

			if ($row['dominion'] === self::DOMINION_WEAK) {
			$weak[$id] = true;
			}

			$matches = $paths[$path];
			foreach ($matches as $match => $ignored) {
			$path_hits[$match][] = array(
			'id' => $id,
			'excluded' => $excluded,
			'length' => $length,
			);
			}
			}

			// For each path, process the matching package paths to figure out which
			// packages actually own it.
			$path_packages = array();
			foreach ($path_hits as $match => $hits) {
			$hits = isort($hits, 'length');

			$packages = array();
			foreach ($hits as $hit) {
			$package_id = $hit['id'];
			if ($hit['excluded']) {
			unset($packages[$package_id]);
			} else {
			$packages[$package_id] = $hit;
			}
			}

			$path_packages[$match] = $packages;
			}

			// Remove packages with weak dominion rules that should cede control to
			// a more specific package.
			if ($weak) {
			foreach ($path_packages as $match => $packages) {

			// Group packages by length.
			$length_map = array();
			foreach ($packages as $package_id => $package) {
			$length_map[$package['length']][$package_id] = $package;
			}

			// For each path length, remove all weak packages if there are any
			// strong packages of the same length. This makes sure that if there
			// are one or more strong claims on a particular path, only those
			// claims stand.
			foreach ($length_map as $package_list) {
			$any_strong = false;
			foreach ($package_list as $package_id => $package) {
			if (!isset($weak[$package_id])) {
			$any_strong = true;
			break;
			}
			}

			if ($any_strong) {
			foreach ($package_list as $package_id => $package) {
			if (isset($weak[$package_id])) {
			unset($packages[$package_id]);
			}
			}
			}
			}

			$packages = isort($packages, 'length');
			$packages = array_reverse($packages, true);

			$best_length = null;
			foreach ($packages as $package_id => $package) {
			// If this is the first package we've encountered, note its length.
			// We're iterating over the packages from longest to shortest match,
			// so packages of this length always have the best claim on the path.
			if ($best_length === null) {
			$best_length = $package['length'];
			}

			// If this package has the same length as the best length, its claim
			// stands.
			if ($package['length'] === $best_length) {
			continue;
			}

			// If this is a weak package and does not have the best length,
			// cede its claim to the stronger package.
			if (isset($weak[$package_id])) {
			unset($packages[$package_id]);
			}
			}

			$path_packages[$match] = $packages;
			}
			}

			// For each package that owns at least one path, identify the longest
			// path it owns.
			$package_lengths = array();
			foreach ($path_packages as $match => $hits) {
			foreach ($hits as $hit) {
			$length = $hit['length'];
			$id = $hit['id'];
			if (empty($package_lengths[$id])) {
			$package_lengths[$id] = $length;
			} else {
			$package_lengths[$id] = max($package_lengths[$id], $length);
			}
			}
			}

			return $package_lengths;
			}

			public static function splitPath($path) {
			$trailing_slash = preg_match('@/$@', $path) ? '/' : '';
			$path = trim($path, '/');
			$parts = explode('/', $path);

			$result = array();
			while (count($parts)) {
			$result[] = '/'.implode('/', $parts).$trailing_slash;
			$trailing_slash = '/';
			array_pop($parts);
			}
			$result[] = '/';

			return array_reverse($result);
			}

			public function attachPaths(array $paths) {
			assert_instances_of($paths, 'PhabricatorOwnersPath');
			$this->paths = $paths;

			// Drop this cache if we're attaching new paths.
			$this->pathRepositoryMap = array();

			return $this;
			}

			public function getPaths() {
			return $this->assertAttached($this->paths);
			}

			public function getPathsForRepository($repository_phid) {
			if (isset($this->pathRepositoryMap[$repository_phid])) {
			return $this->pathRepositoryMap[$repository_phid];
			}

			$map = array();
			foreach ($this->getPaths() as $path) {
			if ($path->getRepositoryPHID() == $repository_phid) {
			$map[] = $path;
			}
			}

			$this->pathRepositoryMap[$repository_phid] = $map;

			return $this->pathRepositoryMap[$repository_phid];
			}

			public function attachOwners(array $owners) {
			assert_instances_of($owners, 'PhabricatorOwnersOwner');
			$this->owners = $owners;
			return $this;
			}

			public function getOwners() {
			return $this->assertAttached($this->owners);
			}

			public function getOwnerPHIDs() {
			return mpull($this->getOwners(), 'getUserPHID');
			}

			public function isOwnerPHID($phid) {
			if (!$phid) {
			return false;
			}

			$owner_phids = $this->getOwnerPHIDs();
			$owner_phids = array_fuse($owner_phids);

			return isset($owner_phids[$phid]);
			}

			public function getMonogram() {
			return 'O'.$this->getID();
			}

			public function getURI() {
			// TODO: Move these to "/O123" for consistency.
			return '/owners/package/'.$this->getID().'/';
			}

			/* -( PhabricatorPolicyInterface )----------------------------------------- */


			public function getCapabilities() {
			return array(
			PhabricatorPolicyCapability::CAN_VIEW,
			PhabricatorPolicyCapability::CAN_EDIT,
			);
			}

			public function getPolicy($capability) {
			switch ($capability) {
			case PhabricatorPolicyCapability::CAN_VIEW:
			return $this->getViewPolicy();
			case PhabricatorPolicyCapability::CAN_EDIT:
			return $this->getEditPolicy();
			}
			}

			public function hasAutomaticCapability($capability, PhabricatorUser $viewer) {
			switch ($capability) {
			case PhabricatorPolicyCapability::CAN_VIEW:
			if ($this->isOwnerPHID($viewer->getPHID())) {
			return true;
			}
			break;
			}

			return false;
			}

			public function describeAutomaticCapability($capability) {
			return pht('Owners of a package may always view it.');
			}


			/* -( PhabricatorApplicationTransactionInterface )------------------------- */


			public function getApplicationTransactionEditor() {
			return new PhabricatorOwnersPackageTransactionEditor();
			}

			public function getApplicationTransactionObject() {
			return $this;
			}

			public function getApplicationTransactionTemplate() {
			return new PhabricatorOwnersPackageTransaction();
			}

			public function willRenderTimeline(
			PhabricatorApplicationTransactionView $timeline,
			AphrontRequest $request) {
			return $timeline;
			}


			/* -( PhabricatorCustomFieldInterface )------------------------------------ */


			public function getCustomFieldSpecificationForRole($role) {
			return PhabricatorEnv::getEnvConfig('owners.fields');
			}

			public function getCustomFieldBaseClass() {
			return 'PhabricatorOwnersCustomField';
			}

			public function getCustomFields() {
			return $this->assertAttached($this->customFields);
			}

			public function attachCustomFields(PhabricatorCustomFieldAttachment $fields) {
			$this->customFields = $fields;
			return $this;
			}


			/* -( PhabricatorDestructibleInterface )----------------------------------- */


			public function destroyObjectPermanently(
			PhabricatorDestructionEngine $engine) {

			$this->openTransaction();
			$conn_w = $this->establishConnection('w');

			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE packageID = %d',
			id(new PhabricatorOwnersPath())->getTableName(),
			$this->getID());

			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE packageID = %d',
			id(new PhabricatorOwnersOwner())->getTableName(),
			$this->getID());

			$this->delete();
			$this->saveTransaction();
			}


			/* -( PhabricatorConduitResultInterface )---------------------------------- */


			public function getFieldSpecificationsForConduit() {
			return array(
			id(new PhabricatorConduitSearchFieldSpecification())
			->setKey('name')
			->setType('string')
			->setDescription(pht('The name of the package.')),
			id(new PhabricatorConduitSearchFieldSpecification())
			->setKey('description')
			->setType('string')
			->setDescription(pht('The package description.')),
			id(new PhabricatorConduitSearchFieldSpecification())
			->setKey('status')
			->setType('string')
			->setDescription(pht('Active or archived status of the package.')),
			id(new PhabricatorConduitSearchFieldSpecification())
			->setKey('owners')
			->setType('list<map<string, wild>>')
			->setDescription(pht('List of package owners.')),
			);
			}

			public function getFieldValuesForConduit() {
			$owner_list = array();
			foreach ($this->getOwners() as $owner) {
			$owner_list[] = array(
			'ownerPHID' => $owner->getUserPHID(),
			);
			}

			return array(
			'name' => $this->getName(),
			'description' => $this->getDescription(),
			'status' => $this->getStatus(),
			'owners' => $owner_list,
			);
			}

			public function getConduitSearchAttachments() {
			return array(
			id(new PhabricatorOwnersPathsSearchEngineAttachment())
			->setAttachmentKey('paths'),
			);
			}


			/* -( PhabricatorFulltextInterface )--------------------------------------- */


			public function newFulltextEngine() {
			return new PhabricatorOwnersPackageFulltextEngine();
			}


			/* -( PhabricatorNgramsInterface )----------------------------------------- */


			public function newNgrams() {
			return array(
			id(new PhabricatorOwnersPackageNameNgrams())
			->setValue($this->getName()),
			);
			}

			}

phab/src/applications/owners/storage/PhabricatorOwnersPackage.phpd1842929afaafeature/better-testing

PhabricatorOwnersPackage.phpView Options

phab/src/applications/owners/storage/PhabricatorOwnersPackage.php
d1842929afaafeature/better-testing

PhabricatorOwnersPackage.php
View Options