@justair/justair-library 4.4.2-alpha → 4.4.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@justair/justair-library",
-  "version": "4.4.2-alpha",
+  "version": "4.4.3",
   "description": "JustAir Internal Library",
   "main": "src/index.js", 
   "type": "module",

package/src/models/measurements.js

@@ -8,6 +8,40 @@ const correctionSnapshotSchema = new mongoose.Schema(
   { _id: false } // Not generating separate _id for each correction
 );
 
+// Annotation Schema
+const annotationSchema = new mongoose.Schema(
+  {
+    measurementIdentifier: {
+      type: String,
+      required: true,
+      trim: true,
+      // Examples: "PM2_5", "PM10", "NO2", "O3", etc.
+    },
+    comment: {
+      type: String,
+      required: true,
+      trim: true,
+    },
+    adminId: {
+      type: mongoose.Types.ObjectId,
+      ref: "Admin",
+      required: true,
+    },
+    timestamp: {
+      type: Date,
+      default: Date.now,
+      required: true,
+    },
+  },
+  {
+    _id: true,
+    timestamps: false,
+  }
+);
+
+// Add index for efficient querying
+annotationSchema.index({ measurementIdentifier: 1, timestamp: -1 });
+
 // Audit Schema
 const auditSchema = mongoose.Schema(
   {
@@ -25,6 +59,7 @@ const auditSchema = mongoose.Schema(
       default: {},
     },
     correctedNormalizedMeasurements: Object,
+    annotations: [annotationSchema], // Include annotations in audit trail
   },
   {
     timestamps: true,
@@ -49,6 +84,10 @@ const measurementsSchema = mongoose.Schema(
       default: {},
     },
     correctedNormalizedMeasurements: Object,
+    annotations: {
+      type: [annotationSchema],
+      default: [],
+    },
   },
   {
     timestamps: true,
@@ -58,6 +97,12 @@ const measurementsSchema = mongoose.Schema(
 measurementsSchema.index({ monitorId: 1 });
 measurementsSchema.index({ timeUpdated: 1 });
 
+measurementsSchema.index({
+  "annotations.measurementIdentifier": 1,
+  "annotations.timestamp": -1,
+});
+measurementsSchema.index({ "annotations.userId": 1 });
+
 // Pre-hook to log single document deletions
 measurementsSchema.pre("findOneAndDelete", async function () {
   const docToDelete = await this.model.findOne(this.getFilter());
@@ -74,6 +119,7 @@ measurementsSchema.pre("findOneAndDelete", async function () {
       appliedCorrections: docToDelete.appliedCorrections,
       correctedNormalizedMeasurements:
         docToDelete.correctedNormalizedMeasurements,
+      annotations: docToDelete.annotations, // Include annotations in audit
       deletedAt: new Date(),
     });
     await auditLog.save();
@@ -97,6 +143,7 @@ measurementsSchema.pre("deleteMany", async function () {
       flags: doc.flags,
       appliedCorrections: doc.appliedCorrections,
       correctedNormalizedMeasurements: doc.correctedNormalizedMeasurements,
+      annotations: doc.annotations, // Include annotations in audit
       deletedAt: new Date(),
     }));
 
@@ -122,6 +169,7 @@ measurementsSchema.pre("deleteOne", async function () {
       appliedCorrections: docToDelete.appliedCorrections,
       correctedNormalizedMeasurements:
         docToDelete.correctedNormalizedMeasurements,
+      annotations: docToDelete.annotations, // Include annotations in audit
       deletedAt: new Date(),
     });
     await auditLog.save();
@@ -144,6 +192,7 @@ measurementsSchema.pre("findOneAndUpdate", async function () {
       appliedCorrections: docToUpdate.appliedCorrections,
       correctedNormalizedMeasurements:
         docToUpdate.correctedNormalizedMeasurements,
+      annotations: docToUpdate.annotations, // Include annotations in audit
       deletedAt: null, // No deletion happening, so set it to null or undefined
     });
     await auditLog.save();
@@ -165,6 +214,7 @@ measurementsSchema.pre("updateMany", async function () {
       flags: doc.flags,
       appliedCorrections: doc.appliedCorrections,
       correctedNormalizedMeasurements: doc.correctedNormalizedMeasurements,
+      annotations: doc.annotations, // Include annotations in audit
       deletedAt: null, // No deletion happening
     }));
 
@@ -174,4 +224,10 @@ measurementsSchema.pre("updateMany", async function () {
 
 const Measurements = mongoose.model("Measurements", measurementsSchema);
 
-export { measurementsSchema, Measurements, Audit, auditSchema };
+export {
+  measurementsSchema,
+  Measurements,
+  Audit,
+  auditSchema,
+  annotationSchema,
+};