package com.yourusername.skinrestorer512.storage; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.google.gson.reflect.TypeToken; import com.yourusername.skinrestorer512.SkinRestorer512; import org.bukkit.configuration.file.YamlConfiguration; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.sql.*; import java.util.HashMap; import java.util.Map; import java.util.UUID; public class StorageManager { private final SkinRestorer512 plugin; private final String storageType; private Connection mysqlConnection; private File storageFile; private Map dataMap; private final Gson gson; public StorageManager(SkinRestorer512 plugin) { this.plugin = plugin; this.storageType = plugin.getConfig().getString("storage-type", "YAML"); this.dataMap = new HashMap<>(); this.gson = new GsonBuilder().setPrettyPrinting().create(); } public void initialize() { switch (storageType.toUpperCase()) { case "MYSQL": initializeMySQL(); break; case "JSON": initializeJSON(); break; default: initializeYAML(); break; } loadAllData(); } private void initializeMySQL() { try { String host = plugin.getConfig().getString("mysql.host"); int port = plugin.getConfig().getInt("mysql.port"); String database = plugin.getConfig().getString("mysql.database"); String username = plugin.getConfig().getString("mysql.username"); String password = plugin.getConfig().getString("mysql.password"); String url = String.format("jdbc:mysql://%s:%d/%s", host, port, database); mysqlConnection = DriverManager.getConnection(url, username, password); Statement statement = mysqlConnection.createStatement(); statement.executeUpdate( "CREATE TABLE IF NOT EXISTS skins (" + "uuid VARCHAR(36) PRIMARY KEY," + "value TEXT NOT NULL," + "signature TEXT NOT NULL," + "timestamp BIGINT NOT NULL" + ")" ); statement.close(); plugin.getLogger().info("MySQL storage initialized!"); } catch (SQLException e) { plugin.getLogger().severe("Failed to initialize MySQL storage!"); e.printStackTrace(); initializeYAML(); } } private void initializeYAML() { storageFile = new File(plugin.getDataFolder(), "skins.yml"); if (!storageFile.exists()) { try { storageFile.createNewFile(); } catch (Exception e) { e.printStackTrace(); } } plugin.getLogger().info("YAML storage initialized!"); } private void initializeJSON() { storageFile = new File(plugin.getDataFolder(), "skins.json"); if (!storageFile.exists()) { try { storageFile.createNewFile(); FileWriter writer = new FileWriter(storageFile); writer.write("{}"); writer.close(); } catch (Exception e) { e.printStackTrace(); } } plugin.getLogger().info("JSON storage initialized!"); } private void loadAllData() { switch (storageType.toUpperCase()) { case "MYSQL": loadFromMySQL(); break; case "JSON": loadFromJSON(); break; default: loadFromYAML(); break; } } private void loadFromMySQL() { try { Statement statement = mysqlConnection.createStatement(); ResultSet rs = statement.executeQuery("SELECT * FROM skins"); while (rs.next()) { UUID uuid = UUID.fromString(rs.getString("uuid")); String value = rs.getString("value"); String signature = rs.getString("signature"); long timestamp = rs.getLong("timestamp"); dataMap.put(uuid, new SkinData(value, signature, timestamp)); } rs.close(); statement.close(); plugin.getLogger().info("Loaded " + dataMap.size() + " skins from MySQL!"); } catch (SQLException e) { e.printStackTrace(); } } private void loadFromYAML() { YamlConfiguration config = YamlConfiguration.loadConfiguration(storageFile); for (String key : config.getKeys(false)) { try { UUID uuid = UUID.fromString(key); String value = config.getString(key + ".value"); String signature = config.getString(key + ".signature"); long timestamp = config.getLong(key + ".timestamp"); dataMap.put(uuid, new SkinData(value, signature, timestamp)); } catch (Exception e) { plugin.getLogger().warning("Failed to load skin for " + key); } } plugin.getLogger().info("Loaded " + dataMap.size() + " skins from YAML!"); } private void loadFromJSON() { try { FileReader reader = new FileReader(storageFile); Map> jsonMap = gson.fromJson( reader, new TypeToken>>(){}.getType() ); reader.close(); if (jsonMap != null) { for (Map.Entry> entry : jsonMap.entrySet()) { UUID uuid = UUID.fromString(entry.getKey()); Map data = entry.getValue(); String value = (String) data.get("value"); String signature = (String) data.get("signature"); long timestamp = ((Double) data.get("timestamp")).longValue(); dataMap.put(uuid, new SkinData(value, signature, timestamp)); } } plugin.getLogger().info("Loaded " + dataMap.size() + " skins from JSON!"); } catch (Exception e) { e.printStackTrace(); } } public void saveSkin(UUID uuid, SkinData skinData) { dataMap.put(uuid, skinData); switch (storageType.toUpperCase()) { case "MYSQL": saveToMySQL(uuid, skinData); break; case "JSON": saveToJSON(); break; default: saveToYAML(); break; } } private void saveToMySQL(UUID uuid, SkinData skinData) { try { PreparedStatement statement = mysqlConnection.prepareStatement( "INSERT INTO skins (uuid, value, signature, timestamp) VALUES (?, ?, ?, ?) " + "ON DUPLICATE KEY UPDATE value = ?, signature = ?, timestamp = ?" ); statement.setString(1, uuid.toString()); statement.setString(2, skinData.getValue()); statement.setString(3, skinData.getSignature()); statement.setLong(4, skinData.getTimestamp()); statement.setString(5, skinData.getValue()); statement.setString(6, skinData.getSignature()); statement.setLong(7, skinData.getTimestamp()); statement.executeUpdate(); statement.close(); } catch (SQLException e) { e.printStackTrace(); } } private void saveToYAML() { YamlConfiguration config = new YamlConfiguration(); for (Map.Entry entry : dataMap.entrySet()) { String key = entry.getKey().toString(); SkinData data = entry.getValue(); config.set(key + ".value", data.getValue()); config.set(key + ".signature", data.getSignature()); config.set(key + ".timestamp", data.getTimestamp()); } try { config.save(storageFile); } catch (Exception e) { e.printStackTrace(); } } private void saveToJSON() { Map> jsonMap = new HashMap<>(); for (Map.Entry entry : dataMap.entrySet()) { Map data = new HashMap<>(); data.put("value", entry.getValue().getValue()); data.put("signature", entry.getValue().getSignature()); data.put("timestamp", entry.getValue().getTimestamp()); jsonMap.put(entry.getKey().toString(), data); } try { FileWriter writer = new FileWriter(storageFile); gson.toJson(jsonMap, writer); writer.close(); } catch (Exception e) { e.printStackTrace(); } } public SkinData getSkin(UUID uuid) { return dataMap.get(uuid); } public void removeSkin(UUID uuid) { dataMap.remove(uuid); switch (storageType.toUpperCase()) { case "MYSQL": removeFromMySQL(uuid); break; case "JSON": saveToJSON(); break; default: saveToYAML(); break; } } private void removeFromMySQL(UUID uuid) { try { PreparedStatement statement = mysqlConnection.prepareStatement( "DELETE FROM skins WHERE uuid = ?" ); statement.setString(1, uuid.toString()); statement.executeUpdate(); statement.close(); } catch (SQLException e) { e.printStackTrace(); } } public void close() { if (mysqlConnection != null) { try { mysqlConnection.close(); } catch (SQLException e) { e.printStackTrace(); } } } }