dbgate/plugins/dbgate-plugin-firebird/src/backend/Analyser.js

const _ = require('lodash');
const sql = require('./sql');
const {
  getDataTypeString,
  getTriggerTiming,
  getTriggerEventType,
  getFormattedDefaultValue,
  getTriggerCreateSql,
} = require('./helpers');

const { DatabaseAnalyser } = require('dbgate-tools');

class Analyser extends DatabaseAnalyser {
  constructor(dbhan, driver, version) {
    super(dbhan, driver, version);
  }

  createQuery(resFileName, typeFields) {
    if (!sql[resFileName]) throw new Error(`Missing analyse file ${resFileName}`);
    return super.createQuery(sql[resFileName], typeFields);
  }

  async _runAnalysis() {
    const tablesResult = await this.analyserQuery('tables', ['tables']);
    const columnsResult = await this.analyserQuery('columns', ['tables', 'views']);
    const triggersResult = await this.analyserQuery('triggers', ['triggers']);
    const primaryKeysResult = await this.analyserQuery('primaryKeys', ['primaryKeys']);
    const foreignKeysResult = await this.analyserQuery('foreignKeys', ['foreignKeys']);
    const functionsResults = await this.analyserQuery('functions', ['functions']);
    const functionParametersResults = await this.analyserQuery('functionParameters', ['functions']);
    const proceduresResults = await this.analyserQuery('procedures', ['procedures']);
    const procedureParametersResults = await this.analyserQuery('procedureParameters', ['procedures']);
    const viewsResults = await this.analyserQuery('views', ['views']);
    const unqiuesResults = await this.analyserQuery('uniques', ['tables']);
    const indexesResults = await this.analyserQuery('indexes', ['tables']);

    const columns =
      columnsResult.rows?.map(column => ({
        ...column,
        objectId: `tables:${column.columnName}`,
        dataType: getDataTypeString(column),
        defaultValue: getFormattedDefaultValue(column.defaultValue),
      })) ?? [];

    const triggers =
      triggersResult.rows?.map(i => ({
        ...i,
        objectId: `triggers:${i.pureName}`,
        eventType: getTriggerEventType(i.TRIGGERTYPE),
        triggerTiming: getTriggerTiming(i.TRIGGERTYPE),
        createSql: getTriggerCreateSql(i),
      })) ?? [];

    const primaryKeys =
      primaryKeysResult.rows?.map(primaryKey => ({
        ...primaryKey,
        objectId: `tables:${primaryKey.pureName}`,
      })) ?? [];

    const foreignKeys =
      foreignKeysResult.rows?.map(foreignKey => ({
        ...foreignKey,
        objectId: `tables:${foreignKey.pureName}`,
      })) ?? [];

    const functions =
      functionsResults.rows?.map(func => ({
        ...func,
        objectId: `functions:${func.pureName}`,
        returnType: functionParametersResults.rows?.filter(
          param => param.owningObjectName === func.pureName && param.parameterMode === 'RETURN'
        )?.dataType,
        parameters: functionParametersResults.rows
          ?.filter(param => param.owningObjectName === func.pureName)
          .map(param => ({
            ...param,
            dataType: getDataTypeString(param),
          })),
      })) ?? [];

    const uniques =
      unqiuesResults.rows?.map(unique => ({
        pureName: unique.pureName,
        constraintName: unique.constraintName,
        constraintType: unique.constraintType,
        columns: [
          {
            columnName: unique.columnName,
            isDescending: unique.isDescending,
          },
        ],
      })) ?? [];

    const indexesGrouped = _.groupBy(indexesResults.rows, 'constraintName');
    const indexes = Object.values(indexesGrouped).map(indexGroup => ({
      pureName: indexGroup[0].pureName,
      constraintName: indexGroup[0].constraintName,
      constraintType: indexGroup[0].constraintType,
      columns: indexGroup.map(index => ({
        columnName: index.columnName,
        isDescending: index.isDescending,
      })),
    }));

    const procedures =
      proceduresResults.rows?.map(proc => ({
        ...proc,
        objectId: `procedures:${proc.pureName}`,
        parameters: procedureParametersResults.rows
          ?.filter(param => param.owningObjectName === proc.pureName)
          .map(param => ({
            ...param,
            dataType: getDataTypeString(param),
          })),
      })) ?? [];

    const tables =
      tablesResult.rows?.map(table => ({
        ...table,
        objectId: `tables:${table.pureName}`,
        columns: columns.filter(column => column.tableName === table.pureName),
        primaryKey: DatabaseAnalyser.extractPrimaryKeys(table, primaryKeys),
        foreignKeys: DatabaseAnalyser.extractForeignKeys(table, foreignKeys),
        uniques: uniques.filter(unique => unique.pureName === table.pureName),
        indexes: indexes.filter(index => index.pureName === table.pureName),
      })) ?? [];
    // console.log(uniques);

    const views =
      viewsResults.rows?.map(view => ({
        ...view,
        objectId: `views:${view.pureName}`,
        columns: columns.filter(column => column.tableName === view.pureName),
      })) ?? [];

    return {
      views,
      tables,
      triggers,
      functions,
      procedures,
    };
  }

  async _computeSingleObjectId() {
    const { typeField, pureName } = this.singleObjectFilter;
    console.log('Computing single object ID for', typeField, pureName);
    this.singleObjectId = `${typeField}:${pureName}`;
  }
}

module.exports = Analyser;