juwan-frontend/app/(account)/wallet/page.tsx

"use client"

import {
  ArrowDownLeft,
  ArrowUpRight,
  CreditCard,
  DollarSign,
  RefreshCw,
  Wallet,
} from "lucide-react"
import { Badge } from "@/components/ui/badge"
import { Button } from "@/components/ui/button"
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: "充值",
  payment: "支付",
  income: "收入",
  withdrawal: "提现",
  refund: "退款",
}

const typeIcons: Record<string, typeof ArrowUpRight> = {
  topup: ArrowDownLeft,
  payment: ArrowUpRight,
  income: ArrowDownLeft,
  withdrawal: ArrowUpRight,
  refund: ArrowDownLeft,
}

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>

      <Card>
        <CardContent className="pt-6">
          <div className="flex items-center justify-between">
            <div>
              <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">
            {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>

      {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">
          <CardTitle className="text-base">交易记录</CardTitle>
          <Button variant="ghost" size="sm">
            <RefreshCw className="mr-1 h-3.5 w-3.5" />
            刷新
          </Button>
        </CardHeader>
        <CardContent className="space-y-3">
          {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 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>
  )
}