blsmedia/bot
2
0
Fork 0
Code Activity

✨ Customizable allowed origins

Browse Source
This commit is contained in:
Baptiste Arnaud
2024-01-17 09:04:07 +01:00
parent b2f8cd44b8
commit 8771def9a1
12 changed files with 151 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
apps/viewer/src/features/chat/api/continueChat.ts
View File

@ -28,7 +28,7 @@ export const continueChat = publicProcedure
})
)
.output(continueChatResponseSchema)
.mutation(async ({ input: { sessionId, message } }) => {
.mutation(async ({ input: { sessionId, message }, ctx: { res, origin } }) => {
const session = await getSession(sessionId)
if (!session) {
@ -49,6 +49,15 @@ export const continueChat = publicProcedure
message: 'Session expired. You need to start a new session.',
})
if (
session?.state.allowedOrigins &&
session.state.allowedOrigins.length > 0
) {
if (origin && session.state.allowedOrigins.includes(origin))
res.setHeader('Access-Control-Allow-Origin', origin)
else res.removeHeader('Access-Control-Allow-Origin')
}
const {
messages,
input,