package dev.ninja.lemonac.utils; import javax.net.ssl.HttpsURLConnection; import java.awt.Color; import java.io.IOException; import java.io.OutputStream; import java.net.URL; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; public class DiscordWebhook { private final String url; private String content; private String username; private String avatarUrl; private final List embeds = new ArrayList<>(); public DiscordWebhook(String url) { this.url = url; } public void setContent(String content) { this.content = content; } public void setUsername(String username) { this.username = username; } public void setAvatarUrl(String avatarUrl) { this.avatarUrl = avatarUrl; } public void addEmbed(EmbedObject embed) { this.embeds.add(embed); } public void execute() throws IOException { if (this.content == null && this.embeds.isEmpty()) { throw new IllegalArgumentException("Content or embeds must be set"); } Map json = new HashMap<>(); if (this.content != null) { json.put("content", this.content); } if (this.username != null) { json.put("username", this.username); } if (this.avatarUrl != null) { json.put("avatar_url", this.avatarUrl); } if (!this.embeds.isEmpty()) { List> embedObjects = new ArrayList<>(); for (EmbedObject embed : this.embeds) { embedObjects.add(embed.toJson()); } json.put("embeds", embedObjects); } URL url = new URL(this.url); HttpsURLConnection connection = (HttpsURLConnection) url.openConnection(); connection.addRequestProperty("Content-Type", "application/json"); connection.addRequestProperty("User-Agent", "LemonAC-Webhook"); connection.setDoOutput(true); connection.setRequestMethod("POST"); OutputStream stream = connection.getOutputStream(); stream.write(toJson(json).getBytes()); stream.flush(); stream.close(); connection.getInputStream().close(); connection.disconnect(); } private String toJson(Map map) { StringBuilder json = new StringBuilder("{"); int count = 0; for (Map.Entry entry : map.entrySet()) { json.append("\"").append(entry.getKey()).append("\":"); json.append(valueToJson(entry.getValue())); if (++count < map.size()) { json.append(","); } } json.append("}"); return json.toString(); } private String valueToJson(Object value) { if (value instanceof String) { return "\"" + ((String) value).replace("\"", "\\\"") + "\""; } else if (value instanceof Integer || value instanceof Long || value instanceof Double || value instanceof Boolean) { return String.valueOf(value); } else if (value instanceof Map) { return toJson((Map) value); } else if (value instanceof List) { StringBuilder json = new StringBuilder("["); List list = (List) value; for (int i = 0; i < list.size(); i++) { json.append(valueToJson(list.get(i))); if (i < list.size() - 1) { json.append(","); } } json.append("]"); return json.toString(); } return "null"; } public static class EmbedObject { private String title; private String description; private String url; private Color color; private Footer footer; private final List fields = new ArrayList<>(); public void setTitle(String title) { this.title = title; } public void setDescription(String description) { this.description = description; } public void setUrl(String url) { this.url = url; } public void setColor(Color color) { this.color = color; } public void setColor(int color) { this.color = new Color(color); } public void setFooter(String text, String icon) { this.footer = new Footer(text, icon); } public void addField(String name, String value, boolean inline) { this.fields.add(new Field(name, value, inline)); } private Map toJson() { Map json = new HashMap<>(); if (title != null) json.put("title", title); if (description != null) json.put("description", description); if (url != null) json.put("url", url); if (color != null) json.put("color", color.getRGB() & 0xFFFFFF); if (footer != null) { Map footerJson = new HashMap<>(); footerJson.put("text", footer.text); if (footer.iconUrl != null) footerJson.put("icon_url", footer.iconUrl); json.put("footer", footerJson); } if (!fields.isEmpty()) { List> fieldList = new ArrayList<>(); for (Field field : fields) { Map fieldJson = new HashMap<>(); fieldJson.put("name", field.name); fieldJson.put("value", field.value); fieldJson.put("inline", field.inline); fieldList.add(fieldJson); } json.put("fields", fieldList); } return json; } private static class Footer { private final String text; private final String iconUrl; private Footer(String text, String iconUrl) { this.text = text; this.iconUrl = iconUrl; } } private static class Field { private final String name; private final String value; private final boolean inline; private Field(String name, String value, boolean inline) { this.name = name; this.value = value; this.inline = inline; } } } }