客户管理添加修改金额操作

src/app/(root)/cust/page.tsx

@@ -24,6 +24,7 @@ import {
   SelectValue,
 } from "@/components/ui/select"
 import type { Cust } from "@/models/cust"
+import { BalanceDialog } from "./balanceDialog"
 import { AddUserDialog } from "./create"
 import { UpdateDialog } from "./update"
 
@@ -67,6 +68,8 @@ export default function UserPage() {
   const [isAddDialogOpen, setIsAddDialogOpen] = useState(false)
   const [isEditDialogOpen, setIsEditDialogOpen] = useState(false)
   const [currentEditUser, setCurrentEditUser] = useState<Cust | null>(null)
+  const [balanceDialog, setBalanceDialog] = useState(false)
+  const [balance, setBalance] = useState<Cust | null>(null)
   const { control, handleSubmit, reset } = useForm<FormValues>({
     resolver: zodResolver(filterSchema),
     defaultValues: {
@@ -258,7 +261,7 @@ export default function UserPage() {
                   2: "代理商注册",
                   3: "代理商添加",
                 }
-                return sourceMap[row.original.source] ?? "未知"
+                return sourceMap[row.original.source] ?? "官网注册"
               },
             },
             {
@@ -352,15 +355,27 @@ export default function UserPage() {
               header: "操作",
               cell: ({ row }) => {
                 return (
-                  <Button
-                    size="sm"
-                    onClick={() => {
-                      setCurrentEditUser(row.original)
-                      setIsEditDialogOpen(true)
-                    }}
-                  >
-                    修改
-                  </Button>
+                  <div className="flex gap-2">
+                    <Button
+                      size="sm"
+                      variant="outline"
+                      onClick={() => {
+                        setBalance(row.original)
+                        setBalanceDialog(true)
+                      }}
+                    >
+                      修改余额
+                    </Button>
+                    <Button
+                      size="sm"
+                      onClick={() => {
+                        setCurrentEditUser(row.original)
+                        setIsEditDialogOpen(true)
+                      }}
+                    >
+                      修改
+                    </Button>
+                  </div>
                 )
               },
             },
@@ -380,6 +395,13 @@ export default function UserPage() {
         currentUser={currentEditUser}
         onSuccess={refreshTable}
       />
+
+      <BalanceDialog
+        open={balanceDialog}
+        onOpenChange={setBalanceDialog}
+        currentUser={balance}
+        onSuccess={refreshTable}
+      />
     </div>
   )
 }