import { NextRequest, NextResponse } from "next/server"
import { getServerSession } from "next-auth"
import { authOptions } from "@/lib/auth"
import { prisma } from "@/lib/prisma"
import { z } from "zod"

const createTreeSchema = z.object({
  name: z.string().min(1, "家族树名称不能为空"),
  description: z.string().optional(),
})

// 获取用户的所有家族树
export async function GET(req: NextRequest) {
  try {
    const session = await getServerSession(authOptions)
    
    if (!session?.user?.id) {
      return NextResponse.json({ error: "未授权" }, { status: 401 })
    }

    // 使用并行查询优化性能
    const [ownedTrees, collaboratedRelations] = await Promise.all([
      // 1. 查询拥有的家族树 - 只选择必要字段
      prisma.familyTree.findMany({
        where: { ownerId: session.user.id },
        select: {
          id: true,
          name: true,
          description: true,
          rootMemberId: true,
          ownerId: true,
          createdAt: true,
          updatedAt: true,
          owner: {
            select: { id: true, name: true, email: true }
          },
          _count: {
            select: { members: true, collaborators: true }
          }
        },
        orderBy: { updatedAt: 'desc' }
      }),
      // 2. 查询协作的家族树
      prisma.treeCollaborator.findMany({
        where: { userId: session.user.id },
        select: {
          role: true,
          tree: {
            select: {
              id: true,
              name: true,
              description: true,
              rootMemberId: true,
              ownerId: true,
              createdAt: true,
              updatedAt: true,
              owner: {
                select: { id: true, name: true, email: true }
              },
              _count: {
                select: { members: true, collaborators: true }
              }
            }
          }
        },
        orderBy: { invitedAt: 'desc' }
      })
    ])

    // 3. 格式化数据
    const formattedOwnedTrees = ownedTrees.map(tree => ({
      ...tree,
      currentUserRole: "OWNER" as const
    }))

    const formattedCollaboratedTrees = collaboratedRelations.map(relation => ({
      ...relation.tree,
      currentUserRole: relation.role
    }))

    // 4. 合并列表
    const trees = [...formattedOwnedTrees, ...formattedCollaboratedTrees]

    return NextResponse.json({ trees })
  } catch (error) {
    console.error("获取家族树失败:", error)
    return NextResponse.json(
      { error: "获取家族树失败" },
      { status: 500 }
    )
  }
}

// 创建新家族树
export async function POST(req: NextRequest) {
  try {
    const session = await getServerSession(authOptions)
    
    if (!session?.user?.id) {
      return NextResponse.json({ error: "未授权" }, { status: 401 })
    }

    const body = await req.json()
    const { name, description } = createTreeSchema.parse(body)

    const tree = await prisma.familyTree.create({
      data: {
        name,
        description,
        ownerId: session.user.id,
      },
      include: {
        owner: {
          select: {
            id: true,
            name: true,
            email: true,
          }
        }
      }
    })

    return NextResponse.json({ tree }, { status: 201 })
  } catch (error) {
    if (error instanceof z.ZodError) {
      return NextResponse.json(
        { error: error.errors[0].message },
        { status: 400 }
      )
    }
    
    console.error("创建家族树失败:", error)
    return NextResponse.json(
      { error: "创建家族树失败" },
      { status: 500 }
    )
  }
}