103 lines
3.2 KiB
JavaScript
103 lines
3.2 KiB
JavaScript
import each from 'async/each';
|
|
import moment from 'moment';
|
|
|
|
import Project from '../models/project';
|
|
import { resolvePathToFile } from '../utils/filePath';
|
|
import { deleteObjectsFromS3 } from './aws.controller';
|
|
|
|
// Bug -> timestamps don't get created, but it seems like this will
|
|
// be fixed in mongoose soon
|
|
// https://github.com/Automattic/mongoose/issues/4049
|
|
export function createFile(req, res) {
|
|
Project.findByIdAndUpdate(req.params.project_id,
|
|
{
|
|
$push: {
|
|
files: req.body
|
|
}
|
|
},
|
|
{
|
|
new: true
|
|
}, (err, updatedProject) => {
|
|
if (err) {
|
|
console.log(err);
|
|
res.json({ success: false });
|
|
return;
|
|
}
|
|
const newFile = updatedProject.files[updatedProject.files.length - 1];
|
|
updatedProject.files.id(req.body.parentId).children.push(newFile.id);
|
|
updatedProject.save((innerErr) => {
|
|
if (innerErr) {
|
|
console.log(innerErr);
|
|
res.json({ success: false });
|
|
return;
|
|
}
|
|
res.json(updatedProject.files[updatedProject.files.length - 1]);
|
|
});
|
|
});
|
|
}
|
|
|
|
function getAllDescendantIds(files, nodeId) {
|
|
return files.find(file => file.id === nodeId).children
|
|
.reduce((acc, childId) => (
|
|
[...acc, childId, ...getAllDescendantIds(files, childId)]
|
|
), []);
|
|
}
|
|
|
|
function deleteMany(files, ids) {
|
|
const objectKeys = [];
|
|
|
|
each(ids, (id, cb) => {
|
|
if (files.id(id).url) {
|
|
if (!process.env.S3_DATE || (process.env.S3_DATE && moment(process.env.S3_DATE) < moment(files.id(id).createdAt))) {
|
|
const urlComponents = files.id(id).url.split('/');
|
|
const key = urlComponents.pop();
|
|
const userId = urlComponents.pop();
|
|
objectKeys.push(`${userId}/${key}`);
|
|
}
|
|
}
|
|
files.id(id).remove();
|
|
cb();
|
|
}, (err) => {
|
|
deleteObjectsFromS3(objectKeys);
|
|
});
|
|
}
|
|
|
|
function deleteChild(files, parentId, id) {
|
|
return files.map((file) => {
|
|
if (file.id === parentId) {
|
|
file.children = file.children.filter(child => child !== id);
|
|
return file;
|
|
}
|
|
return file;
|
|
});
|
|
}
|
|
|
|
export function deleteFile(req, res) {
|
|
Project.findById(req.params.project_id, (err, project) => {
|
|
const idsToDelete = getAllDescendantIds(project.files, req.params.file_id);
|
|
deleteMany(project.files, [req.params.file_id, ...idsToDelete]);
|
|
project.files = deleteChild(project.files, req.query.parentId, req.params.file_id);
|
|
// project.files.id(req.params.file_id).remove();
|
|
// const childrenArray = project.files.id(req.query.parentId).children;
|
|
// project.files.id(req.query.parentId).children = childrenArray.filter(id => id !== req.params.file_id);
|
|
project.save((innerErr) => {
|
|
res.json(project.files);
|
|
});
|
|
});
|
|
}
|
|
|
|
export function getFileContent(req, res) {
|
|
Project.findById(req.params.project_id, (err, project) => {
|
|
if (err) {
|
|
res.status(404).send({ success: false, message: 'Project with that id does not exist.' });
|
|
return;
|
|
}
|
|
const filePath = req.params[0];
|
|
const resolvedFile = resolvePathToFile(filePath, project.files);
|
|
if (!resolvedFile) {
|
|
res.status(404).send({ success: false, message: 'File with that name and path does not exist.' });
|
|
return;
|
|
}
|
|
res.send(resolvedFile.content);
|
|
});
|
|
}
|