"use client" import { useState } from "react" import type { FamilyMember } from "@/types/family" import { cn } from "@/lib/utils" import { useAvatarCache } from "@/hooks/use-avatar-cache" import { CircleUser, CircleUserRound, Heart, UserPlus, Users, Baby } from "lucide-react" import { MemberNameWithStatus } from "@/components/member-name-with-status" import { useRouter } from "next/navigation" import { useFamily } from "@/context/family-context" import { ContextMenu, ContextMenuContent, ContextMenuItem, ContextMenuSeparator, ContextMenuLabel, ContextMenuTrigger, } from "@/components/ui/context-menu" import { GenerationWarningDialog } from "@/components/tree/generation-warning-dialog" import { shouldShowGenerationWarning } from "@/lib/generation-utils" interface FamilyNodeProps { member: FamilyMember spouse?: FamilyMember spouses?: FamilyMember[] isRoot?: boolean onSelect?: (member: FamilyMember) => void isHighlighted?: boolean highlightedMemberId?: string | null relationMode?: boolean isSelected?: boolean selectedMembers?: string[] canEdit?: boolean onAddRelation?: (type: "child" | "spouse") => void // 折叠功能 hasChildren?: boolean isCollapsed?: boolean onToggleCollapse?: () => void } // 格式化日期显示(只显示年份) function formatYear(dateString?: string): string { if (!dateString) return '' const year = dateString.split('-')[0] return year } // 计算年龄 function calculateAge(birthDate?: string, deathDate?: string): number | null { if (!birthDate) return null const birth = new Date(birthDate) const end = deathDate ? new Date(deathDate) : new Date() let age = end.getFullYear() - birth.getFullYear() const monthDiff = end.getMonth() - birth.getMonth() if (monthDiff < 0 || (monthDiff === 0 && end.getDate() < birth.getDate())) { age-- } return age } // 单个人员卡片组件 function PersonCard({ member, isRoot, isSpouse, onSelect, isHighlighted, relationMode, isSelected, canEdit = true, onAddRelation, hasChildren = false, isCollapsed = false, onToggleCollapse }: { member: FamilyMember isRoot?: boolean isSpouse?: boolean onSelect?: (member: FamilyMember) => void isHighlighted?: boolean relationMode?: boolean isSelected?: boolean canEdit?: boolean onAddRelation?: (type: "child" | "spouse") => void hasChildren?: boolean isCollapsed?: boolean onToggleCollapse?: () => void }) { const { avatarUrl: avatarBlobUrl } = useAvatarCache(member?.avatarUrl) const router = useRouter() const { currentTree } = useFamily() // 代数重新计算警告对话框状态 const [showWarningDialog, setShowWarningDialog] = useState(false) const [pendingRelationType, setPendingRelationType] = useState<'father' | 'mother' | null>(null) const birthYear = formatYear(member.birthDate) const deathYear = formatYear(member.deathDate) const age = calculateAge(member.birthDate, member.deathDate) const lifeSpan = birthYear ? deathYear ? `${birthYear}-${deathYear}` : `${birthYear}-` : '' const ageText = age !== null ? member.deathDate ? `享年${age}岁` : `${age}岁` : '' // 判断可以添加的关系 const canAddFather = !member.fatherId const canAddMother = !member.motherId const canAddSpouse = true // 配偶总是可以添加 const canAddChild = true // 子女总是可以添加 // 构建新增成员的 URL const buildAddUrl = (type: 'father' | 'mother' | 'spouse' | 'child') => { const params = new URLSearchParams() if (currentTree?.id) { params.set('treeId', currentTree.id) } switch (type) { case 'father': params.set('childId', member.id) params.set('gender', 'MALE') // 如果是第1代成员,新父母的世代也设为1(后端会重新计算所有成员世代) params.set('generation', member.generation === 1 ? '1' : (member.generation - 1).toString()) // 如果已有母亲,设为配偶 if (member.motherId) { params.set('spouseId', member.motherId) } break case 'mother': params.set('childId', member.id) params.set('gender', 'FEMALE') // 如果是第1代成员,新父母的世代也设为1(后端会重新计算所有成员世代) params.set('generation', member.generation === 1 ? '1' : (member.generation - 1).toString()) // 如果已有父亲,设为配偶 if (member.fatherId) { params.set('spouseId', member.fatherId) } break case 'spouse': params.set('spouseId', member.id) params.set('generation', member.generation.toString()) // 配偶性别与当前成员相反 params.set('gender', member.gender === 'MALE' ? 'FEMALE' : 'MALE') // 如果当前成员有子女,将他们也作为新配偶的子女 if (member.childrenIds && member.childrenIds.length > 0) { // 传递所有子女ID member.childrenIds.forEach(id => { params.append('childrenIds', id) }) } break case 'child': // 根据当前成员性别设置父/母 if (member.gender === 'MALE') { params.set('fatherId', member.id) // 如果父亲有配偶,设置配偶为母亲 if (member.spouseIds && member.spouseIds.length > 0) { params.set('motherId', member.spouseIds[0]) } } else { params.set('motherId', member.id) // 如果母亲有配偶,设置配偶为父亲 if (member.spouseIds && member.spouseIds.length > 0) { params.set('fatherId', member.spouseIds[0]) } } params.set('generation', (member.generation + 1).toString()) break } return `/members/new?${params.toString()}` } const handleAddMember = (type: 'father' | 'mother' | 'spouse' | 'child') => { // 检查是否需要显示代数重新计算警告 if ((type === 'father' || type === 'mother') && shouldShowGenerationWarning(member, type)) { setPendingRelationType(type) setShowWarningDialog(true) return } const url = buildAddUrl(type) // 使用 setTimeout 延迟跳转,确保菜单关闭后再执行 setTimeout(() => { window.location.href = url }, 100) } // 确认添加父母后的处理 const handleConfirmAddParent = () => { if (pendingRelationType) { const url = buildAddUrl(pendingRelationType) setTimeout(() => { window.location.href = url }, 100) setPendingRelationType(null) } } // 是否显示右键菜单(有可添加的选项时才显示) const hasAddOptions = canEdit && (canAddFather || canAddMother || canAddSpouse || canAddChild) // 庄严正式的中国风配色 const isDead = !!member.deathDate const isMale = member.gender === 'MALE' // 使用 isRoot prop 而不是 isFounder 字段 const isFounder = isRoot === true const cardContent = (