feat(search): lint & beautify, update content type

This commit is contained in:
ItzCrazyKns
2025-03-30 21:12:04 +05:30
parent 5d60ab1139
commit 90e303f737

View File

@ -146,7 +146,10 @@ export const POST = async (req: Request) => {
} }
} catch (error) { } catch (error) {
reject( reject(
Response.json({ message: 'Error parsing data' }, { status: 500 }), Response.json(
{ message: 'Error parsing data' },
{ status: 500 },
),
); );
} }
}); });
@ -157,7 +160,10 @@ export const POST = async (req: Request) => {
emitter.on('error', (error: any) => { emitter.on('error', (error: any) => {
reject( reject(
Response.json({ message: 'Search error', error }, { status: 500 }), Response.json(
{ message: 'Search error', error },
{ status: 500 },
),
); );
}); });
}, },
@ -165,52 +171,56 @@ export const POST = async (req: Request) => {
} }
const encoder = new TextEncoder(); const encoder = new TextEncoder();
// Create an AbortController to handle cancellation
const abortController = new AbortController(); const abortController = new AbortController();
const { signal } = abortController; const { signal } = abortController;
const stream = new ReadableStream({ const stream = new ReadableStream({
start(controller) { start(controller) {
let sources: any[] = []; let sources: any[] = [];
// Send an initial message to keep the connection alive controller.enqueue(
controller.enqueue(encoder.encode(JSON.stringify({ encoder.encode(
type: 'init', JSON.stringify({
data: 'Stream connected' type: 'init',
}) + '\n')); data: 'Stream connected',
}) + '\n',
),
);
// Set up cleanup function for when client disconnects
signal.addEventListener('abort', () => { signal.addEventListener('abort', () => {
// Remove all listeners from emitter to prevent memory leaks
emitter.removeAllListeners(); emitter.removeAllListeners();
// Close the controller if it's still active
try { try {
controller.close(); controller.close();
} catch (error) { } catch (error) {}
// Controller might already be closed
}
}); });
emitter.on('data', (data: string) => { emitter.on('data', (data: string) => {
// Check if request has been cancelled before processing
if (signal.aborted) return; if (signal.aborted) return;
try { try {
const parsedData = JSON.parse(data); const parsedData = JSON.parse(data);
if (parsedData.type === 'response') { if (parsedData.type === 'response') {
controller.enqueue(encoder.encode(JSON.stringify({ controller.enqueue(
type: 'response', encoder.encode(
data: parsedData.data JSON.stringify({
}) + '\n')); type: 'response',
data: parsedData.data,
}) + '\n',
),
);
} else if (parsedData.type === 'sources') { } else if (parsedData.type === 'sources') {
sources = parsedData.data; sources = parsedData.data;
controller.enqueue(encoder.encode(JSON.stringify({ controller.enqueue(
type: 'sources', encoder.encode(
data: sources JSON.stringify({
}) + '\n')); type: 'sources',
data: sources,
}) + '\n',
),
);
} }
} catch (error) { } catch (error) {
controller.error(error); controller.error(error);
@ -218,32 +228,34 @@ export const POST = async (req: Request) => {
}); });
emitter.on('end', () => { emitter.on('end', () => {
// Check if request has been cancelled before processing
if (signal.aborted) return; if (signal.aborted) return;
controller.enqueue(encoder.encode(JSON.stringify({ controller.enqueue(
type: 'done' encoder.encode(
}) + '\n')); JSON.stringify({
type: 'done',
}) + '\n',
),
);
controller.close(); controller.close();
}); });
emitter.on('error', (error: any) => { emitter.on('error', (error: any) => {
// Check if request has been cancelled before processing
if (signal.aborted) return; if (signal.aborted) return;
controller.error(error); controller.error(error);
}); });
}, },
cancel() { cancel() {
abortController.abort(); abortController.abort();
} },
}); });
return new Response(stream, { return new Response(stream, {
headers: { headers: {
'Content-Type': 'application/json', 'Content-Type': 'text/event-stream',
'Cache-Control': 'no-cache, no-transform', 'Cache-Control': 'no-cache, no-transform',
'Connection': 'keep-alive', Connection: 'keep-alive',
}, },
}); });
} catch (err: any) { } catch (err: any) {