Merge pull request #524 from cyber-excel10/fix/issue-230-seed-script

fix: add seedDeterministic script to populate dev database via CLI

Author: ritik4ever

README.md

@@ -414,10 +414,42 @@ Open:
 - Frontend: `http://localhost:3000`
 - Backend: `http://localhost:3001`
 
-Build:
+### Seed the dev database
 
+Wipes and repopulates the local SQLite database with deterministic campaigns and pledges — useful for a fresh, reproducible local state.
+
+```bash
+cd backend
+npm run seed
+
+# Seed a custom number of campaigns (default: 3)
+npm run seed -- --count 10
+```
+
+Seeded campaign IDs are printed to stdout. The first 3 campaigns always match a fixed
+set (open, funded, claimed status). Any additional campaigns beyond that cycle
+deterministically through open/funded/claimed statuses so the seed is reproducible run to run.
+
+Build:eed the dev database
++
++Wipes and repopulates the local SQLite database with deterministic campaigns and pledges — useful for a fresh, reproducible local state.
++
++```bash
++cd backend
++npm run seed
++
++# Seed a custom number of campaigns (default: 3)
++npm run seed -- --count 10
++```
++
++Seeded campaign IDs are printed to stdout. The first 3 campaigns always match a fixed
++set (open, funded, claimed status). Any additional campaigns beyond that cycle
++deterministically through open/funded/claimed statuses so the seed is reproducible run to run.
++
+ Build:
+ 
 ```bash
-npm run build
+ npm run build
 ```
 
 ### Local development with Docker (hot-reload)

backend/src/services/seedDeterministic.ts

@@ -15,7 +15,15 @@ type SeedCampaign = {
   claimedAt: number | null;
 };
 
