105 lines
2.5 KiB
JavaScript
105 lines
2.5 KiB
JavaScript
const path = require("path");
|
|
const _ = require("lodash");
|
|
require("dotenv").config({
|
|
path: path.resolve(
|
|
process.cwd(),
|
|
`.env.${process.env.NODE_ENV || "development"}`
|
|
),
|
|
});
|
|
|
|
const { io } = require("../../server");
|
|
const { admin } = require("../firebase/firebase-handler");
|
|
const CdkJobExport = require("../cdk/cdk-job-export").default;
|
|
const { isArray } = require("lodash");
|
|
const logger = require("../utils/logger");
|
|
|
|
io.use(function (socket, next) {
|
|
try {
|
|
if (socket.handshake.auth.token) {
|
|
admin
|
|
.auth()
|
|
.verifyIdToken(socket.handshake.auth.token)
|
|
.then((user) => {
|
|
socket.user = user;
|
|
next();
|
|
})
|
|
.catch((error) => {
|
|
next(new Error("Authentication error", JSON.stringify(error)));
|
|
});
|
|
} else {
|
|
next(new Error("Authentication error - no authorization token."));
|
|
}
|
|
} catch (error) {
|
|
console.log("Uncaught connection error:::", error);
|
|
logger.log("websocket-connection-error", "error", null, null, {
|
|
token: socket.handshake.auth.token,
|
|
...error,
|
|
});
|
|
next(new Error(`Authentication error ${error}`));
|
|
}
|
|
});
|
|
|
|
io.on("connection", (socket) => {
|
|
socket.log_level = "TRACE";
|
|
createLogEvent(socket, "DEBUG", `Connected and Authenticated.`);
|
|
|
|
socket.on("set-log-level", (level) => {
|
|
socket.log_level = level;
|
|
createLogEvent(socket, "DEBUG", `Updated log level to ${level}`);
|
|
});
|
|
|
|
socket.on("cdk-export-job", (jobid) => {
|
|
CdkJobExport(socket, jobid);
|
|
});
|
|
|
|
socket.on("disconnect", () => {
|
|
createLogEvent(socket, "DEBUG", `User disconnected.`);
|
|
});
|
|
});
|
|
|
|
function createLogEvent(socket, level, message) {
|
|
if (LogLevelHierarchy(socket.log_level) >= LogLevelHierarchy(level)) {
|
|
console.log(
|
|
`[WS LOG EVENT] ${level} - ${new Date()} - ${socket.user.email} - ${
|
|
socket.id
|
|
} - ${message}`
|
|
);
|
|
socket.emit("log-event", {
|
|
timestamp: new Date(),
|
|
level,
|
|
message,
|
|
});
|
|
|
|
logger.log("ws-log-event", level, socket.user.email, socket.recordid, {
|
|
wsmessage: message,
|
|
});
|
|
|
|
if (socket.logEvents && isArray(socket.logEvents)) {
|
|
socket.logEvents.push({
|
|
timestamp: new Date(),
|
|
level,
|
|
message,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
function LogLevelHierarchy(level) {
|
|
switch (level) {
|
|
case "TRACE":
|
|
return 5;
|
|
case "DEBUG":
|
|
return 4;
|
|
case "INFO":
|
|
return 3;
|
|
case "WARNING":
|
|
return 2;
|
|
case "ERROR":
|
|
return 1;
|
|
default:
|
|
return 3;
|
|
}
|
|
}
|
|
|
|
exports.createLogEvent = createLogEvent;
|