"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 = (
onSelect?.(member)} > {/* 顶部装饰条 */}
{/* 选中打钩标记 */} {isSelected && (
)} {/* 世代标签 - 现代胶囊风格 */}
第{member.generation}世
{/* 头像 - 圆形带光晕效果 */}
{/* 光晕背景 */}
{avatarBlobUrl ? ( {member.fullName} ) : ( member.gender === 'FEMALE' ? ( ) : ( ) )}
{/* 在线/已故状态指示器 */}
{/* 姓名和信息 */}
{member.courtesyName && (
字 {member.courtesyName}
)} {lifeSpan && (
{lifeSpan}
)} {ageText && (
{ageText}
)}
{/* 折叠/展开按钮 - 更现代的样式 */} {hasChildren && onToggleCollapse && ( )}
) // 如果没有可添加的选项或在关系模式下,直接返回卡片 if (!hasAddOptions || relationMode) { return cardContent } // 有可添加选项时,包裹右键菜单 return ( <> {cardContent} 为 {member.fullName} 添加 {canAddFather && ( { e.preventDefault() handleAddMember('father') }} > 添加父亲 )} {canAddMother && ( { e.preventDefault() handleAddMember('mother') }} > 添加母亲 )} {(canAddFather || canAddMother) && (canAddSpouse || canAddChild) && ( )} {canAddSpouse && ( { e.preventDefault() if (onAddRelation) { onAddRelation('spouse') } else { handleAddMember('spouse') } }} > 添加配偶 )} {canAddChild && ( { e.preventDefault() if (onAddRelation) { onAddRelation('child') } else { handleAddMember('child') } }} > 添加子女 )} {/* 代数重新计算警告对话框 */} {pendingRelationType && ( )} ) } export function FamilyNode({ member, spouse, spouses, isRoot, onSelect, isHighlighted, highlightedMemberId, relationMode, isSelected, selectedMembers = [], canEdit = true, onAddRelation, hasChildren = false, isCollapsed = false, onToggleCollapse }: FamilyNodeProps) { // 优先使用 spouses 数组,如果没有则使用单个 spouse const spouseList = spouses || (spouse ? [spouse] : []) if (spouseList.length === 0) { return ( ) } // 有配偶时,合并到一个卡片中显示(类似D3布局) const isMale = member.gender === 'MALE' const isDead = !!member.deathDate // 检查是否有任何配偶被高亮,如果是则整个卡片高亮 const isAnySpouseHighlighted = spouseList.some(sp => sp.id === highlightedMemberId) const isCardHighlighted = isHighlighted || isAnySpouseHighlighted // 使用 isRoot prop 而不是 isFounder 字段 const isFounder = isRoot === true const avatarUrl = member.avatarUrl // 计算生卒年份 const birthYear = formatYear(member.birthDate) const deathYear = formatYear(member.deathDate) const lifeSpan = birthYear ? `${birthYear}${deathYear ? ` — ${deathYear}` : ' —'}` : '' const age = calculateAge(member.birthDate, member.deathDate) const ageText = age !== null ? (isDead ? `享年${age}岁` : `${age}岁`) : '' return (
{/* 顶部装饰条 */}
{/* 世代标签 - 居中显示 */} {member.generation && (
第{member.generation}世
)} {/* 内容区域 */}
{/* 主成员 */}
onSelect?.(member)} > {/* 选中对钩标记 */} {isSelected && (
)} {/* 头像 */}
{avatarUrl ? ( {member.fullName} ) : isMale ? ( ) : ( )}
{/* 状态指示器 */}
{/* 姓名 */}
{lifeSpan && (
{lifeSpan}
)} {ageText && (
{ageText}
)}
{/* 配偶们 */} {spouseList.map((sp, idx) => { const spIsMale = sp.gender === 'MALE' const spIsDead = !!sp.deathDate const spAvatarUrl = sp.avatarUrl const spBirthYear = formatYear(sp.birthDate) const spDeathYear = formatYear(sp.deathDate) const spLifeSpan = spBirthYear ? `${spBirthYear}${spDeathYear ? `-${spDeathYear}` : '-'}` : '' const spAge = calculateAge(sp.birthDate, sp.deathDate) const spAgeText = spAge !== null ? (spIsDead ? `享年${spAge}岁` : `${spAge}岁`) : '' const spIsSelected = selectedMembers.includes(sp.id) return (
{/* 连接线 */} {idx === 0 && (
)} {/* 配偶卡片 */}
onSelect?.(sp)} > {/* 选中对钩标记 */} {spIsSelected && (
)} {/* 头像 */}
{spAvatarUrl ? ( {sp.fullName} ) : spIsMale ? ( ) : ( )}
{/* 状态指示器 */}
{/* 姓名 */}
{spLifeSpan && (
{spLifeSpan}
)} {spAgeText && (
{spAgeText}
)}
) })}
{/* 折叠/展开按钮 */} {hasChildren && onToggleCollapse && ( )}
) }