From 6a0152611d09aaca1c48e30d2ff1afb5006eb50c Mon Sep 17 00:00:00 2001
From: Lea <lea@floofball.dev>
Date: Sun, 16 Jun 2024 17:28:24 +0200
Subject: [PATCH] hgberdtzhgij

---
 src/index.ts | 35 ++++++++++++++++++++++++++++-------
 1 file changed, 28 insertions(+), 7 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 75065b3..c0488eb 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,5 +1,5 @@
 import path from "path";
-import { fileURLToPath } from "url";
+import { fileURLToPath, URL } from "url";
 import fs from "fs/promises";
 import axios from "axios";
 import { createWriteStream } from "fs";
@@ -38,6 +38,8 @@ type CommonImages = {
     }
 }
 
+const fsExists = async (path: string) => (await fs.stat(path).catch((e) => false)) !== false;
+
 async function downloadToPath(url: string, path: string) {
     const stream = createWriteStream(path);
     const response = await axios.get(url, { responseType: "stream" });
@@ -50,15 +52,15 @@ async function downloadToPath(url: string, path: string) {
  * unless it is already cached.
  */
 async function fetchScript(makerId: string) {
-    const cacheDir   = path.join(dirname, "..", "cache", makerId);
+    const cacheDir   = path.join(dirname, "..", "cache", "picrew", makerId);
     const htmlPath   = path.join(cacheDir, "index.html");
     const scriptPath = path.join(cacheDir, "script.js");
 
     await fs.mkdir(cacheDir, { recursive: true })
         .catch((e) => { if (e?.code != "EEXIST") throw e; }); // Ignore "already exists" error
 
-    const htmlExists = (await fs.stat(htmlPath).catch((e) => false)) !== false;
-    const scriptExists = (await fs.stat(scriptPath).catch((e) => false)) !== false;
+    const htmlExists = await fsExists(htmlPath);
+    const scriptExists = await fsExists(scriptPath);
 
     if (!htmlExists) {
         await downloadToPath(`https://picrew.me/en/image_maker/${encodeURIComponent(makerId)}`, htmlPath);
@@ -83,7 +85,7 @@ async function fetchScript(makerId: string) {
 }
 
 async function extractDataFromScript(makerId: string) {
-    const code = await fs.readFile(path.join(dirname, "..", "cache", makerId, "script.js"), { encoding: "utf8" });
+    const code = await fs.readFile(path.join(dirname, "..", "cache", "picrew", makerId, "script.js"), { encoding: "utf8" });
     const isolate = new ivm.Isolate();
     const context = await isolate.createContext();
     const script = await isolate.compileScript(code);
@@ -97,6 +99,27 @@ async function extractDataFromScript(makerId: string) {
     };
 }
 
+/**
+ * Download an asset if it's not cached already, then
+ * return it as buffer.
+ */
+async function getAssetBuffer(url: string) {
+    // https://cdn.picrew.me/app/image_maker/{makerId}/{pId}/{random_string}.png
+    const [ makerId, cId, filename ] = new URL(url).pathname.split("/").slice(-3);
+
+    const dir = path.join(dirname, "..", "cache", "assets", makerId, cId);
+    const filePath = path.join(dir, filename);
+
+    await fs.mkdir(dir, { recursive: true })
+        .catch((e) => { if (e?.code != "EEXIST") throw e; }); // Ignore "already exists" error
+
+    if (!await fsExists(filePath)) {
+        await downloadToPath(url, filePath);
+    }
+
+    return await fs.readFile(filePath);
+}
+
 await fetchScript("1904634");
 const { config, commonImages, cdnRoot } = await extractDataFromScript("1904634");
 
@@ -138,5 +161,3 @@ for (const c in config.pList) {
         items.push({ itemId: id.toString(), layers: layers });
     }
 }
-
-console.log(layerColors);