MusareFork/backend/logic/migration/migrations/migration2.js

import async from "async";

/**
 * Migration 2
 *
 * Updates the document version 1 stations to add the includedPlaylists and excludedPlaylists properties, and to create a station playlist and link that playlist with the playlist2 property.
 *
 * @param {object} MigrationModule - the MigrationModule
 * @returns {Promise} - returns promise
 */
export default async function migrate(MigrationModule) {
	const stationModel = await MigrationModule.runJob("GET_MODEL", { modelName: "station" }, this);
	const playlistModel = await MigrationModule.runJob("GET_MODEL", { modelName: "playlist" }, this);

	return new Promise((resolve, reject) => {
		async.waterfall(
			[
				next => {
					this.log("INFO", `Migration 2. Finding stations with document version 1.`);
					stationModel.find({ documentVersion: 1 }, (err, stations) => {
						this.log("INFO", `Migration 2. Found ${stations.length} stations with document version 1.`);

						next(
							null,
							stations.map(station => station._doc)
						);
					});
				},

				(stations, next) => {
					async.eachLimit(
						stations,
						1,
						(station, next) => {
							this.log("INFO", `Migration 2. Creating station playlist for station ${station._id}.`);
							playlistModel.create(
								{
									isUserModifiable: false,
									displayName: `Station - ${station.displayName}`,
									songs: [],
									createdBy: station.type === "official" ? "Musare" : station.createdBy,
									createdFor: `${station._id}`,
									createdAt: Date.now(),
									type: "station",
									documentVersion: 1
								},
								(err, playlist2) => {
									if (err) next(err);
									else {
										this.log("INFO", `Migration 2. Updating station ${station._id}.`);
										stationModel.updateOne(
											{ _id: station._id },
											{
												$set: {
													playlist2: playlist2._id,
													includedPlaylists: [],
													excludedPlaylists: [],
													playlist: station.type === "official" ? [] : station.playlist,
													genres: [],
													documentVersion: 2
												}
											},
											(err, res) => {
												if (err) next(err);
												else {
													this.log(
														"INFO",
														`Migration 2. Updating station ${station._id} done. Matched: ${res.n}, modified: ${res.nModified}, ok: ${res.ok}.`
													);
												}
											}
										);
										next();
									}
								}
							);
						},
						next
					);
				}
			],
			(err, response) => {
				if (err) {
					reject(new Error(err));
				} else {
					resolve(response);
				}
			}
		);
	});
}