thelounge/client/js/socket.js

"use strict";

const $ = require("jquery");
const io = require("socket.io-client");
const utils = require("./utils");
const path = window.location.pathname + "socket.io/";
const status = $("#loading-page-message, #connection-error");

const socket = io({
	transports: $(document.body).data("transports"),
	path: path,
	autoConnect: false,
	reconnection: !$(document.body).hasClass("public"),
});

socket.on("disconnect", handleDisconnect);
socket.on("connect_error", handleDisconnect);
socket.on("error", handleDisconnect);

socket.on("reconnecting", function(attempt) {
	status.text(`Reconnecting… (attempt ${attempt})`);
});

socket.on("connecting", function() {
	status.text("Connecting…");
});

socket.on("connect", function() {
	// Clear send buffer when reconnecting, socket.io would emit these
	// immediately upon connection and it will have no effect, so we ensure
	// nothing is sent to the server that might have happened.
	socket.sendBuffer = [];

	status.text("Finalizing connection…");
});

socket.on("authorized", function() {
	status.text("Loading messages…");
});

function handleDisconnect(data) {
	const message = data.message || data;

	status.text(`Waiting to reconnect… (${message})`).addClass("shown");
	$(".show-more-button, #input").prop("disabled", true);
	$("#submit").hide();

	// If the server shuts down, socket.io skips reconnection
	// and we have to manually call connect to start the process
	if (socket.io.skipReconnect) {
		utils.requestIdleCallback(() => socket.connect(), 2000);
	}
}

module.exports = socket;