0.3.0.5

2025-11-30 13:09:52 +08:00
parent 0281885f73
commit 89f1e061d4
146 changed files with 692 additions and 356 deletions
@@ -3,8 +3,18 @@
 import type { FamilyMember } from "@/types/family"
 import { cn } from "@/lib/utils"
 import { useAvatarCache } from "@/hooks/use-avatar-cache"
-import { CircleUser, CircleUserRound, Heart } from "lucide-react"
+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"

 interface FamilyNodeProps {
  member: FamilyMember
@@ -15,6 +25,7 @@ interface FamilyNodeProps {
  isHighlighted?: boolean
  relationMode?: boolean
  isSelected?: boolean
+  canEdit?: boolean
 }

 // 格式化日期显示（只显示年份）
@@ -42,8 +53,28 @@ function calculateAge(birthDate?: string, deathDate?: string): number | null {
 }

 // 单个人员卡片组件
-function PersonCard({ member, isRoot, isSpouse, onSelect, isHighlighted, relationMode, isSelected }: { member: FamilyMember; isRoot?: boolean; isSpouse?: boolean; onSelect?: (member: FamilyMember) => void; isHighlighted?: boolean; relationMode?: boolean; isSelected?: boolean }) {
+function PersonCard({ 
+  member, 
+  isRoot, 
+  isSpouse, 
+  onSelect, 
+  isHighlighted, 
+  relationMode, 
+  isSelected,
+  canEdit = true 
+}: { 
+  member: FamilyMember
+  isRoot?: boolean
+  isSpouse?: boolean
+  onSelect?: (member: FamilyMember) => void
+  isHighlighted?: boolean
+  relationMode?: boolean
+  isSelected?: boolean
+  canEdit?: boolean
+}) {
  const { avatarUrl: avatarBlobUrl } = useAvatarCache(member?.avatarUrl)
+  const router = useRouter()
+  const { currentTree } = useFamily()
  
  const birthYear = formatYear(member.birthDate)
  const deathYear = formatYear(member.deathDate)
@@ -60,8 +91,92 @@ function PersonCard({ member, isRoot, isSpouse, onSelect, isHighlighted, relatio
      ? `享年${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')
+        params.set('generation', (member.generation - 1).toString())
+        // 如果已有母亲，设为配偶
+        if (member.motherId) {
+          params.set('spouseId', member.motherId)
+        }
+        break
+      case 'mother':
+        params.set('childId', member.id)
+        params.set('gender', 'FEMALE')
+        params.set('generation', (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') => {
+    const url = buildAddUrl(type)
+    console.log('handleAddMember called:', type, url)
+    // 使用 setTimeout 延迟跳转，确保菜单关闭后再执行
+    setTimeout(() => {
+      console.log('Navigating to:', url)
+      window.location.href = url
+    }, 100)
+  }
+
+  // 是否显示右键菜单（有可添加的选项时才显示）
+  const hasAddOptions = canEdit && (canAddFather || canAddMother || canAddSpouse || canAddChild)
  
-  return (
+  // 调试日志
+  console.log('PersonCard render:', member.fullName, { canEdit, hasAddOptions, relationMode })
+
+  const cardContent = (
    <div
      className={cn(
        "family-node-card relative flex flex-col items-center gap-2 p-3 rounded-lg transition-all cursor-pointer hover:scale-105 shadow-sm w-32",
@@ -126,24 +241,100 @@ function PersonCard({ member, isRoot, isSpouse, onSelect, isHighlighted, relatio
      </div>
    </div>
  )
+
+  // 如果没有可添加的选项或在关系模式下，直接返回卡片
+  if (!hasAddOptions || relationMode) {
+    return cardContent
+  }
+
+  // 有可添加选项时，包裹右键菜单
+  return (
+    <ContextMenu>
+      <ContextMenuTrigger>
+        {cardContent}
+      </ContextMenuTrigger>
+      <ContextMenuContent className="w-48">
+        <ContextMenuLabel className="text-xs text-muted-foreground">
+          为 {member.fullName} 添加
+        </ContextMenuLabel>
+        <ContextMenuSeparator />
+        
+        {canAddFather && (
+          <ContextMenuItem 
+            className="gap-2 cursor-pointer"
+            onPointerDown={(e) => {
+              e.preventDefault()
+              handleAddMember('father')
+            }}
+          >
+            <UserPlus className="h-4 w-4 text-blue-500" />
+            添加父亲
+          </ContextMenuItem>
+        )}
+        
+        {canAddMother && (
+          <ContextMenuItem 
+            className="gap-2 cursor-pointer"
+            onPointerDown={(e) => {
+              e.preventDefault()
+              handleAddMember('mother')
+            }}
+          >
+            <UserPlus className="h-4 w-4 text-pink-500" />
+            添加母亲
+          </ContextMenuItem>
+        )}
+        
+        {(canAddFather || canAddMother) && (canAddSpouse || canAddChild) && (
+          <ContextMenuSeparator />
+        )}
+        
+        {canAddSpouse && (
+          <ContextMenuItem 
+            className="gap-2 cursor-pointer"
+            onPointerDown={(e) => {
+              e.preventDefault()
+              handleAddMember('spouse')
+            }}
+          >
+            <Users className="h-4 w-4 text-red-500" />
+            添加配偶
+          </ContextMenuItem>
+        )}
+        
+        {canAddChild && (
+          <ContextMenuItem 
+            className="gap-2 cursor-pointer"
+            onPointerDown={(e) => {
+              e.preventDefault()
+              handleAddMember('child')
+            }}
+          >
+            <Baby className="h-4 w-4 text-green-500" />
+            添加子女
+          </ContextMenuItem>
+        )}
+      </ContextMenuContent>
+    </ContextMenu>
+  )
 }

-export function FamilyNode({ member, spouse, spouses, isRoot, onSelect, isHighlighted, relationMode, isSelected }: FamilyNodeProps) {
+export function FamilyNode({ member, spouse, spouses, isRoot, onSelect, isHighlighted, relationMode, isSelected, canEdit = true }: FamilyNodeProps) {
  // 优先使用 spouses 数组，如果没有则使用单个 spouse
  const spouseList = spouses || (spouse ? [spouse] : [])
  
  if (spouseList.length === 0) {
-    return <PersonCard member={member} isRoot={isRoot} onSelect={onSelect} isHighlighted={isHighlighted} relationMode={relationMode} isSelected={isSelected} />
+    return <PersonCard member={member} isRoot={isRoot} onSelect={onSelect} isHighlighted={isHighlighted} relationMode={relationMode} isSelected={isSelected} canEdit={canEdit} />
  }
  
  // 有配偶时，并排显示所有配偶
  return (
    <div className="flex items-center gap-2">
-      <PersonCard member={member} isRoot={isRoot} onSelect={onSelect} isHighlighted={isHighlighted} relationMode={relationMode} isSelected={isSelected} />
+      <PersonCard member={member} isRoot={isRoot} onSelect={onSelect} isHighlighted={isHighlighted} relationMode={relationMode} isSelected={isSelected} canEdit={canEdit} />
      {spouseList.map((spouseMember, index) => (
        <div key={spouseMember.id} className="flex items-center gap-2">
          <Heart className="h-4 w-4 text-red-400 flex-shrink-0" />
-          <PersonCard member={spouseMember} isSpouse={true} onSelect={onSelect} relationMode={relationMode} />
+          <PersonCard member={spouseMember} isSpouse={true} onSelect={onSelect} relationMode={relationMode} canEdit={canEdit} />
        </div>
      ))}
    </div>
@@ -4,6 +4,7 @@ import { FamilyNode } from "./family-node"
 import { useFamily } from "@/context/family-context"
 import { useEffect, useRef, useState, useCallback, useMemo, memo } from "react"
 import { useRouter } from "next/navigation"
+import { permissions } from "@/lib/permissions"

 interface TreeLayoutProps {
  rootId: string
@@ -26,7 +27,10 @@ export function TreeLayout({
  selectedMembers = [],
  onMemberClick
 }: TreeLayoutProps) {
-  const { getMember, treeData, highlightedMemberId } = useFamily()
+  const { getMember, treeData, highlightedMemberId, currentTree } = useFamily()
+  
+  // 检查用户是否有创建权限
+  const canEdit = permissions.canCreate(currentTree?.currentUserRole)
  const root = getMember(rootId)
  const router = useRouter()
  const containerRef = useRef<HTMLDivElement>(null)
@@ -192,6 +196,7 @@ export function TreeLayout({
            isHighlighted={member.id === highlightedMemberId}
            relationMode={relationMode}
            isSelected={selectedMembers.includes(member.id)}
+            canEdit={canEdit}
          />
        </div>