import { Context } from '@nocobase/actions'; import lodash from 'lodash'; import UiSchemaRepository from '../repository'; import { ActionParams } from '@nocobase/resourcer'; const getRepositoryFromCtx = (ctx: Context) => { return ctx.db.getCollection('uiSchemas').repository as UiSchemaRepository; }; const callRepositoryMethod = (method, paramsKey: 'resourceIndex' | 'values', optionsBuilder?) => { return async (ctx, next) => { const params = lodash.get(ctx.action.params, paramsKey); const options = optionsBuilder ? optionsBuilder(ctx.action.params) : {}; const repository = getRepositoryFromCtx(ctx); const returnValue = await repository[method](params, options); ctx.body = returnValue || { result: 'ok', }; await next(); }; }; function parseInsertAdjacentValues(values) { if (lodash.has(values, 'schema')) { return values; } return { schema: values, wrap: null }; } export const uiSchemaActions = { getJsonSchema: callRepositoryMethod('getJsonSchema', 'resourceIndex', (params: ActionParams) => { return { includeAsyncNode: params?.includeAsyncNode, }; }), getProperties: callRepositoryMethod('getProperties', 'resourceIndex'), insert: callRepositoryMethod('insert', 'values'), insertNewSchema: callRepositoryMethod('insertNewSchema', 'values'), remove: callRepositoryMethod('remove', 'resourceIndex'), patch: callRepositoryMethod('patch', 'values'), clearAncestor: callRepositoryMethod('clearAncestor', 'resourceIndex'), async insertAdjacent(ctx: Context, next) { const { resourceIndex, position, values, removeParentsIfNoChildren, breakRemoveOn } = ctx.action.params; const repository = getRepositoryFromCtx(ctx); const { schema, wrap } = parseInsertAdjacentValues(values); ctx.body = await repository.insertAdjacent(position, resourceIndex, schema, { removeParentsIfNoChildren, breakRemoveOn, wrap, }); await next(); }, insertBeforeBegin: insertPositionActionBuilder('beforeBegin'), insertAfterBegin: insertPositionActionBuilder('afterBegin'), insertBeforeEnd: insertPositionActionBuilder('beforeEnd'), insertAfterEnd: insertPositionActionBuilder('afterEnd'), }; function insertPositionActionBuilder(position: 'beforeBegin' | 'afterBegin' | 'beforeEnd' | 'afterEnd') { return async function (ctx: Context, next) { const { resourceIndex, values, removeParentsIfNoChildren, breakRemoveOn } = ctx.action.params; const repository = getRepositoryFromCtx(ctx); const { schema, wrap } = parseInsertAdjacentValues(values); ctx.body = await repository.insertAdjacent(position, resourceIndex, schema, { removeParentsIfNoChildren, breakRemoveOn, wrap, }); await next(); }; }