termlog/index.js
2021-08-24 15:53:22 +07:00

84 lines
2.1 KiB
JavaScript
Executable File

let ws = null;
// https://stackoverflow.com/a/44782052/7539840
const defaultConsole = Object.assign(Object.create(Object.getPrototypeOf(console)), console);
const configure = (conn) => {
// skip if already configured
if (console._tsconsole_configured) return;
console._tsconsole_configured = true;
console.log = (...args) => {
sendWhenConnected(conn, JSON.stringify({ type: 'log', data: Array.from(args) }));
defaultConsole.log.apply(defaultConsole, args);
};
console.error = (...args) => {
sendWhenConnected(conn, JSON.stringify({ type: 'error', data: Array.from(args) }));
defaultConsole.error.apply(defaultConsole, args);
};
console.warn = (...args) => {
sendWhenConnected(conn, JSON.stringify({ type: 'warn', data: Array.from(args) }));
defaultConsole.warn.apply(defaultConsole, args);
};
console.debug = (...args) => {
sendWhenConnected(conn, JSON.stringify({ type: 'debug', data: Array.from(args) }));
defaultConsole.debug.apply(defaultConsole, args);
};
}
const release = () => {
console = defaultConsole;
ws = null;
}
const termsole = (options = {}) => {
if (ws) return; // already running
options = {
host: "localhost",
port: 3456,
ssl: false,
...options,
}
ws = new WebSocket(`${options.ssl ? "wss" : "ws"}://${options.host}:${options.port}`);
ws.onopen = () => {
configure(ws);
console.log('[TCONSOLE]: Connected');
};
ws.onclose = (event) => {
release();
console.log("[TCONSOLE]: Disconnected", event.message);
}
ws.onerror = (event) => {
release();
console.error("[TCONSOLE]: Disconnected", event.message);
}
}
// *** Utils *** //
const sendWhenConnected = (ws, msg, n = 0, maxTries = 100) => {
// Closing or closed
if (ws.readyState === 2 || ws.readyState === 3) return;
// try sending
setTimeout(() => {
if (ws.readyState === 1) {
ws.send(msg);
} else if (n < maxTries) {
sendWhenConnected(ws, msg, n + 1);
} else{
console.error("Exceed tries to send message: ", msg);
}
}, 10); // wait 10 milisecond for the connection...
}
export default termsole;