<?php
/**
 * @author  Magediary
 * @package Magediary_ProductQuestion
 */

namespace Magediary\ProductQuestion\Model\ResourceModel\Question;

use Magento\Framework\Model\ResourceModel\Db\Collection\AbstractCollection;

class Collection extends AbstractCollection
{
    /**
     * @var string
     */
    protected $_idFieldName = 'question_id';

    /**
     * Define resource model
     *
     * @return void
     */
    protected function _construct()
    {
        $this->_init(
            \Magediary\ProductQuestion\Model\Question::class,
            \Magediary\ProductQuestion\Model\ResourceModel\Question::class
        );
    }

    /**
     * Add field filter to collection
     *
     * @param string|array $field
     * @param null|string|array $condition
     * @return $this
     */
    public function addFieldToFilter($field, $condition = null)
    {
        if (is_array($field)) {
            if (count($field) > 1) {
                return parent::addFieldToFilter($field, $condition);
            } elseif (count($field) === 1) {
                $field = $field[0];
                $condition = $condition[0] ?? $condition;
            }
        }

        if ($field === 'product_id') {
            if (is_array($condition)) {
                $condition = array_shift($condition);
            }
            return $this->addProductFilter($condition);
        }

        return parent::addFieldToFilter($field, $condition);
    }

    /**
     * Add product filter
     *
     * @param int $pkValue
     * @return $this
     */
    public function addProductFilter($pkValue)
    {
        $this->getSelect()
            ->join(
                ['p' => $this->getResource()->getTable('magediary_question_product')],
                "p.question_id = main_table.question_id AND p.product_id = $pkValue",
                ['product_id']
            );
        return $this;
    }

    /**
     * Add customer filter
     *
     * @param mixed $customer
     * @return $this
     */
    public function addCustomerFilter($customer)
    {
        $inCond = $this->getConnection()->prepareSqlCondition('main_table.customer_id', ['in' => $customer]);
        $this->getSelect()->where($inCond);
        return $this;
    }

    /**
     * Add status filter
     *
     * @param int|string $status
     * @return $this
     */
    public function addStatusFilter($status)
    {
        $inCond = $this->getConnection()->prepareSqlCondition('main_table.status', ['in' => $status]);
        $this->getSelect()->where($inCond);
        return $this;
    }

    /**
     * Add store filter
     *
     * @param int|int[] $storeId
     * @return $this
     */
    public function addStoreFilter($storeId)
    {
        $inCond = $this->getConnection()->prepareSqlCondition('main_table.store_id', ['in' => [0, $storeId]]);
        $this->getSelect()->where($inCond);
        return $this;
    }

    /**
     * Set date order
     *
     * @param string $dir
     * @return $this
     */
    public function setDateOrder($dir = 'DESC')
    {
        $this->setOrder('main_table.created_at', $dir);
        return $this;
    }
}