-
Notifications
You must be signed in to change notification settings - Fork 26.1k
/
_middleware.js
119 lines (103 loc) · 2.95 KB
/
_middleware.js
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
115
116
117
118
119
import { createElement } from 'react'
import { renderToString } from 'react-dom/server.browser'
import { NextResponse } from 'next/server'
export async function middleware(request, ev) {
// eslint-disable-next-line no-undef
const { readable, writable } = new TransformStream()
const url = request.nextUrl
const writer = writable.getWriter()
const encoder = new TextEncoder()
const next = NextResponse.next()
// Header based on query param
if (url.searchParams.get('nested-header') === 'true') {
next.headers.set('x-nested-header', 'valid')
}
// Ensure deep can append to this value
if (url.searchParams.get('append-me') === 'true') {
next.headers.append('x-append-me', 'top')
}
// Ensure deep can append to this value
if (url.searchParams.get('cookie-me') === 'true') {
next.headers.append('set-cookie', 'bar=chocochip')
}
// Sends a header
if (url.pathname === '/responses/header') {
next.headers.set('x-first-header', 'valid')
return next
}
// Streams a basic response
if (url.pathname === '/responses/stream-a-response') {
ev.waitUntil(
(async () => {
writer.write(encoder.encode('this is a streamed '))
writer.write(encoder.encode('response'))
writer.close()
})()
)
return new Response(readable)
}
if (url.pathname === '/responses/bad-status') {
return new Response('Auth required', {
headers: { 'WWW-Authenticate': 'Basic realm="Secure Area"' },
status: 401,
})
}
if (url.pathname === '/responses/stream-long') {
ev.waitUntil(
(async () => {
writer.write(encoder.encode('this is a streamed '.repeat(10)))
await sleep(2000)
writer.write(encoder.encode('after 2 seconds '.repeat(10)))
await sleep(2000)
writer.write(encoder.encode('after 4 seconds '.repeat(10)))
await sleep(2000)
writer.close()
})()
)
return new Response(readable)
}
// Sends response
if (url.pathname === '/responses/send-response') {
return new Response(JSON.stringify({ message: 'hi!' }))
}
// Render React component
if (url.pathname === '/responses/react') {
return new Response(
renderToString(
createElement(
'h1',
{},
'SSR with React! Hello, ' + url.searchParams.get('name')
)
)
)
}
// Stream React component
if (url.pathname === '/responses/react-stream') {
ev.waitUntil(
(async () => {
writer.write(
encoder.encode(
renderToString(createElement('h1', {}, 'I am a stream'))
)
)
await sleep(500)
writer.write(
encoder.encode(
renderToString(createElement('p', {}, 'I am another stream'))
)
)
writer.close()
})()
)
return new Response(readable)
}
return next
}
function sleep(time) {
return new Promise((resolve) => {
setTimeout(() => {
resolve()
}, time)
})
}