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

import archiver from 'archiver';
import format from 'date-fns/format';
import isUrl from 'is-url';
import jsdom, { serializeDocument } from 'jsdom';
import isBefore from 'date-fns/is_before';
import isAfter from 'date-fns/is_after';
import request from 'request';
import slugify from 'slugify';
import Project from '../models/project';
import User from '../models/user';
import { resolvePathToFile } from '../utils/filePath';
import generateFileSystemSafeName from '../utils/generateFileSystemSafeName';
import { deleteObjectsFromS3, getObjectKey } from './aws.controller';

export function createProject(req, res) {
  let projectValues = {
    user: req.user._id
  };

  projectValues = Object.assign(projectValues, req.body);

  Project.create(projectValues, (err, newProject) => {
    if (err) {
      res.json({ success: false });
      return;
    }
    Project.populate(
      newProject,
      { path: 'user', select: 'username' },
      (innerErr, newProjectWithUser) => {
        if (innerErr) {
          res.json({ success: false });
          return;
        }
        res.json(newProjectWithUser);
      }
    );
  });
}

export function updateProject(req, res) {
  Project.findById(req.params.project_id, (findProjectErr, project) => {
    if (!project.user.equals(req.user._id)) {
      res.status(403).send({ success: false, message: 'Session does not match owner of project.' });
      return;
    }
    if (req.body.updatedAt && isAfter(new Date(project.updatedAt), req.body.updatedAt)) {
      res.status(409).send({ success: false, message: 'Attempted to save stale version of project.' });
      return;
    }
    Project.findByIdAndUpdate(
      req.params.project_id,
      {
        $set: req.body
      },
      {
        new: true
      }
    )
      .populate('user', 'username')
      .exec((updateProjectErr, updatedProject) => {
        if (updateProjectErr) {
          console.log(updateProjectErr);
          res.json({ success: false });
          return;
        }
        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, savedProject) => {
            if (innerErr) {
              console.log(innerErr);
              res.json({ success: false });
              return;
            }
            res.json(savedProject);
          });
        } else {
          res.json(updatedProject);
        }
      });
  });
}

export function getProject(req, res) {
  const projectId = req.params.project_id;
  Project.findById(projectId)
    .populate('user', 'username')
    .exec((err, project) => { // eslint-disable-line
      if (err) {
        return res.status(404).send({ message: 'Project with that id does not exist' });
      } else if (!project) {
        Project.findOne({ slug: projectId })
          .populate('user', 'username')
          .exec((innerErr, projectBySlug) => {
            if (innerErr || !projectBySlug) {
              return res.status(404).send({ message: 'Project with that id does not exist' });
            }
            return res.json(projectBySlug);
          });
      } else {
        return res.json(project);
      }
    });
}

function deleteFilesFromS3(files) {
  deleteObjectsFromS3(files.filter((file) => {
    if (file.url) {
      if (!process.env.S3_DATE || (
        process.env.S3_DATE &&
        isBefore(new Date(process.env.S3_DATE), new Date(file.createdAt)))) {
        return true;
      }
    }
    return false;
  })
    .map(file => getObjectKey(file.url)));
}

export function deleteProject(req, res) {
  Project.findById(req.params.project_id, (findProjectErr, project) => {
    if (!project.user.equals(req.user._id)) {
      res.status(403).json({ success: false, message: 'Session does not match owner of project.' });
      return;
    }
    deleteFilesFromS3(project.files);
    Project.remove({ _id: req.params.project_id }, (removeProjectError) => {
      if (removeProjectError) {
        res.status(404).send({ message: 'Project with that id does not exist' });
        return;
      }
      res.json({ success: true });
    });
  });
}

export function getProjectsForUserId(userId) {
  return new Promise((resolve, reject) => {
    Project.find({ user: userId })
      .sort('-createdAt')
      .select('name files id createdAt updatedAt')
      .exec((err, projects) => {
        if (err) {
          console.log(err);
        }
        resolve(projects);
      });
  });
}

