tachybase_todo/packages/plugins/@tachybase/plugin-users/src/server/migrations/20240722114514-add-users-specialRole.ts
TomyJan 7756c18cb1 perf: user delete permission judgement (#1363)
Reviewed-on: daoyoucloud/tachybase#1363
Reviewed-by: sealday <zhanglin@daoyoucloud.com>
Co-authored-by: TomyJan <TomyJan6@gmail.com>
Co-committed-by: TomyJan <TomyJan6@gmail.com>
2024-07-22 15:38:34 +08:00

56 lines
1.4 KiB
TypeScript

import { DataTypes } from '@tachybase/database';
import { Migration } from '@tachybase/server';
export default class AddUsersSpecialRoleMigration extends Migration {
on = 'beforeLoad';
appVersion = '<0.21.88';
async up() {
const collection = this.db.collection({
name: 'users',
fields: [
{
type: 'string',
name: 'specialRole',
},
],
});
const tableNameWithSchema = collection.getTableNameWithSchema();
const field = collection.getField('specialRole');
const exists = await field.existsInDb();
if (!exists) {
await this.db.sequelize.getQueryInterface().addColumn(tableNameWithSchema, field.columnName(), {
type: DataTypes.STRING,
});
}
try {
await this.db.sequelize.getQueryInterface().addConstraint(tableNameWithSchema, {
type: 'unique',
fields: [field.columnName()],
});
} catch (error) {
throw new Error(error);
}
const repo = this.context.db.getRepository('users');
const user = await repo.findOne({
filter: {
email: process.env.INIT_ROOT_EMAIL || 'admin@tachybase.com',
},
});
if (user) {
await repo.update({
values: {
specialRole: 'root',
},
filter: {
id: user.id,
},
});
} else {
throw new Error('Root user not found');
}
}
async down() {}
}