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

import path from 'path';
import * as actions from './actions/users';
import * as middlewares from './middlewares';
import { Collection } from '@nocobase/database';
import { Plugin, PluginOptions } from '@nocobase/server';

export default class PluginUsers extends Plugin {
  async beforeLoad() {
    this.app.on('installing', async () => {
      const User = this.db.getCollection('users');
      await User.model.create({
        nickname: 'Super Admin',
        email: process.env.ADMIN_EMAIL || 'admin@nocobase.com',
        password: process.env.ADMIN_PASSWORD || 'admin123',
      });
    });

    this.db.on('users.afterCreateWithAssociations', async (model, options) => {
      const { transaction } = options;

      const defaultRole = await this.app.db.getRepository('roles').findOne({
        filter: {
          default: true,
        },
        transaction,
      });

      if (defaultRole && (await model.countRoles({ transaction })) == 0) {
        await model.addRoles(defaultRole, { transaction });
      }
    });

    this.db.on('afterDefineCollection', (collection: Collection) => {
      let { createdBy, updatedBy } = collection.options;
      if (createdBy === true) {
        collection.setField('createdById', {
          type: 'context',
          dataType: 'integer',
          dataIndex: 'state.currentUser.id',
          createOnly: true,
        });
        collection.setField('createdBy', {
          type: 'belongsTo',
          target: 'users',
          foreignKey: 'createdById',
          targetKey: 'id',
        });
      }
      if (updatedBy === true) {
        collection.setField('updatedById', {
          type: 'context',
          dataType: 'integer',
          dataIndex: 'state.currentUser.id',
        });
        collection.setField('updatedBy', {
          type: 'belongsTo',
          target: 'users',
          foreignKey: 'updatedById',
          targetKey: 'id',
        });
      }
    });

    for (const [key, action] of Object.entries(actions)) {
      this.app.resourcer.registerActionHandler(`users:${key}`, action);
    }

    this.app.resourcer.use(middlewares.parseToken({}));
  }

  async load() {
    await this.db.import({
      directory: path.resolve(__dirname, 'collections'),
    });
  }
}