BE #6 Save image info after s3 upload

Author: Satyajit Dey

controllers/binaries.controller.js

@@ -3,6 +3,8 @@ const AWS = require('aws-sdk');
 const mime = require('mime-types');
 const {v4: uuidv4} = require('uuid');
 
+const ImageModel = require('../models/images.model');
+
 AWS.config.update({
     accessKeyId: process.env.IAM_ACCESS_KEY,
     secretAccessKey: process.env.IAM_SECRET_KEY,
@@ -14,7 +16,8 @@ const s3 = new AWS.S3();
 exports.uploadImage = (req, res) => {
     let fileInfo = {
         source: req.file.path,
-        key:  `${uuidv4()}---${req.file.filename}`,
+        key: `${uuidv4()}---${req.file.filename}`,
+        caption: req.body.caption,
         contentType: mime.lookup(req.file.path),
         author: req.jwt.name,
         authorEmail: req.jwt.email
@@ -46,11 +49,24 @@ let uploadFile = (fileInfo, res) => {
             }
         };
 
+        //Upload image to S3 bucket
         s3.upload(params, (err, data) => {
             if (err) {
                 return res.status(500).send({error: err});
             }
 
+            let keys = fileInfo.key.split('---');
+            const imageData = {
+                _id: keys[0],
+                fileName: keys[1],
+                caption: fileInfo.caption,
+                author: fileInfo.author
+            };
+
+            //After successful upload save the image info to Image Mongo model
+            ImageModel.saveImage(imageData);
+
+            //After successful upload delete the local image
             fs.unlink(fileInfo.source, (err) => {
                 if (err) throw err;
                 console.log(`${fileInfo.source} deleted.`)

models/images.model.js

@@ -1,9 +1,30 @@
 const mongoose = require('../services/db.service').mongoose;
 
 const imageSchema = new mongoose.Schema({
-    imageId: {type: String, required: true, trim: true},
+    _id: {type: String},
     fileName: {type: String, required: true, trim: true},
-    storage: {type: String, required: true, trim: true},
     caption: {type: String},
+    author: {type: String, trim: true},
+    credits: [String],
+    imageType: {type: String},
+    copyright: {type: String, trim: true},
     createdAt: {type: Date, default: Date.now},
-});
+    status: {type: String, default: 'published'}
+});
+
+const Images = mongoose.model('Image', imageSchema);
+
+exports.saveImage = (imageData) => {
+    const image = new Images(imageData);
+    return image.save();
+};
+
+exports.findById = (id) => {
+    Images.findById(id).then(result => {
+        return result;
+    });
+};
+
+exports.findByAuthor = (email) => {
+    return Images.find({author: email});
+};