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

import Project from '../models/project';

// 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);
        return res.json({ success: false });
      }
      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);
          return 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) {
  ids.forEach(id => {
    files.id(id).remove();
  });
}

function deleteChild(files, parentId, id) {
  files = 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]);
    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);
    })
  });
}