wip: code cleanup
This commit is contained in:
@@ -0,0 +1,206 @@
|
||||
package eu.mhsl.minecraft.pixelblocks;
|
||||
|
||||
import eu.mhsl.minecraft.pixelblocks.utils.Direction;
|
||||
import eu.mhsl.minecraft.pixelblocks.pixelblock.PixelBlock;
|
||||
import org.bukkit.Bukkit;
|
||||
import org.bukkit.Location;
|
||||
import org.bukkit.entity.Entity;
|
||||
import org.bukkit.entity.EntityType;
|
||||
|
||||
import java.sql.*;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
|
||||
public class PixelBlockDatabase {
|
||||
public Connection db;
|
||||
|
||||
public PixelBlockDatabase(String url) {
|
||||
try {
|
||||
this.db = DriverManager.getConnection(url);
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException("Failed to load Database", e);
|
||||
}
|
||||
}
|
||||
|
||||
public Statement getStatement() throws SQLException {
|
||||
return this.db.createStatement();
|
||||
}
|
||||
|
||||
public void removePixelBlock(PixelBlock pixelBlock) {
|
||||
Bukkit.getScheduler().runTask(PixelBlocksPlugin.plugin, () -> {
|
||||
|
||||
try {
|
||||
PreparedStatement prep = this.db.prepareStatement("DELETE FROM pixelblocks WHERE uuid = ?");
|
||||
prep.setString(1, pixelBlock.blockUUID.toString());
|
||||
prep.executeUpdate();
|
||||
|
||||
prep.close();
|
||||
} catch (SQLException e) {
|
||||
System.err.println(e.getMessage());
|
||||
}
|
||||
|
||||
});
|
||||
}
|
||||
|
||||
public void savePixelBlock(PixelBlock pixelBlock) {
|
||||
Bukkit.getScheduler().runTask(PixelBlocksPlugin.plugin, () -> {
|
||||
List<UUID> uuids = new ArrayList<>();
|
||||
|
||||
try (var statement = getStatement()) {
|
||||
ResultSet pixelBlocksResult = statement.executeQuery("SELECT * FROM pixelblocks;");
|
||||
while (pixelBlocksResult.next()) {
|
||||
uuids.add(UUID.fromString(pixelBlocksResult.getString("uuid")));
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
System.err.println(e.getMessage());
|
||||
}
|
||||
|
||||
if(!uuids.contains(pixelBlock.blockUUID)) {
|
||||
|
||||
try {
|
||||
PreparedStatement prep = this.db.prepareStatement(
|
||||
"INSERT INTO pixelblocks(uuid, owner, " +
|
||||
"locationWorldName, locationX, locationY, locationZ, " +
|
||||
"entryLocationWorldName, entryLocationX, entryLocationY, entryLocationZ, direction) " +
|
||||
"VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);"
|
||||
);
|
||||
prep.setString(1, pixelBlock.blockUUID.toString());
|
||||
prep.setString(2, pixelBlock.ownerUUID.toString());
|
||||
|
||||
prep.setString(3, pixelBlock.pixelBlockLocation.getWorld().getName());
|
||||
prep.setDouble(4, pixelBlock.pixelBlockLocation.getX());
|
||||
prep.setDouble(5, pixelBlock.pixelBlockLocation.getY());
|
||||
prep.setDouble(6, pixelBlock.pixelBlockLocation.getZ());
|
||||
|
||||
if(pixelBlock.lastEntryLocation != null) {
|
||||
prep.setString(7, pixelBlock.lastEntryLocation.getWorld().getName());
|
||||
prep.setDouble(8, pixelBlock.lastEntryLocation.getX());
|
||||
prep.setDouble(9, pixelBlock.lastEntryLocation.getY());
|
||||
prep.setDouble(10, pixelBlock.lastEntryLocation.getZ());
|
||||
} else {
|
||||
prep.setString(7, Bukkit.getWorlds().getFirst().getName());
|
||||
prep.setDouble(8, Bukkit.getWorlds().getFirst().getSpawnLocation().getX());
|
||||
prep.setDouble(9, Bukkit.getWorlds().getFirst().getSpawnLocation().getY());
|
||||
prep.setDouble(10, Bukkit.getWorlds().getFirst().getSpawnLocation().getZ());
|
||||
}
|
||||
|
||||
prep.setString(11, pixelBlock.facingDirection.toString());
|
||||
|
||||
prep.executeUpdate();
|
||||
prep.close();
|
||||
} catch (SQLException e) {
|
||||
System.err.println(e.getMessage());
|
||||
}
|
||||
|
||||
} else {
|
||||
try {
|
||||
PreparedStatement prep = this.db.prepareStatement(
|
||||
"UPDATE pixelblocks " +
|
||||
"SET owner=?, locationWorldName=?, locationX=?, locationY=?, locationZ=?," +
|
||||
"entryLocationWorldName=?, entryLocationX=?, entryLocationY=?, entryLocationZ=?, direction=? " +
|
||||
"WHERE uuid=?;"
|
||||
);
|
||||
prep.setString(1, pixelBlock.ownerUUID.toString());
|
||||
|
||||
prep.setString(2, pixelBlock.pixelBlockLocation.getWorld().getName());
|
||||
prep.setDouble(3, pixelBlock.pixelBlockLocation.getX());
|
||||
prep.setDouble(4, pixelBlock.pixelBlockLocation.getY());
|
||||
prep.setDouble(5, pixelBlock.pixelBlockLocation.getZ());
|
||||
|
||||
if(pixelBlock.lastEntryLocation != null) {
|
||||
prep.setString(6, pixelBlock.lastEntryLocation.getWorld().getName());
|
||||
prep.setDouble(7, pixelBlock.lastEntryLocation.getX());
|
||||
prep.setDouble(8, pixelBlock.lastEntryLocation.getY());
|
||||
prep.setDouble(9, pixelBlock.lastEntryLocation.getZ());
|
||||
} else {
|
||||
prep.setString(6, Bukkit.getWorlds().getFirst().getName());
|
||||
prep.setDouble(7, Bukkit.getWorlds().getFirst().getSpawnLocation().getX());
|
||||
prep.setDouble(8, Bukkit.getWorlds().getFirst().getSpawnLocation().getY());
|
||||
prep.setDouble(9, Bukkit.getWorlds().getFirst().getSpawnLocation().getZ());
|
||||
}
|
||||
|
||||
prep.setString(10, pixelBlock.blockUUID.toString());
|
||||
prep.setString(11, pixelBlock.facingDirection.toString());
|
||||
|
||||
prep.executeUpdate();
|
||||
prep.close();
|
||||
} catch (SQLException e) {
|
||||
System.err.println(e.getMessage());
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
public void loadPixelBlocks() {
|
||||
List<Entity> entities = new ArrayList<>();
|
||||
entities.addAll(Bukkit.getWorlds().get(0).getEntities().stream()
|
||||
.filter(entity -> entity.getType().equals(EntityType.BLOCK_DISPLAY) || entity.getType().equals(EntityType.INTERACTION) || entity.getType().equals(EntityType.ITEM_DISPLAY))
|
||||
.toList());
|
||||
entities.addAll(Bukkit.getWorlds().get(1).getEntities().stream()
|
||||
.filter(entity -> entity.getType().equals(EntityType.BLOCK_DISPLAY) || entity.getType().equals(EntityType.INTERACTION) || entity.getType().equals(EntityType.ITEM_DISPLAY))
|
||||
.toList());
|
||||
entities.addAll(Bukkit.getWorlds().get(2).getEntities().stream()
|
||||
.filter(entity -> entity.getType().equals(EntityType.BLOCK_DISPLAY) || entity.getType().equals(EntityType.INTERACTION) || entity.getType().equals(EntityType.ITEM_DISPLAY))
|
||||
.toList());
|
||||
|
||||
String sql = "CREATE TABLE IF NOT EXISTS pixelblocks (" +
|
||||
"uuid CHAR(36) PRIMARY KEY, " +
|
||||
"owner CHAR(36), " +
|
||||
"locationWorldName CHAR(36), " +
|
||||
"locationX DOUBLE, " +
|
||||
"locationY DOUBLE, " +
|
||||
"locationZ DOUBLE, " +
|
||||
"entryLocationWorldName CHAR(36), " +
|
||||
"entryLocationX DOUBLE, " +
|
||||
"entryLocationY DOUBLE, " +
|
||||
"entryLocationZ DOUBLE, " +
|
||||
"direction CHAR(36)" +
|
||||
");";
|
||||
|
||||
try (var statement = getStatement()) {
|
||||
statement.execute(sql);
|
||||
ResultSet pixelBlocksResult = statement.executeQuery("SELECT * FROM pixelblocks;");
|
||||
|
||||
while (pixelBlocksResult.next()) {
|
||||
Location newPixelBlockLocation = new Location(
|
||||
Bukkit.getWorld(pixelBlocksResult.getString("locationWorldName")),
|
||||
pixelBlocksResult.getDouble("locationX"),
|
||||
pixelBlocksResult.getDouble("locationY"),
|
||||
pixelBlocksResult.getDouble("locationZ")
|
||||
);
|
||||
|
||||
Location newEntryLocation = new Location(
|
||||
Bukkit.getWorld(pixelBlocksResult.getString("entryLocationWorldName")),
|
||||
pixelBlocksResult.getDouble("entryLocationX"),
|
||||
pixelBlocksResult.getDouble("entryLocationY"),
|
||||
pixelBlocksResult.getDouble("entryLocationZ")
|
||||
);
|
||||
|
||||
if(!newPixelBlockLocation.getChunk().isEntitiesLoaded()) {
|
||||
newPixelBlockLocation.getChunk().load(true);
|
||||
newPixelBlockLocation.getChunk().getEntities();
|
||||
}
|
||||
|
||||
entities.stream().filter(entity -> entity
|
||||
.getLocation()
|
||||
.add(0, PixelBlocksPlugin.configuration.hitboxOffset(), 0)
|
||||
.toBlockLocation()
|
||||
.equals(newPixelBlockLocation))
|
||||
.forEach(Entity::remove);
|
||||
|
||||
PixelBlock newPixelBlock = new PixelBlock(
|
||||
newPixelBlockLocation,
|
||||
UUID.fromString(pixelBlocksResult.getString("owner")),
|
||||
UUID.fromString(pixelBlocksResult.getString("uuid"))
|
||||
);
|
||||
newPixelBlock.lastEntryLocation = newEntryLocation;
|
||||
newPixelBlock.place(newPixelBlockLocation, Direction.valueOf(pixelBlocksResult.getString("direction")));
|
||||
}
|
||||
|
||||
} catch (SQLException e) {
|
||||
System.err.println(e.getMessage());
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user