suimeng
/
honojs-middleware
mirror of https://github.com/honojs/middleware.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(react-renderer): fix overwriting headers when stream is enabled (#419) 

* fix(react-renderer): fix overwriting headers when stream is enabled

* Create changeset

* Add test to set headers with stream enaabled
pull/423/head
yoshikouki 2024-03-17 10:40:58 +09:00 committed by GitHub
parent a5e59c4199
commit 6eaf2a14cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 31 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.changeset/twenty-windows-raise.md 100644
View File

@ -0,0 +1,5 @@
---
'@hono/react-renderer': patch
---
fix: overwriting headers when stream is enabled

18
packages/react-renderer/src/react-renderer.ts
View File

@ -30,15 +30,15 @@ const createRenderer =
React.createElement(RequestContext.Provider, { value: c }, node),
options.readableStreamOptions
)
return c.body(stream, {
headers:
options.stream === true
? {
'Transfer-Encoding': 'chunked',
'Content-Type': 'text/html; charset=UTF-8',
}
: options.stream,
})
if (options.stream === true) {
c.header('Transfer-Encoding', 'chunked')
c.header('Content-Type', 'text/html; charset=UTF-8')
} else {
for (const [key, value] of Object.entries(options.stream)) {
c.header(key, value)
}
}
return c.body(stream)
} else {
const docType =
typeof options?.docType === 'string'

17
packages/react-renderer/test/react-renderer.test.tsx
View File

@ -105,6 +105,23 @@ describe('Basic', () => {
'<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd"><html><body><h1>Hello</h1></body></html>'
)
})
it('Should return as streaming content with headers added in a handler', async () => {
const app = new Hono()
app.use(reactRenderer(({ children }) => <>{children}</>, { stream: true }))
app.get('/', (c) => {
c.header('X-Message-Set', 'Hello')
c.header('X-Message-Append', 'Hello', { append: true })
return c.render(<h1>Hi</h1>, { title: 'Hi' })
})
const res = await app.request('/')
expect(res.status).toBe(200)
expect(res.headers.get('Transfer-Encoding')).toBe('chunked')
expect(res.headers.get('Content-Type')).toBe('text/html; charset=UTF-8')
expect(res.headers.get('X-Message-Set')).toBe('Hello')
expect(res.headers.get('X-Message-Append')).toBe('Hello')
expect(await res.text()).toBe('<h1>Hi</h1>')
})
})
describe('Streaming', () => {