90 lines
2.3 KiB
TypeScript
90 lines
2.3 KiB
TypeScript
import { getPayload } from "payload";
|
|
import configPromise from "./payload.config";
|
|
import fs from "fs";
|
|
import path from "path";
|
|
import { fileURLToPath } from "url";
|
|
|
|
const __filename = fileURLToPath(import.meta.url);
|
|
const __dirname = path.dirname(__filename);
|
|
|
|
async function run() {
|
|
try {
|
|
let payload;
|
|
let retries = 5;
|
|
while (retries > 0) {
|
|
try {
|
|
console.log(
|
|
`Connecting to database (URI: ${process.env.DATABASE_URI || "default"})...`,
|
|
);
|
|
payload = await getPayload({ config: configPromise });
|
|
break;
|
|
} catch (e: any) {
|
|
if (
|
|
e.code === "ECONNREFUSED" ||
|
|
e.code === "ENOTFOUND" ||
|
|
e.message?.includes("ECONNREFUSED") ||
|
|
e.message?.includes("ENOTFOUND") ||
|
|
e.message?.includes("cannot connect to Postgres")
|
|
) {
|
|
console.log(
|
|
`Database not ready (${e.code || "UNKNOWN"}), retrying in 3 seconds... (${retries} retries left)`,
|
|
);
|
|
retries--;
|
|
await new Promise((res) => setTimeout(res, 3000));
|
|
} else {
|
|
console.error("Fatal connection error:", e);
|
|
throw e;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!payload) {
|
|
throw new Error(
|
|
"Failed to connect to the database after multiple retries.",
|
|
);
|
|
}
|
|
|
|
const existing = await payload.find({
|
|
collection: "context-files",
|
|
limit: 0,
|
|
});
|
|
|
|
if (existing.totalDocs > 0) {
|
|
console.log("Context collection already populated. Skipping seed.");
|
|
process.exit(0);
|
|
}
|
|
|
|
const seedDir = path.resolve(
|
|
__dirname,
|
|
"src/payload/collections/ContextFiles/seed",
|
|
);
|
|
if (!fs.existsSync(seedDir)) {
|
|
console.log(`Seed directory not found at ${seedDir}`);
|
|
process.exit(0);
|
|
}
|
|
|
|
const files = fs.readdirSync(seedDir).filter((f) => f.endsWith(".md"));
|
|
let count = 0;
|
|
|
|
for (const file of files) {
|
|
const content = fs.readFileSync(path.join(seedDir, file), "utf8");
|
|
await payload.create({
|
|
collection: "context-files",
|
|
data: {
|
|
filename: file,
|
|
content: content,
|
|
},
|
|
});
|
|
count++;
|
|
}
|
|
|
|
console.log(`Seeded ${count} context files.`);
|
|
process.exit(0);
|
|
} catch (e) {
|
|
console.error("Seeding failed:", e);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
run();
|