heavycaffeiner/ayaya
1
0
Fork 0
mirror of https://github.com/heavycaffeiner/ayaya.git synced 2025-10-28 04:44:56 +09:00
Code Issues Packages Projects Releases Wiki Activity

init (v0.0.1)

Browse Source
This commit is contained in:
Dong Hyun Kim
2025-06-10 12:01:21 +09:00
commit cd9b0aad4f
15 changed files with 748 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
src/main/java/dev/ptnr/AyayaBot.java Normal file
View File

@@ -0,0 +1,100 @@
package dev.ptnr;
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.JDA;
import net.dv8tion.jda.api.JDABuilder;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
import net.dv8tion.jda.api.events.session.ReadyEvent;
import net.dv8tion.jda.api.hooks.ListenerAdapter;
import net.dv8tion.jda.api.interactions.InteractionContextType;
import net.dv8tion.jda.api.interactions.commands.OptionType;
import net.dv8tion.jda.api.interactions.commands.build.Commands;
import net.dv8tion.jda.api.interactions.commands.build.OptionData;
import net.dv8tion.jda.api.requests.restaction.CommandListUpdateAction;
import net.dv8tion.jda.api.utils.FileUpload;
import java.util.List;
public class AyayaBot {
public static void main(String[] args) {
String token = System.getenv("DISCORD_TOKEN");
JDA jda = JDABuilder.createDefault(token)
.addEventListeners(new BotListener())
.build();
try {
jda.awaitReady();
} catch (InterruptedException e) {
System.err.println("[ERR] In AyayaBot.main() > " + e.getMessage());
}
}
}
class BotListener extends ListenerAdapter {
@Override
public void onReady(ReadyEvent event) {
JDA jda = event.getJDA();
CommandListUpdateAction commands = jda.updateCommands();
commands.addCommands(
Commands.slash("hitomi", "Get Image from ID (Hitomi)")
.addOptions(new OptionData(OptionType.INTEGER, "id", "gallery id")
.setRequiredRange(0, 9999999).setRequired(true))
.setContexts(InteractionContextType.GUILD)
);
commands.queue();
List<Guild> guildList = jda.getGuilds();
for (Guild guild : guildList) {
System.out.println("Contain Guild : " + guild.getName());
// System.out.println("ID: " + guild.get);
}
System.out.println(jda.getSelfUser().getName() + " is Ready.");
}
@Override
public void onSlashCommandInteraction(SlashCommandInteractionEvent event) {
if (event.getGuild() == null) return;
switch (event.getName()) {
case "hitomi":
hitomi(event, event.getOption("id").getAsInt());
break;
}
}
public void hitomi(SlashCommandInteractionEvent event, Integer id) {
MessageChannel channel = event.getChannel();
event.reply("Message Sent.").setEphemeral(true).queue();
EmbedBuilder eb = new EmbedBuilder();
eb.setTitle("품번 : " + id);
// eb.setDescription("This image has been added to the bot.");
// eb.addBlankField(false);
byte[] webpData = AyayaUtils.GetFileFromUrl(Hitomi.GetHitomiData(id));
if (webpData == null) {
channel.sendMessage("URL에서 이미지를 가져오지 못했습니다. URL이나 서버 상태를 확인해주세요.").queue();
return;
}
byte[] pngData = AyayaUtils.ConvertWebpToPng(webpData);
if (pngData == null) {
channel.sendMessage("이미지를 PNG로 변환하는 데 실패했습니다. 파일이 올바른 WebP 형식이 아닐 수 있습니다.").queue();
return;
}
FileUpload fileUpload = FileUpload.fromData(pngData, "1.png");
eb.setImage("attachment://" + "1.png");
channel.sendFiles(fileUpload).setEmbeds(eb.build()).queue();
}
}

54
src/main/java/dev/ptnr/AyayaUtils.java Normal file
View File

