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
122
123
|
import { useWebSocket } from "@/hooks/useWs";
import { useState } from "react";
type WidgetProps = {
url: string;
protocols?: string | string[];
};
export default function WebSocketWidget({ url, protocols }: WidgetProps) {
const {
status,
retryCount,
lastMessage,
error,
bufferedAmount,
send,
reconnectNow,
close,
} = useWebSocket({
url,
protocols,
onMessage: (ev) => {
// Example: auto reply to pings
console.log(ev.data, "ws event");
if (
typeof ev.data === "string" &&
// ev.data.toLowerCase().includes("ping")
ev.data.toLowerCase().trim() == "ping"
) {
try {
console.log("sending pong");
send("pong");
} catch {}
}
},
});
const [outbound, setOutbound] = useState("");
return (
<div className="w-full max-w-xl mx-auto p-4 grid gap-3">
<header className="flex items-center justify-between">
<h1 className="text-xl font-semibold">WebSocketWidget</h1>
<span className="text-sm px-2 py-1 rounded-full border">
{status.toUpperCase()} {retryCount ? `(retry ${retryCount})` : ""}
</span>
</header>
<div className="text-sm text-gray-600">
<div>
<b>URL:</b> {url}
</div>
<div>
<b>Buffered:</b> {bufferedAmount} bytes
</div>
{error && (
<div className="text-red-600">
<b>Error:</b>{" "}
{"message" in error
? (error as any).message
: String(error.type || "error")}
</div>
)}
</div>
<div className="p-3 rounded-2xl border bg-gray-50 min-h-[4rem] font-mono text-sm break-words">
<div className="opacity-70">Last message:</div>
<div>
{lastMessage
? typeof lastMessage.data === "string"
? lastMessage.data
: "(binary)"
: "—"}
</div>
</div>
<form
className="flex gap-2"
onSubmit={(e) => {
e.preventDefault();
if (!outbound) return;
send(outbound);
setOutbound("");
}}
>
<input
className="flex-1 px-3 py-2 rounded-xl border"
placeholder="Type message…"
value={outbound}
onChange={(e) => setOutbound(e.target.value)}
/>
<button type="submit" className="px-3 py-2 rounded-xl border">
Send
</button>
</form>
<div className="flex gap-2">
<button className="px-3 py-2 rounded-xl border" onClick={reconnectNow}>
Reconnect
</button>
<button className="px-3 py-2 rounded-xl border" onClick={() => close()}>
Close
</button>
</div>
<details className="mt-2">
<summary className="cursor-pointer">Usage</summary>
<pre className="text-xs bg-gray-100 p-2 rounded-xl overflow-auto">
{`import WebSocketWidget from "./WebSocketWidget";
export default function App() {
return (
<div className="p-6">
<WebSocketWidget url="wss://echo.websocket.events" />
</div>
);
}
`}
</pre>
</details>
</div>
);
}
|
