138 lines
3.7 KiB
JavaScript
138 lines
3.7 KiB
JavaScript
import mongoose from 'mongoose';
|
|
import User from '../models/user';
|
|
import Project from '../models/project';
|
|
import Collection from '../models/collection';
|
|
import { moveObjectToUserInS3 } from '../controllers/aws.controller';
|
|
import mail from '../utils/mail';
|
|
import { renderAccountConsolidation } from '../views/mail';
|
|
|
|
|
|
const mongoConnectionString = process.env.MONGO_URL;
|
|
const { ObjectId } = mongoose.Types;
|
|
// Connect to MongoDB
|
|
mongoose.Promise = global.Promise;
|
|
mongoose.connect(mongoConnectionString, { useNewUrlParser: true, useUnifiedTopology: true });
|
|
mongoose.set('useCreateIndex', true);
|
|
mongoose.connection.on('error', () => {
|
|
console.error('MongoDB Connection Error. Please make sure that MongoDB is running.');
|
|
process.exit(1);
|
|
});
|
|
|
|
/*
|
|
* Requires the MongoDB Node.js Driver
|
|
* https://mongodb.github.io/node-mongodb-native
|
|
*/
|
|
|
|
const agg = [
|
|
{
|
|
$project: {
|
|
email: {
|
|
$toLower: [
|
|
'$email'
|
|
]
|
|
}
|
|
}
|
|
}, {
|
|
$group: {
|
|
_id: '$email',
|
|
total: {
|
|
$sum: 1
|
|
}
|
|
}
|
|
}, {
|
|
$match: {
|
|
total: {
|
|
$gt: 1
|
|
}
|
|
}
|
|
}, {
|
|
$sort: {
|
|
total: -1
|
|
}
|
|
}
|
|
];
|
|
|
|
|
|
// steps to make this work
|
|
// iterate through the results
|
|
// check if any files are on AWS
|
|
// if so, move them to the right user bucket
|
|
// then, update the user to currentUser
|
|
// then, after updating all of the projects
|
|
// also update the collections
|
|
// delete other users
|
|
// update user email so it is all lowercase
|
|
// then, send the email
|
|
// then, figure out how to iterate through all of the users.
|
|
|
|
let currentUser = null;
|
|
let duplicates = null;
|
|
User.aggregate(agg).then((result) => {
|
|
console.log(result);
|
|
const email = result[0]._id;
|
|
return User.find({ email }).collation({ locale: 'en', strength: 2 })
|
|
.sort({ createdAt: 1 }).exec();
|
|
}).then((result) => {
|
|
[currentUser, ...duplicates] = result;
|
|
console.log('Current User: ', currentUser._id, ' ', currentUser.email);
|
|
duplicates = duplicates.map(dup => dup._id);
|
|
console.log('Duplicates: ', duplicates);
|
|
return Project.find({
|
|
user: { $in: duplicates }
|
|
}).exec();
|
|
}).then((sketches) => {
|
|
const saveSketchPromises = [];
|
|
sketches.forEach((sketch) => {
|
|
const moveSketchFilesPromises = [];
|
|
sketch.files.forEach((file) => {
|
|
if (file.url && file.url.includes(process.env.S3_BUCKET_URL_BASE)) {
|
|
const fileSavePromise = moveObjectToUserInS3(file.url, currentUser._id)
|
|
.then((newUrl) => {
|
|
file.url = newUrl;
|
|
});
|
|
moveSketchFilesPromises.push(fileSavePromise);
|
|
}
|
|
});
|
|
const sketchSavePromise = Promise.all(moveSketchFilesPromises).then(() => {
|
|
sketch.user = ObjectId(currentUser._id);
|
|
return sketch.save();
|
|
});
|
|
saveSketchPromises.push(sketchSavePromise);
|
|
});
|
|
return Promise.all(saveSketchPromises);
|
|
}).then(() => {
|
|
console.log('Moved and updated all sketches.');
|
|
return Collection.updateMany(
|
|
{ owner: { $in: duplicates } },
|
|
{ $set: { owner: ObjectId(currentUser.id) } }
|
|
);
|
|
}).then(() => {
|
|
console.log('Moved and updated all collections.');
|
|
return User.deleteMany({ _id: { $in: duplicates } });
|
|
}).then(() => {
|
|
console.log('Deleted other user accounts.');
|
|
currentUser.email = currentUser.email.toLowerCase();
|
|
return currentUser.save();
|
|
}).then(() => {
|
|
console.log('Migrated email to lowercase.');
|
|
// const protocol = process.env.NODE_ENV === 'production' ? 'https' : 'http';
|
|
const mailOptions = renderAccountConsolidation({
|
|
body: {
|
|
domain: 'https://editor.p5js.org',
|
|
username: currentUser.username,
|
|
email: currentUser.email
|
|
},
|
|
to: currentUser.email,
|
|
});
|
|
|
|
mail.send(mailOptions, (mailErr, result) => {
|
|
console.log('Sent email.');
|
|
process.exit(0);
|
|
});
|
|
});
|
|
|
|
// ).then((result) => {
|
|
// console.log(result);
|
|
// });
|
|
|