@@ -0,0 +1,54 @@
package dev.ptnr;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
public class AyayaUtils {
public static byte[] GetFileFromUrl(String url) {
HttpURLConnection conn = null;
try {
URL u = new URL(url);
conn = (HttpURLConnection) u.openConnection();
conn.setRequestMethod("GET");
conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Safari/537.36");
conn.setRequestProperty("Accept", "image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8");
conn.setRequestProperty("Referer", "https://hitomi.la/");
if (conn.getResponseCode() != HttpURLConnection.HTTP_OK) {
return null;
}
try (InputStream in = conn.getInputStream(); ByteArrayOutputStream out = new ByteArrayOutputStream()) {
byte[] buffer = new byte[4096];
int n;
while ((n = in.read(buffer)) != -1) {
out.write(buffer, 0, n);
}
return out.toByteArray();
}
} catch (IOException e) {
e.printStackTrace();
return null;
} finally {
if (conn != null) {
conn.disconnect();
}
}
}
public static byte[] ConvertWebpToPng(byte[] webpBytes) {
try (ByteArrayInputStream bais = new ByteArrayInputStream(webpBytes);
ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
BufferedImage image = ImageIO.read(bais);
if (image == null) return null;
ImageIO.write(image, "png", baos);
return baos.toByteArray();
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
}

135
src/main/java/dev/ptnr/Hitomi.java Normal file
View File

@@ -0,0 +1,135 @@
package dev.ptnr;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.regex.*;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
public class Hitomi {
private static String cdnUrl = ".gold-usergeneratedcontent.net";
private static String baseCDNUrl = "https://ltn" + cdnUrl;
private static Map<String, Boolean> getGalleryData(Integer galleryId) throws Exception {
Map<String, Boolean> galleryData = new LinkedHashMap<>();
String baseUrl = baseCDNUrl + "/galleries/" + galleryId + ".js";
String rawData = new String(AyayaUtils.GetFileFromUrl(baseUrl));
if (rawData.length() == 3) {
throw new Exception("Failed to Get JS (code " + rawData + ")");
}
rawData = rawData.replace("var galleryinfo = ", "");
ObjectMapper objectMapper = new ObjectMapper();
try {
JsonNode rootNode = objectMapper.readTree(rawData);
JsonNode files = rootNode.get("files");
for (JsonNode node : files) {
String fileHash = node.get("hash").asText();
Boolean hasAvif = node.get("hasavif").asBoolean();
galleryData.put(fileHash, hasAvif);
}
} catch (JsonProcessingException e) {
System.out.println("[ERR] In Hitomi.getGalleryData() > " + e.getMessage());
throw e;
}
return galleryData;
}
private static String getIdFromGG(String regex, String data) {
Matcher matcher = Pattern.compile(regex).matcher(data);
String regexData = "";
matcher.find();
regexData = matcher.group(1);
return regexData;
}
private static Map<String, Integer> getOffsetsFromGG(String regex, String data, Integer key) {
Matcher matcher = Pattern.compile(regex).matcher(data);
Map<String, Integer> offsets = new HashMap<>();
while (matcher.find()) {
offsets.put(matcher.group(1), key);
}
return offsets;
}
private static String getGGJS() {
String baseUrl = baseCDNUrl + "/gg.js";
try {
return new String(AyayaUtils.GetFileFromUrl(baseUrl));
} catch (Exception e) {
System.out.println("[ERR] In Hitomi.GetGGJS() > " + e.getMessage());
return "";
}
}
private static String getImageUrl(String rawGG, String hash, Boolean hasAvif) {
String DefaultSubdomainKey = getIdFromGG("var o = (\\d)", rawGG);
String SubdomainKey = getIdFromGG("o = (\\d); break;", rawGG);
String CommonImagePath = getIdFromGG("b: '(.+)'", rawGG);
Map<String, Integer> Offsets = getOffsetsFromGG("case (\\d+):", rawGG, Integer.parseInt(SubdomainKey));
String ImageUrl = "";
Integer SubdomainOffset = Integer.parseInt(DefaultSubdomainKey) + 1;
// ex) hash == 415e8c1bfbac7cb6192b6f7c14768c196dde9029233126ca5f108d6aa99818de
// 8de to slice hash[-1:] / hash[-3:-1] (e / 8d)
// -> s == hash[-1:] + hash[-3:-1] == "e8d"
String SubKey = hash.substring(hash.length() - 1, hash.length()) + hash.substring(hash.length() - 3, hash.length() - 1);
// s to parseInt(s, 16) == 3725 == ImageId
Integer ImageId = Integer.parseInt(SubKey, 16);
if (Offsets.containsKey(ImageId.toString())) SubdomainOffset = Integer.parseInt(SubdomainKey) + 1;
// a{key}.gold-usergeneratedcontent.net/1749488401/3725/415e8c1bfbac7cb6192b6f7c14768c196dde9029233126ca5f108d6aa99818de.avif
// domain number (a1, a2 / w1, w2..)
// webp = w1, w2 / avif = a1, a2
// in gg.js case -> o(SubdomainKey) + 1 == 1, out of ggjs case -> o(DefaultSubdomainKey) + 1 == 2
// hasavif flag 1 -> a1, a2 / flag 0 -> w1, w2
// if (hasAvif) ImageUrl = "https://a" + SubdomainOffset + cdnUrl + "/" + CommonImagePath + ImageId + "/" + hash + ".avif";
// else ImageUrl = "https://w" + SubdomainOffset + cdnUrl + "/" + CommonImagePath + ImageId + "/" + hash + ".webp";
ImageUrl = "https://w" + SubdomainOffset + cdnUrl + "/" + CommonImagePath + ImageId + "/" + hash + ".webp";
// System.out.println("ImageUrl: " + ImageUrl);
return ImageUrl;
}
public static String GetHitomiData(Integer galleryId) {
Map<String, Boolean> galleryData = new LinkedHashMap<>();
ArrayList<String> images = new ArrayList<>();
try {
galleryData = getGalleryData(galleryId);
} catch (Exception e) {
System.err.println("In Hitomi.main() > " + e.getMessage());
return "";
}
String GGJS = getGGJS();
galleryData.forEach((hash, hasAvif) -> {
images.add(getImageUrl(GGJS, hash, hasAvif));
});
return images.get(0);
}
public static void main(String[] args) {
// GetHitomiData(3390541);
}
}