nocobase/packages/core/database/src/sync-runner.ts

import { InheritedCollection } from './inherited-collection';
import lodash from 'lodash';

export class SyncRunner {
  static async syncInheritModel(model: any, options: any) {
    const { transaction } = options;

    const inheritedCollection = model.collection as InheritedCollection;
    const db = inheritedCollection.context.database;
    const schemaName = db.options.schema || 'public';

    const dialect = db.sequelize.getDialect();

    const queryInterface = db.sequelize.getQueryInterface();

    if (dialect != 'postgres') {
      throw new Error('Inherit model is only supported on postgres');
    }

    const parents = inheritedCollection.parents;

    if (!parents) {
      throw new Error(
        `Inherit model ${inheritedCollection.name} can't be created without parents, parents option is ${lodash
          .castArray(inheritedCollection.options.inherits)
          .join(', ')}`,
      );
    }

    const parentTables = parents.map((parent) => parent.model.tableName);

    const tableName = model.tableName;

    const schemaTableName = db.utils.addSchema(tableName);
    const quoteTableName = db.utils.quoteTable(tableName);

    const attributes = model.tableAttributes;

    const childAttributes = lodash.pickBy(attributes, (value) => {
      return !value.inherit;
    });

    let maxSequenceVal = 0;
    let maxSequenceName;

    if (childAttributes.id && childAttributes.id.autoIncrement) {
      for (const parent of parentTables) {
        const sequenceNameResult = await queryInterface.sequelize.query(
          `SELECT column_default
           FROM information_schema.columns
           WHERE table_name = '${parent}'
            and table_schema = '${schemaName}'
             and "column_name" = 'id';`,
          {
            transaction,
          },
        );

        if (!sequenceNameResult[0].length) {
          continue;
        }

        const columnDefault = sequenceNameResult[0][0]['column_default'];

        if (!columnDefault) {
          throw new Error(`Can't find sequence name of ${parent}`);
        }

        const regex = new RegExp(/nextval\('(.*)'::regclass\)/);
        const match = regex.exec(columnDefault);

        const sequenceName = match[1];

        const sequenceCurrentValResult = await queryInterface.sequelize.query(
          `select last_value
           from ${sequenceName}`,
          {
            transaction,
          },
        );

        const sequenceCurrentVal = parseInt(sequenceCurrentValResult[0][0]['last_value']);

        if (sequenceCurrentVal > maxSequenceVal) {
          maxSequenceName = sequenceName;
          maxSequenceVal = sequenceCurrentVal;
        }
      }
    }

    await this.createTable(schemaTableName, childAttributes, options, model, parentTables, db);

    if (maxSequenceName) {
      const parentsDeep = Array.from(db.inheritanceMap.getParents(inheritedCollection.name)).map(
        (parent) => db.getCollection(parent).model.tableName,
      );

      const sequenceTables = [...parentsDeep, tableName.toString()];

      for (const sequenceTable of sequenceTables) {
        const queryName =
          Boolean(sequenceTable.match(/[A-Z]/)) && !sequenceTable.includes(`"`) ? `"${sequenceTable}"` : sequenceTable;

        const idColumnQuery = await queryInterface.sequelize.query(
          `
          SELECT column_name
FROM information_schema.columns
WHERE table_name='${queryName}' and column_name='id' and table_schema = '${schemaName}';
          `,
          {
            transaction,
          },
        );

        if (idColumnQuery[0].length == 0) {
          continue;
        }

        await queryInterface.sequelize.query(
          `alter table "${schemaName}"."${sequenceTable}"
            alter column id set default nextval('${maxSequenceName}')`,
          {
            transaction,
          },
        );
      }
    }

    if (options.alter) {
      const columns = await queryInterface.describeTable(tableName, options);

      for (const attribute in childAttributes) {
        const columnName = childAttributes[attribute].field;

        if (!columns[columnName]) {
          await queryInterface.addColumn(tableName, columnName, childAttributes[columnName], options);
        }
      }
    }
  }

  static async createTable(tableName, attributes, options, model, parentTables, db) {
    let sql = '';

    options = { ...options };

    if (options && options.uniqueKeys) {
      lodash.forOwn(options.uniqueKeys, (uniqueKey) => {
        if (uniqueKey.customIndex === undefined) {
          uniqueKey.customIndex = true;
        }
      });
    }

    if (model) {
      options.uniqueKeys = options.uniqueKeys || model.uniqueKeys;
    }

    const queryGenerator = model.queryGenerator;

    attributes = lodash.mapValues(attributes, (attribute) => model.sequelize.normalizeAttribute(attribute));

    attributes = queryGenerator.attributesToSQL(attributes, { table: tableName, context: 'createTable' });

    sql = `${queryGenerator.createTableQuery(tableName, attributes, options)}`.replace(
      ';',
      ` INHERITS (${parentTables
        .map((t) => {
          return db.utils.quoteTable(db.utils.addSchema(t, db.options.schema));
        })
        .join(', ')});`,
    );

    return await model.sequelize.query(sql, options);
  }
}