From 45b46debbb8cbc4999226cc7e005ca3a8dcba6e3 Mon Sep 17 00:00:00 2001
From: Steven Vandevelde <steven@tokono.ma>
Date: Thu, 29 May 2025 11:58:36 +0200
Subject: [PATCH] chore: allow passing streams to the metadata processor (not
 sure this actually works)

---
 .../orchestrator/input-cache/_applet.astro    |  2 +-
 .../processor/metadata-fetcher/_applet.astro  | 27 ++++++++++++-------
 .../processor/metadata-fetcher/_manifest.json |  6 +++--
 3 files changed, 22 insertions(+), 13 deletions(-)

diff --git a/src/pages/orchestrator/input-cache/_applet.astro b/src/pages/orchestrator/input-cache/_applet.astro
index c0c3432..1ed7d45 100644
--- a/src/pages/orchestrator/input-cache/_applet.astro
+++ b/src/pages/orchestrator/input-cache/_applet.astro
@@ -1,7 +1,7 @@
 <script>
   import { applets } from "@web-applets/sdk";
 
-  import type { Output, Track, TrackStats, TrackTags } from "@applets/core/types.d.ts";
+  import type { Output, Track } from "@applets/core/types.d.ts";
   import { applet, waitUntilAppletIsReady } from "@scripts/theme";
 
   ////////////////////////////////////////////
diff --git a/src/pages/processor/metadata-fetcher/_applet.astro b/src/pages/processor/metadata-fetcher/_applet.astro
index 638cfed..4c39f95 100644
--- a/src/pages/processor/metadata-fetcher/_applet.astro
+++ b/src/pages/processor/metadata-fetcher/_applet.astro
@@ -21,39 +21,46 @@
   ////////////////////////////////////////////
   context.setActionHandler("extract", extract);
 
-  async function extract({ urls }: { urls: Urls }) {
+  async function extract(args: { mimeType?: string; stream?: ReadableStream; urls?: Urls }) {
     // Construct records
     // TODO: Use other metadata lib as fallback: https://github.com/buzz/mediainfo.js
-    const { stats, tags } = await musicMetadataTags(urls, false);
+    const { stats, tags } = await musicMetadataTags(args, false);
 
     // Fin
     return { stats, tags };
   }
 
   // 🛠️
-  async function musicMetadataTags(urls: Urls, covers: boolean = false): Promise<Extraction> {
-    const uri = URI.parse(urls.get);
-    const pathParts = uri.path?.split("/");
+  async function musicMetadataTags(
+    { mimeType, stream, urls }: { mimeType?: string; stream?: ReadableStream; urls?: Urls },
+    covers: boolean = false,
+  ): Promise<Extraction> {
+    const uri = urls ? URI.parse(urls.get) : undefined;
+    const pathParts = uri?.path?.split("/");
     const filename = pathParts?.[pathParts.length - 1];
 
     let meta;
 
-    if (urls.get.startsWith("blob:")) {
-      const mimeType = filename?.includes(".")
+    if (urls?.get.startsWith("blob:")) {
+      const mimeFallback = filename?.includes(".")
         ? contentType(filename.split(".").reverse()[0])
         : undefined;
 
       const resp = await fetch(urls.get);
       const stream = resp.body;
 
-      meta = await parseWebStream(stream, { mimeType });
-    } else {
+      meta = await parseWebStream(stream, { mimeType: mimeType || mimeFallback });
+    } else if (urls) {
       const httpClient = new HTTP_TOKENIZER.HttpClient(urls.head, { resolveUrl: false });
       httpClient.resolvedUrl = urls.get;
 
       const tokenizer = await RANGE_TOKENIZER.tokenizer(httpClient);
 
       meta = await parseFromTokenizer(tokenizer, { skipCovers: covers });
+    } else if (stream) {
+      meta = await parseWebStream(stream, { mimeType });
+    } else {
+      throw new Error("Missing args, need either some urls or a stream.");
     }
 
     const stats: TrackStats = {
@@ -65,7 +72,7 @@
       artist: meta.common.artist,
       disc: { no: meta.common.disk.no || 1, of: meta.common.disk.of ?? undefined },
       genre: Array.isArray(meta.common.genre) ? meta.common.genre[0] : meta.common.genre,
-      title: meta.common.title || filename || urls.head,
+      title: meta.common.title || filename || urls?.head || "Unknown",
       track: { no: meta.common.track.no || 1, of: meta.common.track.of ?? undefined },
       year: meta.common.year,
     };
diff --git a/src/pages/processor/metadata-fetcher/_manifest.json b/src/pages/processor/metadata-fetcher/_manifest.json
index 891d14b..612918d 100644
--- a/src/pages/processor/metadata-fetcher/_manifest.json
+++ b/src/pages/processor/metadata-fetcher/_manifest.json
@@ -9,6 +9,9 @@
       "params_schema": {
         "type": "object",
         "properties": {
+          "stream": {
+            "type": "object"
+          },
           "urls": {
             "type": "object",
             "properties": {
@@ -17,8 +20,7 @@
             },
             "required": ["get", "head"]
           }
-        },
-        "required": ["urls"]
+        }
       }
     }
   }
-- 
2.51.2