45 lines
1.4 KiB
JavaScript
45 lines
1.4 KiB
JavaScript
|
import path from "path";
|
||
|
|
import { FILE_STORAGE_LOCATION } from "../../consts/filePaths";
|
||
|
|
import FileModel from "../../models/FileModel";
|
||
|
|
import { createFileIO } from "./ioHelpers";
|
||
|
|
import { connectMongo } from "./mongoHelpers";
|
||
|
|
import { randomString } from "./stringHelpers";
|
||
|
|
|
||
|
|
connectMongo();
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Returns the the files object in mongo for the given id
|
||
|
|
*
|
||
|
|
* @param {string} fileId The files id
|
||
|
|
* @return The file object or null if not found
|
||
|
|
*/
|
||
|
|
export async function getFile(fileId) {
|
||
|
|
return await FileModel.findOne({ fileId: fileId });
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Creates the file object in mongo and stores it to the storage location
|
||
|
|
*
|
||
|
|
* @param {UserModel} uploader The user who uploaded the file
|
||
|
|
* @param {[]} fileData The file data for the upload
|
||
|
|
*/
|
||
|
|
export async function createFile(uploader, fileName, buffer, contentType) {
|
||
|
|
const fileId = randomString(process.env.FILE_ID_LENGTH);
|
||
|
|
const extention = fileName.split(".")[1].toLowerCase();
|
||
|
|
// Todo: Check if the file was actually saved to
|
||
|
|
// disk and create a return type so we can notify the user what happened
|
||
|
|
await createFileIO(
|
||
|
|
`${FILE_STORAGE_LOCATION}${path.sep}${uploader.uploadKey}`,
|
||
|
|
`${fileId}.${extention}`,
|
||
|
|
buffer
|
||
|
|
);
|
||
|
|
const file = await FileModel.create({
|
||
|
|
uploader: uploader._id,
|
||
|
|
fileId: fileId,
|
||
|
|
uploadDate: new Date(),
|
||
|
|
contentType: contentType,
|
||
|
|
});
|
||
|
|
await file.save();
|
||
|
|
return `${fileId}.${extention}`;
|
||
|
|
}
|