-import * as express from 'express'
-import * as multer from 'multer'
+import express, { RequestHandler } from 'express'
+import multer, { diskStorage } from 'multer'
+import { getLowercaseExtension } from '@shared/core-utils'
+import { HttpStatusCode } from '../../shared/models/http/http-error-codes'
+import { CONFIG } from '../initializers/config'
import { REMOTE_SCHEME } from '../initializers/constants'
-import { logger } from './logger'
-import { deleteFileAsync, generateRandomString } from './utils'
-import { extname } from 'path'
import { isArray } from './custom-validators/misc'
-import { CONFIG } from '../initializers/config'
+import { logger } from './logger'
+import { deleteFileAndCatch, generateRandomString } from './utils'
+import { getExtFromMimetype } from './video'
function buildNSFWFilter (res?: express.Response, paramNSFW?: string) {
if (paramNSFW === 'true') return true
if (paramNSFW === 'false') return false
if (paramNSFW === 'both') return undefined
- if (res && res.locals.oauth) {
+ if (res?.locals.oauth) {
const user = res.locals.oauth.token.User
// User does not want NSFW videos
return null
}
-function cleanUpReqFiles (req: { files: { [ fieldname: string ]: Express.Multer.File[] } | Express.Multer.File[] }) {
- const files = req.files
+function cleanUpReqFiles (req: express.Request) {
+ const filesObject = req.files
+ if (!filesObject) return
- if (!files) return
-
- if (isArray(files)) {
- (files as Express.Multer.File[]).forEach(f => deleteFileAsync(f.path))
+ if (isArray(filesObject)) {
+ filesObject.forEach(f => deleteFileAndCatch(f.path))
return
}
- for (const key of Object.keys(files)) {
- const file = files[ key ]
+ for (const key of Object.keys(filesObject)) {
+ const files = filesObject[key]
- if (isArray(file)) file.forEach(f => deleteFileAsync(f.path))
- else deleteFileAsync(file.path)
+ files.forEach(f => deleteFileAndCatch(f.path))
}
}
return host
}
-function badRequest (req: express.Request, res: express.Response) {
- return res.type('json').status(400).end()
+function badRequest (_req: express.Request, res: express.Response) {
+ return res.type('json')
+ .status(HttpStatusCode.BAD_REQUEST_400)
+ .end()
}
function createReqFiles (
fieldNames: string[],
- mimeTypes: { [ id: string ]: string },
- destinations: { [ fieldName: string ]: string }
-) {
- const storage = multer.diskStorage({
+ mimeTypes: { [id: string]: string | string[] },
+ destination = CONFIG.STORAGE.TMP_DIR
+): RequestHandler {
+ const storage = diskStorage({
destination: (req, file, cb) => {
- cb(null, destinations[ file.fieldname ])
+ cb(null, destination)
},
- filename: async (req, file, cb) => {
- let extension: string
- const fileExtension = extname(file.originalname)
- const extensionFromMimetype = mimeTypes[ file.mimetype ]
-
- // Take the file extension if we don't understand the mime type
- // We have the OGG/OGV exception too because firefox sends a bad mime type when sending an OGG file
- if (fileExtension === '.ogg' || fileExtension === '.ogv' || !extensionFromMimetype) {
- extension = fileExtension
- } else {
- extension = extensionFromMimetype
- }
-
- let randomString = ''
-
- try {
- randomString = await generateRandomString(16)
- } catch (err) {
- logger.error('Cannot generate random string for file name.', { err })
- randomString = 'fake-random-string'
- }
-
- cb(null, randomString + extension)
+ filename: (req, file, cb) => {
+ return generateReqFilename(file, mimeTypes, cb)
}
})
- let fields: { name: string, maxCount: number }[] = []
+ const fields: { name: string, maxCount: number }[] = []
for (const fieldName of fieldNames) {
fields.push({
name: fieldName,
return multer({ storage }).fields(fields)
}
+function createAnyReqFiles (
+ mimeTypes: { [id: string]: string | string[] },
+ fileFilter: (req: express.Request, file: Express.Multer.File, cb: (err: Error, result: boolean) => void) => void
+): RequestHandler {
+ const storage = diskStorage({
+ destination: (req, file, cb) => {
+ cb(null, CONFIG.STORAGE.TMP_DIR)
+ },
+
+ filename: (req, file, cb) => {
+ return generateReqFilename(file, mimeTypes, cb)
+ }
+ })
+
+ return multer({ storage, fileFilter }).any()
+}
+
function isUserAbleToSearchRemoteURI (res: express.Response) {
const user = res.locals.oauth ? res.locals.oauth.token.User : undefined
(CONFIG.SEARCH.REMOTE_URI.USERS === true && user !== undefined)
}
+function getCountVideos (req: express.Request) {
+ return req.query.skipCount !== true
+}
+
// ---------------------------------------------------------------------------
export {
buildNSFWFilter,
getHostWithPort,
+ createAnyReqFiles,
isUserAbleToSearchRemoteURI,
badRequest,
createReqFiles,
- cleanUpReqFiles
+ cleanUpReqFiles,
+ getCountVideos
+}
+
+// ---------------------------------------------------------------------------
+
+async function generateReqFilename (
+ file: Express.Multer.File,
+ mimeTypes: { [id: string]: string | string[] },
+ cb: (err: Error, name: string) => void
+) {
+ let extension: string
+ const fileExtension = getLowercaseExtension(file.originalname)
+ const extensionFromMimetype = getExtFromMimetype(mimeTypes, file.mimetype)
+
+ // Take the file extension if we don't understand the mime type
+ if (!extensionFromMimetype) {
+ extension = fileExtension
+ } else {
+ // Take the first available extension for this mimetype
+ extension = extensionFromMimetype
+ }
+
+ let randomString = ''
+
+ try {
+ randomString = await generateRandomString(16)
+ } catch (err) {
+ logger.error('Cannot generate random string for file name.', { err })
+ randomString = 'fake-random-string'
+ }
+
+ cb(null, randomString + extension)
}