1 files changed, 100 insertions, 0 deletions
diff --git a/src/lib/socket.ts b/src/lib/socket.ts
new file mode 100644
index 00000000..02d441bf
--- /dev/null
+++ b/src/lib/socket.ts
@@ -0,0 +1,100 @@
+import { invalidate } from "$app/navigation";
+import { navigating } from "$app/stores";
+import { PUBLIC_SOCKET } from "$env/static/public";
+import { event, invoice, last, request } from "$lib/store";
+import { sat, sleep, success, wait } from "$lib/utils";
+import { get } from "svelte/store";
+
+let socket;
+let token;
+
+export const auth = () => token && send("login", token);
+
+export const send = async (type, data) => {
+	try {
+		await wait(() => socket && socket.readyState === 1, 1000, 10);
+	} catch (e: any) {
+		if (e.message === "timeout") reconnectToWebsocket();
+	}
+
+	await wait(() => socket.readyState === 1, 1000, 10);
+	socket.send(JSON.stringify({ type, data }));
+};
+
+export const messages = (data) => ({
+	id() {
+		last.set(Date.now());
+	},
+	invoice() {
+		invoice.set(data);
+	},
+	event() {
+		event.set(data);
+	},
+
+	request() {
+		request.set(data);
+		invalidate("app:invoice");
+	},
+
+	async payment() {
+		const { amount, confirmed } = data;
+		invalidate("app:user");
+		invalidate("app:invoice");
+		invalidate("app:payments");
+
+		if (amount > 0) {
+			success(`${confirmed ? "Received" : "Detected"} ${sat(amount)}!`);
+		}
+	},
+
+	connected: auth,
+});
+
+const initialReconnectDelay = 1000;
+const maxReconnectDelay = 16000;
+
+let currentReconnectDelay = initialReconnectDelay;
+
+export function connect(t) {
+	token = t;
+
+	if (socket) return auth();
+
+	socket = new WebSocket(PUBLIC_SOCKET);
+	socket.addEventListener("open", onWebsocketOpen);
+	socket.addEventListener("close", onWebsocketClose);
+	socket.addEventListener("message", onWebsocketMessage);
+}
+
+export function close() {
+	if (socket) socket.close();
+}
+
+async function onWebsocketMessage(msg) {
+	const { type, data } = JSON.parse(msg.data);
+	if (get(navigating)) await sleep(2000);
+
+	if (messages(data)[type]) messages(data)[type]();
+}
+
+function onWebsocketOpen() {
+	currentReconnectDelay = initialReconnectDelay;
+}
+
+function onWebsocketClose() {
+	socket = null;
+	setTimeout(
+		() => {
+			reconnectToWebsocket();
+		},
+		currentReconnectDelay + Math.floor(Math.random() * 3000),
+	);
+}
+
+function reconnectToWebsocket() {
+	if (currentReconnectDelay < maxReconnectDelay) {
+		currentReconnectDelay *= 2;
+	}
+	connect(token);
+}