p5.js-web-editor/server/migrations/s3UnderUser.js

/* eslint-disable */
import s3 from 's3';
import path from 'path';
import mongoose from 'mongoose';
import User from '../models/user';
import Project from '../models/project';
require('dotenv').config({path: path.resolve('.env')});
mongoose.connect('mongodb://localhost:27017/p5js-web-editor');
mongoose.connection.on('error', () => {
  console.error('MongoDB Connection Error. Please make sure that MongoDB is running.');
  process.exit(1);
});

let client = s3.createClient({
  maxAsyncS3: 20,
  s3RetryCount: 3,
  s3RetryDelay: 1000,
  multipartUploadThreshold: 20971520, // this is the default (20 MB)
  multipartUploadSize: 15728640, // this is the default (15 MB)
  s3Options: {
    accessKeyId: `${process.env.AWS_ACCESS_KEY}`,
    secretAccessKey: `${process.env.AWS_SECRET_KEY}`,
    region: `${process.env.AWS_REGION}`
  },
});

Project.find({}, (err, projects) => {
  projects.forEach((project, projectIndex) => {
    if (!project.user) return;
    const userId = project.user.valueOf();
    project.files.forEach((file, fileIndex) => {
      if (file.url && file.url.includes(process.env.S3_BUCKET) && !file.url.includes(userId)) {
        const key = file.url.split('/').pop();
        console.log(key);
        const params = {
          Bucket: `${process.env.S3_BUCKET}`,
          CopySource: `${process.env.S3_BUCKET}/${key}`,
          Key: `${userId}/${key}`
        };
        try {
          client.moveObject(params)
          .on('err', (err) => {
            console.log(err);
          })
          .on('end', () => {
            file.url = (process.env.S3_BUCKET_URL_BASE ||
                        `https://s3-${process.env.AWS_REGION}.amazonaws.com/${process.env.S3_BUCKET}`) + `/${userId}/${key}`;
            project.save((err, savedProject) => {
              console.log(`updated file ${key}`);
            });
          });
        } catch(e) {
          console.log(e);
        }
      }
    });
  });
});