package com.serverscout.storage; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.google.gson.JsonArray; import com.google.gson.JsonObject; import com.serverscout.ServerScout; import com.serverscout.data.DiscoveredServer; import com.serverscout.data.Category; import net.fabricmc.loader.api.FabricLoader; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Path; import java.util.ArrayList; import java.util.List; public class ServerStorage { private static final Gson GSON = new GsonBuilder().setPrettyPrinting().create(); private static final Path CONFIG_DIR = FabricLoader.getInstance().getConfigDir().resolve("serverscout"); private static final Path SERVERS_FILE = CONFIG_DIR.resolve("discovered_servers.json"); public static final Path FAVICON_DIR = CONFIG_DIR.resolve("favicons"); private final List servers = new ArrayList<>(); private final List categories = new ArrayList<>(); public void load() { try { Files.createDirectories(CONFIG_DIR); Files.createDirectories(FAVICON_DIR); if (Files.exists(SERVERS_FILE)) { String json = Files.readString(SERVERS_FILE); JsonObject root = GSON.fromJson(json, JsonObject.class); if (root.has("categories")) { JsonArray categoriesArray = root.getAsJsonArray("categories"); categories.clear(); for (int i = 0; i < categoriesArray.size(); i++) { JsonObject catObj = categoriesArray.get(i).getAsJsonObject(); String id = catObj.get("id").getAsString(); String displayName = catObj.get("displayName").getAsString(); int color = catObj.get("color").getAsInt(); int sortOrder = catObj.has("sortOrder") ? catObj.get("sortOrder").getAsInt() : 0; categories.add(new Category(id, displayName, color, sortOrder)); } } JsonArray serversArray = root.getAsJsonArray("servers"); servers.clear(); for (int i = 0; i < serversArray.size(); i++) { JsonObject serverObj = serversArray.get(i).getAsJsonObject(); String name = serverObj.get("name").getAsString(); String address = serverObj.get("address").getAsString(); String categoryId = serverObj.has("categoryId") ? serverObj.get("categoryId").getAsString() : "uncategorized"; long lastPinged = serverObj.has("lastPinged") ? serverObj.get("lastPinged").getAsLong() : 0L; boolean favorite = serverObj.has("favorite") && serverObj.get("favorite").getAsBoolean(); servers.add(new DiscoveredServer(name, address, categoryId, lastPinged, favorite)); } ServerScout.LOGGER.info("Loaded {} discovered servers and {} categories", servers.size(), categories.size()); } if (categories.isEmpty()) { initializeDefaultCategories(); } } catch (IOException e) { ServerScout.LOGGER.error("Failed to load discovered servers", e); } } public void save() { try { Files.createDirectories(CONFIG_DIR); JsonObject root = new JsonObject(); JsonArray categoriesArray = new JsonArray(); for (Category category : categories) { JsonObject catObj = new JsonObject(); catObj.addProperty("id", category.getId()); catObj.addProperty("displayName", category.getDisplayName()); catObj.addProperty("color", category.getColor()); catObj.addProperty("sortOrder", category.getSortOrder()); categoriesArray.add(catObj); } root.add("categories", categoriesArray); JsonArray serversArray = new JsonArray(); for (DiscoveredServer server : servers) { JsonObject serverObj = new JsonObject(); serverObj.addProperty("name", server.getName()); serverObj.addProperty("address", server.getAddress()); serverObj.addProperty("categoryId", server.getCategoryId()); serverObj.addProperty("lastPinged", server.getLastPinged()); serverObj.addProperty("favorite", server.isFavorite()); serversArray.add(serverObj); } root.add("servers", serversArray); Files.writeString(SERVERS_FILE, GSON.toJson(root)); } catch (IOException e) { ServerScout.LOGGER.error("Failed to save discovered servers", e); } } public List getServers() { return servers; } public void addServer(DiscoveredServer server) { if (servers.stream().noneMatch(s -> s.getAddress().equals(server.getAddress()))) { servers.add(server); save(); } } public void removeServer(DiscoveredServer server) { servers.remove(server); save(); } public void updateServer(DiscoveredServer server) { save(); } private void initializeDefaultCategories() { categories.add(new Category("favorites", "Favorites", 0xFFD700, 0)); categories.add(new Category("uncategorized", "Uncategorized", 0x808080, 1)); categories.add(new Category("survival", "Survival", 0x4CAF50, 2)); categories.add(new Category("pvp", "PvP", 0xF44336, 3)); categories.add(new Category("creative", "Creative", 0x2196F3, 4)); categories.add(new Category("minigames", "Minigames", 0xFF9800, 5)); categories.add(new Category("modded", "Modded", 0x9C27B0, 6)); categories.add(new Category("prison", "Prison", 0x795548, 7)); save(); } public List getCategories() { return categories; } public Category getCategoryById(String id) { return categories.stream() .filter(c -> c.getId().equals(id)) .findFirst() .orElse(categories.get(0)); } public void addCategory(Category category) { if (categories.stream().noneMatch(c -> c.getId().equals(category.getId()))) { categories.add(category); save(); } } public void removeCategory(Category category) { if (!isBuiltInCategory(category.getId())) { for (DiscoveredServer server : servers) { if (server.getCategoryId().equals(category.getId())) { server.setCategoryId("uncategorized"); } } categories.remove(category); save(); } } public void updateCategory(Category category) { save(); } public List getServersByCategory(String categoryId) { if (categoryId == null || categoryId.equals("all")) { return servers; } return servers.stream() .filter(s -> s.getCategoryId().equals(categoryId)) .toList(); } public String normalizeAddress(String address) { address = address.toLowerCase().trim(); if (!address.contains(":")) { address = address + ":25565"; } return address; } public boolean isBuiltInCategory(String categoryId) { return categoryId.equals("favorites") || categoryId.equals("uncategorized"); } public List getFavoriteServers() { return servers.stream() .filter(DiscoveredServer::isFavorite) .toList(); } public List searchServers(String query) { if (query == null || query.trim().isEmpty()) { return servers; } String lowerQuery = query.toLowerCase().trim(); return servers.stream() .filter(s -> s.getName().toLowerCase().contains(lowerQuery) || s.getAddress().toLowerCase().contains(lowerQuery)) .toList(); } public List getServersByCategoryAndSearch(String categoryId, String searchQuery) { List filtered; if (categoryId == null || categoryId.equals("all")) { filtered = servers; } else if (categoryId.equals("favorites")) { filtered = getFavoriteServers(); } else { filtered = servers.stream() .filter(s -> s.getCategoryId().equals(categoryId)) .toList(); } if (searchQuery != null && !searchQuery.trim().isEmpty()) { String lowerQuery = searchQuery.toLowerCase().trim(); filtered = filtered.stream() .filter(s -> s.getName().toLowerCase().contains(lowerQuery) || s.getAddress().toLowerCase().contains(lowerQuery)) .toList(); } return filtered; } }