nocobase/packages/plugin-collections/src/server.ts

import path from 'path';
import { Application } from '@nocobase/server';
import { registerModels, Table } from '@nocobase/database';
import * as models from './models';
import { createOrUpdate, findAll } from './actions';
import { create } from './actions/fields';

export default async function (this: Application, options = {}) {
  const database = this.database;
  registerModels(models);
  database.import({
    directory: path.resolve(__dirname, 'collections'),
  });
  const [Collection, Field] = database.getModels(['collections', 'fields']);
  Field.beforeCreate(async (model) => {
    if (!model.get('name')) {
      model.set('name', model.get('key'));
    }
    if (!model.get('collection_name') && model.get('parentKey')) {
      const field = await Field.findByPk(model.get('parentKey'));
      if (field) {
        const { target } = field.get('options') || {};
        if (target) {
          model.set('collection_name', target);
        }
      }
    }
  });
  Field.beforeUpdate(async (model) => {
    if (!model.get('collection_name') && model.get('parentKey')) {
      const field = await Field.findByPk(model.get('parentKey'));
      if (field) {
        const { target } = field.get('options') || {};
        if (target) {
          model.set('collection_name', target);
        }
      }
    }
  });
  Field.afterCreate(async (model) => {
    console.log('afterCreate');
    if (model.get('interface') !== 'subTable') {
      return;
    }
    const { target } = model.get('options') || {};
    // const uiSchemaKey = model.get('ui_schema_key');
    // console.log({ uiSchemaKey })
    try {
      let collection = await Collection.findOne({
        where: {
          name: target,
        },
      });
      if (!collection) {
        collection = await Collection.create({
          name: target,
          // ui_schema_key: uiSchemaKey,
        });
      }
      // if (model.get('ui_schema_key')) {
      //   collection.set('ui_schema_key', model.get('ui_schema_key'));
      //   await collection.save({ hooks: false });
      // }
      await collection.migrate();
    } catch (error) {
      throw error;
    }
  });
  Field.afterUpdate(async (model) => {
    console.log('afterUpdate');
    if (model.get('interface') !== 'subTable') {
      return;
    }
    const { target } = model.get('options') || {};
    try {
      let collection = await Collection.findOne({
        where: {
          name: target,
        },
      });
      if (!collection) {
        collection = await Collection.create({
          name: target,
        });
      }
      // if (model.get('ui_schema_key')) {
      //   collection.set('ui_schema_key', model.get('ui_schema_key'));
      //   await collection.save({ hooks: false });
      // }
      await collection.migrate();
    } catch (error) {
      throw error;
    }
  });
  this.resourcer.registerActionHandler('collections.fields:create', create);
  this.resourcer.registerActionHandler('collections:findAll', findAll);
  this.resourcer.registerActionHandler('collections:createOrUpdate', createOrUpdate);
}