@wzyjs/next 0.2.60 → 0.2.62

package/dist/Base.zmodel

@@ -36,7 +36,7 @@ plugin trpc {
     importCreateRouter = "@/api/trpc/trpc"
     importProcedure = "@/api/trpc/procedures"
     zodSchemasImport = '@/api/generated/zod'
-    generateModelActions = 'findMany, findUnique, create, update, delete, deleteMany'
+    generateModelActions = 'findMany, findUnique, create, update, updateMany, delete, deleteMany, count'
 }
 
 plugin crud_tags {

package/dist/plugin/crud/useHabitGroupCRUD.ts

@@ -1,40 +1,68 @@
-// @ts-nocheck
+import React, { useState } from 'react'
 
 // @ts-ignore
 import { message } from '@/components'
 
 // @ts-ignore
-import { api } from '@/api/react'
+import { trpc as api } from '@/api/react'
 
-interface Option {
-  showTip?: boolean,
+import type { Prisma } from '@prisma/client'
+import type { UseTRPCQueryOptions } from '@trpc/react-query/shared'
+
+// 定义 Prisma 查询相关类型
+type FindManyArgs = Prisma.HabitGroupFindManyArgs;
+type FindUniqueArgs = Prisma.HabitGroupFindUniqueArgs;
+type GetPayload<T extends FindManyArgs | FindUniqueArgs> = Prisma.HabitGroupGetPayload<T>;
+
+// 定义 SelectSubset 类型
+type SelectSubset<T extends FindManyArgs | FindUniqueArgs, A> = Prisma.SelectSubset<T, A>;
+type UseOptions<T> = UseTRPCQueryOptions<T, T, Error>;
+
+// 定义 Option 接口
+interface Option<TList extends FindManyArgs = FindManyArgs, TInfo extends FindUniqueArgs = FindUniqueArgs> {
+  showTip?: boolean
   list?: false | {
-    query?: Parameters<typeof api.habitGroup.findMany.useQuery>[0],
-    option?: Parameters<typeof api.habitGroup.findMany.useQuery>[1],
-  },
-  info?: false | {
-    query?: Parameters<typeof api.habitGroup.findUnique.useQuery>[0],
-    option?: Parameters<typeof api.habitGroup.findUnique.useQuery>[1],
-  },
-  create?: Parameters<typeof api.habitGroup.create.useMutation>[0],
-  update?: Parameters<typeof api.habitGroup.update.useMutation>[0],
-  remove?: Parameters<typeof api.habitGroup.update.useMutation>[0],
-  delete?: Parameters<typeof api.habitGroup.delete.useMutation>[0],
+    query?: SelectSubset<TList, FindManyArgs>
+    option?: UseOptions<GetPayload<TList>[]>
+  }
+  info?: {
+    query: SelectSubset<TInfo, FindUniqueArgs>
+    option?: UseOptions<GetPayload<TInfo> | null>
+  }
+  create?: Parameters<typeof api.habitGroup.create.useMutation>[0]
+  update?: Parameters<typeof api.habitGroup.update.useMutation>[0]
+  remove?: Parameters<typeof api.habitGroup.update.useMutation>[0]
+  delete?: Parameters<typeof api.habitGroup.delete.useMutation>[0]
+}
+
+const r = (q: any) => {
+  if (!q) {
+    return
+  }
+  const query = JSON.parse(JSON.stringify(q))
+  delete query?.skip
+  delete query?.take
+  return query
 }
 
-export const useHabitGroupCRUD = (option: Option = {}) => {
+export const useHabitGroupCRUD = <TList extends FindManyArgs = FindManyArgs, TInfo extends FindUniqueArgs = FindUniqueArgs>(option: Option<TList, TInfo> = {}) => {
   const { list, info, create, update, remove, delete: deleteOption, showTip = true } = option
 
   const apiUtils = api.useUtils()
 
-  const listState = api.habitGroup.findMany.useQuery(list ? list.query : {}, {
-    enabled: typeof list === 'boolean' ? list : true,
-    ...list?.option,
+  const listState = api.habitGroup.findMany.useQuery(list === false ? undefined : list?.query, {
+    enabled: list !== false,
+    ...(list ? list.option : undefined),
+  })
+
+  const countState = api.habitGroup.count.useQuery(list === false ? {} as any : r(list?.query), {
+    enabled: list !== false,
+    ...(list ? list.option : undefined),
   })
 
-  const infoState = api.habitGroup.findUnique.useQuery(info ? info.query : { where: { id: '' } }, {
-    enabled: !!(info && info.query),
-    ...info?.option,
+  const infoState = api.habitGroup.findUnique.useQuery(info?.query || {} as any, {
+    enabled: !!info,
+    ...(info ? info.option : undefined),
   })
 
   const onSuccess = (tip: string) => {
@@ -44,6 +72,7 @@ export const useHabitGroupCRUD = (option: Option = {}) => {
     }
     if (list) {
       listState.refetch()
+      countState.refetch()
     }
   }
 
@@ -58,7 +87,7 @@ export const useHabitGroupCRUD = (option: Option = {}) => {
   })
 
   const createState = {
-    ...baseCreateState as unknown as typeof baseCreateState,
+    ...(baseCreateState as unknown as typeof baseCreateState),
     mutate: (data: Parameters<typeof baseCreateState.mutate>[0]['data']) => {
       return baseCreateState.mutate({ data })
     },
@@ -78,7 +107,7 @@ export const useHabitGroupCRUD = (option: Option = {}) => {
   })
 
   const updateState = {
-    ...baseUpdateState as unknown as typeof baseUpdateState,
+    ...(baseUpdateState as unknown as typeof baseUpdateState),
     mutate: (id: string, data: Parameters<typeof baseUpdateState.mutate>[0]['data']) => {
       return baseUpdateState.mutate({
         where: { id },
@@ -104,7 +133,7 @@ export const useHabitGroupCRUD = (option: Option = {}) => {
   })
 
   const removeState = {
-    ...baseRemoveState as unknown as typeof baseRemoveState,
+    ...(baseRemoveState as unknown as typeof baseRemoveState),
     mutate: (id: string) => {
       return baseRemoveState.mutate({
         where: { id },
@@ -130,7 +159,7 @@ export const useHabitGroupCRUD = (option: Option = {}) => {
   })
 
   const deleteState = {
-    ...baseDeleteState as unknown as typeof baseDeleteState,
+    ...(baseDeleteState as unknown as typeof baseDeleteState),
     mutate: (id: string) => {
       return baseDeleteState.mutate({
         where: { id },
@@ -143,13 +172,17 @@ export const useHabitGroupCRUD = (option: Option = {}) => {
     },
   }
 
+  const [query, setQuery] = useState()
+
   return {
     listState,
+    countState,
     infoState,
     createState,
     removeState,
     updateState,
     deleteState,
+
     apiUtils,
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wzyjs/next",
-  "version": "0.2.60",
+  "version": "0.2.62",
   "description": "description",
   "author": "wzy",
   "main": "dist/index.js",
@@ -17,7 +17,7 @@
   "dependencies": {
     "@zenstackhq/sdk": "^2.12.3"
   },
-  "gitHead": "81b5a3fc804541ed989d13f40098effd6540bfa5",
+  "gitHead": "d9493fc26e494769edcb3aab4d193e5d7ab68bc1",
   "publishConfig": {
     "access": "public"
   }