<?php
namespace Inani\Larapoll\Traits;

use Illuminate\Support\Collection;
use Inani\Larapoll\Exceptions\CheckedOptionsException;
use Inani\Larapoll\Exceptions\OptionsInvalidNumberProvidedException;
use Inani\Larapoll\Exceptions\OptionsNotProvidedException;
use Inani\Larapoll\Exceptions\RemoveVotedOptionException;
use Inani\Larapoll\Option;
use InvalidArgumentException;

trait PollManipulator
{

    /**
     * attach new options
     *
     * @param $options
     * @return mixed
     */
    public function attach($options)
    {
        if (is_array($options)) {
            $newOptions = [];

            foreach ($options as $option) {
                if (is_string($option)) {
                    $newOptions[] = new Option([
                        'name' => $option
                    ]);
                } else {
                    throw new InvalidArgumentException("Array arguments must be composed of Strings values");
                }
            }
            return $this->options()->saveMany($newOptions);
        }

        if (is_string($options)) {
            return $this->options()->save(
                new Option([
                    'name' => $options
                ])
            );
        }

        throw new InvalidArgumentException("Invalid Argument provided");
    }


    /**
     * Remove a (list of elements)
     *
     * @param $options
     * @return bool
     * @throws CheckedOptionsException
     * @throws OptionsInvalidNumberProvidedException
     * @throws OptionsNotProvidedException
     * @throws RemoveVotedOptionException
     */
    public function detach($options)
    {
        // Prepare the elements as an array
        $options = is_array($options) ? $options : func_get_args();

        $oldOptions = [];
        $elements = $this->options()->pluck('id');
        foreach ($options as $option) {
            if (is_numeric($option) && is_int(intval($option))) {
                $option = intval($option);
                $option = Option::findOrFail($option);
                if ($option->isVoted())
                    throw new RemoveVotedOptionException();
                if ($this->containsAndNotVoted($elements, $option->id)) {
                    $oldOptions[] = $option->id;
                }
            } else if ($option instanceof Option) {
                if ($option->isVoted())
                    throw new RemoveVotedOptionException();
                if ($this->containsAndNotVoted($elements, $option->getKey())) {
                    $oldOptions = $option->getKey();
                }
            } else {
                throw new InvalidArgumentException("Array arguments must be composed of ids or option object values");
            }
        }

        // verify the number of options
        $diff = ($elements->count() - count($oldOptions));
        if ($diff == 0)
            throw new OptionsNotProvidedException();
        if ($diff == 1)
            throw new OptionsInvalidNumberProvidedException();

        if ($this->isRadio()) {
            $count = count($oldOptions);
            return Option::destroy($oldOptions) == $count;
        }

        // checkbox case

        if ($diff <= $this->maxCheck)
            throw new CheckedOptionsException();

        $count = count($oldOptions);

        return Option::destroy($oldOptions) == $count;
    }

    /**
     * Check if the option already exists in the poll
     *
     * @param Collection $elements
     * @param $against
     * @return bool
     */
    private function containsAndNotVoted(Collection $elements, $against)
    {
        return $elements->contains($against);
    }

    /**
     * Close the poll
     *
     * @return mixed
     */
    public function lock()
    {
        // Update the total of votes column
        foreach ($this->options()->get() as $option) {
            $option->updateTotalVotes();
        }

        $this->isClosed = now();

        return $this->save();
    }

    /**
     *  Unlock the closed poll
     *
     * @return mixed
     */
    public function unLock()
    {
        $this->isClosed = null;

        return $this->save();
    }

    /**
     *  enable show results
     *
     * @return mixed
     */
    public function enableShowResults()
    {
        $this->canVoterSeeResult = 1;

        return $this->save();
    }

    /**
     *  disable show results
     *
     * @return mixed
     */
    public function disableShowResults()
    {
        $this->canVoterSeeResult = 0;

        return $this->save();
    }

    /**
     * Set the number of checkable options
     *
     * @param $count
     * @return $this
     */
    public function canSelect($count)
    {
        $this->maxCheck = $count;

        return $this->save();
    }

    /**
     * Remove a Poll
     *
     * @return mixed
     */
    public function remove()
    {
        return $this->delete();
    }
}