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 })
    }

    // 1. 查询拥有的家族树
    const ownedTrees = await prisma.familyTree.findMany({
      where: { ownerId: session.user.id },
      include: {
        owner: {
          select: { id: true, name: true, email: true }
        },
        _count: {
          select: { members: true, collaborators: true }
        }
      },
      orderBy: { updatedAt: 'desc' }
    })

    // 2. 查询协作的家族树
    const collaboratedRelations = await prisma.treeCollaborator.findMany({
      where: { userId: session.user.id },
      include: {
        tree: {
          include: {
            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"
    }))

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

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

    console.log('🌳 GET /api/trees 返回数据:', {
      userId: session.user.id,
      ownedCount: formattedOwnedTrees.length,
      collaboratedCount: formattedCollaboratedTrees.length,
      totalCount: trees.length,
      trees: trees.map(t => ({ id: t.id, name: t.name, role: t.currentUserRole }))
    })

    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 }
    )
  }
}