-const SEED_CAMPAIGNS: SeedCampaign[] = [
+type SeedPledge = {
+  campaignId: string;
+  contributor: string;
+  amount: number;
+  assetCode: string;
+  createdAt: number;
+};
+
+const BASE_CAMPAIGNS: SeedCampaign[] = [
   {
     id: '1',
     creator: `G${'A'.repeat(55)}`,
@@ -54,10 +62,107 @@ const SEED_CAMPAIGNS: SeedCampaign[] = [
   },
 ];
 
-export function seedDeterministicState(): void {
+const BASE_PLEDGES: SeedPledge[] = [
+  { campaignId: '1', contributor: `G${'D'.repeat(55)}`, amount: 100, assetCode: 'USDC', createdAt: FIXED_NOW - 250 },
+  { campaignId: '2', contributor: `G${'E'.repeat(55)}`, amount: 250, assetCode: 'XLM', createdAt: FIXED_NOW - 150 },
+];
+
+// Deterministic status/asset rotation used to extend past the 3 base campaigns.
+// Each generated campaign gets one matching pledge, same as the base set.
+const EXTRA_STATUS_CYCLE: Array<'open' | 'funded' | 'claimed'> = ['open', 'funded', 'claimed'];
+const EXTRA_ASSET_CYCLE = ['USDC', 'XLM'];
+
+function letterFor(index: number): string {
+  // A-Z, then AA, BB... style wrap for very large counts (index is 0-based
+  // continuing on from the 3 base campaigns, which already used A/B/C).
+  const cycleIndex = index % 26;
+  const letter = String.fromCharCode(65 + cycleIndex);
+  return letter;
+}
+
+function buildExtraCampaign(index: number): { campaign: SeedCampaign; pledge: SeedPledge } {
+  // index is 0-based for campaigns beyond the 3 base ones, so real id = index + 4
+  const id = String(index + 4);
+  const letter = letterFor(index + 3); // continue the A/B/C sequence
+  const creator = `G${letter.repeat(55)}`;
+  const status = EXTRA_STATUS_CYCLE[index % EXTRA_STATUS_CYCLE.length];
+  const assetCode = EXTRA_ASSET_CYCLE[index % EXTRA_ASSET_CYCLE.length];
+
+  const targetAmount = 200 + index * 50;
+  let pledgedAmount: number;
+  let deadline: number;
+  let claimedAt: number | null;
+
+  if (status === 'open') {
+    pledgedAmount = Math.floor(targetAmount * 0.4);
+    deadline = FIXED_NOW + 86_400 + index * 1_000;
+    claimedAt = null;
+  } else if (status === 'funded') {
+    pledgedAmount = targetAmount;
+    deadline = FIXED_NOW + 43_200 + index * 1_000;
+    claimedAt = null;
+  } else {
+    pledgedAmount = targetAmount;
+    deadline = FIXED_NOW - 600 - index * 1_000;
+    claimedAt = FIXED_NOW - 100 - index * 1_000;
+  }
+
+  const campaign: SeedCampaign = {
+    id,
+    creator,
+    title: `Deterministic campaign ${id} (${status})`,
+    description: `Generated deterministic campaign seed #${id} for ${status} status checks.`,
+    assetCode,
+    targetAmount,
+    pledgedAmount,
+    deadline,
+    createdAt: FIXED_NOW - 300 - index * 100,
+    claimedAt,
+  };
+
+  const pledgeLetter = letterFor(index + 3 + 100); // distinct pool from campaign creators
+  const pledge: SeedPledge = {
+    campaignId: id,
+    contributor: `G${pledgeLetter.repeat(55)}`,
+    amount: pledgedAmount > 0 ? pledgedAmount : 1,
+    assetCode,
+    createdAt: FIXED_NOW - 250 - index * 100,
+  };
+
+  return { campaign, pledge };
+}
+
+function buildSeedSet(count: number): { campaigns: SeedCampaign[]; pledges: SeedPledge[] } {
+  if (count <= 0) {
+    throw new Error('count must be a positive integer');
+  }
+
+  if (count <= BASE_CAMPAIGNS.length) {
+    const campaigns = BASE_CAMPAIGNS.slice(0, count);
+    const ids = new Set(campaigns.map((c) => c.id));
+    const pledges = BASE_PLEDGES.filter((p) => ids.has(p.campaignId));
+    return { campaigns, pledges };
+  }
+
+  const extraCount = count - BASE_CAMPAIGNS.length;
+  const extras = Array.from({ length: extraCount }, (_, i) => buildExtraCampaign(i));
+
+  return {
+    campaigns: [...BASE_CAMPAIGNS, ...extras.map((e) => e.campaign)],
+    pledges: [...BASE_PLEDGES, ...extras.map((e) => e.pledge)],
+  };
+}
+
+/**
+ * Wipes and repopulates the dev database with `count` deterministic campaigns
+ * (default 3). Returns the seeded campaign IDs in insertion order.
+ */
+export function seedDeterministicState(count: number = BASE_CAMPAIGNS.length): string[] {
   initDb();
   const db = getDb();
 
+  const { campaigns, pledges } = buildSeedSet(count);
+
   db.prepare(`DELETE FROM campaign_events`).run();
   db.prepare(`DELETE FROM pledges`).run();
   db.prepare(`DELETE FROM campaigns`).run();
@@ -68,7 +173,7 @@ export function seedDeterministicState(): void {
     ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, NULL)`,
   );
 
-  for (const campaign of SEED_CAMPAIGNS) {
+  for (const campaign of campaigns) {
     insertCampaign.run(
       campaign.id,
       campaign.creator,
@@ -83,19 +188,32 @@ export function seedDeterministicState(): void {
     );
   }
 
-  db.prepare(
+  const insertPledge = db.prepare(
     `INSERT INTO pledges (campaign_id, contributor, amount, asset_code, created_at, refunded_at, transaction_hash)
      VALUES (?, ?, ?, ?, ?, NULL, NULL)`,
-  ).run('1', `G${'D'.repeat(55)}`, 100, 'USDC', FIXED_NOW - 250);
+  );
 
-  db.prepare(
-    `INSERT INTO pledges (campaign_id, contributor, amount, asset_code, created_at, refunded_at, transaction_hash)
-     VALUES (?, ?, ?, ?, ?, NULL, NULL)`,
-  ).run('2', `G${'E'.repeat(55)}`, 250, 'XLM', FIXED_NOW - 150);
-}
+  for (const pledge of pledges) {
+    insertPledge.run(pledge.campaignId, pledge.contributor, pledge.amount, pledge.assetCode, pledge.createdAt);
+  }
 
-if (require.main === module) {
-  seedDeterministicState();
-  // eslint-disable-next-line no-console
-  console.log('Deterministic database seed complete.');
+  return campaigns.map((c) => c.id);
 }
+
+export function parseCountArg(argv: string[]): number {
+  const flagIndex = argv.findIndex((arg) => arg === '--count' || arg.startsWith('--count='));
+  if (flagIndex === -1) {
+    return BASE_CAMPAIGNS.length;
+  }
+
+  const raw = argv[flagIndex].includes('=')
+    ? argv[flagIndex].split('=')[1]
+    : argv[flagIndex + 1];
+
+  const parsed = Number(raw);
+  if (!Number.isInteger(parsed) || parsed <= 0) {
+    throw new Error(`Invalid --count value: "${raw}". Must be a positive integer.`);
+  }
+
+  return parsed;
+}