PhabricatorSearchAttachController.php
No OneTemporary
Actions

Size

6 KB

Referenced Files

None

Subscribers

None

PhabricatorSearchAttachController.php
View Options

	<?php

	/*
	* Copyright 2011 Facebook, Inc.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	class PhabricatorSearchAttachController extends PhabricatorSearchController {

	private $phid;
	private $type;

	public function willProcessRequest(array $data) {
	$this->phid = $data['phid'];
	$this->type = $data['type'];
	}

	public function processRequest() {

	$request = $this->getRequest();
	$user = $request->getUser();

	$handles = id(new PhabricatorObjectHandleData(array($this->phid)))
	->loadHandles();
	$handle = $handles[$this->phid];

	$object_phid = $this->phid;
	$object_type = $handle->getType();
	$attach_type = $this->type;


	// Load the object we're going to attach/detach stuff from. This is the
	// object that triggered the action, e.g. the revision you clicked
	// "Edit Maniphest Tasks" on.
	$object = null;
	switch ($object_type) {
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$object = id(new DifferentialRevision())->loadOneWhere(
	'phid = %s',
	$this->phid);
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$object = id(new ManiphestTask())->loadOneWhere(
	'phid = %s',
	$this->phid);
	break;
	}

	if (!$object) {
	return new Aphront404Response();
	}

	if ($request->isFormPost()) {
	$phids = explode(';', $request->getStr('phids'));
	$phids = array_filter($phids);
	$phids = array_values($phids);
	// sort() so that removing [X, Y] and then adding [Y, X] is correctly
	// detected as a no-op.
	sort($phids);

	$old_phids = $object->getAttachedPHIDs($attach_type);
	sort($old_phids);

	if (($phids \|\| $old_phids) && ($phids !== $old_phids)) {

	// Load all the objects we're attaching or detaching from the main
	// object.
	switch ($attach_type) {
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$attach_objs = id(new DifferentialRevision())->loadAllWhere(
	'phid IN (%Ls)',
	array_merge($phids, $old_phids));
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$attach_objs = id(new ManiphestTask())->loadAllWhere(
	'phid IN (%Ls)',
	array_merge($phids, $old_phids));
	break;
	}

	$attach_objs = mpull($attach_objs, null, 'getPHID');

	// Remove PHIDs which don't actually exist, to prevent silliness.
	$phids = array_keys(array_select_keys($attach_objs, $phids));
	if ($phids) {
	$phids = array_combine($phids, $phids);
	}

	// Update the primary object.
	switch ($object_type) {
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$object->setAttachedPHIDs($attach_type, $phids);
	$object->save();
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$this->applyTaskTransaction(
	$object,
	$attach_type,
	$phids);
	break;
	}

	// Loop through all of the attached/detached objects and update them.
	foreach ($attach_objs as $phid => $attach_obj) {
	$attached_phids = $attach_obj->getAttachedPHIDs($object_type);
	// Figure out if we're attaching or detaching this object.
	if (isset($phids[$phid])) {
	$attached_phids[] = $object_phid;
	} else {
	$attached_phids = array_fill_keys($attached_phids, true);
	unset($attached_phids[$object_phid]);
	$attached_phids = array_keys($attached_phids);
	}
	switch ($attach_type) {
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$attach_obj->setAttachedPHIDs($object_type, $attached_phids);
	$attach_obj->save();
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$this->applyTaskTransaction(
	$attach_obj,
	$object_type,
	$attached_phids);
	break;
	}
	}
	}

	return id(new AphrontReloadResponse())->setURI($handle->getURI());
	} else {
	$phids = $object->getAttachedPHIDs($attach_type);
	}

	switch ($attach_type) {
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$noun = 'Revisions';
	$selected = 'created';
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$noun = 'Tasks';
	$selected = 'assigned';
	break;
	}

	$handles = id(new PhabricatorObjectHandleData($phids))
	->loadHandles();

	$obj_dialog = new PhabricatorObjectSelectorDialog();
	$obj_dialog
	->setUser($user)
	->setHandles($handles)
	->setFilters(array(
	'assigned' => 'Assigned to Me',
	'created' => 'Created By Me',
	'open' => 'All Open '.$noun,
	'all' => 'All '.$noun,
	))
	->setSelectedFilter($selected)
	->setCancelURI($handle->getURI())
	->setSearchURI('/search/select/'.$attach_type.'/')
	->setNoun($noun);

	$dialog = $obj_dialog->buildDialog();

	return id(new AphrontDialogResponse())->setDialog($dialog);
	}

	private function applyTaskTransaction(
	ManiphestTask $task,
	$attach_type,
	array $new_phids) {

	$user = $this->getRequest()->getUser();

	$editor = new ManiphestTransactionEditor();
	$type = ManiphestTransactionType::TYPE_ATTACH;

	$transaction = new ManiphestTransaction();
	$transaction->setAuthorPHID($user->getPHID());
	$transaction->setTransactionType($type);

	$new = $task->getAttached();
	$new[$attach_type] = array_fill_keys($new_phids, array());

	$transaction->setNewValue($new);
	$editor->applyTransactions($task, array($transaction));
	}
	}

File Metadata

Mime Type: text/x-php
Expires: Sat, May 3, 5:36 PM (1 d, 12 h)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 62365
Default Alt Text: PhabricatorSearchAttachController.php (6 KB)

PhabricatorSearchAttachController.phpNo OneTemporaryActions

PhabricatorSearchAttachController.phpView Options

File Metadata

Event Timeline

PhabricatorSearchAttachController.php
No OneTemporary
Actions

PhabricatorSearchAttachController.php
View Options