// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.builtin.adventure.farming;

import com.hypixel.hytale.component.Holder;
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import com.hypixel.hytale.server.core.universe.world.chunk.BlockComponentChunk;
import com.hypixel.hytale.server.core.universe.world.events.ChunkPreLoadProcessEvent;
import com.hypixel.hytale.event.EventPriority;
import com.hypixel.hytale.component.system.ISystem;
import com.hypixel.hytale.builtin.adventure.farming.config.stages.spread.DirectionalGrowthBehaviour;
import com.hypixel.hytale.builtin.adventure.farming.config.stages.spread.SpreadGrowthBehaviour;
import com.hypixel.hytale.builtin.adventure.farming.config.stages.spread.SpreadFarmingStageData;
import com.hypixel.hytale.builtin.adventure.farming.config.stages.PrefabFarmingStageData;
import com.hypixel.hytale.builtin.adventure.farming.config.stages.BlockStateFarmingStageData;
import com.hypixel.hytale.builtin.adventure.farming.config.stages.BlockTypeFarmingStageData;
import com.hypixel.hytale.codec.lookup.StringCodecMapCodec;
import com.hypixel.hytale.server.core.asset.type.blocktype.config.farming.FarmingStageData;
import com.hypixel.hytale.builtin.adventure.farming.config.modifiers.WaterGrowthModifierAsset;
import com.hypixel.hytale.builtin.adventure.farming.config.modifiers.LightLevelGrowthModifierAsset;
import com.hypixel.hytale.builtin.adventure.farming.config.modifiers.FertilizerGrowthModifierAsset;
import com.hypixel.hytale.builtin.adventure.farming.interactions.UseCaptureCrateInteraction;
import com.hypixel.hytale.builtin.adventure.farming.interactions.UseCoopInteraction;
import com.hypixel.hytale.builtin.adventure.farming.interactions.UseWateringCanInteraction;
import com.hypixel.hytale.builtin.adventure.farming.interactions.ChangeFarmingStageInteraction;
import com.hypixel.hytale.builtin.adventure.farming.interactions.FertilizeSoilInteraction;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.builtin.adventure.farming.interactions.HarvestCropInteraction;
import com.hypixel.hytale.server.core.modules.interaction.interaction.config.Interaction;
import com.hypixel.hytale.builtin.tagset.config.NPCGroup;
import com.hypixel.hytale.server.core.asset.type.item.config.ItemDropList;
import com.hypixel.hytale.builtin.adventure.farming.config.FarmingCoopAsset;
import java.util.function.Function;
import com.hypixel.hytale.server.core.asset.type.weather.config.Weather;
import com.hypixel.hytale.assetstore.codec.AssetCodec;
import com.hypixel.hytale.assetstore.map.DefaultAssetMap;
import com.hypixel.hytale.server.core.asset.type.blocktype.config.farming.GrowthModifierAsset;
import com.hypixel.hytale.assetstore.AssetStore;
import com.hypixel.hytale.server.core.asset.HytaleAssetStore;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.plugin.JavaPluginInit;
import com.hypixel.hytale.builtin.adventure.farming.component.CoopResidentComponent;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.builtin.adventure.farming.states.CoopBlock;
import com.hypixel.hytale.builtin.adventure.farming.states.FarmingBlockState;
import com.hypixel.hytale.builtin.adventure.farming.states.FarmingBlock;
import com.hypixel.hytale.builtin.adventure.farming.states.TilledSoilBlock;
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.server.core.plugin.JavaPlugin;

public class FarmingPlugin extends JavaPlugin
{
    protected static FarmingPlugin instance;
    private ComponentType<ChunkStore, TilledSoilBlock> tiledSoilBlockComponentType;
    private ComponentType<ChunkStore, FarmingBlock> farmingBlockComponentType;
    private ComponentType<ChunkStore, FarmingBlockState> farmingBlockStateComponentType;
    private ComponentType<ChunkStore, CoopBlock> coopBlockStateComponentType;
    private ComponentType<EntityStore, CoopResidentComponent> coopResidentComponentType;
    
    public static FarmingPlugin get() {
        return FarmingPlugin.instance;
    }
    
    public FarmingPlugin(@Nonnull final JavaPluginInit init) {
        super(init);
    }
    
