mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-16 08:20:14 +01:00
dbe4dd96b4
* 🧹 chore: Update logger imports to use @librechat/data-schemas across multiple files and remove unused sleep function from queue.js (#9930) * chore: Replace local isEnabled utility with @librechat/api import across multiple files, update test files * chore: Replace local logger import with @librechat/data-schemas logger in countTokens.js and fork.js * chore: Update logs volume path in docker-compose.yml to correct directory * chore: import order of isEnabled in static.js
71 lines
2.6 KiB
JavaScript
71 lines
2.6 KiB
JavaScript
const fs = require('fs');
|
|
const path = require('path');
|
|
const sharp = require('sharp');
|
|
const { logger } = require('@librechat/data-schemas');
|
|
const { getStrategyFunctions } = require('../strategies');
|
|
const { resizeImageBuffer } = require('./resize');
|
|
|
|
/**
|
|
* Converts an image file or buffer to target output type with specified resolution.
|
|
*
|
|
* @param {ServerRequest} req - The request object, containing user and app configuration data.
|
|
* @param {Buffer | Express.Multer.File} file - The file object, containing either a path or a buffer.
|
|
* @param {'low' | 'high'} [resolution='high'] - The desired resolution for the output image.
|
|
* @param {string} [basename=''] - The basename of the input file, if it is a buffer.
|
|
* @returns {Promise<{filepath: string, bytes: number, width: number, height: number}>} An object containing the path, size, and dimensions of the converted image.
|
|
* @throws Throws an error if there is an issue during the conversion process.
|
|
*/
|
|
async function convertImage(req, file, resolution = 'high', basename = '') {
|
|
try {
|
|
const appConfig = req.config;
|
|
let inputBuffer;
|
|
let outputBuffer;
|
|
let extension = path.extname(file.path ?? basename).toLowerCase();
|
|
|
|
// Check if the input is a buffer or a file path
|
|
if (Buffer.isBuffer(file)) {
|
|
inputBuffer = file;
|
|
} else if (file && file.path) {
|
|
const inputFilePath = file.path;
|
|
inputBuffer = await fs.promises.readFile(inputFilePath);
|
|
} else {
|
|
throw new Error('Invalid input: file must be a buffer or contain a valid path.');
|
|
}
|
|
|
|
// Resize the image buffer
|
|
const {
|
|
buffer: resizedBuffer,
|
|
width,
|
|
height,
|
|
} = await resizeImageBuffer(inputBuffer, resolution);
|
|
|
|
// Check if the file is already in target format; if it isn't, convert it:
|
|
const targetExtension = `.${appConfig.imageOutputType}`;
|
|
if (extension === targetExtension) {
|
|
outputBuffer = resizedBuffer;
|
|
} else {
|
|
outputBuffer = await sharp(resizedBuffer).toFormat(appConfig.imageOutputType).toBuffer();
|
|
extension = targetExtension;
|
|
}
|
|
|
|
// Generate a new filename for the output file
|
|
const newFileName =
|
|
path.basename(file.path ?? basename, path.extname(file.path ?? basename)) + extension;
|
|
|
|
const { saveBuffer } = getStrategyFunctions(appConfig.fileStrategy);
|
|
|
|
const savedFilePath = await saveBuffer({
|
|
userId: req.user.id,
|
|
buffer: outputBuffer,
|
|
fileName: newFileName,
|
|
});
|
|
|
|
const bytes = Buffer.byteLength(outputBuffer);
|
|
return { filepath: savedFilePath, bytes, width, height };
|
|
} catch (err) {
|
|
logger.error(err);
|
|
throw err;
|
|
}
|
|
}
|
|
|
|
module.exports = { convertImage };
|