using stream_fetch in App

app/utils/stream.ts

@@ -12,9 +12,55 @@ type ResponseEvent = {
   };
 };
 
+type StreamResponse = {
+  request_id: number;
+  status: number;
+  status_text: string;
+  headers: Record<string, string>;
+};
+
 export function fetch(url: string, options?: RequestInit): Promise<any> {
   if (window.__TAURI__) {
     const { signal, method = "GET", headers = {}, body = [] } = options || {};
+    let unlisten: Function | undefined;
+    let request_id = 0;
+    const ts = new TransformStream();
+    const writer = ts.writable.getWriter();
+
+    const close = () => {
+      unlisten && unlisten();
+      writer.ready.then(() => {
+        try {
+          writer.releaseLock();
+        } catch (e) {
+          console.error(e);
+        }
+        ts.writable.close();
+      });
+    };
+
+    if (signal) {
+      signal.addEventListener("abort", () => close());
+    }
+    // @ts-ignore 2. listen response multi times, and write to Response.body
+    window.__TAURI__.event
+      .listen("stream-response", (e: ResponseEvent) => {
+        const { request_id: rid, chunk, status } = e?.payload || {};
+        if (request_id != rid) {
+          return;
+        }
+        if (chunk) {
+          writer &&
+            writer.ready.then(() => {
+              writer && writer.write(new Uint8Array(chunk));
+            });
+        } else if (status === 0) {
+          // end of body
+          close();
+        }
+      })
+      .then((u: Function) => (unlisten = u));
+
     return window.__TAURI__
       .invoke("stream_fetch", {
         method,
@@ -26,61 +72,11 @@ export function fetch(url: string, options?: RequestInit): Promise<any> {
             ? Array.from(new TextEncoder().encode(body))
             : [],
       })
-      .then(
-        (res: {
-          request_id: number;
-          status: number;
-          status_text: string;
-          headers: Record<string, string>;
-        }) => {
-          const { request_id, status, status_text: statusText, headers } = res;
-          console.log("send request_id", request_id, status, statusText);
-          let unlisten: Function | undefined;
-          const ts = new TransformStream();
-          const writer = ts.writable.getWriter();
-
-          const close = () => {
-            unlisten && unlisten();
-            writer.ready.then(() => {
-              try {
-                writer.releaseLock();
-              } catch (e) {
-                console.error(e);
-              }
-              ts.writable.close();
-            });
-          };
-
-          const response = new Response(ts.readable, {
-            status,
-            statusText,
-            headers,
-          });
-          if (signal) {
-            signal.addEventListener("abort", () => close());
-          }
-          // @ts-ignore 2. listen response multi times, and write to Response.body
-          window.__TAURI__.event
-            .listen("stream-response", (e: ResponseEvent) => {
-              const { id, payload } = e;
-              const { request_id: rid, chunk, status } = payload;
-              if (request_id != rid) {
-                return;
-              }
-              if (chunk) {
-                writer &&
-                  writer.ready.then(() => {
-                    writer && writer.write(new Uint8Array(chunk));
-                  });
-              } else if (status === 0) {
-                // end of body
-                close();
-              }
-            })
-            .then((u: Function) => (unlisten = u));
-          return response;
-        },
-      )
+      .then((res: StreamResponse) => {
+        request_id = res.request_id;
+        const { status, status_text: statusText, headers } = res;
+        return new Response(ts.readable, { status, statusText, headers });
+      })
       .catch((e) => {
         console.error("stream error", e);
         throw e;
@@ -88,7 +84,3 @@ export function fetch(url: string, options?: RequestInit): Promise<any> {
   }
   return window.fetch(url, options);
 }
-
-if (undefined !== window) {
-  window.tauriFetch = fetch;
-}

src-tauri/src/stream.rs

@@ -66,9 +66,7 @@ pub async fn stream_fetch(
   let res = response_future.await;
   let response = match res {
     Ok(res) => {
-      println!("Error: {:?}", res);
       // get response and emit to client
-      // .register_uri_scheme_protocol("stream", move |app_handle, request| {
       let mut headers = HashMap::new();
       for (name, value) in res.headers() {
         headers.insert(
@@ -84,7 +82,7 @@ pub async fn stream_fetch(
         while let Some(chunk) = stream.next().await {
           match chunk {
             Ok(bytes) => {
-              println!("chunk: {:?}", bytes);
+              // println!("chunk: {:?}", bytes);
               window.emit(event_name, ChunkPayload{ request_id, chunk: bytes }).unwrap();
             }
             Err(err) => {