@ak--47/dungeon-master 1.0.0 → 1.2.0

package/dungeons/{fintech.js → vertical/fintech.js}

@@ -1,7 +1,7 @@
 import dayjs from "dayjs";
 import utc from "dayjs/plugin/utc.js";
 import "dotenv/config";
-import * as u from "../lib/utils/utils.js";
+import * as u from "../../lib/utils/utils.js";
 
 const SEED = "harness-fintech";
 dayjs.extend(utc);
@@ -9,7 +9,7 @@ const chance = u.initChance(SEED);
 const num_users = 5_000;
 const days = 100;
 
-/** @typedef  {import("../types.d.ts").Dungeon} Config */
+/** @typedef  {import("../../types").Dungeon} Config */
 
 /**
  * ===================================================================
@@ -151,11 +151,24 @@ const config = {
 
 	hasAvatar: true,
 
-	batchSize: 2_500_000,
 	concurrency: 1,
 	writeToDisk: false,
 
-	scdProps: {},
+	scdProps: {
+		account_tier: {
+			values: ["basic", "plus", "premium"],
+			frequency: "month",
+			timing: "fuzzy",
+			max: 6
+		},
+		risk_category: {
+			values: ["low", "medium", "high", "critical"],
+			frequency: "month",
+			timing: "fixed",
+			max: 8,
+			type: "household_id"
+		}
+	},
 
 	funnels: [
 		{
@@ -221,7 +234,7 @@ const config = {
 			weight: 1,
 			isFirstEvent: true,
 			properties: {
-				"account_type": u.pickAWinner(["personal", "business", "personal"]),
+				"account_type": ["personal", "business", "personal"],
 				"signup_channel": ["app", "web", "referral", "branch"],
 			}
 		},
@@ -231,6 +244,7 @@ const config = {
 			properties: {
 				"session_duration_sec": u.weighNumRange(10, 600, 0.3, 60),
 				"pages_viewed": u.weighNumRange(1, 15, 0.5, 3),
+				"month_end_anxiety": [false],
 			}
 		},
 		{
@@ -239,6 +253,7 @@ const config = {
 			properties: {
 				"account_balance": u.weighNumRange(0, 50000, 0.8, 2500),
 				"account_type": ["checking", "savings", "investment"],
+				"month_end_check": [false],
 			}
 		},
 		{
@@ -249,6 +264,8 @@ const config = {
 				"amount": u.weighNumRange(1, 5000, 0.3, 50),
 				"merchant_category": ["grocery", "restaurant", "gas", "retail", "online", "subscription", "utilities"],
 				"payment_method": ["debit", "credit", "contactless", "online"],
+				"payday": [false],
+				"fraud_sequence": [false],
 			}
 		},
 		{
@@ -258,6 +275,7 @@ const config = {
 				"transfer_type": ["internal", "external", "p2p", "wire"],
 				"amount": u.weighNumRange(10, 10000, 0.3, 200),
 				"recipient_type": ["friend", "family", "business", "self"],
+				"post_payday_spending": [false],
 			}
 		},
 		{
@@ -266,7 +284,9 @@ const config = {
 			properties: {
 				"bill_type": ["rent", "utilities", "phone", "insurance", "subscription", "loan_payment"],
 				"amount": u.weighNumRange(20, 3000, 0.5, 150),
-				"auto_pay": u.pickAWinner([true, false], 0.4),
+				"auto_pay": [false, false, false, true, true],
+				"missed_payment": [false],
+				"manual_payment": [false],
 			}
 		},
 		{
@@ -292,6 +312,7 @@ const config = {
 				"goal_type": ["emergency", "vacation", "car", "home", "education", "retirement"],
 				"target_amount": u.weighNumRange(500, 50000, 0.3, 5000),
 				"monthly_contribution": u.weighNumRange(25, 2000, 0.5, 200),
+				"budget_discipline": [false],
 			}
 		},
 		{
@@ -300,7 +321,9 @@ const config = {
 			properties: {
 				"investment_type": ["stocks", "etf", "crypto", "bonds", "mutual_fund"],
 				"amount": u.weighNumRange(10, 10000, 0.3, 250),
-				"action": u.pickAWinner(["buy", "sell", "buy"]),
+				"action": ["buy", "sell", "buy"],
+				"premium_returns": [false],
+				"budget_discipline": [false],
 			}
 		},
 		{
@@ -308,6 +331,7 @@ const config = {
 			weight: 1,
 			properties: {
 				"reason": ["lost", "stolen", "suspicious_activity", "travel"],
+				"fraud_sequence": [false],
 			}
 		},
 		{
@@ -316,6 +340,7 @@ const config = {
 			properties: {
 				"dispute_amount": u.weighNumRange(10, 2000, 0.5, 100),
 				"reason": ["unauthorized", "duplicate", "not_received", "damaged", "wrong_amount"],
+				"fraud_sequence": [false],
 			}
 		},
 		{
@@ -340,8 +365,8 @@ const config = {
 			weight: 2,
 			properties: {
 				"old_tier": ["basic", "plus", "premium"],
-				"new_tier": u.pickAWinner(["plus", "premium", "premium"]),
-				"monthly_fee": u.pickAWinner([4.99, 9.99, 14.99]),
+				"new_tier": ["plus", "premium", "premium"],
+				"monthly_fee": [4.99, 9.99, 14.99],
 			}
 		},
 		{
@@ -350,7 +375,8 @@ const config = {
 			properties: {
 				"channel": ["chat", "phone", "email", "in_app"],
 				"issue_type": ["transaction", "account", "card", "transfer", "technical"],
-				"resolved": u.pickAWinner([true, false], 0.8),
+				"resolved": [false, true, true, true, true],
+				"fraud_sequence": [false],
 			}
 		},
 		{
@@ -358,7 +384,7 @@ const config = {
 			weight: 10,
 			properties: {
 				"notification_type": ["transaction", "low_balance", "bill_due", "reward", "security", "promo"],
-				"action_taken": u.pickAWinner([true, false], 0.6),
+				"action_taken": [false, false, true, true, true],
 			}
 		},
 		{
@@ -367,13 +393,15 @@ const config = {
 			properties: {
 				"reward_type": ["cashback", "points", "discount", "partner_offer"],
 				"value": u.weighNumRange(1, 100, 0.5, 10),
+				"premium_reward": [false],
 			}
 		}
 	],
 
 	superProps: {
-		account_tier: u.pickAWinner(["basic", "basic", "basic", "plus", "plus", "premium"]),
+		account_tier: ["basic", "basic", "basic", "plus", "plus", "premium"],
 		platform: ["ios", "android", "web"],
+		low_balance_churn: [false],
 	},
 
 	userProps: {
@@ -381,7 +409,13 @@ const config = {
 		"income_bracket": ["under_30k", "30k_50k", "50k_75k", "75k_100k", "100k_150k", "over_150k"],
 		"account_age_months": u.weighNumRange(1, 60, 0.5, 12),
 		"total_balance": u.weighNumRange(0, 100000, 0.3, 5000),
-		"has_direct_deposit": u.pickAWinner([true, false], 0.6),
+		"has_direct_deposit": [false, false, true, true, true],
+		"account_segment": ["personal"],
+		"employee_count": [0],
+		"annual_revenue": [0],
+		"industry": [""],
+		"age_range": [""],
+		"life_stage": [""],
 	},
 
 	groupKeys: [
@@ -393,7 +427,7 @@ const config = {
 			"household_size": u.weighNumRange(1, 6),
 			"combined_income": u.weighNumRange(20000, 300000, 0.3, 75000),
 			"financial_health_score": u.weighNumRange(1, 100, 1, 65),
-			"primary_bank": u.pickAWinner(["NexBank_only", "multi_bank", "NexBank_only"]),
+			"primary_bank": ["NexBank_only", "multi_bank", "NexBank_only"],
 		}
 	},
 
@@ -568,48 +602,62 @@ const config = {
 					const burstCount = chance.integer({ min: 3, max: 5 });
 					const fraudEvents = [];
 
+					// Find template events for cloning
+					const txnTemplate = userEvents.find(e => e.event === "transaction completed");
+					const cardTemplate = userEvents.find(e => e.event === "card locked");
+					const disputeTemplate = userEvents.find(e => e.event === "dispute filed");
+					const supportTemplate = userEvents.find(e => e.event === "support contacted");
+
 					for (let i = 0; i < burstCount; i++) {
+						if (txnTemplate) {
+							fraudEvents.push({
+								...txnTemplate,
+								time: midTime.add(i * 10, "minutes").toISOString(),
+								user_id: distinctId,
+								transaction_type: "purchase",
+								amount: chance.integer({ min: 500, max: 3000 }),
+								merchant_category: chance.pickone(["online", "retail"]),
+								payment_method: "credit",
+								fraud_sequence: true,
+							});
+						}
+					}
+
+					// Card locked after the burst
+					if (cardTemplate) {
 						fraudEvents.push({
-							event: "transaction completed",
-							time: midTime.add(i * 10, "minutes").toISOString(),
+							...cardTemplate,
+							time: midTime.add(burstCount * 10 + 5, "minutes").toISOString(),
 							user_id: distinctId,
-							transaction_type: "purchase",
-							amount: chance.integer({ min: 500, max: 3000 }),
-							merchant_category: chance.pickone(["online", "retail"]),
-							payment_method: "credit",
+							reason: "suspicious_activity",
 							fraud_sequence: true,
 						});
 					}
 
-					// Card locked after the burst
-					fraudEvents.push({
-						event: "card locked",
-						time: midTime.add(burstCount * 10 + 5, "minutes").toISOString(),
-						user_id: distinctId,
-						reason: "suspicious_activity",
-						fraud_sequence: true,
-					});
-
 					// Dispute filed shortly after
-					fraudEvents.push({
-						event: "dispute filed",
-						time: midTime.add(burstCount * 10 + 30, "minutes").toISOString(),
-						user_id: distinctId,
-						dispute_amount: chance.integer({ min: 500, max: 3000 }),
-						reason: "unauthorized",
-						fraud_sequence: true,
-					});
+					if (disputeTemplate) {
+						fraudEvents.push({
+							...disputeTemplate,
+							time: midTime.add(burstCount * 10 + 30, "minutes").toISOString(),
+							user_id: distinctId,
+							dispute_amount: chance.integer({ min: 500, max: 3000 }),
+							reason: "unauthorized",
+							fraud_sequence: true,
+						});
+					}
 
 					// Support contacted
-					fraudEvents.push({
-						event: "support contacted",
-						time: midTime.add(burstCount * 10 + 45, "minutes").toISOString(),
-						user_id: distinctId,
-						channel: "phone",
-						issue_type: "card",
-						resolved: true,
-						fraud_sequence: true,
-					});
+					if (supportTemplate) {
+						fraudEvents.push({
+							...supportTemplate,
+							time: midTime.add(burstCount * 10 + 45, "minutes").toISOString(),
+							user_id: distinctId,
+							channel: "phone",
+							issue_type: "card",
+							resolved: true,
+							fraud_sequence: true,
+						});
+					}
 
 					// Splice all fraud events into the user's timeline
 					userEvents.splice(midIdx + 1, 0, ...fraudEvents);
@@ -674,16 +722,19 @@ const config = {
 
 					// Splice extra savings goal events (budget-conscious users set more goals)
 					if (event.event === "budget created" && chance.bool({ likelihood: 50 })) {
-						const extraGoal = {
-							event: "savings goal set",
-							time: eventTime.add(chance.integer({ min: 1, max: 7 }), "days").toISOString(),
-							user_id: event.user_id,
-							goal_type: chance.pickone(["emergency", "vacation", "car", "home"]),
-							target_amount: chance.integer({ min: 1000, max: 20000 }),
-							monthly_contribution: chance.integer({ min: 100, max: 800 }),
-							budget_discipline: true,
-						};
-						userEvents.splice(idx + 1, 0, extraGoal);
+						const savingsTemplate = userEvents.find(e => e.event === "savings goal set");
+						if (savingsTemplate) {
+							const extraGoal = {
+								...savingsTemplate,
+								time: eventTime.add(chance.integer({ min: 1, max: 7 }), "days").toISOString(),
+								user_id: event.user_id,
+								goal_type: chance.pickone(["emergency", "vacation", "car", "home"]),
+								target_amount: chance.integer({ min: 1000, max: 20000 }),
+								monthly_contribution: chance.integer({ min: 100, max: 800 }),
+								budget_discipline: true,
+							};
+							userEvents.splice(idx + 1, 0, extraGoal);
+						}
 					}
 				});
 			}