75 lines
2.3 KiB
JavaScript
75 lines
2.3 KiB
JavaScript
"use strict";
|
|
var __defProp = Object.defineProperty;
|
|
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
var __export = (target, all) => {
|
|
for (var name in all)
|
|
__defProp(target, name, { get: all[name], enumerable: true });
|
|
};
|
|
var __copyProps = (to, from, except, desc) => {
|
|
if (from && typeof from === "object" || typeof from === "function") {
|
|
for (let key of __getOwnPropNames(from))
|
|
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
}
|
|
return to;
|
|
};
|
|
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
var sse_exports = {};
|
|
__export(sse_exports, {
|
|
SSEStreamingApi: () => SSEStreamingApi,
|
|
streamSSE: () => streamSSE
|
|
});
|
|
module.exports = __toCommonJS(sse_exports);
|
|
var import_stream = require("../../utils/stream");
|
|
class SSEStreamingApi extends import_stream.StreamingApi {
|
|
constructor(writable, readable) {
|
|
super(writable, readable);
|
|
}
|
|
async writeSSE(message) {
|
|
const data = message.data.split("\n").map((line) => {
|
|
return `data: ${line}`;
|
|
}).join("\n");
|
|
const sseData = [
|
|
message.event && `event: ${message.event}`,
|
|
data,
|
|
message.id && `id: ${message.id}`,
|
|
message.retry && `retry: ${message.retry}`
|
|
].filter(Boolean).join("\n") + "\n\n";
|
|
await this.write(sseData);
|
|
}
|
|
}
|
|
const run = async (stream, cb, onError) => {
|
|
try {
|
|
await cb(stream);
|
|
} catch (e) {
|
|
if (e instanceof Error && onError) {
|
|
await onError(e, stream);
|
|
await stream.writeSSE({
|
|
event: "error",
|
|
data: e.message
|
|
});
|
|
} else {
|
|
console.error(e);
|
|
}
|
|
} finally {
|
|
stream.close();
|
|
}
|
|
};
|
|
const streamSSE = (c, cb, onError) => {
|
|
const { readable, writable } = new TransformStream();
|
|
const stream = new SSEStreamingApi(writable, readable);
|
|
c.header("Transfer-Encoding", "chunked");
|
|
c.header("Content-Type", "text/event-stream");
|
|
c.header("Cache-Control", "no-cache");
|
|
c.header("Connection", "keep-alive");
|
|
run(stream, cb, onError);
|
|
return c.newResponse(stream.responseReadable);
|
|
};
|
|
// Annotate the CommonJS export names for ESM import in node:
|
|
0 && (module.exports = {
|
|
SSEStreamingApi,
|
|
streamSSE
|
|
});
|