package com.vpsbot.manager; import com.google.gson.Gson; import com.google.gson.JsonArray; import com.google.gson.JsonObject; import com.vpsbot.VPSDeployBot; import com.vpsbot.models.Droplet; import lombok.Getter; import okhttp3.*; import java.io.IOException; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; @Getter public class VPSManager { private final VPSDeployBot bot; private final OkHttpClient httpClient; private final Gson gson; private final Map userCooldowns; private final Map> userDroplets; private static final String API_BASE = "https://api.digitalocean.com/v2"; private static final MediaType JSON = MediaType.get("application/json; charset=utf-8"); public VPSManager(VPSDeployBot bot) { this.bot = bot; this.httpClient = new OkHttpClient(); this.gson = new Gson(); this.userCooldowns = new ConcurrentHashMap<>(); this.userDroplets = new ConcurrentHashMap<>(); } public Droplet createDroplet(String name, String region, String size, String image, List tags, String userId) throws IOException { JsonObject dropletData = new JsonObject(); dropletData.addProperty("name", name); dropletData.addProperty("region", region); dropletData.addProperty("size", size); dropletData.addProperty("image", image); JsonArray tagsArray = new JsonArray(); tags.forEach(tagsArray::add); tagsArray.add("deployed-by-" + userId); dropletData.add("tags", tagsArray); RequestBody body = RequestBody.create(gson.toJson(dropletData), JSON); Request request = new Request.Builder() .url(API_BASE + "/droplets") .post(body) .addHeader("Authorization", "Bearer " + bot.getConfigManager().getDigitalOceanToken()) .addHeader("Content-Type", "application/json") .build(); try (Response response = httpClient.newCall(request).execute()) { if (!response.isSuccessful()) { throw new IOException("Failed to create droplet: " + response.code() + " - " + response.message()); } JsonObject responseJson = gson.fromJson(response.body().string(), JsonObject.class); JsonObject dropletJson = responseJson.getAsJsonObject("droplet"); Droplet droplet = parseDroplet(dropletJson); userDroplets.computeIfAbsent(userId, k -> new ArrayList<>()).add(droplet.getId()); return droplet; } } public List listDroplets(String userId) throws IOException { Request request = new Request.Builder() .url(API_BASE + "/droplets?tag_name=deployed-by-" + userId) .get() .addHeader("Authorization", "Bearer " + bot.getConfigManager().getDigitalOceanToken()) .build(); try (Response response = httpClient.newCall(request).execute()) { if (!response.isSuccessful()) { throw new IOException("Failed to list droplets: " + response.code()); } JsonObject responseJson = gson.fromJson(response.body().string(), JsonObject.class); JsonArray dropletsArray = responseJson.getAsJsonArray("droplets"); List droplets = new ArrayList<>(); for (int i = 0; i < dropletsArray.size(); i++) { droplets.add(parseDroplet(dropletsArray.get(i).getAsJsonObject())); } return droplets; } } public Droplet getDroplet(int dropletId) throws IOException { Request request = new Request.Builder() .url(API_BASE + "/droplets/" + dropletId) .get() .addHeader("Authorization", "Bearer " + bot.getConfigManager().getDigitalOceanToken()) .build(); try (Response response = httpClient.newCall(request).execute()) { if (!response.isSuccessful()) { throw new IOException("Failed to get droplet: " + response.code()); } JsonObject responseJson = gson.fromJson(response.body().string(), JsonObject.class); return parseDroplet(responseJson.getAsJsonObject("droplet")); } } public boolean deleteDroplet(int dropletId, String userId) throws IOException { List userDropletIds = userDroplets.get(userId); if (userDropletIds == null || !userDropletIds.contains(dropletId)) { return false; } Request request = new Request.Builder() .url(API_BASE + "/droplets/" + dropletId) .delete() .addHeader("Authorization", "Bearer " + bot.getConfigManager().getDigitalOceanToken()) .build(); try (Response response = httpClient.newCall(request).execute()) { if (response.code() == 204) { userDropletIds.remove(Integer.valueOf(dropletId)); return true; } return false; } } public boolean isOnCooldown(String userId) { Long lastDeployment = userCooldowns.get(userId); if (lastDeployment == null) { return false; } long cooldownMs = bot.getConfigManager().getDeploymentCooldown() * 1000L; return System.currentTimeMillis() - lastDeployment < cooldownMs; } public long getCooldownRemaining(String userId) { Long lastDeployment = userCooldowns.get(userId); if (lastDeployment == null) { return 0; } long cooldownMs = bot.getConfigManager().getDeploymentCooldown() * 1000L; long remaining = cooldownMs - (System.currentTimeMillis() - lastDeployment); return Math.max(0, remaining / 1000); } public void setCooldown(String userId) { userCooldowns.put(userId, System.currentTimeMillis()); } public int getUserDropletCount(String userId) { List droplets = userDroplets.get(userId); return droplets != null ? droplets.size() : 0; } private Droplet parseDroplet(JsonObject json) { int id = json.get("id").getAsInt(); String name = json.get("name").getAsString(); String status = json.get("status").getAsString(); JsonObject region = json.getAsJsonObject("region"); String regionName = region.get("slug").getAsString(); JsonObject size = json.getAsJsonObject("size"); String sizeName = size.get("slug").getAsString(); JsonObject image = json.getAsJsonObject("image"); String imageName = image.get("slug").getAsString(); String ipAddress = null; if (json.has("networks")) { JsonObject networks = json.getAsJsonObject("networks"); if (networks.has("v4")) { JsonArray v4Networks = networks.getAsJsonArray("v4"); if (v4Networks.size() > 0) { JsonObject firstNetwork = v4Networks.get(0).getAsJsonObject(); ipAddress = firstNetwork.get("ip_address").getAsString(); } } } return new Droplet(id, name, status, regionName, sizeName, imageName, ipAddress); } }