|
1 | 1 | import type { GlobalRole } from '@n8n/permissions';
|
2 | 2 |
|
| 3 | +import { ApiKey } from '@/databases/entities/api-key'; |
3 | 4 | import type { MigrationContext, ReversibleMigration } from '@/databases/types';
|
4 | 5 | import { getApiKeyScopesForRole } from '@/public-api/permissions.ee';
|
5 | 6 |
|
6 | 7 | type ApiKeyWithRole = { id: string; role: GlobalRole };
|
7 | 8 |
|
8 | 9 | export class AddScopesColumnToApiKeys1742918400000 implements ReversibleMigration {
|
9 |
| - async up({ runQuery, escape, schemaBuilder: { addColumns, column } }: MigrationContext) { |
| 10 | + async up({ |
| 11 | + runQuery, |
| 12 | + escape, |
| 13 | + queryRunner, |
| 14 | + schemaBuilder: { addColumns, column }, |
| 15 | + }: MigrationContext) { |
10 | 16 | await addColumns('user_api_keys', [column('scopes').json]);
|
11 | 17 |
|
12 | 18 | const userApiKeysTable = escape.tableName('user_api_keys');
|
13 | 19 | const userTable = escape.tableName('user');
|
14 | 20 | const idColumn = escape.columnName('id');
|
15 | 21 | const userIdColumn = escape.columnName('userId');
|
16 | 22 | const roleColumn = escape.columnName('role');
|
17 |
| - const scopesColumn = escape.columnName('scopes'); |
18 | 23 |
|
19 | 24 | const apiKeysWithRoles = await runQuery<ApiKeyWithRole[]>(
|
20 | 25 | `SELECT ${userApiKeysTable}.${idColumn} AS id, ${userTable}.${roleColumn} AS role FROM ${userApiKeysTable} JOIN ${userTable} ON ${userTable}.${idColumn} = ${userApiKeysTable}.${userIdColumn}`,
|
21 | 26 | );
|
22 | 27 |
|
23 | 28 | for (const { id, role } of apiKeysWithRoles) {
|
24 |
| - const scopes = JSON.stringify(getApiKeyScopesForRole(role)); |
25 |
| - |
26 |
| - await runQuery( |
27 |
| - `UPDATE ${userApiKeysTable} SET ${scopesColumn} = '${scopes}' WHERE ${idColumn} = "${id}"`, |
28 |
| - ); |
| 29 | + const scopes = getApiKeyScopesForRole(role); |
| 30 | + await queryRunner.manager.update(ApiKey, { id }, { scopes }); |
29 | 31 | }
|
30 | 32 | }
|
31 | 33 |
|
|
0 commit comments