Button.php

<?php

namespace Gridito;

/**
 * Action button
 *
 * @author Jan Marek
 * @license MIT
 */
class Button extends BaseButton
{
	/** @var bool */
	private $ajax = false;

	/** @var string|callback|null */
	private $confirmationQuestion = null;



	/**
	 * Is ajax?
	 * @return bool
	 */
	public function isAjax()
	{
		return $this->ajax;
	}



	/**
	 * Set ajax mode
	 * @param bool $ajax ajax
	 * @return Button
	 */
	public function setAjax($ajax)
	{
		$this->ajax = (bool) $ajax;
		return $this;
	}



	/**
	 * Get confirmation question
	 * @param mixed $row row
	 * @return string|callback|null
	 */
	public function getConfirmationQuestion($row)
	{
		if (is_callable($this->confirmationQuestion)) {
			return call_user_func($this->confirmationQuestion, $row);
		} else {
			return $this->confirmationQuestion;
		}
	}



	/**
	 * Set confirmation question
	 * @param string|callback|null $confirmationQuestion confirmation question
	 * @return Button
	 */
	public function setConfirmationQuestion($confirmationQuestion)
	{
		$this->confirmationQuestion = $confirmationQuestion;
		return $this;
	}



	/**
	 * Handle click signal
	 * @param string $token security token
	 * @param mixed $uniqueId primary key
	 */
	public function handleClick($token, $uniqueId = null)
	{
		parent::handleClick($token, $uniqueId);

		if ($this->getPresenter()->isAjax()) {
			$this->getGrid()->invalidateControl();
		} else {
			$this->getGrid()->redirect('this');
		}
	}



	/**
	 * Create button element
	 * @param mixed $row row
	 * @return Nette\Web\Html
	 */
	public function createButton($row = null)
	{
		$el = parent::createButton($row);
		$el->class[] = $this->isAjax() ? $this->getGrid()->getAjaxClass() : null;
		$el->data('gridito-question', $this->getConfirmationQuestion($row));

		return $el;
	}

}