diff --git a/src/migrate.ts b/src/migrate.ts index ed2b3710db..ceaec4185d 100644 --- a/src/migrate.ts +++ b/src/migrate.ts @@ -19,6 +19,7 @@ import { Following } from './models/entities/following'; import { genId } from './misc/gen-id'; import { Poll } from './models/entities/poll'; import { PollVote } from './models/entities/poll-vote'; +import { NoteFavorite } from './models/entities/note-favorite'; const u = (config as any).mongodb.user ? encodeURIComponent((config as any).mongodb.user) : null; const p = (config as any).mongodb.pass ? encodeURIComponent((config as any).mongodb.pass) : null; @@ -49,6 +50,7 @@ const _DriveFolder = db.get('driveFolders'); const _Note = db.get('notes'); const _Following = db.get('following'); const _PollVote = db.get('pollVotes'); +const _Favorite = db.get('favorites'); const getDriveFileBucket = async (): Promise => { const db = await nativeDbConn(); const bucket = new mongo.GridFSBucket(db, { @@ -66,6 +68,7 @@ async function main() { const Followings = getRepository(Following); const Polls = getRepository(Poll); const PollVotes = getRepository(PollVote); + const NoteFavorites = getRepository(NoteFavorite); async function migrateUser(user: any) { await Users.insert({ @@ -237,12 +240,21 @@ async function main() { await PollVotes.save({ id: vote._id.toHexString(), createdAt: vote.createdAt, - noteId: vote.note.id.toHexString(), - userId: vote.user.id.toHexString(), + noteId: vote.noteId.toHexString(), + userId: vote.userId.toHexString(), choice: vote.choice }); } + async function migrateNoteFavorite(favorite: any) { + await NoteFavorites.save({ + id: favorite._id.toHexString(), + createdAt: favorite.createdAt, + noteId: favorite.noteId.toHexString(), + userId: favorite.userId.toHexString(), + }); + } + const allUsersCount = await _User.count(); for (let i = 0; i < allUsersCount; i++) { const user = await _User.findOne({}, { @@ -330,6 +342,20 @@ async function main() { console.error(e); } } + + const allNoteFavoritesCount = await _Favorite.count(); + for (let i = 0; i < allNoteFavoritesCount; i++) { + const favorite = await _Favorite.findOne({}, { + skip: i + }); + try { + await migrateNoteFavorite(favorite); + console.log(`FAVORITE (${i + 1}/${allNoteFavoritesCount}) ${favorite._id} ${chalk.green('DONE')}`); + } catch (e) { + console.log(`FAVORITE (${i + 1}/${allNoteFavoritesCount}) ${favorite._id} ${chalk.red('ERR')}`); + console.error(e); + } + } } main();