123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193
import { Status } from 'https://deno.land/std@0.93.0/http/http_status.ts'import type { BufReader, BufWriter } from 'https://deno.land/std@0.93.0/io/bufio.ts'import type { MultipartFormData } from 'https://deno.land/std@0.93.0/mime/multipart.ts'import { MultipartReader } from 'https://deno.land/std@0.93.0/mime/multipart.ts'import log from '../shared/log.ts'import type { APIRequest, ServerRequest, ServerResponse } from '../types.ts'import compress from './compress.ts'
type Response = {  status: number  headers: Headers  done: boolean}
export class Request implements APIRequest {  #req: ServerRequest  #params: Record<string, string>  #query: URLSearchParams  #cookies: ReadonlyMap<string, string>  #resp: Response
  constructor(req: ServerRequest, params: Record<string, string>, query: URLSearchParams) {    this.#req = req    this.#params = params    this.#query = query    const cookies = new Map()    this.headers.get('cookie')?.split(';').forEach(cookie => {      const p = cookie.split('=')      if (p.length >= 2) {        cookies.set(p.shift()!, decodeURI(p.join('=')))      }    })    this.#cookies = cookies    this.#resp = {      status: 200,      headers: new Headers({        Server: 'Aleph.js',      }),      done: false    }  }
  get url(): string {    return this.#req.url  }
  get method(): string {    return this.#req.method  }
  get hostname(): string {    return (this.#req.conn.remoteAddr as Deno.NetAddr).hostname  }
  get headers(): Headers {    return this.#req.headers  }
  get conn(): Deno.Conn {    return this.#req.conn  }
  get r(): BufReader {    return this.#req.r  }
  get w(): BufWriter {    return this.#req.w  }
  get body(): Deno.Reader {    return this.#req.body  }
  async respond(r: ServerResponse): Promise<void> {    return this.#req.respond(r)  }
  get params(): Record<string, string> {    return this.#params  }
  get query(): URLSearchParams {    return this.#query  }
  get cookies(): ReadonlyMap<string, string> {    return this.#cookies  }
  async readBody(type?: 'raw'): Promise<Uint8Array>  async readBody(type: 'text'): Promise<string>  async readBody(type: 'json'): Promise<any>  async readBody(type: 'form'): Promise<MultipartFormData>  async readBody(type?: string): Promise<any> {    switch (type) {      case 'text': {        const buff: Uint8Array = await Deno.readAll(this.body)        const encoded = new TextDecoder('utf-8').decode(buff)        return encoded      }      case 'json': {        const buff: Uint8Array = await Deno.readAll(this.body)        const encoded = new TextDecoder('utf-8').decode(buff)        const data = JSON.parse(encoded)        return data      }      case 'form': {        const contentType = this.headers.get('content-type') as string        const reader = new MultipartReader(this.body, contentType)        return reader.readForm()      }      default: {        return await Deno.readAll(this.body)      }    }  }
  addHeader(key: string, value: string): this {    this.#resp.headers.append(key, value)    return this  }
  setHeader(key: string, value: string): this {    this.#resp.headers.set(key, value)    return this  }
  removeHeader(key: string): this {    this.#resp.headers.delete(key)    return this  }
  status(code: number): this {    this.#resp.status = code    return this  }
  redirect(url: string, status: Status = Status.Found): this {    // "back" is an alias for the referrer.    if (url == "back") {      url = this.#resp.headers.get("Referrer") || "/"    }    this.#resp.status = status    this.#resp.headers.set("Location", encodeURI(url))    return this  }
  async send(data?: string | Uint8Array | ArrayBuffer, contentType?: string): Promise<void> {    if (this.#resp.done) {      log.warn('ServerRequest: repeat respond calls')      return Promise.resolve()    }
    let body = new Uint8Array()    if (typeof data === 'string') {      body = new TextEncoder().encode(data)    } else if (data instanceof Uint8Array) {      body = data    } else if (data instanceof ArrayBuffer) {      body = new Uint8Array(data)    }    if (contentType) {      this.#resp.headers.set('Content-Type', contentType)    } else if (this.#resp.headers.has('Content-Type')) {      contentType = this.#resp.headers.get('Content-Type')!    } else if (typeof data === 'string' && data.length > 0) {      contentType = 'text/plain; charset=utf-8'      this.#resp.headers.set('Content-Type', contentType)    }    if (contentType) {      body = compress.apply(this.#req, this.#resp, contentType, body)    }    if (!this.#resp.headers.has('Date')) {      this.#resp.headers.set('Date', (new Date).toUTCString())    }    this.#resp.done = true    try {      await this.respond({        status: this.#resp.status,        headers: this.#resp.headers,        body      })    } catch (err) {      log.warn('ServerRequest.respond:', err.message)    }  }
  json(data: any, replacer?: (this: any, key: string, value: any) => any, space?: string | number): Promise<void> {    return this.send(JSON.stringify(data, replacer, space), 'application/json; charset=utf-8')  }}