@wordrhyme/auto-crud-server 0.2.0 → 0.4.0

package/dist/index.js

@@ -373,9 +373,58 @@ const dataTableConfig = {
 
 //#endregion
 //#region src/routers/_factory.ts
-/**
-* 过滤器项 Schema
-*/
+function resolveSoftDelete(option) {
+	if (!option) return null;
+	if (option === true) return {
+		column: "deletedAt",
+		getValue: () => /* @__PURE__ */ new Date()
+	};
+	if (typeof option === "string") return {
+		column: option,
+		getValue: () => /* @__PURE__ */ new Date()
+	};
+	const config = option;
+	return {
+		column: config.column,
+		getValue: config.value ?? (() => /* @__PURE__ */ new Date())
+	};
+}
+function resolveConfig(config) {
+	const mode = config.mode || "declarative";
+	if (mode === "factory") {
+		const factoryConfig = config;
+		return {
+			procedureFactory: factoryConfig.procedureFactory,
+			guard: void 0,
+			getScope: () => void 0,
+			getInject: (ctx, op) => factoryConfig.inject?.(ctx, op) ?? {},
+			checkAuthorize: async () => true
+		};
+	}
+	if (mode === "procedures") {
+		const proceduresConfig = config;
+		const defaultProc = proceduresConfig.defaultProcedure ?? publicProcedure;
+		return {
+			procedureFactory: (op) => proceduresConfig.procedures[op] ?? defaultProc,
+			guard: void 0,
+			getScope: () => void 0,
+			getInject: (ctx, op) => proceduresConfig.inject?.(ctx, op) ?? {},
+			checkAuthorize: async () => true
+		};
+	}
+	const declarativeConfig = config;
+	const baseProcedure = declarativeConfig.procedure ?? publicProcedure;
+	return {
+		procedureFactory: () => baseProcedure,
+		guard: declarativeConfig.guard,
+		getScope: (ctx, tbl, op) => declarativeConfig.scope?.(ctx, tbl, op),
+		getInject: (ctx, op) => declarativeConfig.inject?.(ctx, op) ?? {},
+		checkAuthorize: async (ctx, resource, op) => {
+			if (declarativeConfig.authorize) return declarativeConfig.authorize(ctx, resource, op);
+			return true;
+		}
+	};
+}
 const filterItemSchema = z.object({
 	id: z.string(),
 	value: z.union([z.string(), z.array(z.string())]),
@@ -383,9 +432,6 @@ const filterItemSchema = z.object({
 	operator: z.enum(dataTableConfig.operators),
 	filterId: z.string()
 });
-/**
-* 列表查询输入 Schema
-*/
 const listInputSchema = z.object({
 	page: z.number().min(1).default(1),
 	perPage: z.number().min(1).max(100).default(10),
@@ -396,98 +442,178 @@ const listInputSchema = z.object({
 	filters: z.array(filterItemSchema).optional(),
 	joinOperator: z.enum(["and", "or"]).default("and")
 });
-/**
-* 验证列是否在白名单中
-*/
-function validateColumn(columnId, allowedColumns, columnType) {
+function validateColumn(columnId, allowedColumns) {
 	if (!allowedColumns || allowedColumns.length === 0) return true;
 	return allowedColumns.includes(columnId);
 }
-/**
-* 创建通用 CRUD Router
-*/
 function createCrudRouter(config) {
-	const { table, insertSchema, updateSchema, idField = "id", filterableColumns, sortableColumns, maxBatchSize = 100, procedure: baseProcedure = publicProcedure, authorize } = config;
+	const { table, insertSchema, updateSchema, idField = "id", filterableColumns, sortableColumns, maxBatchSize = 100, softDelete: softDeleteOption, hooks = {} } = config;
+	const resolved = resolveConfig(config);
+	const softDelete = resolveSoftDelete(softDeleteOption);
+	const h = hooks;
 	const getIdColumn = () => table[idField];
-	const withAuthorize = async (operation, input) => {
-		if (authorize) {
-			if (!await authorize({
-				operation,
-				input
-			})) throw new Error(`Unauthorized: ${operation} operation not allowed`);
+	const getSoftDeleteColumn = () => softDelete ? table[softDelete.column] : null;
+	const buildWhere = (ctx, operation, additionalCondition) => {
+		const conditions = [];
+		const scopeCondition = resolved.getScope(ctx, table, operation);
+		if (scopeCondition) conditions.push(scopeCondition);
+		if (softDelete && (operation === "list" || operation === "get")) {
+			const col = getSoftDeleteColumn();
+			if (col) conditions.push(isNull(col));
+		}
+		if (additionalCondition) conditions.push(additionalCondition);
+		if (conditions.length === 0) return void 0;
+		if (conditions.length === 1) return conditions[0];
+		return and(...conditions);
+	};
+	const withGuard = async (ctx, operation) => {
+		if (resolved.guard) {
+			if (!await resolved.guard(ctx, operation)) throw new Error(`Forbidden: ${operation} not allowed`);
 		}
 	};
-	return router({
-		list: baseProcedure.input(listInputSchema).query(async ({ ctx, input }) => {
-			await withAuthorize("list", input);
-			const offset = (input.page - 1) * input.perPage;
-			const validatedFilters = input.filters?.filter((filter) => validateColumn(filter.id, filterableColumns, "filter"));
-			const where = validatedFilters?.length ? filterColumns({
+	const withAuthorize = async (ctx, resource, operation) => {
+		if (!resource) return;
+		if (!await resolved.checkAuthorize(ctx, resource, operation)) throw new Error(`Forbidden: Cannot ${operation} this resource`);
+	};
+	const procedures = {
+		list: resolved.procedureFactory("list").input(listInputSchema).query(async ({ ctx, input }) => {
+			await withGuard(ctx, "list");
+			let processedInput = input;
+			if (h.beforeList) {
+				const result = await h.beforeList(ctx, processedInput);
+				if (result) processedInput = result;
+			}
+			const offset = (processedInput.page - 1) * processedInput.perPage;
+			const validatedFilters = processedInput.filters?.filter((filter) => validateColumn(filter.id, filterableColumns));
+			const where = buildWhere(ctx, "list", validatedFilters?.length ? filterColumns({
 				table,
 				filters: validatedFilters,
-				joinOperator: input.joinOperator
-			}) : void 0;
+				joinOperator: processedInput.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, "sort")) {
+			if (processedInput.sort?.length) {
+				const sortField = processedInput.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);
+			const data = await query.limit(processedInput.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 {
+			let listResult = {
 				data,
 				total: count,
-				page: input.page,
-				perPage: input.perPage,
-				pageCount: Math.ceil(count / input.perPage)
+				page: processedInput.page,
+				perPage: processedInput.perPage,
+				pageCount: Math.ceil(count / processedInput.perPage)
 			};
+			if (h.afterList) {
+				const result = await h.afterList(ctx, listResult);
+				if (result) listResult = result;
+			}
+			return listResult;
 		}),
-		getById: baseProcedure.input(z.string()).query(async ({ ctx, input }) => {
-			await withAuthorize("get", input);
-			const column = getIdColumn();
-			const [item] = await ctx.db.select().from(table).where(eq(column, input));
-			return item ?? null;
+		getById: resolved.procedureFactory("get").input(z.string()).query(async ({ ctx, input }) => {
+			await withGuard(ctx, "get");
+			if (h.beforeGet) await h.beforeGet(ctx, input);
+			const where = buildWhere(ctx, "get", eq(getIdColumn(), input));
+			const [item] = await ctx.db.select().from(table).where(where);
+			await withAuthorize(ctx, item, "get");
+			let result = item ?? null;
+			if (h.afterGet) {
+				const hookResult = await h.afterGet(ctx, result);
+				if (hookResult !== void 0) result = hookResult;
+			}
+			return result;
 		}),
-		create: baseProcedure.input(insertSchema).mutation(async ({ ctx, input }) => {
-			await withAuthorize("create", input);
-			const [created] = await ctx.db.insert(table).values(input).returning();
+		create: resolved.procedureFactory("create").input(insertSchema).mutation(async ({ ctx, input }) => {
+			await withGuard(ctx, "create");
+			let processedData = input;
+			if (h.beforeCreate) {
+				const result = await h.beforeCreate(ctx, processedData);
+				if (result) processedData = result;
+			}
+			const injectData = resolved.getInject(ctx, "create");
+			const data = {
+				...processedData,
+				...injectData
+			};
+			const [created] = await ctx.db.insert(table).values(data).returning();
+			if (h.afterCreate) await h.afterCreate(ctx, created);
 			return created;
 		}),
-		update: baseProcedure.input(z.object({
+		update: resolved.procedureFactory("update").input(z.object({
 			id: z.string(),
 			data: updateSchema
 		})).mutation(async ({ ctx, input }) => {
-			await withAuthorize("update", input);
-			const column = getIdColumn();
-			const [updated] = await ctx.db.update(table).set(input.data).where(eq(column, input.id)).returning();
+			await withGuard(ctx, "update");
+			const where = buildWhere(ctx, "update", eq(getIdColumn(), input.id));
+			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");
+			let processedData = input.data;
+			if (h.beforeUpdate) {
+				const result = await h.beforeUpdate(ctx, input.id, processedData, existing);
+				if (result) processedData = result;
+			}
+			const injectData = resolved.getInject(ctx, "update");
+			const data = {
+				...processedData,
+				...injectData
+			};
+			const [updated] = await ctx.db.update(table).set(data).where(where).returning();
+			if (h.afterUpdate) await h.afterUpdate(ctx, updated);
 			return updated;
 		}),
-		delete: baseProcedure.input(z.string()).mutation(async ({ ctx, input }) => {
-			await withAuthorize("delete", input);
-			const column = getIdColumn();
-			const [deleted] = await ctx.db.delete(table).where(eq(column, input)).returning();
+		delete: resolved.procedureFactory("delete").input(z.string()).mutation(async ({ ctx, input }) => {
+			await withGuard(ctx, "delete");
+			const where = buildWhere(ctx, "delete", eq(getIdColumn(), input));
+			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 (h.beforeDelete) await h.beforeDelete(ctx, input, existing);
+			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();
+			if (h.afterDelete) await h.afterDelete(ctx, deleted);
 			return deleted;
 		}),
-		deleteMany: baseProcedure.input(z.array(z.string()).max(maxBatchSize, `Maximum ${maxBatchSize} items allowed`)).mutation(async ({ ctx, input }) => {
-			await withAuthorize("deleteMany", input);
-			const column = getIdColumn();
-			return { deleted: (await ctx.db.delete(table).where(inArray(column, input)).returning()).length };
+		deleteMany: resolved.procedureFactory("deleteMany").input(z.array(z.string()).max(maxBatchSize)).mutation(async ({ ctx, input }) => {
+			await withGuard(ctx, "deleteMany");
+			if (h.beforeDeleteMany) await h.beforeDeleteMany(ctx, input);
+			const where = buildWhere(ctx, "deleteMany", inArray(getIdColumn(), input));
+			let result;
+			if (softDelete) result = { deleted: (await ctx.db.update(table).set({ [softDelete.column]: softDelete.getValue() }).where(where).returning()).length };
+			else result = { deleted: (await ctx.db.delete(table).where(where).returning()).length };
+			if (h.afterDeleteMany) await h.afterDeleteMany(ctx, result);
+			return result;
 		}),
-		updateMany: baseProcedure.input(z.object({
-			ids: z.array(z.string()).max(maxBatchSize, `Maximum ${maxBatchSize} items allowed`),
+		updateMany: resolved.procedureFactory("updateMany").input(z.object({
+			ids: z.array(z.string()).max(maxBatchSize),
 			data: updateSchema
 		})).mutation(async ({ ctx, input }) => {
-			await withAuthorize("updateMany", input);
-			const column = getIdColumn();
-			return { updated: (await ctx.db.update(table).set(input.data).where(inArray(column, input.ids)).returning()).length };
+			await withGuard(ctx, "updateMany");
+			let processedData = input.data;
+			if (h.beforeUpdateMany) {
+				const result$1 = await h.beforeUpdateMany(ctx, input.ids, processedData);
+				if (result$1) processedData = result$1;
+			}
+			const where = buildWhere(ctx, "updateMany", inArray(getIdColumn(), input.ids));
+			const injectData = resolved.getInject(ctx, "update");
+			const data = {
+				...processedData,
+				...injectData
+			};
+			const result = { updated: (await ctx.db.update(table).set(data).where(where).returning()).length };
+			if (h.afterUpdateMany) await h.afterUpdateMany(ctx, result);
+			return result;
 		})
-	});
+	};
+	const crudRouter = router(procedures);
+	return Object.assign(crudRouter, { procedures });
 }
 
 //#endregion

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@wordrhyme/auto-crud-server",
   "type": "module",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "tRPC server utilities for auto-crud - automatic CRUD routers for Drizzle ORM",
   "author": "wordrhyme",
   "license": "MIT",
@@ -46,11 +46,11 @@
     "tsdown": "^0.15.12",
     "typescript": "^5.9.3",
     "zod": "^4.1.13",
-    "@internal/prettier-config": "0.0.1",
-    "@internal/vitest-config": "0.1.0",
     "@internal/eslint-config": "0.3.0",
+    "@internal/tsconfig": "0.1.0",
+    "@internal/prettier-config": "0.0.1",
     "@internal/tsdown-config": "0.1.0",
-    "@internal/tsconfig": "0.1.0"
+    "@internal/vitest-config": "0.1.0"
   },
   "prettier": "@internal/prettier-config",
   "scripts": {