From: Chocobozzz <florian.bigard@gmail.com>
Date: Sat, 9 Jan 2016 11:59:13 +0000 (+0100)
Subject: Fix upload of file which don't have any extension
X-Git-Tag: v0.0.1-alpha~1056
X-Git-Url: https://git.librecmc.org/?a=commitdiff_plain;h=e78a20c7f914423736f7b5b0bf981418e43b6c36;p=oweals%2Fpeertube.git

Fix upload of file which don't have any extension
---

diff --git a/routes/api/v1/videos.js b/routes/api/v1/videos.js
index ae70d1640..95c1dffc7 100644
--- a/routes/api/v1/videos.js
+++ b/routes/api/v1/videos.js
@@ -3,6 +3,7 @@
 
   var express = require('express')
   var config = require('config')
+  var crypto = require('crypto')
   var multer = require('multer')
   var router = express.Router()
 
@@ -58,7 +59,24 @@
     })
   }
 
-  var reqFiles = multer({ dest: uploads }).fields([{ name: 'input_video', maxCount: 1 }])
+  // multer configuration
+  var storage = multer.diskStorage({
+    destination: function (req, file, cb) {
+      cb(null, uploads)
+    },
+
+    filename: function (req, file, cb) {
+      var extension = ''
+      if (file.mimetype === 'video/webm') extension = 'webm'
+      else if (file.mimetype === 'video/mp4') extension = 'mp4'
+      else if (file.mimetype === 'video/ogg') extension = 'ogv'
+      crypto.pseudoRandomBytes(16, function (err, raw) {
+        var fieldname = err ? undefined : raw.toString('hex')
+        cb(null, fieldname + '.' + extension)
+      })
+    }
+  })
+  var reqFiles = multer({ storage: storage }).fields([{ name: 'input_video', maxCount: 1 }])
 
   router.get('/', miscMiddleware.cache(false), listVideos)
   router.post('/', reqFiles, reqValidator.videosAdd, miscMiddleware.cache(false), addVideos)