@wordrhyme/auto-crud-server 0.3.0 → 0.6.0

package/dist/index.js

@@ -447,9 +447,10 @@ function validateColumn(columnId, allowedColumns) {
 	return allowedColumns.includes(columnId);
 }
 function createCrudRouter(config) {
-	const { table, insertSchema, updateSchema, idField = "id", filterableColumns, sortableColumns, maxBatchSize = 100, softDelete: softDeleteOption } = config;
+	const { table, insertSchema, updateSchema, idField = "id", filterableColumns, sortableColumns, maxBatchSize = 100, softDelete: softDeleteOption, middleware = {} } = config;
 	const resolved = resolveConfig(config);
 	const softDelete = resolveSoftDelete(softDeleteOption);
+	const m = middleware;
 	const getIdColumn = () => table[idField];
 	const getSoftDeleteColumn = () => softDelete ? table[softDelete.column] : null;
 	const buildWhere = (ctx, operation, additionalCondition) => {
@@ -477,50 +478,75 @@ function createCrudRouter(config) {
 	const procedures = {
 		list: resolved.procedureFactory("list").input(listInputSchema).query(async ({ ctx, input }) => {
 			await withGuard(ctx, "list");
-			const offset = (input.page - 1) * input.perPage;
-			const validatedFilters = input.filters?.filter((filter) => validateColumn(filter.id, filterableColumns));
-			const where = buildWhere(ctx, "list", validatedFilters?.length ? filterColumns({
-				table,
-				filters: validatedFilters,
-				joinOperator: input.joinOperator
-			}) : void 0);
-			let query = ctx.db.select().from(table).$dynamic();
-			if (where) query = query.where(where);
-			if (input.sort?.length) {
-				const sortField = input.sort[0];
-				if (sortField && validateColumn(sortField.id, sortableColumns)) {
-					const column = table[sortField.id];
-					if (column) query = query.orderBy(sortField.desc ? desc(column) : asc(column));
+			const doList = async (listInput$1) => {
+				const offset = (listInput$1.page - 1) * listInput$1.perPage;
+				const validatedFilters = listInput$1.filters?.filter((filter) => validateColumn(filter.id, filterableColumns));
+				const where = buildWhere(ctx, "list", validatedFilters?.length ? filterColumns({
+					table,
+					filters: validatedFilters,
+					joinOperator: listInput$1.joinOperator
+				}) : void 0);
+				let query = ctx.db.select().from(table).$dynamic();
+				if (where) query = query.where(where);
+				if (listInput$1.sort?.length) {
+					const sortField = listInput$1.sort[0];
+					if (sortField && validateColumn(sortField.id, sortableColumns)) {
+						const column = table[sortField.id];
+						if (column) query = query.orderBy(sortField.desc ? desc(column) : asc(column));
+					}
 				}
-			}
-			const data = await query.limit(input.perPage).offset(offset);
-			let countQuery = ctx.db.select({ count: sql`count(*)::int` }).from(table).$dynamic();
-			if (where) countQuery = countQuery.where(where);
-			const count = (await countQuery)[0]?.count ?? 0;
-			return {
-				data,
-				total: count,
-				page: input.page,
-				perPage: input.perPage,
-				pageCount: Math.ceil(count / input.perPage)
+				const data = await query.limit(listInput$1.perPage).offset(offset);
+				let countQuery = ctx.db.select({ count: sql`count(*)::int` }).from(table).$dynamic();
+				if (where) countQuery = countQuery.where(where);
+				const count = (await countQuery)[0]?.count ?? 0;
+				return {
+					data,
+					total: count,
+					page: listInput$1.page,
+					perPage: listInput$1.perPage,
+					pageCount: Math.ceil(count / listInput$1.perPage)
+				};
 			};
+			const listInput = input;
+			if (m.list) return m.list({
+				ctx,
+				input: listInput,
+				next: async (modifiedInput) => doList(modifiedInput ?? listInput)
+			});
+			return doList(listInput);
 		}),
 		getById: resolved.procedureFactory("get").input(z.string()).query(async ({ ctx, input }) => {
 			await withGuard(ctx, "get");
-			const where = buildWhere(ctx, "get", eq(getIdColumn(), input));
-			const [item] = await ctx.db.select().from(table).where(where);
-			await withAuthorize(ctx, item, "get");
-			return item ?? null;
+			const doGet = async () => {
+				const where = buildWhere(ctx, "get", eq(getIdColumn(), input));
+				const [item] = await ctx.db.select().from(table).where(where);
+				await withAuthorize(ctx, item, "get");
+				return item ?? null;
+			};
+			if (m.get) return m.get({
+				ctx,
+				id: input,
+				next: doGet
+			});
+			return doGet();
 		}),
 		create: resolved.procedureFactory("create").input(insertSchema).mutation(async ({ ctx, input }) => {
 			await withGuard(ctx, "create");
-			const injectData = resolved.getInject(ctx, "create");
-			const data = {
-				...input,
-				...injectData
+			const doCreate = async (inputData) => {
+				const injectData = resolved.getInject(ctx, "create");
+				const data = {
+					...inputData,
+					...injectData
+				};
+				const [created] = await ctx.db.insert(table).values(data).returning();
+				return created;
 			};
-			const [created] = await ctx.db.insert(table).values(data).returning();
-			return created;
+			if (m.create) return m.create({
+				ctx,
+				input,
+				next: async (modifiedInput) => doCreate(modifiedInput ?? input)
+			});
+			return doCreate(input);
 		}),
 		update: resolved.procedureFactory("update").input(z.object({
 			id: z.string(),
@@ -531,13 +557,23 @@ function createCrudRouter(config) {
 			const [existing] = await ctx.db.select().from(table).where(where);
 			await withAuthorize(ctx, existing, "update");
 			if (!existing) throw new Error("Resource not found or access denied");
-			const injectData = resolved.getInject(ctx, "update");
-			const data = {
-				...input.data,
-				...injectData
+			const doUpdate = async (updateData) => {
+				const injectData = resolved.getInject(ctx, "update");
+				const data = {
+					...updateData,
+					...injectData
+				};
+				const [updated] = await ctx.db.update(table).set(data).where(where).returning();
+				return updated;
 			};
-			const [updated] = await ctx.db.update(table).set(data).where(where).returning();
-			return updated;
+			if (m.update) return m.update({
+				ctx,
+				id: input.id,
+				data: input.data,
+				existing,
+				next: async (modifiedData) => doUpdate(modifiedData ?? input.data)
+			});
+			return doUpdate(input.data);
 		}),
 		delete: resolved.procedureFactory("delete").input(z.string()).mutation(async ({ ctx, input }) => {
 			await withGuard(ctx, "delete");
@@ -545,18 +581,33 @@ function createCrudRouter(config) {
 			const [existing] = await ctx.db.select().from(table).where(where);
 			await withAuthorize(ctx, existing, "delete");
 			if (!existing) throw new Error("Resource not found or access denied");
-			if (softDelete) {
-				const [deleted$1] = await ctx.db.update(table).set({ [softDelete.column]: softDelete.getValue() }).where(where).returning();
-				return deleted$1;
-			}
-			const [deleted] = await ctx.db.delete(table).where(where).returning();
-			return deleted;
+			const doDelete = async () => {
+				let deleted;
+				if (softDelete) [deleted] = await ctx.db.update(table).set({ [softDelete.column]: softDelete.getValue() }).where(where).returning();
+				else [deleted] = await ctx.db.delete(table).where(where).returning();
+				return deleted;
+			};
+			if (m.delete) return m.delete({
+				ctx,
+				id: input,
+				existing,
+				next: doDelete
+			});
+			return doDelete();
 		}),
 		deleteMany: resolved.procedureFactory("deleteMany").input(z.array(z.string()).max(maxBatchSize)).mutation(async ({ ctx, input }) => {
 			await withGuard(ctx, "deleteMany");
 			const where = buildWhere(ctx, "deleteMany", inArray(getIdColumn(), input));
-			if (softDelete) return { deleted: (await ctx.db.update(table).set({ [softDelete.column]: softDelete.getValue() }).where(where).returning()).length };
-			return { deleted: (await ctx.db.delete(table).where(where).returning()).length };
+			const doDeleteMany = async () => {
+				if (softDelete) return { deleted: (await ctx.db.update(table).set({ [softDelete.column]: softDelete.getValue() }).where(where).returning()).length };
+				else return { deleted: (await ctx.db.delete(table).where(where).returning()).length };
+			};
+			if (m.deleteMany) return m.deleteMany({
+				ctx,
+				ids: input,
+				next: doDeleteMany
+			});
+			return doDeleteMany();
 		}),
 		updateMany: resolved.procedureFactory("updateMany").input(z.object({
 			ids: z.array(z.string()).max(maxBatchSize),
@@ -564,21 +615,145 @@ function createCrudRouter(config) {
 		})).mutation(async ({ ctx, input }) => {
 			await withGuard(ctx, "updateMany");
 			const where = buildWhere(ctx, "updateMany", inArray(getIdColumn(), input.ids));
-			const injectData = resolved.getInject(ctx, "update");
-			const data = {
-				...input.data,
-				...injectData
+			const doUpdateMany = async (updateData) => {
+				const injectData = resolved.getInject(ctx, "update");
+				const data = {
+					...updateData,
+					...injectData
+				};
+				return { updated: (await ctx.db.update(table).set(data).where(where).returning()).length };
 			};
-			return { updated: (await ctx.db.update(table).set(data).where(where).returning()).length };
+			if (m.updateMany) return m.updateMany({
+				ctx,
+				ids: input.ids,
+				data: input.data,
+				next: async (modifiedData) => doUpdateMany(modifiedData ?? input.data)
+			});
+			return doUpdateMany(input.data);
 		})
 	};
 	const crudRouter = router(procedures);
 	return Object.assign(crudRouter, { procedures });
 }
 
+//#endregion
+//#region src/lib/middleware-helpers.ts
+/**
+* 创建 after-only 的 middleware（最常见场景）
+* 在操作完成后执行副作用，不修改返回值
+*
+* @example
+* middleware: {
+*   create: afterMiddleware(async (ctx, result) => {
+*     await sendEmail(result);
+*     await logAudit(ctx.user, 'create', result);
+*   }),
+* }
+*/
+function afterMiddleware(fn) {
+	return async ({ ctx, next }) => {
+		const result = await next();
+		await fn(ctx, result);
+		return result;
+	};
+}
+/**
+* 创建 after-only 的 middleware，可修改返回值
+*
+* @example
+* middleware: {
+*   get: afterMiddlewareTransform(async (ctx, result) => {
+*     return { ...result, computed: calculateSomething(result) };
+*   }),
+* }
+*/
+function afterMiddlewareTransform(fn) {
+	return async ({ ctx, next }) => {
+		return fn(ctx, await next());
+	};
+}
+/**
+* 创建 before-only 的 middleware
+* 在操作执行前修改输入数据
+*
+* @example
+* middleware: {
+*   create: beforeMiddleware(async (ctx, input) => {
+*     return { ...input, slug: slugify(input.title), createdBy: ctx.user.id };
+*   }),
+* }
+*/
+function beforeMiddleware(fn) {
+	return async ({ ctx, input, next }) => {
+		return next(await fn(ctx, input));
+	};
+}
+/**
+* 组合多个 middleware 为一个
+*
+* @example
+* middleware: {
+*   create: composeMiddleware(
+*     beforeMiddleware((ctx, input) => ({ ...input, slug: slugify(input.title) })),
+*     afterMiddleware((ctx, result) => sendEmail(result)),
+*   ),
+* }
+*/
+function composeMiddleware(...middlewares) {
+	return async (params) => {
+		let index = 0;
+		const executeNext = async (input) => {
+			if (index >= middlewares.length) return params.next(input);
+			const currentMiddleware = middlewares[index++];
+			return currentMiddleware({
+				ctx: params.ctx,
+				input,
+				next: executeNext
+			});
+		};
+		return executeNext(params.input);
+	};
+}
+/**
+* 创建 list 操作的 after middleware
+*/
+function afterList(fn) {
+	return afterMiddleware(fn);
+}
+/**
+* 创建 list 操作的 before middleware
+*/
+function beforeList(fn) {
+	return beforeMiddleware(fn);
+}
+/**
+* 创建 create 操作的 after middleware
+*/
+function afterCreate(fn) {
+	return afterMiddleware(fn);
+}
+/**
+* 创建 create 操作的 before middleware
+*/
+function beforeCreate(fn) {
+	return beforeMiddleware(fn);
+}
+/**
+* 创建 update 操作的 after middleware
+*/
+function afterUpdate(fn) {
+	return afterMiddleware(fn);
+}
+/**
+* 创建 delete 操作的 after middleware
+*/
+function afterDelete(fn) {
+	return afterMiddleware(fn);
+}
+
 //#endregion
 //#region src/routers/index.ts
 const appRouter = router({});
 
 //#endregion
-export { appRouter, createCrudRouter, publicProcedure, router };
+export { afterCreate, afterDelete, afterList, afterMiddleware, afterMiddlewareTransform, afterUpdate, appRouter, beforeCreate, beforeList, beforeMiddleware, composeMiddleware, createCrudRouter, publicProcedure, router };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@wordrhyme/auto-crud-server",
   "type": "module",
-  "version": "0.3.0",
+  "version": "0.6.0",
   "description": "tRPC server utilities for auto-crud - automatic CRUD routers for Drizzle ORM",
   "author": "wordrhyme",
   "license": "MIT",