uptime-kuma/src/mixins/socket.js

import { io } from "socket.io-client";
import { useToast } from "vue-toastification";
const toast = useToast()

let socket;

export default {

    data() {
        return {
            info: { },
            socket: {
                token: null,
                firstConnect: true,
                connected: false,
                connectCount: 0,
            },
            remember: (localStorage.remember !== "0"),
            allowLoginDialog: false,        // Allowed to show login dialog, but "loggedIn" have to be true too. This exists because prevent the login dialog show 0.1s in first before the socket server auth-ed.
            loggedIn: false,
            monitorList: { },
            heartbeatList: { },
            importantHeartbeatList: { },
            avgPingList: { },
            uptimeList: { },
            certInfoList: {},
            notificationList: [],
            connectionErrorMsg: "Cannot connect to the socket server. Reconnecting...",
        }
    },

    created() {
        window.addEventListener("resize", this.onResize);

        let wsHost;
        const env = process.env.NODE_ENV || "production";
        if (env === "development" || localStorage.dev === "dev") {
            wsHost = ":3001"
        } else {
            wsHost = ""
        }

        socket = io(wsHost, {
            transports: ["websocket"],
        });

        socket.on("info", (info) => {
            this.info = info;
        });

        socket.on("setup", (monitorID, data) => {
            this.$router.push("/setup")
        });

        socket.on("autoLogin", (monitorID, data) => {
            this.loggedIn = true;
            this.storage().token = "autoLogin";
            this.allowLoginDialog = false;
        });

        socket.on("monitorList", (data) => {
            // Add Helper function
            Object.entries(data).forEach(([monitorID, monitor]) => {
                monitor.getUrl = () => {
                    try {
                        return new URL(monitor.url);
                    } catch (_) {
                        return null;
                    }
                };
            });
            this.monitorList = data;
        });

        socket.on("notificationList", (data) => {
            this.notificationList = data;
        });

        socket.on("heartbeat", (data) => {
            if (! (data.monitorID in this.heartbeatList)) {
                this.heartbeatList[data.monitorID] = [];
            }

            this.heartbeatList[data.monitorID].push(data)

            // Add to important list if it is important
            // Also toast
            if (data.important) {

                if (data.status === 0) {
                    toast.error(`[${this.monitorList[data.monitorID].name}] [DOWN] ${data.msg}`, {
                        timeout: false,
                    });
                } else if (data.status === 1) {
                    toast.success(`[${this.monitorList[data.monitorID].name}] [Up] ${data.msg}`, {
                        timeout: 20000,
                    });
                } else {
                    toast(`[${this.monitorList[data.monitorID].name}] ${data.msg}`);
                }

                if (! (data.monitorID in this.importantHeartbeatList)) {
                    this.importantHeartbeatList[data.monitorID] = [];
                }

                this.importantHeartbeatList[data.monitorID].unshift(data)
            }
        });

        socket.on("heartbeatList", (monitorID, data, overwrite = false) => {
            if (! (monitorID in this.heartbeatList) || overwrite) {
                this.heartbeatList[monitorID] = data;
            } else {
                this.heartbeatList[monitorID] = data.concat(this.heartbeatList[monitorID])
            }
        });

        socket.on("avgPing", (monitorID, data) => {
            this.avgPingList[monitorID] = data
        });

        socket.on("uptime", (monitorID, type, data) => {
            this.uptimeList[`${monitorID}_${type}`] = data
        });

        socket.on("certInfo", (monitorID, data) => {
            this.certInfoList[monitorID] = JSON.parse(data)
        });

        socket.on("importantHeartbeatList", (monitorID, data, overwrite) => {
            if (! (monitorID in this.importantHeartbeatList) || overwrite) {
                this.importantHeartbeatList[monitorID] = data;
            } else {
                this.importantHeartbeatList[monitorID] = data.concat(this.importantHeartbeatList[monitorID])
            }
        });

        socket.on("connect_error", (err) => {
            console.error(`Failed to connect to the backend. Socket.io connect_error: ${err.message}`);
            this.connectionErrorMsg = `Cannot connect to the socket server. [${err}] Reconnecting...`;
            this.socket.connected = false;
            this.socket.firstConnect = false;
        });

        socket.on("disconnect", () => {
            console.log("disconnect")
            this.connectionErrorMsg = "Lost connection to the socket server. Reconnecting...";
            this.socket.connected = false;
        });

        socket.on("connect", () => {
            console.log("connect")
            this.socket.connectCount++;
            this.socket.connected = true;

            // Reset Heartbeat list if it is re-connect
            if (this.socket.connectCount >= 2) {
                this.clearData()
            }

            let token = this.storage().token;

            if (token) {
                if (token !== "autoLogin") {
                    this.loginByToken(token)
                } else {

                    // Timeout if it is not actually auto login
                    setTimeout(() => {
                        if (! this.loggedIn) {
                            this.allowLoginDialog = true;
                            this.$root.storage().removeItem("token");
                        }
                    }, 5000);

                }
            } else {
                this.allowLoginDialog = true;
            }

            this.socket.firstConnect = false;
        });

    },

    methods: {

        storage() {
            return (this.remember) ? localStorage : sessionStorage;
        },

        getSocket() {
            return socket;
        },

        toastRes(res) {
            if (res.ok) {
                toast.success(res.msg);
            } else {
                toast.error(res.msg);
            }
        },

        login(username, password, callback) {
            socket.emit("login", {
                username,
                password,
            }, (res) => {

                if (res.ok) {
                    this.storage().token = res.token;
                    this.socket.token = res.token;
                    this.loggedIn = true;

                    // Trigger Chrome Save Password
                    history.pushState({}, "")
                }

                callback(res)
            })
        },

        loginByToken(token) {
            socket.emit("loginByToken", token, (res) => {
                this.allowLoginDialog = true;

                if (! res.ok) {
                    this.logout()
                } else {
                    this.loggedIn = true;
                }
            })
        },

        logout() {
            this.storage().removeItem("token");
            this.socket.token = null;
            this.loggedIn = false;

            this.clearData()
        },

        add(monitor, callback) {
            socket.emit("add", monitor, callback)
        },

        deleteMonitor(monitorID, callback) {
            socket.emit("deleteMonitor", monitorID, callback)
        },

        clearData() {
            console.log("reset heartbeat list")
            this.heartbeatList = {}
            this.importantHeartbeatList = {}
        },

        uploadBackup(uploadedJSON, callback) {
            socket.emit("uploadBackup", uploadedJSON, callback)
        },

        clearEvents(monitorID, callback) {
            socket.emit("clearEvents", monitorID, callback)
        },

        clearHeartbeats(monitorID, callback) {
            socket.emit("clearHeartbeats", monitorID, callback)
        },

        clearStatistics(callback) {
            socket.emit("clearStatistics", callback)
        },
    },

    computed: {

        lastHeartbeatList() {
            let result = {}

            for (let monitorID in this.heartbeatList) {
                let index = this.heartbeatList[monitorID].length - 1;
                result[monitorID] = this.heartbeatList[monitorID][index];
            }

            return result;
        },

        statusList() {
            let result = {}

            let unknown = {
                text: "Unknown",
                color: "secondary",
            }

            for (let monitorID in this.lastHeartbeatList) {
                let lastHeartBeat = this.lastHeartbeatList[monitorID]

                if (! lastHeartBeat) {
                    result[monitorID] = unknown;
                } else if (lastHeartBeat.status === 1) {
                    result[monitorID] = {
                        text: "Up",
                        color: "primary",
                    };
                } else if (lastHeartBeat.status === 0) {
                    result[monitorID] = {
                        text: "Down",
                        color: "danger",
                    };
                } else if (lastHeartBeat.status === 2) {
                    result[monitorID] = {
                        text: "Pending",
                        color: "warning",
                    };
                } else {
                    result[monitorID] = unknown;
                }
            }

            return result;
        },
    },

    watch: {

        // Reload the SPA if the server version is changed.
        "info.version"(to, from) {
            if (from && from !== to) {
                window.location.reload()
            }
        },

        remember() {
            localStorage.remember = (this.remember) ? "1" : "0"
        },

    },

}