import { NextRequest, NextResponse } from 'next/server';
import { getServerSession } from 'next-auth';
import { authOptions } from '@/lib/auth';
import { prisma } from '@/lib/prisma';

// POST - 收藏/取消收藏
export async function POST(request: NextRequest) {
  try {
    const session = await getServerSession(authOptions);

    if (!session?.user || !(session.user as any).id) {
      return NextResponse.json({ error: '未登录' }, { status: 401 });
    }

    const { materialId } = await request.json();

    if (!materialId) {
      return NextResponse.json({ error: '缺少素材ID' }, { status: 400 });
    }

    const userId = (session.user as any).id;

    // 检查素材是否存在
    const material = await prisma.material.findUnique({
      where: { id: materialId },
    });

    if (!material) {
      return NextResponse.json({ error: '素材不存在' }, { status: 404 });
    }

    // 检查是否已收藏
    const existingFavorite = await prisma.materialFavorite.findFirst({
      where: {
        materialId,
        userId,
      },
    });

    if (existingFavorite) {
      // 取消收藏
      await prisma.materialFavorite.delete({
        where: {
          id: existingFavorite.id,
        },
      });

      return NextResponse.json({
        message: '已取消收藏',
        isFavorited: false,
      });
    } else {
      // 收藏
      await prisma.materialFavorite.create({
        data: {
          materialId,
          userId,
        },
      });

      return NextResponse.json({
        message: '已收藏',
        isFavorited: true,
      });
    }
  } catch (error: any) {
    // 兼容旧版本数据库（没有material_favorites表）
    if (error?.meta?.table?.includes('material_favorites')) {
      return NextResponse.json(
        { error: '收藏功能需要先运行数据库迁移' },
        { status: 400 }
      );
    }

    console.error('[Materials] FAVORITE error:', error);
    return NextResponse.json({ error: '操作失败' }, { status: 500 });
  }
}