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

"use client"
import { zodResolver } from "@hookform/resolvers/zod"
import { format } from "date-fns"
import { useRouter } from "next/navigation"
import { Suspense, useCallback, useState } from "react"
import { Controller, useForm } from "react-hook-form"
import { toast } from "sonner"
import { z } from "zod"
import { getPageUser } from "@/actions/user"
import { UpdateDialog } from "@/app/(root)/cust/update"
import { Auth } from "@/components/auth"
import { DataTable } from "@/components/data-table"
import { Page } from "@/components/page"
import { Badge } from "@/components/ui/badge"
import { Button } from "@/components/ui/button"
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu"
import {
  Field,
  FieldError,
  FieldGroup,
  FieldLabel,
} from "@/components/ui/field"
import { Input } from "@/components/ui/input"
import {
  ScopeBalanceActivityReadOfUser,
  ScopeBatchReadOfUser,
  ScopeBillReadOfUser,
  ScopeChannelReadOfUser,
  ScopeResourceRead,
  ScopeTradeReadOfUser,
  ScopeUserWrite,
  ScopeUserWriteBalance,
} from "@/lib/scopes"
import type { User } from "@/models/user"
import { AddUserDialog } from "../../cust/create"

interface UserQueryParams {
  account?: string
  name?: string
}

const filterSchema = z.object({
  phone: z.string().optional(),
  name: z.string().optional(),
})

type FormValues = z.infer<typeof filterSchema>

