#pragma once

#include <Either.h>
#include <JSON.h>
#include <ReferenceCounter.h>

#include "EntityGenerator.h"
#include "GeneratorRule.h"
#include "JNoise.h"
#include "JsonExpression.h"
#include "PlantConfig.h"
#include "StructureCollection.h"

class Block;
class Noise;

class BiomGenerator : public virtual Framework::ReferenceCounter
{
private:
    Framework::Text name;
    Framework::RCArray<StructureTemplateCollection> templates;
    Framework::RCArray<GeneratorRule> rules;
    JBoolExpression* condition;
    Framework::Vec3<int> minStructureOffset;
    Framework::Vec3<int> maxStructureOffset;
    Framework::RCArray<EntityGenerator> entityGenerators;
    Framework::RCArray<PlantConfig> plantConfigs;

public:
    BiomGenerator();
    ~BiomGenerator();

    void initialize(JExpressionMemory* zMemory);
    Framework::Either<Block*, int> generateBlock(
        int x, int y, int z, int dimensionId, Chunk* partialGeneratedChunk);
    bool isApplicable();
    void generateStructures(int x,
        int y,
        int z,
        int dimensionId,
        Framework::Vec3<int>& minPos,
        Framework::Vec3<int>& maxPos,
        Framework::RCArray<GeneratedStructure>* zResult);
    void generatePlants(int x,
        int y,
        int z,
        int dimensionId,
        Chunk* zChunk,
        bool underground,
        bool underwater,
        int seaFluidBlockTypeId);
    void generateEntities(int x, int y, int z, int dimensionId);

    const Framework::RCArray<StructureTemplateCollection>& getTemplates() const;
    Framework::Vec3<int> getMinStructureOffset() const;
    Framework::Vec3<int> getMaxStructureOffset() const;

    void setName(Framework::Text name);
    Framework::Text getName() const;
    void setCondition(JBoolExpression* condition);
    JBoolExpression* getCondition() const;
    void addTemplate(StructureTemplateCollection* collection);
    const Framework::RCArray<StructureTemplateCollection>& getTemplates();
    void addGeneratorRule(GeneratorRule* rule);
    const Framework::RCArray<GeneratorRule>& getGeneratorRules() const;
    void addEntityGenerator(EntityGenerator* generator);
    const Framework::RCArray<EntityGenerator>& getEntityGenerators() const;
    void addPlantConfig(PlantConfig* config);
    const Framework::RCArray<PlantConfig>& getPlantConfigs() const;
};

class BiomGeneratorFactory : public ObjectTypeFactory<BiomGenerator>
{
public:
    BiomGeneratorFactory();
    BiomGenerator* fromJson(Framework::JSON::JSONObject* zJson) const override;
    Framework::JSON::JSONObject* toJsonObject(
        BiomGenerator* zObject) const override;
    JSONObjectValidationBuilder* addToValidator(
        JSONObjectValidationBuilder* builder) const override;
};