eCourt-note/app/api/pdf/route.ts

import { eq } from "drizzle-orm";

import { db } from "@/db";
import { files } from "@/db/schema";
import { fixPdfActualText } from "@/lib/pdf-actual-text";
import { PdfPathError, readPublicPdf } from "@/lib/server-pdf";

// ให้ viewer โหลด PDF ผ่านทางนี้แทนไฟล์ดิบใน public/ — ซ่อม ToUnicode จาก ActualText
// ของ PDF ไทยที่ cmap ของฟอนต์เพี้ยน (สกัดข้อความได้ control char ขยะแทนสระ/วรรณยุกต์)
// **แก้บนสำเนาในหน่วยความจำเท่านั้น** ไฟล์ต้นฉบับไม่ถูกเขียน (invariant #1)
// ไฟล์ปกติไม่มี ActualText = ได้ bytes เดิมกลับมาไม่มี cost การแปลง

const UUID_RE =
  /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;

export async function GET(req: Request) {
  const url = new URL(req.url);
  const fileId = url.searchParams.get("fileId") ?? "";
  if (!UUID_RE.test(fileId)) {
    return new Response("fileId ไม่ถูกต้อง", { status: 400 });
  }

  const [file] = await db.select().from(files).where(eq(files.id, fileId));
  if (!file) return new Response("ไม่พบไฟล์นี้", { status: 404 });

  // Guard path traversal: pdfUrl จาก DB ต้อง resolve แล้วอยู่ใน public/ เท่านั้น (§5)
  let srcBytes: Buffer;
  try {
    srcBytes = readPublicPdf(file.pdfUrl);
  } catch (err) {
    if (err instanceof PdfPathError) return new Response(err.message, { status: 400 });
    throw err;
  }

  const fixed = await fixPdfActualText(srcBytes);
  return new Response(new Uint8Array(fixed), {
    headers: {
      "Content-Type": "application/pdf",
      // ขนาดไฟล์ตัวอย่างเล็ก ซ่อมใหม่ต่อ request (สักหลักสิบ ms) — ไม่ต้องแคช
      "Cache-Control": "no-store",
    },
  });
}