silverstripe-gridfieldextensions/code/GridFieldAddExistingSearchHandler.php

<?php

namespace SilverStripe\Forms\GridField;

use SilverStripe\Control\Controller;
use SilverStripe\Control\RequestHandler;
use SilverStripe\Forms\FieldList;
use SilverStripe\Forms\Form;
use SilverStripe\Forms\FormAction;
use SilverStripe\ORM\DataList;
use SilverStripe\ORM\PaginatedList;

/**
 * Used by {@link GridFieldAddExistingSearchButton} to provide the searching
 * functionality.
 */
class GridFieldAddExistingSearchHandler extends RequestHandler {

	private static $allowed_actions = array(
		'index',
		'add',
		'SearchForm'
	);

	/**
	 * @var GridField
	 */
	protected $grid;

	/**
	 * @var GridFieldAddExistingSearchButton
	 */
	protected $button;

	/**
	 * @var SearchContext
	 */
	protected $context;

	public function __construct($grid, $button) {
		$this->grid    = $grid;
		$this->button  = $button;
		$this->context = singleton($grid->getModelClass())->getDefaultSearchContext();

		parent::__construct();
	}

	public function index() {
		return $this->renderWith('GridFieldAddExistingSearchHandler');
	}

	public function add($request) {
		if(!$id = $request->postVar('id')) {
			$this->httpError(400);
		}

		$list = $this->grid->getList();
		$item = DataList::create($list->dataClass())->byID($id);

		if(!$item) {
			$this->httpError(400);
		}

		$list->add($item);
	}

	/**
	 * @return Form
	 */
	public function SearchForm() {
		$form = new Form(
			$this,
			'SilverStripe\\CMS\\Search\\SearchForm',
			$this->context->getFields(),
			new FieldList(
				FormAction::create('doSearch', _t('GridFieldExtensions.SEARCH', 'Search'))
					->setUseButtonTag(true)
					->addExtraClass('ss-ui-button')
					->setAttribute('data-icon', 'magnifier')
			)
		);

		$form->addExtraClass('stacked add-existing-search-form');
		$form->setFormMethod('GET');

		return $form;
	}

	public function doSearch($data, $form) {
		$list = $this->context->getQuery($data, false, false, $this->getSearchList());
		$list = $list->subtract($this->grid->getList());
		$list = new PaginatedList($list, $this->request);

		$data = $this->customise(array(
			'SearchForm' => $form,
			'Items'      => $list
		));
		return $data->index();
	}

	public function Items() {
		$list = $this->getSearchList();
		$list = $list->subtract($this->grid->getList());
		$list = new PaginatedList($list, $this->request);

		return $list;
	}

	public function Link($action = null) {
		return Controller::join_links($this->grid->Link(), 'add-existing-search', $action);
	}

	/**
	 * @return DataList
	 */
	protected function getSearchList() {
		return $this->button->getSearchList() ?: DataList::create($this->grid->getList()->dataClass());
	}

}