✨ Create endpoints for upload single image and image gallery

Author: dinushchathurya

routes/products.js

@@ -1,6 +1,31 @@
 const express = require('express');
 const router = express.Router();
 const mongoose = require('mongoose');
+const multer = require('multer');
+
+const FILE_TYPE_MAP = {
+    'image/png': 'png',
+    'image/jpeg': 'jpeg',
+    'image/jpg': 'jpg',
+}
+
+const storage = multer.diskStorage({
+    destination: function (req, file, cb) {
+        const isValid = FILE_TYPE_MAP[file.mimetype];
+        let uploadError = new Error('Invalid Image Type');
+        if(isValid){
+            uploadError = null
+        }
+        cb(uploadError, '/public/uploads')
+    },
+    filename: function (req, file, cb) {
+        const fileName = file.originalname.split('').json('-');
+        const extension = FILE_TYPE_MAP[file.mimetype];
+        cb(null, `${fileName}-${Date.now()}.${extension}`)
+    }
+})
+
+const upload  = multer({ storage: storage })
 
 const Product = require('../models/product');
 const Category = require('../models/category');
@@ -30,7 +55,7 @@ router.get('/:id', async (req, res) => {
     res.status(200).send(product)
 })
 
-router.post('/', async (req, res) => {
+router.post('/', upload.single('image'), async (req, res) => {
 
     if(!mongoose.isValidObjectId(req.params.id)){
         res.status(400).send('Invalid Product ID')
@@ -40,11 +65,18 @@ router.post('/', async (req, res) => {
     if (!category)
         return res.status(400).send('Invalid Category')
 
+    const file = req.file;
+    if (!file)
+        return res.status(400).send('No image in the request')
+
+    const fileName = file.filename;
+    const basePath = `${req.protocol}://${req.get('host')}/public/uploads/`;
+
     let product = new Product({
         name: req.body.name,
         description: req.body.description,
         richDescription: req.body.richDescription,
-        image: req.body.image,
+        image: `${basePath}${fileName}`,
         brand: req.body.brand,
         price: req.body.price,
         category: req.body.category,
@@ -120,5 +152,32 @@ router.get('/get/featured/:count', async (req, res) => {
     res.status(200).send(products);
 })
 
+router.put('/gallery-images/:id', upload.array('images', 10), async (req, res) => {
+
+    if (!mongoose.isValidObjectId(req.params.id)) {
+        res.status(400).send('Invalid Product ID')
+    }
+
+    const files = req.files;
+    let imagesPaths = [];
+    const basePath = `${req.protocol}://${req.get('host')}/public/uploads/`;
+    if(files){
+        files.map(file => {
+            imagesPaths.push(`${basePath}${file.fileName}`);
+        })
+    }
+
+    const product = await Product.findByIdAndUpdate(req.params.id, {
+
+        image: imagesPaths,
+    },
+    {
+        new: true
+    })
+
+    if (!product)
+        return res.status(500).send('Product cannot be updated')
+    res.send(product);
+})
 
 module.exports = router;