forked from vercel/next.js
/
request.ts
114 lines (96 loc) · 2.55 KB
/
request.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import type { I18NConfig } from '../../config-shared'
import type { IResult } from 'next/dist/compiled/ua-parser-js'
import { NextURL } from '../next-url'
import { isBot } from '../../utils'
import { toNodeHeaders } from '../utils'
import cookie from 'next/dist/compiled/cookie'
import parseua from 'next/dist/compiled/ua-parser-js'
export const INTERNALS = Symbol('internal request')
export class NextRequest extends Request {
[INTERNALS]: {
cookieParser(): { [key: string]: string }
geo: { city?: string; country?: string; region?: string }
ip?: string
page?: { name?: string; params?: { [key: string]: string } }
ua?: UserAgent | null
url: NextURL
}
constructor(input: Request | string, init: RequestInit = {}) {
super(input, init)
const cookieParser = () => {
const value = this.headers.get('cookie')
return value ? cookie.parse(value) : {}
}
this[INTERNALS] = {
cookieParser,
geo: init.geo || {},
ip: init.ip,
page: init.page,
url: new NextURL(typeof input === 'string' ? input : input.url, {
basePath: init.nextConfig?.basePath,
headers: toNodeHeaders(this.headers),
i18n: init.nextConfig?.i18n,
trailingSlash: init.nextConfig?.trailingSlash,
}),
}
}
public get cookies() {
return this[INTERNALS].cookieParser()
}
public get geo() {
return this[INTERNALS].geo
}
public get ip() {
return this[INTERNALS].ip
}
public get preflight() {
return this.headers.get('x-middleware-preflight')
}
public get nextUrl() {
return this[INTERNALS].url
}
public get page() {
return {
name: this[INTERNALS].page?.name,
params: this[INTERNALS].page?.params,
}
}
public get ua() {
if (typeof this[INTERNALS].ua !== 'undefined') {
return this[INTERNALS].ua || undefined
}
const uaString = this.headers.get('user-agent')
if (!uaString) {
this[INTERNALS].ua = null
return this[INTERNALS].ua || undefined
}
this[INTERNALS].ua = {
...parseua(uaString),
isBot: isBot(uaString),
}
return this[INTERNALS].ua
}
public get url() {
return this[INTERNALS].url.toString()
}
}
export interface RequestInit extends globalThis.RequestInit {
geo?: {
city?: string
country?: string
region?: string
}
ip?: string
nextConfig?: {
basePath?: string
i18n?: I18NConfig | null
trailingSlash?: boolean
}
page?: {
name?: string
params?: { [key: string]: string }
}
}
interface UserAgent extends IResult {
isBot: boolean
}