Update migrate.ts
This commit is contained in:
parent
d4bdb5d327
commit
e98cce9aee
|
@ -24,6 +24,7 @@ import { NoteReaction } from './models/entities/note-reaction';
|
||||||
import { UserPublickey } from './models/entities/user-publickey';
|
import { UserPublickey } from './models/entities/user-publickey';
|
||||||
import { UserKeypair } from './models/entities/user-keypair';
|
import { UserKeypair } from './models/entities/user-keypair';
|
||||||
import { extractPublic } from './crypto_key';
|
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 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;
|
const p = (config as any).mongodb.pass ? encodeURIComponent((config as any).mongodb.pass) : null;
|
||||||
|
@ -56,6 +57,7 @@ const _Following = db.get<any>('following');
|
||||||
const _PollVote = db.get<any>('pollVotes');
|
const _PollVote = db.get<any>('pollVotes');
|
||||||
const _Favorite = db.get<any>('favorites');
|
const _Favorite = db.get<any>('favorites');
|
||||||
const _NoteReaction = db.get<any>('noteReactions');
|
const _NoteReaction = db.get<any>('noteReactions');
|
||||||
|
const _Emoji = db.get<any>('emoji');
|
||||||
const getDriveFileBucket = async (): Promise<mongo.GridFSBucket> => {
|
const getDriveFileBucket = async (): Promise<mongo.GridFSBucket> => {
|
||||||
const db = await nativeDbConn();
|
const db = await nativeDbConn();
|
||||||
const bucket = new mongo.GridFSBucket(db, {
|
const bucket = new mongo.GridFSBucket(db, {
|
||||||
|
@ -77,6 +79,7 @@ async function main() {
|
||||||
const NoteReactions = getRepository(NoteReaction);
|
const NoteReactions = getRepository(NoteReaction);
|
||||||
const UserPublickeys = getRepository(UserPublickey);
|
const UserPublickeys = getRepository(UserPublickey);
|
||||||
const UserKeypairs = getRepository(UserKeypair);
|
const UserKeypairs = getRepository(UserKeypair);
|
||||||
|
const Emojis = getRepository(Emoji);
|
||||||
|
|
||||||
async function migrateUser(user: any) {
|
async function migrateUser(user: any) {
|
||||||
await Users.save({
|
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();
|
const allUsersCount = await _User.count();
|
||||||
for (let i = 0; i < allUsersCount; i++) {
|
for (let i = 0; i < allUsersCount; i++) {
|
||||||
const user = await _User.findOne({}, {
|
const user = await _User.findOne({}, {
|
||||||
|
@ -433,6 +448,20 @@ async function main() {
|
||||||
console.error(e);
|
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();
|
main();
|
||||||
|
|
Loading…
Reference in New Issue