<?php
namespace Affilicious\Provider\Repository\Carbon;

use Affilicious\Common\Generator\Key_Generator_Interface;
use Affilicious\Common\Model\Slug;
use Affilicious\Provider\Model\Provider;
use Affilicious\Provider\Model\Provider_Id;
use Affilicious\Provider\Model\Type;
use Affilicious\Provider\Repository\Provider_Repository_Interface;

if (!defined('ABSPATH')) {
    exit('Not allowed to access pages directly.');
}

/**
 * @since 0.8
 */
class Carbon_Provider_Repository implements Provider_Repository_Interface
{
	/**
	 * @since 0.8
	 * @var string
	 */
    const CURRENT_ID = '_affilicious_provider_current_id';

	/**
	 * @since 0.8
	 * @var string
	 */
    const ID_TEMPLATE = '_affilicious_provider_%s_id';

    /**
     * @since 0.8
     * @var int
     */
    protected $current_id;

    /**
     * @since 0.8
     * @var Provider[]
     */
    protected $providers;

    /**
     * @since 0.8
     * @var Key_Generator_Interface
     */
    protected $key_generator;

    /**
     * @since 0.8
     * @param Key_Generator_Interface $key_generator
     */
    public function __construct(Key_Generator_Interface $key_generator)
    {
        $this->key_generator = $key_generator;

        if(($this->current_id = (int) carbon_get_theme_option(self::CURRENT_ID)) == false) {
            $this->current_id = 1;

            update_option(self::CURRENT_ID, $this->current_id);
        }

        $this->providers = array();
    }

    /**
     * @inheritdoc
     * @since 0.8
     */
    public function store(Provider $provider)
    {
        $provider_id = $this->prepare_provider_id($provider);

        $this->providers[$provider->get_slug()->get_value()] = $provider;

        return $provider_id;
    }

    /**
     * @inheritdoc
     * @since 0.8
     */
    public function delete(Provider_Id $provider_id)
    {
        $provider = $this->find_one_by_id($provider_id);
        if($provider === null) {
            return new \WP_Error('aff_provider_not_found', sprintf(
                'Provider #%s not found in the database.',
                $provider_id->get_value()
            ));
        }

        unset($this->providers[$provider->get_name()->get_value()]);
        $provider->set_id(null);

        return true;
    }

    /**
     * @inheritdoc
     * @since 0.9.16
     */
    public function delete_all()
    {
        $providers = $this->find_all();

        foreach ($providers as $provider) {
            $result = $this->delete($provider->get_id());
            if($result instanceof \WP_Error) {
                return $result;
            }
        }

        return true;
    }

    /**
     * @inheritdoc
     * @since 0.9.16
     */
    public function find(Provider_Id $provider_id)
    {
        foreach ($this->providers as $provider) {
            if($provider->get_id()->is_equal_to($provider_id)) {
                return $provider;
            }
        }

        return null;
    }

    /**
     * @inheritdoc
     * @since 0.9.16
     */
    public function find_by_slug(Slug $slug)
    {
        return isset($this->providers[$slug->get_value()]) ? $this->providers[$slug->get_value()] : null;
    }

    /**
     * @inheritdoc
     * @since 0.8
     */
    public function find_all()
    {
        $providers = array();
        foreach ($this->providers as $provider) {
            $this->prepare_provider_id($provider);
            $providers[] = $provider;
        }

        return $providers;
    }

	/**
	 * @inheritdoc
	 * @since 0.9.7
	 */
    public function find_all_by_type(Type $type)
    {
	    $providers = $this->find_all();

	    $providers = array_filter($providers, function(Provider $provider) use ($type) {
		    return $type->is_equal_to($provider->get_type());
	    });

	    $providers = array_values($providers);

	    return $providers;
    }

	/**
     * @since 0.8
     * @param Slug $slug
     * @return Provider_Id|null
     */
    protected function find_provider_id(Slug $slug)
    {
        $key = $this->key_generator->generate_from_slug($slug);
        $option = sprintf(self::ID_TEMPLATE, $key->get_value());
        $id = carbon_get_theme_option($option);

        return !empty($id) ? new Provider_Id($id) : null;
    }

    /**
     * Store the provider ID for the slug.
     *
     * @since 0.8
     * @param Provider_Id $id
     * @param Slug $slug
     */
    protected function store_provider_id(Provider_Id $id, Slug $slug)
    {
        $key = $this->key_generator->generate_from_slug($slug);
        $option = sprintf(self::ID_TEMPLATE, $key->get_value());
        update_option($option, $id->get_value());
    }

    /**
     * Get the next free provider ID.
     *
     * @since 0.8
     * @return Provider_Id
     */
    protected function get_next_provider_id()
    {
        $id = new Provider_Id($this->current_id);

        $this->current_id++;
        update_option(self::CURRENT_ID, $this->current_id);

        return $id;
    }

    /**
     * Prepare the provider ID.
     *
     * @since 0.8
     * @param Provider $provider
     * @return Provider_Id
     */
    protected function prepare_provider_id(Provider $provider)
    {
        $id = $this->find_provider_id($provider->get_slug());
        if($id === null) {
            $id = $this->get_next_provider_id();
            $this->store_provider_id($id, $provider->get_slug());
        }

        $provider->set_id($id);

        return $id;
    }

    /**
     * @inheritdoc
     * @since 0.8
     */
    public function find_one_by_id(Provider_Id $provider_id)
    {
        $provider = $this->find($provider_id);

        return $provider;
    }

    /**
     * @inheritdoc
     * @since 0.8
     */
    public function find_one_by_slug(Slug $slug)
    {
        $provider = $this->find_by_slug($slug);

        return $provider;
    }
}