feature/IO-2769-Job-Totals-Testing

server/job/job-totals-recorder.js

@@ -0,0 +1,133 @@
+const logger = require("../utils/logger");
+const queries = require("../graphql-client/queries");
+const moment = require("moment");
+const { captureFixture } = require("./utils/seralizeHelper");
+const { TotalsServerSide: totalsServerSideCA } = require("./job-totals"); // Canadian version (imex)
+const { TotalsServerSide: totalsServerSideUS } = require("./job-totals-USA");
+const InstanceMgr = require("../utils/instanceMgr").default;
+const { uploadFileToS3 } = require("../utils/s3");
+
+// requires two buckets be made per env, job-totals-test, job-totals-production, locally it will
+// use `job-totals` in the owncloud stack
+
+/**
+ * Returns the environment prefix based on NODE_ENV
+ * @returns {string}
+ */
+const getEnvPrefix = () => {
+  switch (process.env?.NODE_ENV) {
+    case "test":
+      return "test";
+    case "production":
+      return "production";
+    default:
+      return "test";
+  }
+};
+
+const envPrefix = getEnvPrefix();
+
+const S3_BUCKET_NAME = process.env?.NODE_ENV === "development" ? "imex-job-totals" : `job-totals-${envPrefix}`;
+
+/**
+ * Generates a unique S3 key for the job totals file
+ * @param {string} jobId - The job ID
+ * @returns {string} - S3 key with timestamp
+ */
+const generateS3Key = (jobId) => `${jobId}-${moment().toISOString()}.json`;
+
+/**
+ * Uploads job totals data to S3
+ * @param {object} data - The data to upload
+ * @param {string} jobId - The job ID
+ * @param {object} userInfo - User information for logging
+ * @returns {Promise<string>} - The S3 key
+ */
+const uploadJobTotalsToS3 = async (data, jobId, userInfo) => {
+  const key = generateS3Key(jobId);
+
+  try {
+    await uploadFileToS3({
+      bucketName: S3_BUCKET_NAME,
+      key: key,
+      content: JSON.stringify(data, null, 2),
+      contentType: "application/json"
+    });
+
+    logger.log(`Job totals uploaded successfully to ${key}`, "info", userInfo.email, jobId);
+    return key;
+  } catch (error) {
+    logger.log("Failed to upload job totals to S3", "error", userInfo.email, jobId, {
+      error: error?.message,
+      stack: error?.stack
+    });
+    throw error; // Re-throw for the main handler to catch
+  }
+};
+
+/**
+ * Fetches job data using GraphQL
+ * @param {object} client - GraphQL client
+ * @param {string} token - Bearer token
+ * @param {string} jobId - Job ID to fetch
+ * @returns {Promise<object>} - Job data
+ */
+const fetchJobData = async (client, token, jobId) => {
+  return client
+    .setHeaders({ Authorization: token })
+    .request(queries.GET_JOB_BY_PK, { id: jobId })
+    .then((response) => response.jobs_by_pk);
+};
+
+/**
+ * This function is used to capture job totals json files.
+ * @param {object} req - Express request
+ * @param {object} res - Express response
+ * @returns {Promise<void>}
+ */
+const jobTotalsRecorder = async (req, res) => {
+  const { id: jobId } = req.body;
+  const bearerToken = req.BearerToken;
+  const client = req.userGraphQLClient;
+  const userEmail = req?.user?.email;
+
+  logger.log("Starting job totals recording", "debug", userEmail, jobId);
+
+  try {
+    // Fetch job data
+    const jobData = await fetchJobData(client, bearerToken, jobId);
+
+    // Get the appropriate totals function based on instance
+    const totalsFunction = InstanceMgr({
+      imex: totalsServerSideCA,
+      rome: totalsServerSideUS
+    });
+
+    // Calculate the totals
+    const calculatedTotals = await totalsFunction({ body: { job: jobData, client }, req }, res, true);
+
+    // Prepare data for storage
+    const dataToSave = captureFixture(jobData, calculatedTotals);
+
+    // Upload to S3
+    await uploadJobTotalsToS3(dataToSave, jobId, { email: userEmail });
+
+    res.status(200).json({ success: true, message: "Job totals recorded successfully" });
+  } catch (error) {
+    logger.log("Failed to record job totals", "error", userEmail, jobId, {
+      error: error?.message,
+      stack: error?.stack
+    });
+
+    // Avoid sending response if it's already been sent
+    if (!res.headersSent) {
+      res.status(503).json({
+        success: false,
+        message: "Error processing job totals",
+        error: error.message
+      });
+    }
+  }
+};
+
+module.exports = jobTotalsRecorder;