Adding GridFS

Author: Thomasraf

app.js

@@ -8,6 +8,9 @@ var flash = require('express-flash');
 var cookieParser = require('cookie-parser');
 var router = express.Router();
 var MongoStore = require('connect-mongo')(session);
+const multer = require('multer');
+const GridFsStorage = require('multer-gridfs-storage');
+const Grid = require('gridfs-stream');
 
 require('./models/db');
 // require('./seed/product-seeder');
@@ -29,7 +32,7 @@ const options = {
 const databaseURL = 'mongodb+srv://databaseUser:coronavirus@shophub-mquaf.mongodb.net/ShopHub?retryWrites=true&w=majority';
 mongoose.connect(databaseURL, options);
 
-app.use(express.static('public'));
+app.use(express.static('./public'));
 app.set('views', path.join(__dirname, 'views'));
 app.engine('.hbs', exphbs({defaultLayout: 'main', extname: '.hbs'}));
 app.set('view engine', '.hbs');
@@ -71,6 +74,7 @@ app.use(function(req,res,next){
     next();
 });
 
+
 app.listen(app.get('port'), function(){
   console.log('server started on port ' + app.get('port'));
 });

controllers/addProductController.js

@@ -11,36 +11,46 @@ var Order = require('../models/order');
 var Comment = require('../models/comments');
 
 const multer = require('multer');
-const storage = multer.diskStorage({
-    destination: function(req, file, cb){
-        cb(null, './pictures/')
-    },
-    filename: function(req,file, cb){
-        cb(null, file.originalname);
-    }
+const GridFsStorage = require('multer-gridfs-storage');
+const Grid = require('gridfs-stream');
+
+const databaseURL = 'mongodb+srv://databaseUser:coronavirus@shophub-mquaf.mongodb.net/ShopHub?retryWrites=true&w=majority';
+const conn = mongoose.createConnection(databaseURL);
+let gfs;
+
+conn.once('open', () => {
+    gfs = Grid(conn.db, mongoose.mongo);
+    gfs.collection('uploads');
 })
-const fileFilter = (req,file, cb) => {
-    if(file.mimetype === 'image/jpg' || file.mimetype === 'image/jpeg' ||file.mimetype === 'image/png'){
-        cb(null,true);
-    }
-    else{
-        cb(null,false);
-    }
-}
-const upload = multer({
-    storage: storage,
-    limits: {
-        fileFilter: fileFilter
+
+const storage = new GridFsStorage({
+    url: databaseURL,
+    file: (req, file) => {
+      return new Promise((resolve, reject) => {
+        crypto.randomBytes(16, (err, buf) => {
+          if (err) {
+            return reject(err);
+          }
+          const filename = buf.toString('hex') + path.extname(file.originalname);
+          const fileInfo = {
+            filename: filename,
+            bucketName: 'uploads'
+          };
+          resolve(fileInfo);
+        });
+      });
     }
-})
+  });
+  const upload = multer({ storage });
 
 router.get('/', (req, res) => {
     res.render('addProduct', {
     });
 });
 
-router.post('/',upload.single('productImage') ,(req, res) => {
-    if(req.body.productName == '' || req.body.picture == '' || req.body.price == '' || req.body.description == ''){
+router.post('/', upload.single('picture') ,(req, res) => {
+    console.log(req.file)
+    if(req.body.productName == '' || req.body.price == '' || req.body.description == ''){
         res.render('addProduct',{
             message: "Please complete the necessary information for the product"
         })
@@ -59,8 +69,7 @@ router.post('/',upload.single('productImage') ,(req, res) => {
         }
         else{
             var product = new Product();
-            // product.imagePath = `pictures/${req.body.picture}`;
-            product.imagePath = req.file.path;
+            // product.imagePath = `pictures/${req.file.filename}`;
             product.title = req.body.productName;
             product.price = req.body.price;
             product.description = req.body.description;

package-lock.json

@@ -20,11 +20,12 @@
     "express-handlebars": "^3.1.0",
     "express-session": "^1.17.0",
     "express-validator": "^6.4.0",
+    "gridfs-stream": "^1.1.1",
     "locus": "^2.0.4",
     "mongodb": "^3.5.5",
     "mongoose": "^5.9.5",
     "multer": "^1.4.2",
-    "multer-gridfs-storage": "^4.0.2",
+    "multer-gridfs-storage": "^4.0.3",
     "nodemon": "^2.0.2",
     "react-modal": "^3.11.2"
   },

package.json

@@ -47,7 +47,7 @@ input[type=number]::-webkit-outer-spin-button {
                 <div class="col-lg-6">
                     <div id="ui">
                         <h1 class="text-white">Add a Product</h1>
-                        <form class="form-group text-center" action="/addProduct" method="POST">
+                        <form class="form-group text-center" action="/addProduct" method="POST" enctype="multipart/form-data">
                             <div class="row">
                                 <div class="col-lg-6">
                                     <label>Product Name: </label>