1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 

<?php

/**
 * This file is part of the Lean Mapper library (http://www.leanmapper.com)
 *
 * Copyright (c) 2013 Vojtěch Kohout (aka Tharos)
 *
 * For the full copyright and license information, please view the file
 * license-mit.txt that was distributed with this source code.
 */

namespace LeanMapper;

use Closure;
use DibiFluent;
use LeanMapper\Exception\InvalidArgumentException;

/**
 * DibiFluent with filter support
 *
 * @author Vojtěch Kohout
 */
class Fluent extends DibiFluent
{

    /** @var array */
    public static $masks = array( // fixes missing UNION in dibi
        'SELECT' => array('SELECT', 'DISTINCT', 'FROM', 'WHERE', 'GROUP BY',
            'HAVING', 'ORDER BY', 'LIMIT', 'OFFSET', 'UNION'),
        'UPDATE' => array('UPDATE', 'SET', 'WHERE', 'ORDER BY', 'LIMIT'),
        'INSERT' => array('INSERT', 'INTO', 'VALUES', 'SELECT'),
        'DELETE' => array('DELETE', 'FROM', 'USING', 'WHERE', 'ORDER BY', 'LIMIT'),
    );

    /** @var array */
    private $relatedKeys;


    /**
     * Applies given filter to current statement
     *
     * @param Closure|string $filter
     * @param mixed|null $args
     * @return FilteringResult|null
     */
    public function applyFilter($filter, $args = null)
    {
        $args = func_get_args();
        $args[0] = $this;
        return call_user_func_array(
            $filter instanceof Closure ? $filter : $this->getConnection()->getFilterCallback($filter),
            $args
        );
    }

    /**
     * @param array|null $args
     * @return self
     */
    public function createSelect($args = null)
    {
        return call_user_func_array(array($this->getConnection(), 'select'), func_get_args());
    }

    /**
     * Exports current state
     *
     * @param string|null $clause
     * @param array|null $args
     * @return string
     */
    public function _export($clause = null, $args = null)
    {
        return parent::_export($clause, $args);
    }

    /**
     * @return array|null
     */
    public function getRelatedKeys()
    {
        return $this->relatedKeys;
    }

    /**
     * @param array|null $keys
     * @return self
     * @throws InvalidArgumentException
     */
    public function setRelatedKeys($keys)
    {
        if (!is_array($keys) and $keys !== null) {
            throw new InvalidArgumentException('Invalid related keys given. Expected array or null, ' . gettype($keys) . ' given.');
        }
        $this->relatedKeys = $keys;
        return $this;
    }

}