89 lines
2.4 KiB
JavaScript
Executable File
89 lines
2.4 KiB
JavaScript
Executable File
import { DEFAULT_HOST, DEFAULT_PORT } from "./config";
|
|
let ws = null;
|
|
|
|
const configure = (conn, defaultConsole) => {
|
|
// 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 = (defaultConsole) => {
|
|
console = defaultConsole;
|
|
ws = null;
|
|
}
|
|
|
|
const termsole = (options = {}) => {
|
|
// Ensure termsole doesn't run in production mode
|
|
if (process.env.NODE_ENV && process.env.NODE_ENV !== 'development') return;
|
|
|
|
if (ws) {
|
|
console.log("already running");
|
|
return; // already running
|
|
}
|
|
const defaultConsole = Object.assign(Object.create(Object.getPrototypeOf(console)), console);
|
|
|
|
options = {
|
|
host: DEFAULT_HOST,
|
|
port: DEFAULT_PORT,
|
|
ssl: false,
|
|
...options,
|
|
}
|
|
|
|
ws = new WebSocket(`${options.ssl ? "wss" : "ws"}://${options.host}:${options.port}`);
|
|
ws.onopen = () => {
|
|
configure(ws, defaultConsole);
|
|
console.log('[TCONSOLE]: Connected');
|
|
};
|
|
|
|
ws.onclose = (event) => {
|
|
release(defaultConsole);
|
|
console.log("[TCONSOLE]: Disconnected", event.message);
|
|
}
|
|
|
|
ws.onerror = (event) => {
|
|
release(defaultConsole);
|
|
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;
|