douyin-archive/app/api/search/route.ts

import { json } from "@/lib/json";
import { getFileUrl } from "@/lib/minio";
import { prisma } from "@/lib/prisma"; // 你的 Prisma 客户端实例
import { NextResponse } from "next/server";

export async function GET(req: Request) {
  const { searchParams } = new URL(req.url);
  const q = (searchParams.get("q") || "").trim();
  const page = Math.max(1, Number(searchParams.get("page") || 1));
  const limit = Math.min(
    50,
    Math.max(1, Number(searchParams.get("limit") || 20)),
  );
  const offset = (page - 1) * limit;

  if (!q) {
    return NextResponse.json({ results: [], total: 0 });
  }

  try {
    // 核心查询：Postgres 中文全文检索（视频转写 + 视频desc + 图文desc）+ 相关度排序 + 摘要高亮
    // 说明：
    // - 合并搜索来源：VideoTranscript.transcript_zh_tsv、Video.desc 与 ImagePost.desc
    // - 使用 UNION ALL 合并视频和图文的搜索结果
    // - rank 取两者匹配得分的最大值，snippet 优先展示转写匹配，否则展示 desc 匹配
    const rows = await prisma.$queryRaw<
      {
        id: string;
        awemeId: string;
        type: "video" | "image";
        rank: number;
        snippet: string;
      }[]
    >`
      WITH tsq AS (
        SELECT websearch_to_tsquery('zhcfg', ${q}) AS query
      )
      -- 视频搜索结果
      SELECT
        COALESCE(vt.id, 'desc:' || v.aweme_id) AS id,
        v.aweme_id AS "awemeId",
        'video'::text AS type,
        GREATEST(
          COALESCE(ts_rank(vt.transcript_zh_tsv, tsq.query), 0),
          COALESCE(ts_rank(to_tsvector('zhcfg', v.desc), tsq.query), 0)
        ) AS rank,
        CASE
          WHEN vt.transcript_zh_tsv @@ tsq.query THEN
            ts_headline(
              'zhcfg',
              array_to_string(vt.transcript, ' '),
              tsq.query,
              'StartSel=<mark>,StopSel=</mark>,MaxFragments=2,MinWords=2,MaxWords=20'
            )
          ELSE
            ts_headline(
              'zhcfg',
              v.desc,
              tsq.query,
              'StartSel=<mark>,StopSel=</mark>,MaxFragments=2,MinWords=2,MaxWords=20'
            )
        END AS snippet
      FROM "Video" v
      LEFT JOIN "VideoTranscript" vt ON vt."videoId" = v.aweme_id,
           tsq
      WHERE
        (vt.transcript_zh_tsv @@ tsq.query OR to_tsvector('zhcfg', v.desc) @@ tsq.query)
      
      UNION ALL
      
      -- 图文搜索结果
      SELECT
        'img_desc:' || ip.aweme_id AS id,
        ip.aweme_id AS "awemeId",
        'image'::text AS type,
        ts_rank(to_tsvector('zhcfg', ip.desc), tsq.query) AS rank,
        ts_headline(
          'zhcfg',
          ip.desc,
          tsq.query,
          'StartSel=<mark>,StopSel=</mark>,MaxFragments=2,MinWords=2,MaxWords=20'
        ) AS snippet
      FROM "ImagePost" ip, tsq
      WHERE to_tsvector('zhcfg', ip.desc) @@ tsq.query
      
      ORDER BY rank DESC
      OFFSET ${offset}
      LIMIT ${limit};
    `;

    // 查询总数（参数化，避免注入）
    const totalRows = await prisma.$queryRaw<
      {
        count: number;
      }[]
    >`
      WITH tsq AS (
        SELECT websearch_to_tsquery('zhcfg', ${q}) AS query
      )
      SELECT (
        (SELECT COUNT(*)::int FROM "Video" v
         LEFT JOIN "VideoTranscript" vt ON vt."videoId" = v.aweme_id
         WHERE (vt.transcript_zh_tsv @@ tsq.query OR to_tsvector('zhcfg', v.desc) @@ tsq.query))
        +
        (SELECT COUNT(*)::int FROM "ImagePost" ip
         WHERE to_tsvector('zhcfg', ip.desc) @@ tsq.query)
      ) AS count
      FROM tsq;
    `;

    // 分离视频和图文ID
    const videoIds = rows
      .filter((r) => r.type === "video")
      .map((r) => r.awemeId);
    const imagePostIds = rows
      .filter((r) => r.type === "image")
      .map((r) => r.awemeId);

    // 批量查询视频元信息
    const videos =
      videoIds.length > 0
        ? (
            await prisma.video.findMany({
              where: { aweme_id: { in: videoIds } },
              select: {
                aweme_id: true,
                desc: true,
                cover_url: true,
                video_url: true,
                duration_ms: true,
                author: true,
              },
            })
          ).map((v) => ({
            ...v,
            cover_url: getFileUrl(v.cover_url || ""),
            author: {
              ...v.author,
              avatar_url: getFileUrl(v.author.avatar_url || ""),
            },
            video_url: getFileUrl(v.video_url || ""),
          }))
        : [];

    // 批量查询图文元信息
    const imagePosts =
      imagePostIds.length > 0
        ? (
            await prisma.imagePost.findMany({
              where: { aweme_id: { in: imagePostIds } },
              select: {
                aweme_id: true,
                desc: true,
                author: true,
                images: {
                  orderBy: { order: "asc" },
                  take: 1,
                  select: {
                    url: true,
                    width: true,
                    height: true,
                  },
                },
              },
            })
          ).map((ip) => ({
            ...ip,
            author: {
              ...ip.author,
              avatar_url: getFileUrl(ip.author.avatar_url || ""),
            },
            cover_url: ip.images[0] ? getFileUrl(ip.images[0].url) : null,
          }))
        : [];

    return json({
      results: rows.map((r) => ({
        ...r,
        video:
          r.type === "video"
            ? videos.find((v) => v.aweme_id === r.awemeId)
            : undefined,
        imagePost:
          r.type === "image"
            ? imagePosts.find((ip) => ip.aweme_id === r.awemeId)
            : undefined,
      })),
      total: totalRows?.[0]?.count ?? 0,
      page,
      limit,
    });
  } catch (err) {
    console.error("Search error:", err);
    return NextResponse.json({ error: "Search failed" }, { status: 500 });
  }
}