From 40e20c381bde0dbeb749bb2719573b7c09ddc74c Mon Sep 17 00:00:00 2001
From: ThePotioner <heavycaffeiner@gmail.com>
Date: Thu, 12 Jun 2025 19:55:33 +0900
Subject: [PATCH] Add: writeFile() func

---
 src/main/java/dev/ptnr/AyayaUtils.java | 12 ++++++++++++
 src/main/java/dev/ptnr/Hitomi.java     | 23 +++++++++++++++--------
 2 files changed, 27 insertions(+), 8 deletions(-)

diff --git a/src/main/java/dev/ptnr/AyayaUtils.java b/src/main/java/dev/ptnr/AyayaUtils.java
index 073ca1e..636e15d 100644
--- a/src/main/java/dev/ptnr/AyayaUtils.java
+++ b/src/main/java/dev/ptnr/AyayaUtils.java
@@ -50,4 +50,16 @@ public class AyayaUtils {
             return null;
         }
     }
+
+    public static boolean writeFile(String location, byte[] data) {
+        File file = new File(location);
+        try (FileOutputStream fos = new FileOutputStream(file)) {
+            fos.write(data);
+            fos.close();
+        } catch (IOException e) {
+            System.out.println("[ERR > AyayaUtils.writeFile()] Failed to Convert Webp\n" + e.getMessage());
+            return false;
+        }
+        return true;
+    }
 }
diff --git a/src/main/java/dev/ptnr/Hitomi.java b/src/main/java/dev/ptnr/Hitomi.java
index 610f4fa..66f6736 100644
--- a/src/main/java/dev/ptnr/Hitomi.java
+++ b/src/main/java/dev/ptnr/Hitomi.java
@@ -1,6 +1,7 @@
 package dev.ptnr;
 
-import java.io.UnsupportedEncodingException;
+import java.io.File;
+import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.Map;
@@ -34,12 +35,7 @@ public class Hitomi {
             return null;
         }
         
-        try {
-            rawJson = new String(rawData, "UTF-8").replace("var galleryinfo = ", "");
-        } catch (UnsupportedEncodingException e) {
-            System.out.println("[ERR > Hitomi.GetGalleryData()] Failed to Convert JSON");
-            return null;
-        }
+        rawJson = new String(rawData, StandardCharsets.UTF_8).replace("var galleryinfo = ", "");
 
         try {
             ObjectMapper objectMapper = new ObjectMapper();
@@ -173,11 +169,22 @@ public class Hitomi {
     }
 
     public static void main(String[] args) {
-        HitomiDTO test = GetHitomiData(2997822);
+        HitomiDTO test = GetHitomiData(3193561);
         System.out.println(test.getTagsAsString());
         System.out.println(test.getArtistsAsString());
         System.out.println(test.getGroupsAsString());
         System.out.println(test.getCharactersAsString());
         System.out.println(test.getParodysAsString());
+
+        String downloadPath = "D:\\Hitomi_Test";
+        int idx = 1;
+
+        for (String hash : test.getImageHashList()) {
+            File file = new File(downloadPath + "/" + test.getId());
+            file.mkdirs();
+            byte[] data = AyayaUtils.GetFileFromUrl(getImageUrl(hash));
+            AyayaUtils.writeFile(downloadPath + "/" + test.getId() + "/" + idx + ".webp", data);
+            ++idx;
+        }
     }
 }