crush-level-web/src/app/(main)/contact/contact-page.tsx

'use client'

import { Avatar, AvatarImage, AvatarFallback } from '@/components/ui/avatar'
import { Button } from '@/components/ui/button'
import { InfiniteScrollList } from '@/components/ui/infinite-scroll-list'
import RenderContactStatusText from './components/RenderContactStatusText'
import { useHeartbeatRelationListInfinite } from '@/hooks/useIm'
import { HeartbeatRelationListOutput } from '@/services/im/types'
import { useMemo } from 'react'
import { useRouter } from 'next/navigation'
import AIRelationTag from '@/components/features/AIRelationTag'
import Link from 'next/link'
import { usePrefetchRoutes } from '@/hooks/useGlobalPrefetchRoutes'
import Image from 'next/image'

// 联系人数据类型现在使用API返回的数据结构
type ContactItem = HeartbeatRelationListOutput

// 联系人卡片组件
const ContactCard = ({ contact }: { contact: ContactItem }) => {
  const router = useRouter()

  // 计算年龄
  const age = useMemo(() => {
    if (!contact.birthday) return null
    const birthYear = new Date(contact.birthday).getFullYear()
    const currentYear = new Date().getFullYear()
    return currentYear - birthYear
  }, [contact.birthday])

  // 跳转到聊天页面
  const handleChatClick = () => {
    if (contact.aiId) {
      router.push(`/chat/${contact.aiId}`)
    }
  }

  return (
    <div className="flex w-full items-center justify-between gap-4">
      {/* 用户信息部分 */}
      <div className="flex flex-1 items-center gap-4">
        {/* 头像 */}
        <Link href={`/@${contact.aiId}`} prefetch>
          <Avatar className="size-16">
            <AvatarImage src={contact.headImg} alt={contact.nickname || contact.roleName} />
            <AvatarFallback className="bg-surface-element-normal text-txt-primary-normal txt-title-m">
              {(contact.nickname || contact.roleName || 'A').charAt(0)}
            </AvatarFallback>
          </Avatar>
        </Link>

        {/* 用户详细信息 */}
        <div className="flex flex-1 flex-col gap-2">
          {/* 名字和标签 */}
          <div className="flex items-center gap-2">
            <Link href={`/@${contact.aiId}`} prefetch>
              <h3 className="txt-title-m text-white">{contact.nickname || contact.roleName}</h3>
            </Link>
            {contact.heartbeatLevel && contact.isShow && (
              <AIRelationTag heartbeatLevel={contact.heartbeatLevel} size="small" />
            )}
          </div>

          {/* 心动值和用户信息 */}
          <div className="flex items-center gap-2">
            {/* 心动值 */}
            <div className="flex items-center gap-1">
              <Image
                src="/icons/heart.svg"
                alt="Heart"
                width={12}
                height={12}
                className="h-3 w-3"
              />
              <span className="txt-numMonotype-s">{contact.heartbeatVal || 0}°</span>
            </div>

            {/* 分隔线 */}
            <div className="border-outline-normal h-3 w-0 border-l" />

            {/* 用户详细信息 */}
            <span className="txt-label-m flex-1 overflow-hidden text-ellipsis whitespace-nowrap text-white">
              {[age && `${age}`, contact.characterName, contact.tagName]
                .filter(Boolean)
                .join(' · ')}
            </span>
          </div>
        </div>
      </div>

      {/* 聊天按钮 */}
      <Button className="min-w-[80px]" onClick={handleChatClick}>
        <i className="iconfont icon-Chat !text-[24px]" />
      </Button>
    </div>
  )
}

const ContactsPage = () => {
  // 使用无限查询获取心动关系列表
  const { data, fetchNextPage, hasNextPage, isLoading, isFetchingNextPage, error } =
    useHeartbeatRelationListInfinite()

  // 扁平化所有页面的数据
  const allContacts = useMemo(() => {
    return data?.pages.flatMap((page) => page.datas || []) || []
  }, [data])
  const chatRoutes = useMemo(
    () =>
      allContacts.slice(0, 20).map((contact) => (contact?.aiId ? `/chat/${contact.aiId}` : null)),
    [allContacts]
  )
  usePrefetchRoutes(chatRoutes)

  // 加载状态骨架屏组件
  const ContactSkeleton = () => (
    <div className="flex w-full animate-pulse items-center justify-between gap-4">
      <div className="flex flex-1 items-center gap-4">
        <div className="bg-surface-nest-normal size-16 rounded-full" />
        <div className="flex flex-1 flex-col gap-2">
          <div className="flex items-start gap-2">
            <div className="bg-surface-nest-normal h-6 w-24 rounded" />
            <div className="bg-surface-nest-normal h-5 w-16 rounded" />
          </div>
          <div className="bg-surface-nest-normal h-4 w-48 rounded" />
        </div>
      </div>
      <div className="bg-surface-nest-normal h-10 w-[80px] rounded" />
    </div>
  )

  // 空状态组件
  const EmptyState = () => (
    <div className="flex flex-col items-center justify-center py-16 text-center">
      <Image
        src="/icons/empty.svg"
        alt="Empty"
        width={64}
        height={64}
        className="mb-4 h-16 w-16 opacity-50"
      />
      <h3 className="txt-title-m text-txt-secondary-normal mb-2">No crushes found</h3>
      <p className="txt-body-m text-txt-tertiary-normal">
        Start chatting with AI characters to build your crushes
      </p>
    </div>
  )

  return (
    <div className="mx-auto flex w-[752px] flex-col gap-6 pt-28 pb-[200px]">
      {/* 页面标题 */}
      <h1 className="txt-title-l">My Crushes</h1>

      {/* 心动值统计信息栏 */}
      <RenderContactStatusText hasContacts={allContacts.length > 0} />

      {/* 联系人列表 */}
      <InfiniteScrollList<ContactItem>
        items={allContacts}
        renderItem={(contact) => <ContactCard contact={contact} />}
        getItemKey={(contact) => contact.aiId?.toString() || 'unknown'}
        hasNextPage={!!hasNextPage}
        isLoading={isLoading || isFetchingNextPage}
        fetchNextPage={fetchNextPage}
        columns={1}
        gap={4}
        className="!grid-cols-1" // 强制单列布局
        LoadingSkeleton={ContactSkeleton}
        EmptyComponent={EmptyState}
        hasError={!!error}
        onRetry={() => window.location.reload()}
      />
    </div>
  )
}

export default ContactsPage