import path from 'path'; import Database from '@nocobase/database'; import Resourcer from '@nocobase/resourcer'; import getCollection from './actions/getCollection'; import getView from './actions/getView'; import getRoutes from './actions/getRoutes'; import getPageInfo from './actions/getPageInfo'; import * as rolesPagesActions from './actions/roles.pages'; import getCollections from './actions/getCollections'; export default async function (options = {}) { const database: Database = this.database; const resourcer: Resourcer = this.resourcer; database.import({ directory: path.resolve(__dirname, 'collections'), }); resourcer.registerActionHandler('getCollection', getCollection); resourcer.registerActionHandler('getView', getView); resourcer.registerActionHandler('getPageInfo', getPageInfo); resourcer.registerActionHandler('getCollections', getCollections); resourcer.registerActionHandler('pages:getRoutes', getRoutes); Object.keys(rolesPagesActions).forEach(actionName => { resourcer.registerActionHandler(`roles.pages:${actionName}`, rolesPagesActions[actionName]); }); /* const [Collection, Page, View] = database.getModels(['collections', 'pages', 'views']); async function createCollectionPage(model, options) { // const { // transaction = await database.sequelize.transaction(), // } = options; if (model.get('internal')) { return; } const transaction = await database.sequelize.transaction(); const parent = await Page.findOne({ transaction, where: { path: '/collections', } }); let page = await Page.findOne({ transaction, where: { collection: model.get('name'), path: `/collections/${model.get('name')}`, }, }); if (!page) { page = await Page.create({ type: 'collection', collection: model.get('name'), path: `/collections/${model.get('name')}`, sort: 100, parent_id: parent.id, }, { transaction, }); } page.set({ title: model.get('title'), icon: model.get('icon'), showInMenu: !!model.get('showInDataMenu'), }); await page.save({ transaction, }); await Page.collectionPagesResort({transaction}); await transaction.commit(); } Collection.addHook('afterCreate', createCollectionPage); Collection.addHook('afterUpdate', createCollectionPage); Collection.addHook('afterDestroy', async (model, options) => { const { transaction } = options; // console.log('afterDestroy', model); await Page.destroy({ transaction, where: { path: `/collections/${model.get('name')}`, }, }); await Page.collectionPagesResort({transaction}); }); async function syncViewCollectionPage(model, options) { const transaction = await database.sequelize.transaction(); const parentPath = `/collections/${model.get('collection_name')}`; const currentPath = `${parentPath}/views/${model.get('name')}`; try { const parent = await Page.findOne({ transaction, where: { path: parentPath, }, }); if (!parent) { await transaction.rollback(); return; } let page = await Page.findOne({ transaction, where: { collection: model.get('collection_name'), path: currentPath, }, }); if (!page) { page = await Page.create({ type: 'collection', collection: model.get('collection_name'), path: currentPath, sort: 100, parent_id: parent.id, }, { transaction, }); } page.set({ title: model.get('title'), viewName: model.get('name'), viewId: model.get('id'), // icon: model.get('icon'), showInMenu: !!model.get('showInDataMenu'), }); await page.save({ transaction, }); } catch (error) { await transaction.rollback(); console.error(error); } await transaction.commit(); } View.addHook('beforeValidate', (model) => { if (model.get('default')) { model.set('showInDataMenu', true); } }); View.addHook('afterCreate', syncViewCollectionPage); View.addHook('afterUpdate', syncViewCollectionPage); View.addHook('afterDestroy', async (model, options) => { await Page.destroy({ where: { viewId: model.get('id'), }, }); }); */ }