Merge pull request #1687 from OneCommunityGlobal/aayush-create-line-chart-showing-total-injuries-over-time-backend

Aayush create line chart showing total injuries over time backend

Author: one-community

package-lock.json

@@ -0,0 +1,94 @@
+const mongoose = require('mongoose');
+const InjuryOverTime = require('../../models/bmdashboard/buildingInjuryOverTime');
+
+const parseCommaSeparatedValues = (value, mapper = (item) => item) =>
+  value
+    .split(',')
+    .map((item) => item.trim())
+    .filter(Boolean)
+    .map(mapper);
+
+exports.getInjuryOverTime = async (req, res) => {
+  try {
+    const { projectIds, startDate, endDate, types, departments, severities } = req.query;
+
+    const query = {};
+
+    if (projectIds) {
+      const projectIdArray = parseCommaSeparatedValues(projectIds);
+
+      if (!projectIdArray.every((id) => mongoose.Types.ObjectId.isValid(id))) {
+        return res.status(400).json({ error: 'One or more projectIds are invalid' });
+      }
+
+      query.projectId = {
+        $in: projectIdArray.map((id) => mongoose.Types.ObjectId(id)),
+      };
+    }
+
+    if (startDate && endDate) {
+      query.date = {
+        $gte: new Date(startDate),
+        $lte: new Date(endDate),
+      };
+    }
+
+    if (types) {
+      query.injuryType = { $in: parseCommaSeparatedValues(types) };
+    }
+
+    if (departments) {
+      query.department = { $in: parseCommaSeparatedValues(departments) };
+    }
+
+    if (severities) {
+      query.severity = { $in: parseCommaSeparatedValues(severities) };
+    }
+
+    const injuries = await InjuryOverTime.aggregate([
+      { $match: query },
+      {
+        $group: {
+          _id: {
+            projectId: '$projectId',
+            severity: '$severity',
+            injuryType: '$injuryType',
+            department: '$department',
+            month: {
+              $dateToString: {
+                format: '%Y-%m-01',
+                date: '$date',
+              },
+            },
+          },
+          count: { $sum: '$count' },
+        },
+      },
+      {
+        $project: {
+          _id: 0,
+          projectId: '$_id.projectId',
+          severity: '$_id.severity',
+          injuryType: '$_id.injuryType',
+          department: '$_id.department',
+          date: { $toDate: '$_id.month' },
+          count: 1,
+        },
+      },
+      {
+        $sort: {
+          date: 1,
+          projectId: 1,
+          severity: 1,
+          injuryType: 1,
+          department: 1,
+        },
+      },
+    ]);
+
+    res.status(200).json(injuries);
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ error: 'Internal server error' });
+  }
+};

src/controllers/bmdashboard/injuryOverTimeController.js

@@ -0,0 +1,12 @@
+const mongoose = require('mongoose');
+
+const injuryOverTimeSchema = new mongoose.Schema({
+  projectId: { type: mongoose.Schema.Types.ObjectId, ref: 'buildingProject', required: true },
+  date: { type: Date, required: true },
+  injuryType: { type: String, trim: true },
+  department: { type: String, trim: true },
+  severity: { type: String, trim: true },
+  count: { type: Number, default: 1, min: 0 },
+});
+
+module.exports = mongoose.model('InjuryOverTime', injuryOverTimeSchema, 'injuryOverTime');

src/models/bmdashboard/buildingInjuryOverTime.js

@@ -9,9 +9,13 @@ const {
   getProjectsWithInjuries,
 } = require('../../controllers/bmdashboard/injuryCategoryController');
 
+const { getInjuryOverTime } = require('../../controllers/bmdashboard/injuryOverTimeController');
+
 router.get('/category-breakdown', getCategoryBreakdown);
 router.get('/injury-severities', getUniqueSeverities);
 router.get('/injury-types', getUniqueInjuryTypes);
 router.get('/project-injury', getProjectsWithInjuries);
 
+router.get('/over-time', getInjuryOverTime);
+
 module.exports = router;