-
-
Save itIsMaku/5e4edef950e0b83374f79c9675444866 to your computer and use it in GitHub Desktop.
Java DiscordWebhook class to easily execute Discord Webhooks - Lombok and Code Cleanup
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import lombok.AllArgsConstructor; | |
import lombok.Data; | |
import lombok.Getter; | |
import lombok.RequiredArgsConstructor; | |
import javax.net.ssl.HttpsURLConnection; | |
import java.awt.*; | |
import java.io.IOException; | |
import java.io.OutputStream; | |
import java.lang.reflect.Array; | |
import java.net.URL; | |
import java.util.List; | |
import java.util.*; | |
@RequiredArgsConstructor | |
@Data | |
public class DiscordWebhook { | |
private final String url; | |
private String content; | |
private String username; | |
private String avatarUrl; | |
private boolean tts; | |
private List<EmbedObject> embeds = new ArrayList<>(); | |
public void addEmbed(EmbedObject embed) { | |
this.embeds.add(embed); | |
} | |
public void execute() throws IOException { | |
if (content == null && embeds.isEmpty()) { | |
throw new IllegalArgumentException("Set content or add at least one EmbedObject"); | |
} | |
JSONObject json = new JSONObject(); | |
json.put("content", content); | |
json.put("username", username); | |
json.put("avatar_url", avatarUrl); | |
json.put("tts", tts); | |
if (!embeds.isEmpty()) { | |
List<JSONObject> embedObjects = new ArrayList<>(); | |
for (EmbedObject embed : embeds) { | |
JSONObject jsonEmbed = new JSONObject(); | |
jsonEmbed.put("title", embed.getTitle()); | |
jsonEmbed.put("description", embed.getDescription()); | |
jsonEmbed.put("url", embed.getUrl()); | |
if (embed.getColor() != null) { | |
Color color = embed.getColor(); | |
int rgb = color.getRed(); | |
rgb = (rgb << 8) + color.getGreen(); | |
rgb = (rgb << 8) + color.getBlue(); | |
jsonEmbed.put("color", rgb); | |
} | |
EmbedObject.Footer footer = embed.getFooter(); | |
EmbedObject.Image image = embed.getImage(); | |
EmbedObject.Thumbnail thumbnail = embed.getThumbnail(); | |
EmbedObject.Author author = embed.getAuthor(); | |
List<EmbedObject.Field> fields = embed.getFields(); | |
if (footer != null) { | |
JSONObject jsonFooter = new JSONObject(); | |
jsonFooter.put("text", footer.getText()); | |
jsonFooter.put("icon_url", footer.getIconUrl()); | |
jsonEmbed.put("footer", jsonFooter); | |
} | |
if (image != null) { | |
JSONObject jsonImage = new JSONObject(); | |
jsonImage.put("url", image.getUrl()); | |
jsonEmbed.put("image", jsonImage); | |
} | |
if (thumbnail != null) { | |
JSONObject jsonThumbnail = new JSONObject(); | |
jsonThumbnail.put("url", thumbnail.getUrl()); | |
jsonEmbed.put("thumbnail", jsonThumbnail); | |
} | |
if (author != null) { | |
JSONObject jsonAuthor = new JSONObject(); | |
jsonAuthor.put("name", author.getName()); | |
jsonAuthor.put("url", author.getUrl()); | |
jsonAuthor.put("icon_url", author.getIconUrl()); | |
jsonEmbed.put("author", jsonAuthor); | |
} | |
List<JSONObject> jsonFields = new ArrayList<>(); | |
for (EmbedObject.Field field : fields) { | |
JSONObject jsonField = new JSONObject(); | |
jsonField.put("name", field.getName()); | |
jsonField.put("value", field.getValue()); | |
jsonField.put("inline", field.isInline()); | |
jsonFields.add(jsonField); | |
} | |
jsonEmbed.put("fields", jsonFields.toArray()); | |
embedObjects.add(jsonEmbed); | |
} | |
json.put("embeds", embedObjects.toArray()); | |
} | |
URL url = new URL(this.url); | |
HttpsURLConnection connection = (HttpsURLConnection) url.openConnection(); | |
connection.addRequestProperty("Content-Type", "application/json"); | |
connection.addRequestProperty("User-Agent", "Java-DiscordWebhook-BY-Gelox_"); | |
connection.setDoOutput(true); | |
connection.setRequestMethod("POST"); | |
OutputStream stream = connection.getOutputStream(); | |
stream.write(json.toString().getBytes()); | |
stream.flush(); | |
stream.close(); | |
connection.getInputStream().close(); | |
connection.disconnect(); | |
} | |
@Data | |
public static class EmbedObject { | |
private final List<Field> fields = new ArrayList<>(); | |
private String title; | |
private String description; | |
private String url; | |
private Color color; | |
private Footer footer; | |
private Thumbnail thumbnail; | |
private Image image; | |
private Author author; | |
public EmbedObject setThumbnail(String url) { | |
this.thumbnail = new Thumbnail(url); | |
return this; | |
} | |
public EmbedObject setImage(String url) { | |
this.image = new Image(url); | |
return this; | |
} | |
public EmbedObject setFooter(String text, String icon) { | |
this.footer = new Footer(text, icon); | |
return this; | |
} | |
public EmbedObject setAuthor(String name, String url, String icon) { | |
this.author = new Author(name, url, icon); | |
return this; | |
} | |
public EmbedObject addField(String name, String value, boolean inline) { | |
this.fields.add(new Field(name, value, inline)); | |
return this; | |
} | |
@AllArgsConstructor | |
@Getter | |
private static class Image { | |
private final String url; | |
} | |
@AllArgsConstructor | |
@Getter | |
private static class Author { | |
private final String name; | |
private final String url; | |
private final String iconUrl; | |
} | |
@AllArgsConstructor | |
@Getter | |
private static class Field { | |
private final String name; | |
private final String value; | |
private final boolean inline; | |
} | |
@AllArgsConstructor | |
@Getter | |
private static class Footer { | |
private final String text; | |
private final String iconUrl; | |
} | |
@AllArgsConstructor | |
@Getter | |
private static class Thumbnail { | |
private final String url; | |
} | |
} | |
private static class JSONObject { | |
private final HashMap<String, Object> map = new HashMap<>(); | |
void put(String key, Object value) { | |
if (value != null) { | |
map.put(key, value); | |
} | |
} | |
@Override | |
public String toString() { | |
StringBuilder builder = new StringBuilder(); | |
Set<Map.Entry<String, Object>> entrySet = map.entrySet(); | |
builder.append("{"); | |
int i = 0; | |
for (Map.Entry<String, Object> entry : entrySet) { | |
Object val = entry.getValue(); | |
builder.append(quote(entry.getKey())).append(":"); | |
if (val instanceof String) { | |
builder.append(quote(String.valueOf(val))); | |
} else if (val instanceof Integer) { | |
builder.append(Integer.valueOf(String.valueOf(val))); | |
} else if (val instanceof Boolean) { | |
builder.append(val); | |
} else if (val instanceof JSONObject) { | |
builder.append(val); | |
} else if (val.getClass().isArray()) { | |
builder.append("["); | |
int len = Array.getLength(val); | |
for (int j = 0; j < len; j++) { | |
builder.append(Array.get(val, j).toString()).append(j != len - 1 ? "," : ""); | |
} | |
builder.append("]"); | |
} | |
builder.append(++i == entrySet.size() ? "}" : ","); | |
} | |
return builder.toString(); | |
} | |
private String quote(String string) { | |
return "\"" + string + "\""; | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment