FactoryCraft-Server/FactoryCraft/ItemFilter.h

#pragma once

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

class Item;
class ItemType;
class ItemSlot;
class UIMLItemFilterBuilder;

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 UIMLItemFilterBuilder* getUIML() const = 0;
};

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

public:
    CombinedItemFilter();
    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;
    UIMLItemFilterBuilder* getUIML() const override;
    void addFilter(ItemFilter* filter);
    const Framework::RCArray<ItemFilter>& getFilters() const;
    void setOp(std::function<bool(bool, bool)> op);
    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* toJsonObject(
        CombinedItemFilter* zObject) const override;
    JSONObjectValidationBuilder* addToValidator(
        JSONObjectValidationBuilder* builder) const override;
    const char* getTypeToken() const override;
};

class AnyItemFilter : public ItemFilter
{
public:
    AnyItemFilter();
    bool matchItem(const Item* zItem) const override;
    UIMLItemFilterBuilder* getUIML() const override;
};

class AnyItemFilterFactory : public SubTypeFactory<ItemFilter, AnyItemFilter>
{
public:
    AnyItemFilterFactory();
    AnyItemFilter* fromJson(Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJsonObject(
        AnyItemFilter* zObject) const override;
    JSONObjectValidationBuilder* addToValidator(
        JSONObjectValidationBuilder* builder) const override;
    const char* getTypeToken() const override;
};

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

public:
    TypeItemFilter();
    bool matchItem(const Item* zItem) const override;
    UIMLItemFilterBuilder* getUIML() const override;
    void setType(const ItemType* type);
    const ItemType* zType() const;
};

class TypeItemFilterFactory : public SubTypeFactory<ItemFilter, TypeItemFilter>
{
public:
    TypeItemFilterFactory();
    TypeItemFilter* fromJson(Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJsonObject(
        TypeItemFilter* zObject) const override;
    JSONObjectValidationBuilder* addToValidator(
        JSONObjectValidationBuilder* builder) const override;
    const char* 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;
    UIMLItemFilterBuilder* getUIML() 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;
    UIMLItemFilterBuilder* getUIML() 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;
    UIMLItemFilterBuilder* getUIML() const override;
};

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

public:
    GroupItemFilter();
    bool matchItem(const Item* zItem) const override;
    UIMLItemFilterBuilder* getUIML() const override;
    void addGroup(const Framework::Text group);
    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* toJsonObject(
        GroupItemFilter* zObject) const override;
    JSONObjectValidationBuilder* addToValidator(
        JSONObjectValidationBuilder* builder) const override;
    const char* getTypeToken() const override;
};

class TargetSlotNameItemFilter : public ItemFilter
{
private:
    Framework::Text slotName;

public:
    TargetSlotNameItemFilter(const Framework::Text& slotName);
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
    UIMLItemFilterBuilder* getUIML() const override;
};

class SourceSlotNameItemFilter : public ItemFilter
{
private:
    Framework::Text slotName;

public:
    SourceSlotNameItemFilter(const Framework::Text& slotName);
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
    UIMLItemFilterBuilder* getUIML() const override;
};