    @Override
    protected void setup() {
        FarmingPlugin.instance = this;
        this.getAssetRegistry().register(((HytaleAssetStore.Builder)((HytaleAssetStore.Builder)((HytaleAssetStore.Builder)HytaleAssetStore.builder(GrowthModifierAsset.class, new DefaultAssetMap()).setPath()).setCodec(GrowthModifierAsset.CODEC).loadsAfter(Weather.class)).setKeyFunction(GrowthModifierAsset::getId)).build());
        this.getAssetRegistry().register(((HytaleAssetStore.Builder)((HytaleAssetStore.Builder)((HytaleAssetStore.Builder)HytaleAssetStore.builder(FarmingCoopAsset.class, new DefaultAssetMap()).setPath()).setCodec(FarmingCoopAsset.CODEC).loadsAfter(ItemDropList.class, NPCGroup.class)).setKeyFunction(FarmingCoopAsset::getId)).build());
        this.getCodecRegistry(Interaction.CODEC).register("HarvestCrop", HarvestCropInteraction.class, HarvestCropInteraction.CODEC).register("FertilizeSoil", FertilizeSoilInteraction.class, FertilizeSoilInteraction.CODEC).register("ChangeFarmingStage", ChangeFarmingStageInteraction.class, ChangeFarmingStageInteraction.CODEC).register("UseWateringCan", UseWateringCanInteraction.class, UseWateringCanInteraction.CODEC).register("UseCoop", UseCoopInteraction.class, UseCoopInteraction.CODEC).register("UseCaptureCrate", UseCaptureCrateInteraction.class, UseCaptureCrateInteraction.CODEC);
        this.getCodecRegistry(GrowthModifierAsset.CODEC).register("Fertilizer", FertilizerGrowthModifierAsset.class, FertilizerGrowthModifierAsset.CODEC);
        this.getCodecRegistry(GrowthModifierAsset.CODEC).register("LightLevel", LightLevelGrowthModifierAsset.class, LightLevelGrowthModifierAsset.CODEC);
        this.getCodecRegistry(GrowthModifierAsset.CODEC).register("Water", WaterGrowthModifierAsset.class, WaterGrowthModifierAsset.CODEC);
        this.getCodecRegistry((StringCodecMapCodec<Object, BuilderCodec<BlockTypeFarmingStageData>>)FarmingStageData.CODEC).register("BlockType", BlockTypeFarmingStageData.class, BlockTypeFarmingStageData.CODEC);
        this.getCodecRegistry((StringCodecMapCodec<Object, BuilderCodec<BlockStateFarmingStageData>>)FarmingStageData.CODEC).register("BlockState", BlockStateFarmingStageData.class, BlockStateFarmingStageData.CODEC);
        this.getCodecRegistry((StringCodecMapCodec<Object, BuilderCodec<PrefabFarmingStageData>>)FarmingStageData.CODEC).register("Prefab", PrefabFarmingStageData.class, PrefabFarmingStageData.CODEC);
        this.getCodecRegistry((StringCodecMapCodec<Object, BuilderCodec<SpreadFarmingStageData>>)FarmingStageData.CODEC).register("Spread", SpreadFarmingStageData.class, SpreadFarmingStageData.CODEC);
        this.getCodecRegistry((StringCodecMapCodec<Object, BuilderCodec<DirectionalGrowthBehaviour>>)SpreadGrowthBehaviour.CODEC).register("Directional", DirectionalGrowthBehaviour.class, DirectionalGrowthBehaviour.CODEC);
        this.tiledSoilBlockComponentType = this.getChunkStoreRegistry().registerComponent(TilledSoilBlock.class, "TilledSoil", TilledSoilBlock.CODEC);
        this.farmingBlockComponentType = this.getChunkStoreRegistry().registerComponent(FarmingBlock.class, "FarmingBlock", FarmingBlock.CODEC);
        this.farmingBlockStateComponentType = this.getChunkStoreRegistry().registerComponent(FarmingBlockState.class, "Farming", FarmingBlockState.CODEC);
        this.coopBlockStateComponentType = this.getChunkStoreRegistry().registerComponent(CoopBlock.class, "Coop", CoopBlock.CODEC);
        this.coopResidentComponentType = this.getEntityStoreRegistry().registerComponent(CoopResidentComponent.class, "CoopResident", CoopResidentComponent.CODEC);
        this.getChunkStoreRegistry().registerSystem(new FarmingSystems.OnSoilAdded());
        this.getChunkStoreRegistry().registerSystem(new FarmingSystems.OnFarmBlockAdded());
        this.getChunkStoreRegistry().registerSystem(new FarmingSystems.Ticking());
        this.getChunkStoreRegistry().registerSystem(new FarmingSystems.MigrateFarming());
        this.getChunkStoreRegistry().registerSystem(new FarmingSystems.OnCoopAdded());
        this.getEntityStoreRegistry().registerSystem(new FarmingSystems.CoopResidentEntitySystem());
        this.getEntityStoreRegistry().registerSystem(new FarmingSystems.CoopResidentTicking());
        this.getEventRegistry().registerGlobal(EventPriority.LAST, ChunkPreLoadProcessEvent.class, FarmingPlugin::preventSpreadOnNew);
    }
    
    private static void preventSpreadOnNew(final ChunkPreLoadProcessEvent event) {
        if (!event.isNewlyGenerated()) {
            return;
        }
        final BlockComponentChunk components = event.getHolder().getComponent(BlockComponentChunk.getComponentType());
        if (components == null) {
            return;
        }
        final Int2ObjectMap<Holder<ChunkStore>> holders = components.getEntityHolders();
        holders.values().forEach(v -> {
            final FarmingBlock farming = v.getComponent(FarmingBlock.getComponentType());
            if (farming != null) {
                farming.setSpreadRate(0.0f);
            }
        });
    }
    
    public ComponentType<ChunkStore, TilledSoilBlock> getTiledSoilBlockComponentType() {
        return this.tiledSoilBlockComponentType;
    }
    
    public ComponentType<ChunkStore, FarmingBlock> getFarmingBlockComponentType() {
        return this.farmingBlockComponentType;
    }
    
    public ComponentType<ChunkStore, FarmingBlockState> getFarmingBlockStateComponentType() {
        return this.farmingBlockStateComponentType;
    }
    
    public ComponentType<ChunkStore, CoopBlock> getCoopBlockStateComponentType() {
        return this.coopBlockStateComponentType;
    }
    
    public ComponentType<EntityStore, CoopResidentComponent> getCoopResidentComponentType() {
        return this.coopResidentComponentType;
    }
}