save entry in DB before sending to AWS

server.js

@@ -11,10 +11,11 @@ require('dotenv').config()
 const app = express();
 app.use(cors());
 
-const port = 3000;
+const port = process.env.PORT || 3000;
+const dbFile = process.env.SQLITE_FILE || './sqlite-db.sqlite';
 
 // Open a connection to the SQLite database
-const db = new sqlite3.Database('sqlite-db.sqlite', (err) => {
+const db = new sqlite3.Database(dbFile, (err) => {
   if (err) {
     console.error('Error opening database:', err);
   }
@@ -29,11 +30,11 @@ const s3Client = new S3Client({
   }
 });
 
-const UPLOAD_DIR = 'uploads';
-const upload = multer({ dest: UPLOAD_DIR + '/' });
+const uploadDir = 'uploads';
+const uploadMulter = multer({ dest: uploadDir + '/' });
 
 // POST endpoint to upload an image
-app.post('/image', upload.single('image'), (req, res) => {
+app.post('/image', uploadMulter.single('image'), (req, res) => {
   const file = req.file;
 
   // Read the file from the temporary location
@@ -55,22 +56,11 @@ app.post('/image', upload.single('image'), (req, res) => {
 
     // Upload the file to S3
     try {
-      const command = new PutObjectCommand(params);
-      const response = await s3Client.send(command);
-      if (response.$metadata.httpStatusCode !== 200) {
-        const errorTime = new Date().toISOString();
-        console.error(errorTime, "Error uploading to S3 with bad HTTP status. Metadata:", response.$metadata);
-        res.status(500).send(JSON.stringify({ success: false, message: "Got bad status of " + response.$metadata.httpStatusCode + " from AWS. See server logs at " + errorTime }));
-      } else {
-        const finalUrl = `https://${bucketName}.s3.amazonaws.com/${fileName}`;
-        fs.rm(file.path, (err) => {
-          if (err) {
-            console.error("Error deleting temp file", file.path, "with error:", err);
-          }
-        });
-        // Record the upload in the database
+
+      // record the upload in the database
       const currentDate = new Date().toISOString();
-        const localFile = file.path.startsWith(UPLOAD_DIR + '/') ? file.path.substring(UPLOAD_DIR.length + 1) : file.path;
+      const localFile = file.path.startsWith(uploadDir + '/') ? file.path.substring(uploadDir.length + 1) : file.path;
+      const finalUrl = `https://${bucketName}.s3.amazonaws.com/${fileName}`;
       db.run('INSERT INTO image (date, did, local_file, size, aws_file, url) VALUES (?, ?, ?, ?, ?, ?)', [
         currentDate,
         "UNKNOWN",
@@ -83,6 +73,20 @@ app.post('/image', upload.single('image'), (req, res) => {
           console.error(currentDate, "Error inserting record from", "UNKNOWN", "into database:", dbErr);
         }
       });
+
+      // send to AWS
+      const command = new PutObjectCommand(params);
+      const response = await s3Client.send(command);
+      if (response.$metadata.httpStatusCode !== 200) {
+        const errorTime = new Date().toISOString();
+        console.error(errorTime, "Error uploading to S3 with bad HTTP status. Metadata:", response.$metadata);
+        res.status(500).send(JSON.stringify({ success: false, message: "Got bad status of " + response.$metadata.httpStatusCode + " from AWS. See server logs at " + errorTime }));
+      } else {
+        fs.rm(file.path, (err) => {
+          if (err) {
+            console.error("Error deleting temp file", file.path, "with error:", err);
+          }
+        });
         res.send(JSON.stringify({ success: true, url: finalUrl }));
       }
     } catch (uploadError) {
@@ -103,7 +107,6 @@ process.on('SIGINT', () => {
     if (err) {
       return console.error('Error closing DB connection', err);
     }
-    console.log('Closed DB connection.');
     process.exit(0);
   });
 });