export function getProjectAsset(req, res) {
  Project.findById(req.params.project_id)
    .populate('user', 'username')
    .exec((err, project) => { // eslint-disable-line
      if (err) {
        return res.status(404).send({ message: 'Project with that id does not exist' });
      }
      if (!project) {
        return res.status(404).send({ message: 'Project with that id does not exist' });
      }

      const filePath = req.params[0];
      const resolvedFile = resolvePathToFile(filePath, project.files);
      if (!resolvedFile) {
        return res.status(404).send({ message: 'Asset does not exist' });
      }
      if (!resolvedFile.url) {
        return res.send(resolvedFile.content);
      }
      request({ method: 'GET', url: resolvedFile.url, encoding: null }, (innerErr, response, body) => {
        if (innerErr) {
          return res.status(404).send({ message: 'Asset does not exist' });
        }
        return res.send(body);
      });
    });
}

export function getProjectsForUserName(username) {

}

export function getProjects(req, res) {
  if (req.user) {
    getProjectsForUserId(req.user._id)
      .then((projects) => {
        res.json(projects);
      });
  } else {
    // could just move this to client side
    res.json([]);
  }
}

export function getProjectsForUser(req, res) {
  if (req.params.username) {
    User.findOne({ username: req.params.username }, (err, user) => {
      if (!user) {
        res.status(404).json({ message: 'User with that username does not exist.' });
        return;
      }
      Project.find({ user: user._id })
        .sort('-createdAt')
        .select('name files id createdAt updatedAt')
        .exec((innerErr, projects) => res.json(projects));
    });
  } else {
    // could just move this to client side
    res.json([]);
  }
}

export function projectExists(projectId, callback) {
  Project.findById(projectId, (err, project) => (
    project ? callback(true) : callback(false)
  ));
}

export function projectForUserExists(username, projectId, callback) {
  User.findOne({ username }, (err, user) => {
    if (!user) {
      callback(false);
      return;
    }
    Project.findOne({ _id: projectId, user: user._id }, (innerErr, project) => {
      if (project) {
        callback(true);
        return;
      }
      Project.findOne({ slug: projectId, user: user._id }, (slugError, projectBySlug) => {
        if (projectBySlug) {
          callback(true);
          return;
        }
        callback(false);
      });
    });
  });
}

function bundleExternalLibs(project, zip, callback) {
  const indexHtml = project.files.find(file => file.name === 'index.html');
  let numScriptsResolved = 0;
  let numScriptTags = 0;

  function resolveScriptTagSrc(scriptTag, document) {
    const path = scriptTag.src.split('/');
    const filename = path[path.length - 1];
    const { src } = scriptTag;

    if (!isUrl(src)) {
      numScriptsResolved += 1;
      if (numScriptsResolved === numScriptTags) {
        indexHtml.content = serializeDocument(document);
        callback();
      }
      return;
    }

    request({ method: 'GET', url: src, encoding: null }, (err, response, body) => {
      if (err) {
        console.log(err);
      } else {
        zip.append(body, { name: filename });
        scriptTag.src = filename;
      }

      numScriptsResolved += 1;
      if (numScriptsResolved === numScriptTags) {
        indexHtml.content = serializeDocument(document);
        callback();
      }
    });
  }

  jsdom.env(indexHtml.content, (innerErr, window) => {
    const indexHtmlDoc = window.document;
    const scriptTags = indexHtmlDoc.getElementsByTagName('script');
    numScriptTags = scriptTags.length;
    for (let i = 0; i < numScriptTags; i += 1) {
      resolveScriptTagSrc(scriptTags[i], indexHtmlDoc);
    }
    if (numScriptTags === 0) {
      indexHtml.content = serializeDocument(document);
      callback();
    }
  });
}

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;
  let numCompletedFiles = 0;

  zip.on('error', (err) => {
    res.status(500).send({ error: err.message });
  });

  const currentTime = format(new Date(), 'YYYY_MM_DD_HH_mm_ss');
  project.slug = slugify(project.name, '_');
  res.attachment(`${generateFileSystemSafeName(project.slug)}_${currentTime}.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();
      }
    }
  }

  bundleExternalLibs(project, zip, () => {
    addFileToZip(rootFile, '/');
  });
}

export function downloadProjectAsZip(req, res) {
  Project.findById(req.params.project_id, (err, project) => {
    // save project to some path
    buildZip(project, req, res);
  });
}