#pragma once

#include <Array.h>
#include <functional>
#include <ReferenceCounter.h>
#include <TypeRegistry.h>

class Item;
class ItemType;
class ItemSlot;

class ItemFilter : public virtual Framework::ReferenceCounter
{
public:
    ItemFilter();
    virtual bool matchItem(const Item* zItem) const;
    virtual bool matchSourceSlot(ItemSlot* zSlot) const;
    virtual bool matchTargetSlot(ItemSlot* zSlot) const;
    virtual Framework::Text getLogicUIML() const = 0;
};

class CombinedItemFilter : public ItemFilter
{
private:
    Framework::RCArray<ItemFilter> filters;
    std::function<bool(bool, bool)> op;

public:
    CombinedItemFilter(Framework::RCArray<ItemFilter> filters,
        std::function<bool(bool, bool)> op);
    bool matchItem(const Item* zItem) const override;
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
    Framework::Text getLogicUIML() const override;
    const Framework::RCArray<ItemFilter>& getFilters() const;
    std::function<bool(bool, bool)> getOp() const;
};

class CombinedItemFilterFactory
    : public SubTypeFactory<ItemFilter, CombinedItemFilter>
{
public:
    CombinedItemFilterFactory();
    CombinedItemFilter* fromJson(
        Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJson(
        CombinedItemFilter* zObject) const override;
    Framework::JSON::Validator::JSONValidator* getValidator(
        Framework::JSON::Validator::ObjectValidationBuilder<
            Framework::JSON::Validator::JSONValidator>* builder) const override;
    Framework::Text getTypeToken() const override;
};

class AnyItemFilter : public ItemFilter
{
public:
    AnyItemFilter();
    bool matchItem(const Item* zItem) const override;
    Framework::Text getLogicUIML() const override;
};

class AnyItemFilterFactory : public SubTypeFactory<ItemFilter, AnyItemFilter>
{
public:
    AnyItemFilterFactory();
    AnyItemFilter* fromJson(Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJson(AnyItemFilter* zObject) const override;
    Framework::JSON::Validator::JSONValidator* getValidator(
        Framework::JSON::Validator::ObjectValidationBuilder<
            Framework::JSON::Validator::JSONValidator>* builder) const override;
    Framework::Text getTypeToken() const override;
};

class TypeItemFilter : public ItemFilter
{
private:
    const ItemType* type;

public:
    TypeItemFilter(const ItemType* zType);
    bool matchItem(const Item* zItem) const override;
    Framework::Text getLogicUIML() const override;
    const ItemType* zType() const;
};

class TypeItemFilterFactory : public SubTypeFactory<ItemFilter, TypeItemFilter>
{
public:
    TypeItemFilterFactory();
    TypeItemFilter* fromJson(Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJson(TypeItemFilter* zObject) const override;
    Framework::JSON::Validator::JSONValidator* getValidator(
        Framework::JSON::Validator::ObjectValidationBuilder<
            Framework::JSON::Validator::JSONValidator>* builder) const override;
    Framework::Text getTypeToken() const override;
};

class SpecificSlotFilter : public ItemFilter
{
private:
    int sourceSlotId;
    int targetSlotId;

public:
    SpecificSlotFilter(int sourceSlotId, int targetSlotId);
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
    Framework::Text getLogicUIML() const override;
};

class SourceSlotBlacklistFilter : public ItemFilter
{
private:
    Framework::Array<int> blackList;

public:
    SourceSlotBlacklistFilter();
    void addBlackListSlotId(int id);
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
    Framework::Text getLogicUIML() const override;
};

class TargetSlotBlacklistFilter : public ItemFilter
{
private:
    Framework::Array<int> blackList;

public:
    TargetSlotBlacklistFilter();
    void addBlackListSlotId(int id);
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
    Framework::Text getLogicUIML() const override;
};

class GroupItemFilter : public ItemFilter
{
private:
    Framework::RCArray<Framework::Text> groups;

public:
    GroupItemFilter(Framework::RCArray<Framework::Text> groups);
    bool matchItem(const Item* zItem) const override;
    Framework::Text getLogicUIML() const override;
    const Framework::RCArray<Framework::Text>& getGroups() const;
};

class GroupItemFilterFactory
    : public SubTypeFactory<ItemFilter, GroupItemFilter>
{
public:
    GroupItemFilterFactory();
    GroupItemFilter* fromJson(
        Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJson(
        GroupItemFilter* zObject) const override;
    Framework::JSON::Validator::JSONValidator* getValidator(
        Framework::JSON::Validator::ObjectValidationBuilder<
            Framework::JSON::Validator::JSONValidator>* builder) const override;
    Framework::Text getTypeToken() const override;
};