p5.js-web-editor/server/controllers/project.controller.js

160 lines
4.7 KiB
JavaScript
Raw Normal View History

2016-06-23 22:29:55 +00:00
import Project from '../models/project';
import User from '../models/user';
import archiver from 'archiver';
import request from 'request';
2016-06-17 18:11:52 +00:00
export function createProject(req, res) {
2016-07-15 17:36:33 +00:00
let projectValues = {
2016-07-08 18:57:22 +00:00
user: req.user ? req.user._id : undefined // eslint-disable-line no-underscore-dangle
2016-06-23 22:29:55 +00:00
};
2016-06-20 20:29:08 +00:00
2016-07-15 17:36:33 +00:00
projectValues = Object.assign(projectValues, req.body);
2016-06-23 22:29:55 +00:00
Project.create(projectValues, (err, newProject) => {
if (err) { return res.json({ success: false }); }
2016-07-15 15:54:47 +00:00
Project.populate(newProject,
{ path: 'user', select: 'username' },
2016-07-15 15:54:47 +00:00
(innerErr, newProjectWithUser) => {
if (innerErr) { return res.json({ success: false }); }
return res.json(newProjectWithUser);
});
2016-06-23 22:29:55 +00:00
});
2016-06-18 22:33:49 +00:00
}
export function updateProject(req, res) {
2016-06-28 23:35:56 +00:00
Project.findByIdAndUpdate(req.params.project_id,
2016-06-23 22:29:55 +00:00
{
$set: req.body
2016-07-15 15:54:47 +00:00
})
.populate('user', 'username')
.exec((err, updatedProject) => {
2016-08-24 21:29:44 +00:00
if (err) {
console.log(err);
return res.json({ success: false });
}
if (updatedProject.files.length !== req.body.files.length) {
const oldFileIds = updatedProject.files.map(file => file.id);
const newFileIds = req.body.files.map(file => file.id);
const staleIds = oldFileIds.filter(id => newFileIds.indexOf(id) === -1);
staleIds.forEach(staleId => {
updatedProject.files.id(staleId).remove();
});
updatedProject.save((innerErr) => {
if (innerErr) {
console.log(innerErr);
return res.json({ success: false });
}
return res.json(updatedProject);
});
}
return res.json(updatedProject);
2016-06-23 22:29:55 +00:00
});
2016-06-18 22:33:49 +00:00
}
export function getProject(req, res) {
2016-07-15 15:54:47 +00:00
Project.findById(req.params.project_id)
.populate('user', 'username')
.exec((err, project) => {
if (err) {
return res.status(404).send({ message: 'Project with that id does not exist' });
}
return res.json(project);
});
2016-06-23 22:29:55 +00:00
}
2016-07-01 15:30:40 +00:00
export function deleteProject(req, res) {
Project.remove({ _id: req.params.project_id }, (err) => {
if (err) {
return res.status(404).send({ message: 'Project with that id does not exist' });
}
return res.json({ success: true });
});
}
2016-07-01 15:30:40 +00:00
export function getProjects(req, res) {
if (req.user) {
Project.find({ user: req.user._id }) // eslint-disable-line no-underscore-dangle
2016-07-01 15:30:40 +00:00
.sort('-createdAt')
.select('name files id createdAt updatedAt')
2016-07-01 15:30:40 +00:00
.exec((err, projects) => {
res.json(projects);
});
} else {
// could just move this to client side
return res.json([]);
}
}
export function getProjectsForUser(req, res) {
if (req.params.username) {
User.findOne({ username: req.params.username }, (err, user) => {
if (!user) {
return res.status(404).json({ message: 'User with that username does not exist.' });
} else {
Project.find({ user: user._id }) // eslint-disable-line no-underscore-dangle
.sort('-createdAt')
.select('name files id createdAt updatedAt')
.exec((err, projects) => res.json(projects));
}
return null;
});
} else {
// could just move this to client side
return res.json([]);
}
return null;
}
function buildZip(project, req, res) {
const zip = archiver('zip');
const rootFile = project.files.find(file => file.name === 'root');
const numFiles = project.files.filter(file => file.fileType !== 'folder').length;
const files = project.files;
const projectName = project.name;
let numCompletedFiles = 0;
zip.on('error', function (err) {
res.status(500).send({ error: err.message });
});
res.attachment(`${project.name}.zip`);
zip.pipe(res);
function addFileToZip(file, path) {
if (file.fileType === 'folder') {
const newPath = file.name === 'root' ? path : `${path}${file.name}/`;
file.children.forEach(fileId => {
const childFile = files.find(f => f.id === fileId);
(() => {
addFileToZip(childFile, newPath);
})();
});
} else {
if (file.url) {
request({ method: 'GET', url: file.url, encoding: null }, (err, response, body) => {
zip.append(body, { name: `${path}${file.name}` });
numCompletedFiles += 1;
if (numCompletedFiles === numFiles) {
zip.finalize();
}
});
} else {
zip.append(file.content, { name: `${path}${file.name}` });
numCompletedFiles += 1;
if (numCompletedFiles === numFiles) {
zip.finalize();
}
}
}
}
addFileToZip(rootFile, '/');
}
export function downloadProjectAsZip(req, res) {
Project.findById(req.params.project_id, (err, project) => {
// save project to some path
buildZip(project, req, res);
});
}