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
120
121
|
// https://html.spec.whatwg.org/multipage/server-sent-events.html
import type { EventStreamOptions, EventStream as IEventStream } from "bun";
export function getEventStream() {
class EventStream extends ReadableStream implements IEventStream {
// internal reference to the direct controller.
// we initialize it to a stub that writes to an internal queue.
// this makes it so you can call send() before the stream is started.
#ctrl: ReadableStreamDirectController | null;
// This field is read by `new Response`
$contentType = "text/event-stream";
constructor(opts?: EventStreamOptions) {
var queue: any[] = [];
var started = false;
super({
type: "direct",
pull: controller => {
this.#ctrl = controller;
if (queue.length) {
for (const item of queue) {
controller.write(item);
if (item === null) {
controller.close();
this.#ctrl = null;
return;
}
}
controller.flush();
}
opts?.start?.(this);
started = true;
},
cancel: () => {
if (started) {
opts?.cancel?.(this);
}
this.#ctrl = null;
},
});
this.#ctrl = {
write: buf => queue.push(buf),
flush: () => {},
close: () => {
queue.push(null);
},
} as any;
}
setReconnectionTime(time: number): void {
var ctrl = this.#ctrl!;
if (!ctrl) {
throw new Error("EventStream is closed");
}
ctrl.write("retry:" + time + "\n\n");
}
send(event?: unknown, data?: unknown, id?: number | null | undefined): void {
var ctrl = this.#ctrl;
if (!ctrl) {
throw new Error("EventStream is closed");
}
if (!data) {
data = event;
event = undefined;
} else if (event === "message") {
// According to spec, 'The default event type is "message"'
// This means we can omit this event type.
event = undefined;
}
if (data === undefined) {
throw new TypeError("EventStream.send() requires a data argument");
}
if (id !== undefined) {
this.#writeEventWithId(ctrl, event, data, id);
} else {
this.#writeEvent(ctrl, event, data);
}
ctrl.flush();
}
#writeEvent(ctrl: ReadableStreamDirectController, event: unknown, data: unknown) {
if (event) ctrl.write("event:" + event + "\n");
if (typeof data === "string") {
ctrl.write("data:" + data.replace(/\n/g, "\ndata:") + "\n\n");
} else if ($isTypedArrayView(data) || data instanceof ArrayBuffer) {
// TODO: handle newlines in this buffer
ctrl.write("data:");
ctrl.write(data as BufferSource);
ctrl.write("\n\n");
} else {
ctrl.write("data:" + JSON.stringify(data) + "\n\n");
}
}
#writeEventWithId(ctrl: ReadableStreamDirectController, event: unknown, data: unknown, id: number | null) {
if (event) ctrl.write("event:" + event + "\n");
if (typeof data === "string") {
ctrl.write("data:" + data.replace(/\n/g, "\ndata:") + "\n");
} else if (data instanceof Uint8Array) {
// TODO: handle newlines in this buffer
ctrl.write("data:");
ctrl.write(data);
ctrl.write("\n");
} else {
ctrl.write("data:" + JSON.stringify(data) + "\n");
}
if (id === null) {
ctrl.write("id\n");
} else {
ctrl.write("id:" + id + "\n");
}
}
close() {
this.#ctrl?.close();
}
}
Object.defineProperty(EventStream, "name", { value: "EventStream" });
return EventStream;
}
|
