123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
// Copyright 2019-2020 Yusuke Sakurai. All rights reserved. MIT license.import { promiseInterrupter } from "./_util.ts";import { BodyReader } from "./server.ts";
const nullBuffer = new Uint8Array(1024);
export function closableBodyReader(r: Deno.Reader): BodyReader {  return {    read: r.read,    async close() {      await readUntilEof(r);    },  };}
export async function readUntilEof(reader: Deno.Reader): Promise<number> {  let total = 0;  while (true) {    const result = await reader.read(nullBuffer);    if (result == null) {      break;    }    total += result;  }  return total;}
const kDefaultReadTimeout = 10000; // 10sec
export function timeoutReader(  r: Deno.Reader,  opts?: {    timeout?: number;    cancel?: Promise<void>;  },): Deno.Reader {  if (!opts) return r;  if (opts.timeout === undefined) {    opts.timeout = kDefaultReadTimeout;  }  let timeoutOrCancel = promiseInterrupter(opts);  return {    async read(p: Uint8Array): Promise<number | null> {      return await timeoutOrCancel(r.read(p));    },  };}
export function streamReader(stream: ReadableStream<Uint8Array>): Deno.Reader {  const reader = stream.getReader();  const chunks: Uint8Array[] = [];
  const read = async (p: Uint8Array): Promise<number | null> => {    const set = (bytes: Uint8Array): number => {      p.set(bytes);      return bytes.byteLength;    };    const chunk = chunks.shift();    if (chunk) {      if (chunk.byteLength <= p.byteLength) {        return set(chunk);      } else {        const ret = set(chunk.subarray(0, p.byteLength));        chunks.unshift(chunk.subarray(p.byteLength));        return ret;      }    }    const { value, done } = await reader.read();    if (done || !value) {      return null;    }    if (value.byteLength <= p.byteLength) {      return set(value);    } else {      const ret = set(value.subarray(0, p.byteLength));      chunks.push(value.subarray(p.byteLength));      return ret;    }  };  return { read };}
export function noopReader(): Deno.Reader {  return {    async read() {      return null;    },  };}