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";
import { fileKindFromUrl, imageContentTypeFromUrl } from "@/utils/file-kind";

// ให้ 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;
  }

  // N3: ไฟล์รูปภาพไม่มี ToUnicode ให้ซ่อม — คืน bytes ดิบ + content-type ถูกต้อง
  // (viewer โหลดรูปจาก static pdfUrl ตรง ๆ อยู่แล้ว — branch นี้กัน stray call
  // หลุดเข้า fixPdfActualText ที่จะ throw กับ bytes ที่ไม่ใช่ PDF)
  if (fileKindFromUrl(file.pdfUrl) === "image") {
    return new Response(new Uint8Array(srcBytes), {
      headers: {
        "Content-Type": imageContentTypeFromUrl(file.pdfUrl),
        "Cache-Control": "no-store",
      },
    });
  }

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