PistonCheck.java

/*******************************************************************************
 * Copyright (C) 2014 Travis Ralston (turt2live)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/

package com.turt2live.antishare.bukkit.dev.check;

import com.turt2live.antishare.bukkit.dev.AntiShare;
import com.turt2live.antishare.bukkit.dev.CheckBase;
import org.bukkit.*;
import org.bukkit.block.Block;
import org.bukkit.conversations.ConversationContext;
import org.bukkit.conversations.ConversationFactory;
import org.bukkit.conversations.Prompt;
import org.bukkit.entity.Player;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

// Finds blocks that will fall because of the block under them being broken
public class PistonCheck extends CheckBase implements Prompt {

    private Player player;
    private int nextMaterial = -1;
    private Material[] materials = Material.values();
    private List<Material> breaks = new ArrayList<>();

    public PistonCheck(AntiShare plugin, Player player) {
        super(plugin);
        this.player = player;
    }

    public void begin() {
        Bukkit.broadcastMessage(ChatColor.BLUE + "Running piston sequence...");
        prepareCube(10, new Location(player.getWorld(), 0, 70, 0));
        player.teleport(new Location(player.getWorld(), 0, 72, 0));
        player.beginConversation(new ConversationFactory(plugin)
                .withFirstPrompt(this)
                .withTimeout(60)
                .withLocalEcho(false)
                .buildConversation(player));
    }

    // Return true for 'done'
    private boolean advance() {
        final Location spawn = new Location(player.getWorld(), 0, 70, 0);
        Material test;
        do {
            nextMaterial++; // Go to next material
            if (nextMaterial >= materials.length) {
                plugin.getServer().broadcastMessage(ChatColor.GREEN + "Done");
                writeResults();
                return true;
            }
            test = materials[nextMaterial];
        } while (!test.isBlock() || invalid(test));
        prepareCube(10, spawn);
        player.sendRawMessage(ChatColor.BLUE + "Testing " + test.name());
        spawn.getBlock().setType(test);
        return false;
    }

    private void writeResults() {
        try {
            BufferedWriter writer = new BufferedWriter(new FileWriter(new File(plugin.getDataFolder(), "piston_break.txt"), false));
            writer.write("List<Material> materials = new ArrayList<Material>();\n");

            for (Material material : this.breaks) {
                writer.write("materials.add(Material." + material + ");");
                writer.newLine();
            }

            writer.write("return materials;\n");
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private boolean invalid(Material material) {
        switch (material) {
            case PISTON_EXTENSION:
            case PISTON_MOVING_PIECE:
            case ENDER_PORTAL:
            case PORTAL:
                return true;
        }
        return false;
    }

    @Override
    public String getPromptText(ConversationContext conversationContext) {
        return nextMaterial < 0 ? ChatColor.AQUA + "Enter anything to begin." : ChatColor.AQUA + "Move the block with a piston thing. Does it disappear? [Y/N]";
    }

    @Override
    public boolean blocksForInput(ConversationContext conversationContext) {
        return true; // Need input
    }

    @Override
    public Prompt acceptInput(ConversationContext conversationContext, String s) {
        if (nextMaterial >= 0) {
            if (s.equalsIgnoreCase("yes") || s.equalsIgnoreCase("y")) {
                this.breaks.add(this.materials[nextMaterial]);
                player.sendRawMessage(ChatColor.GREEN + "Added " + this.materials[nextMaterial].name());
            } else {
                player.sendRawMessage(ChatColor.RED + "Not added " + this.materials[nextMaterial].name());
            }
        }
        boolean done = advance();
        if (done) {
            player.sendRawMessage(ChatColor.AQUA + "Done!");
        }
        return done ? null : this;
    }

    private void prepareCube(int radius, Location location) {
        World world = location.getWorld();
        for (int x = 0; x < radius; x++) {
            for (int y = radius; y >= 0; y--) {
                for (int z = 0; z < radius; z++) {
                    // Set ceiling
                    if (y == radius) {
                        Location ceilingBlock = new Location(world, x + location.getBlockX() - 5, y + location.getBlockY() + 1, z + location.getBlockZ() - 5);
                        Block block = ceilingBlock.getBlock();
                        block.setType(Material.STONE);
                    }

                    // Clear area
                    Location possibleBlock = new Location(world, x + location.getBlockX() - 5, y + location.getBlockY(), z + location.getBlockZ() - 5);
                    Block block = possibleBlock.getBlock();
                    block.setType(Material.AIR);

                    // Set floor
                    if (y == 0) {
                        Location floorBlock = new Location(world, x + location.getBlockX() - 5, location.getBlockY() - 1, z + location.getBlockZ() - 5);
                        block = floorBlock.getBlock();
                        block.setType(Material.STONE);
                    }
                }
            }
        }
    }
}