39 lines
1.5 KiB
TypeScript
39 lines
1.5 KiB
TypeScript
import Migration from "../../db/Migration";
|
|
import ModelFactory from "../../db/ModelFactory";
|
|
import User from "../models/User";
|
|
import UserNameComponent from "../models/UserNameComponent";
|
|
import MagicLink from "../models/MagicLink";
|
|
import MagicLinkUserNameComponent from "../models/MagicLinkUserNameComponent";
|
|
import {nanoid} from "nanoid";
|
|
|
|
export default class AddNameToUsersMigration extends Migration {
|
|
public async install(): Promise<void> {
|
|
await this.query(`ALTER TABLE users
|
|
ADD COLUMN name VARCHAR(64) NOT NULL`);
|
|
await this.query(`ALTER TABLE magic_links
|
|
ADD COLUMN username VARCHAR(64) DEFAULT NULL`);
|
|
|
|
// Give every user a random name
|
|
const users = await User.select().get(this.getCurrentConnection());
|
|
const callbacks: (() => Promise<void>)[] = [];
|
|
await Promise.all(users.map(user => {
|
|
user.name = nanoid();
|
|
return user.save(this.getCurrentConnection(), c => callbacks.push(c));
|
|
}));
|
|
await Promise.all(callbacks);
|
|
|
|
await this.query(`ALTER TABLE users
|
|
ADD CONSTRAINT UNIQUE (name)`);
|
|
}
|
|
|
|
public async rollback(): Promise<void> {
|
|
await this.query('ALTER TABLE users DROP COLUMN name');
|
|
await this.query('ALTER TABLE magic_links DROP COLUMN username');
|
|
}
|
|
|
|
public registerModels(): void {
|
|
ModelFactory.get(User).addComponent(UserNameComponent);
|
|
ModelFactory.get(MagicLink).addComponent(MagicLinkUserNameComponent);
|
|
}
|
|
}
|