FactoryCraft-Server/FactoryCraft/ItemFilter.h

#pragma once

#include <Array.h>
#include <functional>
#include <ReferenceCounter.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;
};

class CombinedItemFilter : public ItemFilter
{
private:
    ItemFilter* filterA;
    ItemFilter* filterB;
    std::function<bool(bool, bool)> op;

public:
    CombinedItemFilter(ItemFilter* filterA,
        ItemFilter* filterB,
        std::function<bool(bool, bool)> op);
    ~CombinedItemFilter();
    bool matchItem(const Item* zItem) const override;
    bool matchSourceSlot(ItemSlot* zSlot) const override;
    bool matchTargetSlot(ItemSlot* zSlot) const override;
};

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

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

public:
    TypeItemFilter(const ItemType* zType);
    bool matchItem(const Item* zItem) 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;
};

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;
};