package goodgenerator.blocks.tileEntity;

import static com.gtnewhorizon.structurelib.structure.StructureUtility.*;
import static gregtech.api.enums.HatchElement.*;
import static gregtech.api.enums.Textures.BlockIcons.*;

import java.util.stream.Collectors;
import java.util.stream.IntStream;

import javax.annotation.Nonnull;

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.EnumChatFormatting;
import net.minecraft.util.StatCollector;
import net.minecraftforge.common.util.ForgeDirection;

import org.apache.commons.lang3.tuple.Pair;
import org.jetbrains.annotations.NotNull;

import com.gtnewhorizon.structurelib.alignment.constructable.ISurvivalConstructable;
import com.gtnewhorizon.structurelib.structure.IStructureDefinition;
import com.gtnewhorizon.structurelib.structure.ISurvivalBuildEnvironment;
import com.gtnewhorizon.structurelib.structure.StructureDefinition;

import goodgenerator.api.recipe.GoodGeneratorRecipeMaps;
import goodgenerator.loader.Loaders;
import gregtech.api.GregTechAPI;
import gregtech.api.enums.GTValues;
import gregtech.api.enums.Materials;
import gregtech.api.enums.Textures;
import gregtech.api.interfaces.ITexture;
import gregtech.api.interfaces.metatileentity.IMetaTileEntity;
import gregtech.api.interfaces.tileentity.IGregTechTileEntity;
import gregtech.api.logic.ProcessingLogic;
import gregtech.api.metatileentity.implementations.MTEExtendedPowerMultiBlockBase;
import gregtech.api.recipe.RecipeMap;
import gregtech.api.recipe.check.CheckRecipeResult;
import gregtech.api.recipe.check.CheckRecipeResultRegistry;
import gregtech.api.render.TextureFactory;
import gregtech.api.util.GTRecipe;
import gregtech.api.util.GTStructureUtility;
import gregtech.api.util.GTUtility;
import gregtech.api.util.MultiblockTooltipBuilder;
import gregtech.api.util.OverclockCalculator;

