aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/io/github/moulberry/notenoughupdates/util/HastebinUploader.java
blob: 9cbad4027773b037cc2365aaf0cc0c2b5b880b2b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package io.github.moulberry.notenoughupdates.util;

import com.google.gson.Gson;
import com.google.gson.JsonObject;
import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
import org.apache.commons.io.Charsets;
import org.apache.commons.io.IOUtils;

import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;

/**
 * @author ThatGravyBoat
 */
public class HastebinUploader {
	private static final String UPLOAD_URL = "https://hst.sh/documents";
	private static final String RETURN_URL = "https://hst.sh/";
	private static final String RAW_RETURN_URL = "https://hst.sh/raw/";

	private static final Gson GSON = new Gson();

	/**
	 * @param data the data you want to upload
	 * @param mode the mode in which the thing should return NORMAL = returns the url, RAW = returns the raw url, NO_URL = returns the slug.
	 * @return The url if mode is NORMAL OR RAW, the slug if mode is NO_URL, and null if an error occurred.
	 */
	public static String upload(String data, Mode mode) {
		try {
			byte[] bytes = data.getBytes(StandardCharsets.UTF_8);
			HttpURLConnection connection = (HttpURLConnection) new URL(UPLOAD_URL).openConnection();
			connection.setRequestMethod("POST");
			connection.addRequestProperty(
				"User-Agent",
				"Minecraft Mod (" + NotEnoughUpdates.MODID + "/" + NotEnoughUpdates.VERSION + ")"
			);
			connection.addRequestProperty("Content-Length", String.valueOf(bytes.length));
			connection.setReadTimeout(15000);
			connection.setConnectTimeout(15000);
			connection.setDoOutput(true);
			connection.getOutputStream().write(bytes);
			final JsonObject json =
				GSON.fromJson(IOUtils.toString(connection.getInputStream(), Charsets.UTF_8), JsonObject.class);
			if (!json.has("key")) return null;
			final String key = json.get("key").getAsString();
			switch (mode) {
				case RAW:
					return RAW_RETURN_URL + key;
				case NORMAL:
					return RETURN_URL + key;
				case NO_URL:
					return key;
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;
	}

	public enum Mode {
		NORMAL,
		RAW,
		NO_URL
	}
}