From 86ece33c001cb3eedd5feb57c17ed5d617f23df5 Mon Sep 17 00:00:00 2001
From: zetaloop <zetaloop@outlook.com>
Date: Fri, 20 Feb 2026 22:46:28 +0800
Subject: [PATCH] feat: add role-based wallet view for consumer vs player/owner

---
 app/(account)/wallet/page.tsx | 156 ++++++++++++++++++++++------------
 1 file changed, 104 insertions(+), 52 deletions(-)

diff --git a/app/(account)/wallet/page.tsx b/app/(account)/wallet/page.tsx
index 6a44867..00dec7e 100644
--- a/app/(account)/wallet/page.tsx
+++ b/app/(account)/wallet/page.tsx
@@ -1,3 +1,5 @@
+"use client"
+
 import {
   ArrowDownLeft,
   ArrowUpRight,
@@ -12,6 +14,7 @@ import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card"
 import { Input } from "@/components/ui/input"
 import { Separator } from "@/components/ui/separator"
 import { mockTransactions, walletBalance } from "@/lib/mock-data"
+import { useAuthStore } from "@/store/auth"
 
 const typeLabels: Record<string, string> = {
   topup: "充值",
@@ -30,6 +33,20 @@ const typeIcons: Record<string, typeof ArrowUpRight> = {
 }
 
 export default function WalletPage() {
+  const { currentRole } = useAuthStore()
+  const isConsumer = currentRole === "consumer"
+
+  const filteredTransactions = mockTransactions.filter((tx) => {
+    if (isConsumer) {
+      return ["topup", "payment", "refund"].includes(tx.type)
+    }
+    return ["income", "withdrawal"].includes(tx.type)
+  })
+
+  const incomeBalance = mockTransactions
+    .filter((tx) => tx.type === "income")
+    .reduce((acc, tx) => acc + tx.amount, 0)
+
   return (
     <div className="max-w-2xl space-y-6">
       <h1 className="text-2xl font-bold">钱包</h1>
@@ -38,43 +55,74 @@ export default function WalletPage() {
         <CardContent className="pt-6">
           <div className="flex items-center justify-between">
             <div>
-              <p className="text-sm text-muted-foreground">账户余额</p>
-              <p className="text-3xl font-bold mt-1">¥{walletBalance.toFixed(2)}</p>
+              <p className="text-sm text-muted-foreground">
+                {isConsumer ? "账户余额" : "收入余额"}
+              </p>
+              <p className="text-3xl font-bold mt-1">
+                ¥{isConsumer ? walletBalance.toFixed(2) : incomeBalance.toFixed(2)}
+              </p>
             </div>
             <Wallet className="h-10 w-10 text-muted-foreground" />
           </div>
           <div className="flex gap-2 mt-4">
-            <Button>
-              <DollarSign className="mr-1 h-4 w-4" />
-              充值
-            </Button>
-            <Button variant="outline">
-              <CreditCard className="mr-1 h-4 w-4" />
-              提现
-            </Button>
+            {isConsumer ? (
+              <Button>
+                <DollarSign className="mr-1 h-4 w-4" />
+                充值
+              </Button>
+            ) : (
+              <Button variant="outline">
+                <CreditCard className="mr-1 h-4 w-4" />
+                提现
+              </Button>
+            )}
           </div>
         </CardContent>
       </Card>
 
-      <Card>
-        <CardHeader>
-          <CardTitle className="text-base">快捷充值</CardTitle>
-        </CardHeader>
-        <CardContent className="space-y-4">
-          <div className="grid grid-cols-3 gap-2">
-            {[50, 100, 200, 500, 1000, 2000].map((amount) => (
-              <Button key={`amount-${amount}`} variant="outline" className="h-12">
-                ¥{amount}
-              </Button>
-            ))}
-          </div>
-          <Separator />
-          <div className="flex gap-2">
-            <Input placeholder="自定义金额" type="number" />
-            <Button>充值</Button>
-          </div>
-        </CardContent>
-      </Card>
+      {isConsumer ? (
+        <Card>
+          <CardHeader>
+            <CardTitle className="text-base">快捷充值</CardTitle>
+          </CardHeader>
+          <CardContent className="space-y-4">
+            <div className="grid grid-cols-3 gap-2">
+              {[50, 100, 200, 500, 1000, 2000].map((amount) => (
+                <Button key={`amount-${amount}`} variant="outline" className="h-12">
+                  ¥{amount}
+                </Button>
+              ))}
+            </div>
+            <Separator />
+            <div className="flex gap-2">
+              <Input placeholder="自定义金额" type="number" />
+              <Button>充值</Button>
+            </div>
+          </CardContent>
+        </Card>
+      ) : (
+        <Card>
+          <CardHeader>
+            <CardTitle className="text-base">收入概览</CardTitle>
+          </CardHeader>
+          <CardContent>
+            <div className="grid grid-cols-3 gap-4 text-center">
+              <div>
+                <p className="text-xs text-muted-foreground">本月收入</p>
+                <p className="text-lg font-bold">¥1,280.00</p>
+              </div>
+              <div>
+                <p className="text-xs text-muted-foreground">待结算</p>
+                <p className="text-lg font-bold">¥320.00</p>
+              </div>
+              <div>
+                <p className="text-xs text-muted-foreground">已提现</p>
+                <p className="text-lg font-bold">¥5,400.00</p>
+              </div>
+            </div>
+          </CardContent>
+        </Card>
+      )}
 
       <Card>
         <CardHeader className="flex flex-row items-center justify-between">
@@ -85,33 +133,37 @@ export default function WalletPage() {
           </Button>
         </CardHeader>
         <CardContent className="space-y-3">
-          {mockTransactions.map((tx) => {
-            const Icon = typeIcons[tx.type]
-            const isIncome = tx.amount > 0
-            return (
-              <div key={tx.id} className="flex items-center justify-between">
-                <div className="flex items-center gap-3">
-                  <div className="h-8 w-8 rounded-full bg-muted flex items-center justify-center">
-                    <Icon className="h-4 w-4" />
+          {filteredTransactions.length > 0 ? (
+            filteredTransactions.map((tx) => {
+              const Icon = typeIcons[tx.type]
+              const isIncome = tx.amount > 0
+              return (
+                <div key={tx.id} className="flex items-center justify-between">
+                  <div className="flex items-center gap-3">
+                    <div className="h-8 w-8 rounded-full bg-muted flex items-center justify-center">
+                      <Icon className="h-4 w-4" />
+                    </div>
+                    <div>
+                      <p className="text-sm font-medium">{tx.description}</p>
+                      <p className="text-xs text-muted-foreground">
+                        {new Date(tx.createdAt).toLocaleString("zh-CN")}
+                      </p>
+                    </div>
                   </div>
-                  <div>
-                    <p className="text-sm font-medium">{tx.description}</p>
-                    <p className="text-xs text-muted-foreground">
-                      {new Date(tx.createdAt).toLocaleString("zh-CN")}
+                  <div className="text-right">
+                    <p className={`text-sm font-medium ${isIncome ? "text-green-600" : ""}`}>
+                      {isIncome ? "+" : ""}¥{Math.abs(tx.amount).toFixed(2)}
                     </p>
+                    <Badge variant="outline" className="text-[10px]">
+                      {typeLabels[tx.type]}
+                    </Badge>
                   </div>
                 </div>
-                <div className="text-right">
-                  <p className={`text-sm font-medium ${isIncome ? "text-green-600" : ""}`}>
-                    {isIncome ? "+" : ""}¥{Math.abs(tx.amount).toFixed(2)}
-                  </p>
-                  <Badge variant="outline" className="text-[10px]">
-                    {typeLabels[tx.type]}
-                  </Badge>
-                </div>
-              </div>
-            )
-          })}
+              )
+            })
+          ) : (
+            <div className="text-center py-8 text-muted-foreground text-sm">暂无交易记录</div>
+          )}
         </CardContent>
       </Card>
     </div>