public class MTEComponentAssemblyLine extends MTEExtendedPowerMultiBlockBase<MTEComponentAssemblyLine>
    implements ISurvivalConstructable {

    private int casingTier;
    private double speedBonus;
    protected static final String STRUCTURE_PIECE_MAIN = "main";
    private static final IStructureDefinition<MTEComponentAssemblyLine> STRUCTURE_DEFINITION = StructureDefinition
        .<MTEComponentAssemblyLine>builder()
        .addShape(
            STRUCTURE_PIECE_MAIN,
            new String[][] {
                { "         ", "   III   ", " HHI~IHH ", "HH III HH", "H       H", "H       H", "H  JJJ  H",
                    "H  JJJ  H", "H  N N  H", "HHHHHHHHH" },
                { "         ", " ELHHHLE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "AG  C  GA", "AG     GA", "AG     GA",
                    "AG HHH GA", "AG     GA", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "HG  C  GH", "HG     GH", "HG     GH",
                    "HG HHH GH", "HG     GH", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "AG  C  GA", "AG     GA", "AG     GA",
                    "AG HHH GA", "AG     GA", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "HG  C  GH", "HG     GH", "HG     GH",
                    "HG HHH GH", "HG     GH", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "AG  C  GA", "AG     GA", "AG     GA",
                    "AG HHH GA", "AG     GA", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "HG  C  GH", "HG     GH", "HG     GH",
                    "HG HHH GH", "HG     GH", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "   BBB   ", " EL   LE ", "E GGDGG E", "HGG D GGH", "AG  C  GA", "AG     GA", "AG     GA",
                    "AG HHH GA", "AG     GA", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A  n n  A", "MHHHHHHHM" },
                { "   HBH   ", " EL   LE ", "E       E", "HC     CH", "AC     CA", "AC     CA", "A D   D A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "         ", " ELHHHLE ", "E       E", "H       H", "A       A", "A       A", "A       A",
                    "A  HHH  A", "A       A", "MHHHHHHHM" },
                { "         ", "         ", " HHHHHHH ", "HH     HH", "H       H", "H       H", "H       H",
                    "H       H", "H  KKK  H", "HHHHHHHHH" } })
        .addElement(
            'A',
            ofChain(
                ofBlockUnlocalizedName("bartworks", "BW_GlasBlocks", 5),
                ofBlockUnlocalizedName("bartworks", "BW_GlasBlocks", 13),
                ofBlockUnlocalizedName("bartworks", "BW_GlasBlocks", 14),
                ofBlockUnlocalizedName("bartworks", "BW_GlasBlocks", 15),
                ofBlockUnlocalizedName("bartworks", "BW_GlasBlocks2", 0)))
        .addElement('H', ofBlock(GregTechAPI.sBlockCasings8, 7))
        .addElement('C', ofBlock(GregTechAPI.sBlockCasings2, 5))
        .addElement('D', ofBlock(GregTechAPI.sBlockCasings2, 9))
        .addElement('G', ofBlock(GregTechAPI.sBlockCasings9, 0))
        .addElement('E', ofBlock(GregTechAPI.sBlockCasings9, 1))
        .addElement('F', ofBlock(GregTechAPI.sBlockCasings4, 1))
        .addElement(
            'B',
            ofBlocksTiered(
                (block, meta) -> block == Loaders.componentAssemblylineCasing ? meta : -1,
                IntStream.range(0, 14)
                    .mapToObj(i -> Pair.of(Loaders.componentAssemblylineCasing, i))
                    .collect(Collectors.toList()),
                -2,
                (t, meta) -> t.casingTier = meta,
                t -> t.casingTier))
        .addElement(
            'J',
            GTStructureUtility.buildHatchAdder(MTEComponentAssemblyLine.class)
                .atLeast(InputBus)
                .dot(1)
                .casingIndex(183)
                .buildAndChain(GregTechAPI.sBlockCasings8, 7))
        .addElement(
            'N',
            GTStructureUtility.buildHatchAdder(MTEComponentAssemblyLine.class)
                .atLeast(InputBus)
                .dot(1)
                .casingIndex(183)
                .buildAndChain(GTStructureUtility.ofFrame(Materials.TungstenSteel)))
        .addElement(
            'K',
            GTStructureUtility.buildHatchAdder(MTEComponentAssemblyLine.class)
                .atLeast(OutputBus)
                .dot(2)
                .casingIndex(183)
                .buildAndChain(GregTechAPI.sBlockCasings8, 7))
        .addElement(
            'L',
            GTStructureUtility.buildHatchAdder(MTEComponentAssemblyLine.class)
                .atLeast(Energy, ExoticEnergy)
                .dot(3)
                .casingIndex(183)
                .buildAndChain(GregTechAPI.sBlockCasings8, 7))
        .addElement(
            'I',
            GTStructureUtility.buildHatchAdder(MTEComponentAssemblyLine.class)
                .atLeast(Maintenance)
                .dot(4)
                .casingIndex(183)
                .buildAndChain(GregTechAPI.sBlockCasings8, 7))
        .addElement(
            'M',
            GTStructureUtility.buildHatchAdder(MTEComponentAssemblyLine.class)
                .atLeast(InputHatch)
                .dot(5)
                .casingIndex(183)
                .buildAndChain(GregTechAPI.sBlockCasings8, 7))
        .addElement('n', GTStructureUtility.ofFrame(Materials.TungstenSteel))
        .build();

    public MTEComponentAssemblyLine(int aID, String aName, String aNameRegional) {
        super(aID, aName, aNameRegional);
    }

    public MTEComponentAssemblyLine(String aName) {
        super(aName);
    }

    @Override
    public void construct(ItemStack stackSize, boolean hintsOnly) {
        buildPiece(STRUCTURE_PIECE_MAIN, stackSize, hintsOnly, 4, 2, 0);
    }

    @Override
    public IStructureDefinition<MTEComponentAssemblyLine> getStructureDefinition() {
        return STRUCTURE_DEFINITION;
    }

    @Override
    protected MultiblockTooltipBuilder createTooltip() {
        MultiblockTooltipBuilder tt = new MultiblockTooltipBuilder();
        tt.addMachineType("High-Capacity Component Assembler")
            .addInfo("Assembles basic components (motors, pumps, etc.) in large batches.")
            .addInfo(
                "The " + EnumChatFormatting.BOLD
                    + EnumChatFormatting.YELLOW
                    + "Component Assembly Line Casing "
                    + EnumChatFormatting.RESET
                    + EnumChatFormatting.GRAY
                    + "limits the recipes the machine can perform. See the NEI pages for details.")
            .addInfo("Using casings above the required recipe tier provides a speed bonus:")
            .addInfo(EnumChatFormatting.YELLOW + "Halves recipe time per tier above recipe")
            .addInfo(
                "Supports " + EnumChatFormatting.BLUE
                    + "Tec"
                    + EnumChatFormatting.DARK_BLUE
                    + "Tech"
                    + EnumChatFormatting.GRAY
                    + " laser and multi-amp hatches!")
            .addInfo("Supports overclocking beyond MAX!")
            .addInfo(EnumChatFormatting.ITALIC + "Much more efficient than other competing brands!")
            .addTecTechHatchInfo()
            .beginStructureBlock(9, 10, 33, false)
            .addOtherStructurePart("Borosilicate Glass", "Can be UV tier or higher")
            .addInputBus("Start of conveyor belt", 1)
            .addOutputBus("End of conveyor belt", 2)
            .addEnergyHatch("Second-top layer", 3)
            .addMaintenanceHatch("Around the controller", 4)
            .addInputHatch("Bottom left and right corners", 5)
            .toolTipFinisher(EnumChatFormatting.AQUA + "MadMan310");
        return tt;
    }

    @Override
    public IMetaTileEntity newMetaEntity(IGregTechTileEntity aTileEntity) {
        return new MTEComponentAssemblyLine(mName);
    }

    /**
     * Changes and adds new information to the default info data for the scanner.
     */
    @Override
    public String[] getInfoData() {
        String[] origin = super.getInfoData();
        String[] ret = new String[origin.length + 1];
        System.arraycopy(origin, 0, ret, 0, origin.length);
        ret[origin.length] = StatCollector.translateToLocal("scanner.info.CASS.tier")
            + (casingTier >= 0 ? GTValues.VN[casingTier + 1] : "None!");
        return ret;
    }

    @Override
    public ITexture[] getTexture(IGregTechTileEntity aBaseMetaTileEntity, ForgeDirection side, ForgeDirection facing,
        int colorIndex, boolean aActive, boolean aRedstone) {
        if (side == facing) {
            if (aActive) return new ITexture[] { Textures.BlockIcons.getCasingTextureForId(183),
                TextureFactory.builder()
                    .addIcon(OVERLAY_FRONT_PROCESSING_ARRAY_ACTIVE)
                    .extFacing()
                    .build(),
                TextureFactory.builder()
                    .addIcon(OVERLAY_FRONT_PROCESSING_ARRAY_ACTIVE_GLOW)
                    .extFacing()
                    .glow()
                    .build() };
            return new ITexture[] { Textures.BlockIcons.getCasingTextureForId(183), TextureFactory.builder()
                .addIcon(OVERLAY_FRONT_PROCESSING_ARRAY)
                .extFacing()
                .build(),
                TextureFactory.builder()
                    .addIcon(OVERLAY_FRONT_PROCESSING_ARRAY_GLOW)
                    .extFacing()
                    .glow()
                    .build() };
        }
        return new ITexture[] { Textures.BlockIcons.getCasingTextureForId(183) };
    }

    @Override
    public boolean isCorrectMachinePart(ItemStack aStack) {
        return true;
    }

    @Override
    protected ProcessingLogic createProcessingLogic() {
        return new ProcessingLogic() {

            @NotNull
            @Override
            protected CheckRecipeResult validateRecipe(@NotNull GTRecipe recipe) {
                if (recipe.mSpecialValue > casingTier + 1) {
                    return CheckRecipeResultRegistry.insufficientMachineTier(recipe.mSpecialValue);
                }
                return CheckRecipeResultRegistry.SUCCESSFUL;
            }

            @Nonnull
            @Override
            protected OverclockCalculator createOverclockCalculator(@Nonnull GTRecipe recipe) {
                speedBonus = 1 / Math.pow(2, casingTier + 1 - recipe.mSpecialValue);
                return super.createOverclockCalculator(recipe).setSpeedBoost(speedBonus);
            }
        };
    }

    @Override
    protected void setProcessingLogicPower(ProcessingLogic logic) {
        logic.setAvailableVoltage(getMaxInputEu());
        logic.setAvailableAmperage(1);
    }

    @Override
    public int survivalConstruct(ItemStack stackSize, int elementBudget, ISurvivalBuildEnvironment env) {
        if (mMachine) return -1;
        int realBudget = elementBudget >= 200 ? elementBudget : Math.min(200, elementBudget * 5);
        return survivialBuildPiece(STRUCTURE_PIECE_MAIN, stackSize, 4, 2, 0, realBudget, env, false, true);
    }

    @Override
    public boolean checkMachine(IGregTechTileEntity aBaseMetaTileEntity, ItemStack aStack) {
        casingTier = -2;
        return checkPiece(STRUCTURE_PIECE_MAIN, 4, 2, 0);
    }

    @Override
    public void onScrewdriverRightClick(ForgeDirection side, EntityPlayer aPlayer, float aX, float aY, float aZ) {
        inputSeparation = !inputSeparation;
        GTUtility.sendChatToPlayer(
            aPlayer,
            StatCollector.translateToLocal("GT5U.machines.separatebus") + " " + inputSeparation);
    }

    @Override
    public boolean onWireCutterRightClick(ForgeDirection side, ForgeDirection wrenchingSide, EntityPlayer aPlayer,
        float aX, float aY, float aZ) {
        batchMode = !batchMode;
        if (batchMode) {
            GTUtility.sendChatToPlayer(aPlayer, StatCollector.translateToLocal("misc.BatchModeTextOn"));
        } else {
            GTUtility.sendChatToPlayer(aPlayer, StatCollector.translateToLocal("misc.BatchModeTextOff"));
        }
        return true;
    }

    @Override
    public boolean supportsInputSeparation() {
        return true;
    }

    @Override
    public boolean supportsBatchMode() {
        return true;
    }

    @Override
    public int getMaxEfficiency(ItemStack aStack) {
        return 10000;
    }

    @Override
    public int getDamageToComponent(ItemStack aStack) {
        return 0;
    }

    @Override
    public boolean explodesOnComponentBreak(ItemStack aStack) {
        return false;
    }

    @Override
    public RecipeMap<?> getRecipeMap() {
        return GoodGeneratorRecipeMaps.componentAssemblyLineRecipes;
    }

    @Override
    public void saveNBTData(NBTTagCompound aNBT) {
        super.saveNBTData(aNBT);
        aNBT.setInteger("casingTier", casingTier);
    }

    @Override
    public void loadNBTData(final NBTTagCompound aNBT) {
        super.loadNBTData(aNBT);
        casingTier = aNBT.getInteger("casingTier");
        if (!aNBT.hasKey(INPUT_SEPARATION_NBT_KEY)) {
            inputSeparation = aNBT.getBoolean("mSeparate");
        }
    }
}