77 lines
2.9 KiB
JavaScript
77 lines
2.9 KiB
JavaScript
/// <reference types="./mysql-introspector.d.ts" />
|
|
import { DEFAULT_MIGRATION_LOCK_TABLE, DEFAULT_MIGRATION_TABLE, } from '../../migration/migrator.js';
|
|
import { freeze } from '../../util/object-utils.js';
|
|
import { sql } from '../../raw-builder/sql.js';
|
|
export class MysqlIntrospector {
|
|
#db;
|
|
constructor(db) {
|
|
this.#db = db;
|
|
}
|
|
async getSchemas() {
|
|
let rawSchemas = await this.#db
|
|
.selectFrom('information_schema.schemata')
|
|
.select('schema_name')
|
|
.$castTo()
|
|
.execute();
|
|
return rawSchemas.map((it) => ({ name: it.SCHEMA_NAME }));
|
|
}
|
|
async getTables(options = { withInternalKyselyTables: false }) {
|
|
let query = this.#db
|
|
.selectFrom('information_schema.columns as columns')
|
|
.innerJoin('information_schema.tables as tables', (b) => b
|
|
.onRef('columns.TABLE_CATALOG', '=', 'tables.TABLE_CATALOG')
|
|
.onRef('columns.TABLE_SCHEMA', '=', 'tables.TABLE_SCHEMA')
|
|
.onRef('columns.TABLE_NAME', '=', 'tables.TABLE_NAME'))
|
|
.select([
|
|
'columns.COLUMN_NAME',
|
|
'columns.COLUMN_DEFAULT',
|
|
'columns.TABLE_NAME',
|
|
'columns.TABLE_SCHEMA',
|
|
'tables.TABLE_TYPE',
|
|
'columns.IS_NULLABLE',
|
|
'columns.DATA_TYPE',
|
|
'columns.EXTRA',
|
|
'columns.COLUMN_COMMENT',
|
|
])
|
|
.where('columns.TABLE_SCHEMA', '=', sql `database()`)
|
|
.orderBy('columns.TABLE_NAME')
|
|
.orderBy('columns.ORDINAL_POSITION')
|
|
.$castTo();
|
|
if (!options.withInternalKyselyTables) {
|
|
query = query
|
|
.where('columns.TABLE_NAME', '!=', DEFAULT_MIGRATION_TABLE)
|
|
.where('columns.TABLE_NAME', '!=', DEFAULT_MIGRATION_LOCK_TABLE);
|
|
}
|
|
const rawColumns = await query.execute();
|
|
return this.#parseTableMetadata(rawColumns);
|
|
}
|
|
async getMetadata(options) {
|
|
return {
|
|
tables: await this.getTables(options),
|
|
};
|
|
}
|
|
#parseTableMetadata(columns) {
|
|
return columns.reduce((tables, it) => {
|
|
let table = tables.find((tbl) => tbl.name === it.TABLE_NAME);
|
|
if (!table) {
|
|
table = freeze({
|
|
name: it.TABLE_NAME,
|
|
isView: it.TABLE_TYPE === 'VIEW',
|
|
schema: it.TABLE_SCHEMA,
|
|
columns: [],
|
|
});
|
|
tables.push(table);
|
|
}
|
|
table.columns.push(freeze({
|
|
name: it.COLUMN_NAME,
|
|
dataType: it.DATA_TYPE,
|
|
isNullable: it.IS_NULLABLE === 'YES',
|
|
isAutoIncrementing: it.EXTRA.toLowerCase().includes('auto_increment'),
|
|
hasDefaultValue: it.COLUMN_DEFAULT !== null,
|
|
comment: it.COLUMN_COMMENT === '' ? undefined : it.COLUMN_COMMENT,
|
|
}));
|
|
return tables;
|
|
}, []);
|
|
}
|
|
}
|