Refactor Firestore timestamp usage in functions to utilize FieldValue directly for consistency and clarity

Author: Novapool

functions-node/index.js

@@ -4,25 +4,27 @@ const pdfParse = require('pdf-parse');
 const tmp = require('tmp');
 const fs = require('fs');
 const OpenAI = require('openai');
+const { FieldValue } = require('firebase-admin/firestore'); 
 const { calculateCurrentGrade } = require('./calculateGrade');
 const { predictFinalGrade } = require('./predictGrade');
 const { formatDocumentsData } = require('./formatDocumentsData');
 const { DOCUMENT_TYPES, normalizeDocumentType } = require('./constants/documentTypes');
 
+
 // Initialize Firebase Admin with emulator support
 if (process.env.FUNCTIONS_EMULATOR) {
   // Running in emulator
   admin.initializeApp({
     projectId: 'gradingai',
     storageBucket: 'gradingai.appspot.com',
     // This ensures Firestore uses the emulator
-    databaseURL: 'http://localhost:8080',
   });
   console.log('Firebase Admin initialized for emulator environment');
 } else {
   // Production environment
   admin.initializeApp();
   console.log('Firebase Admin initialized for production environment');
+  const db = getFirestore();
 }
 
 // Export functions
@@ -167,7 +169,7 @@ exports.uploadDocument = functions.https.onCall(async (data, context) => {
         filePath: filePath,
         documentType: normalizedDocType,
         name: documentName,
-        uploadedAt: admin.firestore.FieldValue.serverTimestamp(),
+        uploadedAt: FieldValue.serverTimestamp(),        
         status: 'uploaded'
       });
 
@@ -366,7 +368,7 @@ exports.processPdfUpload = functions.storage.object().onFinalize(async (object)
         filePath: filePath,
         documentType: documentType,
         name: pathParts[3],
-        uploadedAt: admin.firestore.FieldValue.serverTimestamp(),
+        uploadedAt: FieldValue.serverTimestamp(),
         status: 'uploaded'
       });
 
@@ -450,7 +452,7 @@ async function extractTextFromPdf(userId, documentId, filePath) {
     console.log(`Updating Firestore document with extracted text`);
     await docRef.set({
       text: extractedText,
-      lastExtracted: admin.firestore.FieldValue.serverTimestamp(),
+      lastExtracted: FieldValue.serverTimestamp(),
       status: 'extracted',
       pageCount: pdfData.numpages
     }, { merge: true });