eCourt-note/scripts/seed-samples.mjs

41 lines
1.9 KiB
JavaScript
Raw Normal View History

// Sync ตาราง files กับไฟล์จริงใน public/samples/ (idempotent — รันซ้ำได้ทุกครั้ง)
// - INSERT ไฟล์ .pdf ที่ยังไม่มี row (pdf_url = /samples/<ชื่อไฟล์>)
// - DELETE row ที่ชี้ /samples/* แต่ไฟล์หายจาก disk (CASCADE ลบ annotation/bookmark ตาม)
// รันตอน container start ต่อจาก migrate.mjs และรันมือบนเครื่อง dev ได้ผ่าน npm run seed-samples
import fs from "node:fs";
import path from "node:path";
import postgres from "postgres";
const SAMPLES_DIR = path.join(process.cwd(), "public", "samples");
const dbUrl = process.env.DATABASE_URL;
if (!dbUrl) throw new Error("ต้องตั้ง DATABASE_URL ก่อนรัน seed-samples");
const sql = postgres(dbUrl, { prepare: false, max: 1 });
try {
const onDisk = fs
.readdirSync(SAMPLES_DIR)
.filter((f) => f.toLowerCase().endsWith(".pdf"))
.map((f) => ({ title: f, pdfUrl: `/samples/${f}` }));
const diskUrls = new Set(onDisk.map((f) => f.pdfUrl));
const rows = await sql`select id, title, pdf_url from files where pdf_url like '/samples/%'`;
const dbUrls = new Set(rows.map((r) => r.pdf_url));
for (const file of onDisk) {
if (dbUrls.has(file.pdfUrl)) continue;
console.log(`[seed] เพิ่ม ${file.pdfUrl}`);
await sql`insert into files (title, pdf_url) values (${file.title}, ${file.pdfUrl})`;
}
for (const row of rows) {
if (diskUrls.has(row.pdf_url)) continue;
console.log(`[seed] ลบ ${row.pdf_url} (ไฟล์หายจาก image — annotation ที่ผูกอยู่ CASCADE ตามไปด้วย)`);
await sql`delete from files where id = ${row.id}`;
}
console.log(`[seed] sync เสร็จ: ${onDisk.length} ไฟล์ใน samples`);
} finally {
await sql.end();
}