89 lines
2.5 KiB
JavaScript
89 lines
2.5 KiB
JavaScript
|
const fs = require('fs');
|
||
|
|
const path = require('path');
|
||
|
|
const crypto = require('crypto');
|
||
|
|
const multer = require('multer');
|
||
|
|
const { sanitizeFilename } = require('@librechat/api');
|
||
|
|
const {
|
||
|
|
mergeFileConfig,
|
||
|
|
getEndpointFileConfig,
|
||
|
|
fileConfig: defaultFileConfig,
|
||
|
|
} = require('librechat-data-provider');
|
||
|
|
const { getAppConfig } = require('~/server/services/Config');
|
||
|
|
|
||
|
|
const storage = multer.diskStorage({
|
||
|
|
destination: function (req, file, cb) {
|
||
|
|
const appConfig = req.config;
|
||
|
|
const outputPath = path.join(appConfig.paths.uploads, 'temp', req.user.id);
|
||
|
|
if (!fs.existsSync(outputPath)) {
|
||
|
|
fs.mkdirSync(outputPath, { recursive: true });
|
||
|
|
}
|
||
|
|
cb(null, outputPath);
|
||
|
|
},
|
||
|
|
filename: function (req, file, cb) {
|
||
|
|
req.file_id = crypto.randomUUID();
|
||
|
|
file.originalname = decodeURIComponent(file.originalname);
|
||
|
|
const sanitizedFilename = sanitizeFilename(file.originalname);
|
||
|
|
cb(null, sanitizedFilename);
|
||
|
|
},
|
||
|
|
});
|
||
|
|
|
||
|
|
const importFileFilter = (req, file, cb) => {
|
||
|
|
if (file.mimetype === 'application/json') {
|
||
|
|
cb(null, true);
|
||
|
|
} else if (path.extname(file.originalname).toLowerCase() === '.json') {
|
||
|
|
cb(null, true);
|
||
|
|
} else {
|
||
|
|
cb(new Error('Only JSON files are allowed'), false);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
*
|
||
|
|
* @param {import('librechat-data-provider').FileConfig | undefined} customFileConfig
|
||
|
|
*/
|
||
|
|
const createFileFilter = (customFileConfig) => {
|
||
|
|
/**
|
||
|
|
* @param {ServerRequest} req
|
||
|
|
* @param {Express.Multer.File}
|
||
|
|
* @param {import('multer').FileFilterCallback} cb
|
||
|
|
*/
|
||
|
|
const fileFilter = (req, file, cb) => {
|
||
|
|
if (!file) {
|
||
|
|
return cb(new Error('No file provided'), false);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (req.originalUrl.endsWith('/speech/stt') && file.mimetype.startsWith('audio/')) {
|
||
|
|
return cb(null, true);
|
||
|
|
}
|
||
|
|
|
||
|
|
const endpoint = req.body.endpoint;
|
||
|
|
const endpointType = req.body.endpointType;
|
||
|
|
const endpointFileConfig = getEndpointFileConfig({
|
||
|
|
fileConfig: customFileConfig,
|
||
|
|
endpoint,
|
||
|
|
endpointType,
|
||
|
|
});
|
||
|
|
|
||
|
|
if (!defaultFileConfig.checkType(file.mimetype, endpointFileConfig.supportedMimeTypes)) {
|
||
|
|
return cb(new Error('Unsupported file type: ' + file.mimetype), false);
|
||
|
|
}
|
||
|
|
|
||
|
|
cb(null, true);
|
||
|
|
};
|
||
|
|
|
||
|
|
return fileFilter;
|
||
|
|
};
|
||
|
|
|
||
|
|
const createMulterInstance = async () => {
|
||
|
|
const appConfig = await getAppConfig();
|
||
|
|
const fileConfig = mergeFileConfig(appConfig?.fileConfig);
|
||
|
|
const fileFilter = createFileFilter(fileConfig);
|
||
|
|
return multer({
|
||
|
|
storage,
|
||
|
|
fileFilter,
|
||
|
|
limits: { fileSize: fileConfig.serverFileSizeLimit },
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
module.exports = { createMulterInstance, storage, importFileFilter };
|