24 lines
668 B
TypeScript
24 lines
668 B
TypeScript
import { MultipleRelationRepository, Repository } from '@nocobase/database';
|
|
import { Context } from '.';
|
|
|
|
export function getRepositoryFromParams(ctx: Context) {
|
|
const { resourceName, resourceOf } = ctx.action;
|
|
|
|
if (resourceOf) {
|
|
return ctx.db.getRepository<MultipleRelationRepository>(resourceName, resourceOf);
|
|
}
|
|
|
|
return ctx.db.getRepository<Repository>(resourceName);
|
|
}
|
|
|
|
export function RelationRepositoryActionBuilder(method: 'remove' | 'set') {
|
|
return async function (ctx: Context, next) {
|
|
const repository = getRepositoryFromParams(ctx);
|
|
|
|
await repository[method](ctx.action.params.values);
|
|
|
|
ctx.status = 200;
|
|
await next();
|
|
};
|
|
}
|