From e98cce9aee910d43ba8c87628be80b78e20899dc Mon Sep 17 00:00:00 2001 From: syuilo Date: Tue, 9 Apr 2019 19:20:10 +0900 Subject: [PATCH] Update migrate.ts --- src/migrate.ts | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/src/migrate.ts b/src/migrate.ts index 16fbc4bf8..8d36762c8 100644 --- a/src/migrate.ts +++ b/src/migrate.ts @@ -24,6 +24,7 @@ import { NoteReaction } from './models/entities/note-reaction'; import { UserPublickey } from './models/entities/user-publickey'; import { UserKeypair } from './models/entities/user-keypair'; import { extractPublic } from './crypto_key'; +import { Emoji } from './models/entities/emoji'; 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; @@ -56,6 +57,7 @@ const _Following = db.get('following'); const _PollVote = db.get('pollVotes'); const _Favorite = db.get('favorites'); const _NoteReaction = db.get('noteReactions'); +const _Emoji = db.get('emoji'); const getDriveFileBucket = async (): Promise => { const db = await nativeDbConn(); const bucket = new mongo.GridFSBucket(db, { @@ -77,6 +79,7 @@ async function main() { const NoteReactions = getRepository(NoteReaction); const UserPublickeys = getRepository(UserPublickey); const UserKeypairs = getRepository(UserKeypair); + const Emojis = getRepository(Emoji); async function migrateUser(user: any) { await Users.save({ @@ -303,6 +306,18 @@ async function main() { }); } + async function migrateEmoji(emoji: any) { + await NoteReactions.save({ + id: emoji._id.toHexString(), + updatedAt: emoji.createdAt, + aliases: emoji.aliases, + url: emoji.url, + uri: emoji.uri, + host: emoji.host, + name: emoji.name + }); + } + const allUsersCount = await _User.count(); for (let i = 0; i < allUsersCount; i++) { const user = await _User.findOne({}, { @@ -433,6 +448,20 @@ async function main() { console.error(e); } } + + const allEmojisCount = await _Emoji.count(); + for (let i = 0; i < allEmojisCount; i++) { + const emoji = await _Emoji.findOne({}, { + skip: i + }); + try { + await migrateEmoji(emoji); + console.log(`EMOJI (${i + 1}/${allEmojisCount}) ${emoji._id} ${chalk.green('DONE')}`); + } catch (e) { + console.log(`EMOJI (${i + 1}/${allEmojisCount}) ${emoji._id} ${chalk.red('ERR')}`); + console.error(e); + } + } } main();