123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
import type { NextFunction } from './deps.ts'import { App, renderTemplate } from './app.ts'import { Request } from './request.ts'import {  getRequestHeader,  getFreshOrStale,  getAccepts,  getAcceptsCharsets,  getAcceptsEncodings,  getAcceptsLanguages,  checkIfXMLHttpRequest,  getHostname,  getIP,  getIPs,  getProtocol,  getSubdomains,  getRangeFromHeader,  reqIs} from './extensions/req/mod.ts'import {  send,  json,  sendStatus,  setHeader,  setLocationHeader,  end,  sendFile,  getResponseHeader,  append,  setLinksHeader,  setContentType,  formatResponse,  setVaryHeader,  attachment,  download,  setCookie,  clearCookie,  redirect} from './extensions/res/mod.ts'import { getQueryParams } from './utils/parseUrl.ts'import { Response } from './response.ts'
export const extendMiddleware =  <RenderOptions = unknown, Req extends Request = Request, Res extends Response = Response>(app: App) =>  (req: Req, res: Res, next: NextFunction) => {    const { settings } = app
    res.locals = res.locals || Object.create(null)
    // Request extensions    if (settings?.bindAppToReqRes) {      req.app = app      res.app = app    }
    req.query = getQueryParams(req.url)
    req.connection = {      remoteAddress: (req.conn.remoteAddr as Deno.NetAddr).hostname    }
    req.get = getRequestHeader(req)
    req.accepts = getAccepts<Req>(req)    req.acceptsCharsets = getAcceptsCharsets<Req>(req)    req.acceptsEncodings = getAcceptsEncodings<Req>(req)    req.acceptsLanguages = getAcceptsLanguages<Req>(req)
    req.range = getRangeFromHeader(req)    req.xhr = checkIfXMLHttpRequest(req)    req.is = reqIs(req)
    if (settings?.networkExtensions) {      req.protocol = getProtocol<Req>(req)      req.secure = req.protocol === 'https'      req.hostname = getHostname<Req>(req)      req.subdomains = getSubdomains<Req>(req, settings.subdomainOffset)      req.ip = getIP<Req>(req)      req.ips = getIPs<Req>(req)    }
    // Response extensions
    res.end = end(req, res)    res.send = send<Req, Res>(req, res)    res.sendFile = sendFile<Req, Res>(req, res)    res.sendStatus = sendStatus(req, res)    res.json = json<Req, Res>(req, res)    res.setHeader = setHeader<Res>(res)    res.set = setHeader<Res>(res)    res.location = setLocationHeader<Req, Res>(req, res)    res.get = getResponseHeader<Res>(res)    res.append = append<Res>(res)    res.render = renderTemplate<RenderOptions, Res>(res, app)    res.links = setLinksHeader<Res>(res)    res.redirect = redirect<Req, Res, NextFunction>(req, res, next)    res.type = setContentType<Res>(res)    res.format = formatResponse<Req, Res>(req, res, next)    res.vary = setVaryHeader<Res>(res)    res.download = download<Req, Res>(req, res)    res.attachment = attachment<Res>(res)
    res.cookie = setCookie<Req, Res>(req, res)    res.clearCookie = clearCookie<Res>(res)
    Object.defineProperty(req, 'fresh', { get: getFreshOrStale.bind(null, req, res), configurable: true })    req.stale = !req.fresh
    next?.()  }