128 lines
4.8 KiB
TypeScript
128 lines
4.8 KiB
TypeScript
import { NextFunction, Request, Response } from "express";
|
|
import { Client } from "@elastic/elasticsearch";
|
|
import { logMemoryStore } from "../utils/LogMemoryStore";
|
|
|
|
if (!process.env.ELASTICSEARCH_INDEX) {
|
|
throw new Error("Require ELASTICSEARCH_INDEX to store log.");
|
|
}
|
|
|
|
const ELASTICSEARCH_INDEX = process.env.ELASTICSEARCH_INDEX;
|
|
|
|
const LOG_LEVEL_MAP: Record<string, number> = {
|
|
debug: 4,
|
|
info: 3,
|
|
warning: 2,
|
|
error: 1,
|
|
none: 0,
|
|
};
|
|
|
|
const elasticsearch = new Client({
|
|
node: `${process.env.ELASTICSEARCH_PROTOCOL}://${process.env.ELASTICSEARCH_HOST}:${process.env.ELASTICSEARCH_PORT}`,
|
|
});
|
|
|
|
async function logMiddleware(req: Request, res: Response, next: NextFunction) {
|
|
if (!req.url.startsWith("/api/")) return next();
|
|
|
|
let data: any;
|
|
|
|
const originalJson = res.json;
|
|
|
|
res.json = function (v: any) {
|
|
data = v;
|
|
return originalJson.call(this, v);
|
|
};
|
|
const timestamp = new Date().toISOString();
|
|
const start = performance.now();
|
|
|
|
req.app.locals.logData = {};
|
|
|
|
res.on("finish", async () => {
|
|
try {
|
|
if (!req.url.startsWith("/api/")) return;
|
|
let system = "organization";
|
|
if (req.url.startsWith("/api/v1/org/keycloak/log/sso")) system = "inout";
|
|
if (req.url.startsWith("/api/v1/org/metadata/")) system = "master";
|
|
if (req.url.startsWith("/api/v1/org/pos/position/")) system = "master";
|
|
if (req.url.startsWith("/api/v1/org/pos/type/")) system = "master";
|
|
if (req.url.startsWith("/api/v1/org/employee/pos/position/")) system = "master";
|
|
if (req.url.startsWith("/api/v1/org/employee/pos/type/")) system = "master";
|
|
if (req.url.startsWith("/api/v1/org/auth/authRoleAttr/")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/auth/authRole/")) system = "admin";
|
|
// if (req.url.startsWith("/api/v1/org/keycloak")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/pos/admin/master/list")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/super-admin/{id}")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/permission-org/")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/pos/assign/")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/profile/")) system = "registry";
|
|
if (req.url.startsWith("/api/v1/org/profile-employee/")) system = "registry";
|
|
if (req.url.startsWith("/api/v1/org/profile-temp/")) system = "registry";
|
|
if (req.url.startsWith("/api/v1/org/ex/")) system = "retirement";
|
|
|
|
if (req.url.startsWith("/api/v1/org/commandType/admin")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/commandSys/")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/commandSalary/")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/apiKey/")) system = "admin";
|
|
if (req.url.startsWith("/api/v1/org/api-manage/")) system = "admin";
|
|
|
|
if (req.url.startsWith("/api/v1/org/keycloak/")) system = "registry";
|
|
|
|
const level = LOG_LEVEL_MAP[process.env.LOG_LEVEL ?? "debug"] || 4;
|
|
|
|
// // Get profile from cache
|
|
// const profileByKeycloak = await logMemoryStore.getProfileByKeycloak(
|
|
// req.app.locals.logData.userId,
|
|
// );
|
|
|
|
// // Get rootId from cache
|
|
// const rootId = await logMemoryStore.getRootIdByProfileId(profileByKeycloak?.id);
|
|
// // console.log("ancestorDNA:", rootId);
|
|
|
|
// Get rootId from token
|
|
const rootId = req.app.locals.logData?.orgRootDnaId;
|
|
|
|
let _msg = data?.message;
|
|
if (!_msg) {
|
|
if (res.statusCode >= 500) {
|
|
_msg = "ไม่สำเร็จ";
|
|
} else if (res.statusCode >= 400) {
|
|
_msg = "พบข้อผิดพลาด";
|
|
} else if (res.statusCode >= 200) {
|
|
_msg = "สำเร็จ";
|
|
}
|
|
}
|
|
|
|
if (level === 1 && res.statusCode < 500) return;
|
|
if (level === 2 && res.statusCode < 400) return;
|
|
if (level === 3 && res.statusCode < 200) return;
|
|
const obj = {
|
|
logType: res.statusCode >= 500 ? "error" : res.statusCode >= 400 ? "warning" : "info",
|
|
ip: req.ip,
|
|
rootId: rootId ?? null,
|
|
systemName: system,
|
|
startTimeStamp: timestamp,
|
|
endTimeStamp: new Date().toISOString(),
|
|
processTime: performance.now() - start,
|
|
host: req.hostname,
|
|
method: req.method,
|
|
endpoint: req.url,
|
|
responseCode: String(res.statusCode === 304 ? 200 : res.statusCode),
|
|
responseDescription: _msg,
|
|
input: level === 4 ? JSON.stringify(req.body, null, 2) : undefined,
|
|
output: level === 4 ? JSON.stringify(data, null, 2) : undefined,
|
|
...req.app.locals.logData,
|
|
};
|
|
|
|
// Send log to Elasticsearch
|
|
await elasticsearch.index({
|
|
index: ELASTICSEARCH_INDEX,
|
|
document: obj,
|
|
});
|
|
} catch (err) {
|
|
console.error("Error in logMiddleware:", err);
|
|
}
|
|
});
|
|
|
|
return next();
|
|
}
|
|
|
|
export default logMiddleware;
|