export default function UserQueryPage() {
  const [userList, setUserList] = useState<User[]>([])
  const [loading, setLoading] = useState(false)
  const [currentFilters, setCurrentFilters] = useState<UserQueryParams>({})

  const router = useRouter()
  const { control, handleSubmit, reset } = useForm<FormValues>({
    resolver: zodResolver(filterSchema),
    defaultValues: {
      phone: "",
      name: "",
    },
  })

  const fetchUsers = useCallback(async (filters: UserQueryParams = {}) => {
    setLoading(true)
    try {
      const res = await getPageUser(filters)
      if (res.success) {
        const data = Array.isArray(res.data) ? res.data : [res.data]
        setUserList(data)
      } else {
        toast.error(res.message || "获取用户失败")
        setUserList([])
      }
    } catch (error) {
      const message = error instanceof Error ? error.message : error
      toast.error(`获取用户失败: ${message}`)
    } finally {
      setLoading(false)
    }
  }, [])

  const onFilter = handleSubmit((data: FormValues) => {
    const params: UserQueryParams = {}
    if (data.phone?.trim()) params.account = data.phone.trim()
    if (data.name?.trim()) params.name = data.name.trim()
    if (Object.keys(params).length === 0) {
      toast.info("请至少输入一个搜索条件")
      return
    }
    setCurrentFilters(params)
    fetchUsers(params)
  })

  const refreshTable = useCallback(() => {
    if (Object.keys(currentFilters).length > 0) {
      fetchUsers(currentFilters)
    }
  }, [fetchUsers, currentFilters])

  const handleReset = () => {
    reset()
    setCurrentFilters({})
    setUserList([])
  }

  return (
    <Page>
      <form onSubmit={onFilter} className="bg-card p-4 rounded-lg">
        <div className="flex flex-wrap items-end gap-4">
          <Controller
            name="phone"
            control={control}
            render={({ field, fieldState }) => (
              <Field
                data-invalid={fieldState.invalid}
                className="w-40 flex-none"
              >
                <FieldLabel>手机号</FieldLabel>
                <Input {...field} placeholder="请输入手机号" />
                <FieldError>{fieldState.error?.message}</FieldError>
              </Field>
            )}
          />

          <Controller
            name="name"
            control={control}
            render={({ field, fieldState }) => (
              <Field
                data-invalid={fieldState.invalid}
                className="w-40 flex-none"
              >
                <FieldLabel>姓名</FieldLabel>
                <Input {...field} placeholder="请输入姓名" />
                <FieldError>{fieldState.error?.message}</FieldError>
              </Field>
            )}
          />
        </div>

        <FieldGroup className="flex-row justify-start mt-4 gap-2">
          <Auth scope={ScopeUserWrite}>
            <AddUserDialog onSuccess={refreshTable} />
          </Auth>
          <Button type="button" variant="outline" onClick={handleReset}>
            重置
          </Button>
          <Button type="submit">搜索</Button>
        </FieldGroup>
      </form>

      <Suspense>
        <DataTable<User>
          data={userList || []}
          status={loading ? "load" : "done"}
          columns={[
            { header: "手机", accessorKey: "phone" },
            {
              header: "创建时间",
              accessorKey: "created_at",
              cell: ({ row }) =>
                format(new Date(row.original.created_at), "yyyy-MM-dd HH:mm"),
            },
            {
              header: "客户来源",
              accessorKey: "source",
              cell: ({ row }) => {
                const sourceMap: Record<number, string> = {
                  0: "官网注册",
                  1: "管理员添加",
                  2: "代理商注册",
                  3: "代理商添加",
                }
                return sourceMap[row.original.source] ?? "官网注册"
              },
            },
            {
              header: "余额",
              accessorKey: "balance",
              cell: ({ row }) => {
                const balance = Number(row.original.balance) || 0
                return (
                  <span
                    className={
                      balance > 0 ? "text-green-500" : "text-orange-500"
                    }
                  >
                    ￥{balance.toFixed(2)}
                  </span>
                )
              },
            },
            { header: "账号", accessorKey: "username" },
            {
              header: "账号状态",
              accessorKey: "status",
              cell: ({ row }) => (row.original.status === 1 ? "正常" : "禁用"),
            },
            {
              header: "客户经理",
              cell: ({ row }) => row.original.admin?.name || "",
            },
            { header: "姓名", accessorKey: "name" },
            {
              header: "实名状态",
              accessorKey: "id_type",
              cell: ({ row }) => (
                <Badge
                  variant={row.original.id_type === 1 ? "default" : "secondary"}
                  className={
                    row.original.id_type === 1
                      ? "bg-green-100 text-green-800"
                      : "bg-gray-100 text-gray-800"
                  }
                >
                  {row.original.id_type === 1 ? "已认证" : "未认证"}
                </Badge>
              ),
            },
            {
              header: "最后登录时间",
              accessorKey: "last_login",
              cell: ({ row }) =>
                row.original.last_login
                  ? format(
                      new Date(row.original.last_login),
                      "yyyy-MM-dd HH:mm",
                    )
                  : "",
            },
            {
              header: "最后登录IP",
              accessorKey: "last_login_ip",
              cell: ({ row }) => row.original.last_login_ip || "",
            },
            { header: "联系方式", accessorKey: "contact_wechat" },
            {
              id: "action",
              meta: { pin: "right" },
              header: "操作",
              cell: ({ row }) => {
                return (
                  <div className="flex flex-wrap gap-2 w-75">
                    <Auth scope={ScopeUserWriteBalance}>
                      <UpdateDialog
                        user={row.original}
                        onSuccess={refreshTable}
                      />
                    </Auth>
                    <DropdownMenu>
                      <DropdownMenuTrigger asChild>
                        <Button size="sm" variant="outline">
                          打开菜单
                        </Button>
                      </DropdownMenuTrigger>
                      <DropdownMenuContent align="end" className="w-8">
                        <Auth scope={ScopeTradeReadOfUser}>
                          <DropdownMenuItem
                            onClick={() => {
                              router.push(
                                `/client/trade?userId=${row.original.id}`,
                              )
                            }}
                          >
                            交易明细
                          </DropdownMenuItem>
                        </Auth>
                        <Auth scope={ScopeBillReadOfUser}>
                          <DropdownMenuItem
                            onClick={() => {
                              router.push(
                                `/client/billing?userId=${row.original.id}`,
                              )
                            }}
                          >
                            账单详情
                          </DropdownMenuItem>
                        </Auth>
                        <Auth scope={ScopeResourceRead}>
                          <DropdownMenuItem
                            onClick={() => {
                              router.push(
                                `/client/resources?userId=${row.original.id}`,
                              )
                            }}
                          >
                            套餐管理
                          </DropdownMenuItem>
                        </Auth>
                        <Auth scope={ScopeBatchReadOfUser}>
                          <DropdownMenuItem
                            onClick={() => {
                              router.push(
                                `/client/batch?userId=${row.original.id}`,
                              )
                            }}
                          >
                            提取记录
                          </DropdownMenuItem>
                        </Auth>
                        <Auth scope={ScopeChannelReadOfUser}>
                          <DropdownMenuItem
                            onClick={() => {
                              router.push(
                                `/client/channel?userId=${row.original.id}`,
                              )
                            }}
                          >
                            IP管理
                          </DropdownMenuItem>
                        </Auth>
                        <Auth scope={ScopeBalanceActivityReadOfUser}>
                          <DropdownMenuItem
                            onClick={() => {
                              router.push(
                                `/client/balance?userId=${row.original.id}&phone=${row.original.phone}`,
                              )
                            }}
                          >
                            余额明细
                          </DropdownMenuItem>
                        </Auth>
                      </DropdownMenuContent>
                    </DropdownMenu>
                  </div>
                )
              },
            },
          ]}
        />
      </Suspense>
    </Page>
  )
}