yaklang-chrome-extension/public/background.js

let socket;
const connectWebsocket = url => {
    disconnectWebsocket()
    socket = new WebSocket(url);
    socket.onopen = () => {
        chrome.runtime.sendMessage({status: "connected"})
    }
    socket.onclose = () => {
        chrome.runtime.sendMessage({status: "disconnected"})
    }
}

const disconnectWebsocket = () => {
    if (socket) {
        try {
            socket.close()
            socket = null;
        } catch (e) {

        }
    }
}

heartbeat = () => {
    if (socket) {
        if (socket.readyState === WebSocket.OPEN) {
            try {
                socket.send(JSON.stringify({
                    "type": "heartbeat",
                }))
            } catch (e) {
                console.error("Error sending heartbeat:", e);
            }
        } else {
            disconnectWebsocket()
        }
    }
}
heartbeat()
setInterval(heartbeat, 3000)

console.info("Chrome Extenstion Background is loaded")
chrome.runtime.onMessage.addListener(function (msg, sender, sendResponse) {
    if (msg.action === "connectWebsocket") {
        console.info("Start to connect websocket")
        connectWebsocket(msg.url)
    } else if (msg.action === "init") {
        if (socket) {
            chrome.runtime.sendMessage({status: "connected"})
        } else {
            chrome.runtime.sendMessage({status: "initialized"})
        }
    }
})