Function fetch

src/utils/tauri-stream.ts:29–108

(url: string, options?: RequestInit)

};

export function fetch(url: string, options?: RequestInit): Promise<Response> {
  if (!IsTauri) return window.fetch(url, options)
  const {
    signal,
    method = 'GET',
    headers: _headers = {},
    body = []
  } = options || {}
  let unlisten: Function | undefined
  let setRequestId: Function | undefined
  const requestIdPromise = new Promise((resolve) => (setRequestId = resolve))
  const ts = new TransformStream()
  const writer = ts.writable.getWriter()

  let closed = false
  const close = () => {
    if (closed) return
    closed = true
    unlisten && unlisten()
    writer.ready.then(() => {
      writer.close().catch((e) => console.error(e))
    })
  }

  if (signal) {
    signal.addEventListener('abort', () => close())
  }
  // @ts-ignore 2. listen response multi times, and write to Response.body
  listen('stream-response', (e: ResponseEvent) =>
    requestIdPromise.then((request_id) => {
      const { request_id: rid, chunk, status } = e?.payload || {}
      if (request_id != rid) {
        return
      }
      if (chunk) {
        writer.ready.then(() => {
          writer.write(new Uint8Array(chunk))
        })
      } else if (status === 0) {
        // end of body
        close()
      }
    })
  ).then((u: Function) => (unlisten = u))

  const headers = new Headers({
    Accept: 'application/json, text/plain, */*',
    'Accept-Language': 'en-US,en;q=0.9,zh-CN;q=0.8,zh;q=0.7',
    'User-Agent': navigator.userAgent
  })
  for (const [key, value] of new Headers(_headers || {})) {
    headers.set(key, value)
  }
  return invoke('stream_fetch', {
    method: method.toUpperCase(),
    url,
    headers: Object.fromEntries(headers.entries()),
    // TODO FormData