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

package/dungeons/{food.js → vertical/food-delivery.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";
 import * as v from "ak-tools";
 
 const SEED = "harness-food";
@@ -10,7 +10,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 */
 
 /*
  * ═══════════════════════════════════════════════════════════════════════════════
@@ -176,10 +176,23 @@ const config = {
 	hasAdSpend: false,
 	percentUsersBornInDataset: 50,
 	hasAvatar: true,
-	batchSize: 2_500_000,
 	concurrency: 1,
 	writeToDisk: false,
-	scdProps: {},
+	scdProps: {
+		subscription_tier: {
+			values: ["free", "trial", "monthly", "annual"],
+			frequency: "month",
+			timing: "fuzzy",
+			max: 6
+		},
+		restaurant_tier: {
+			values: ["new", "verified", "featured", "premium"],
+			frequency: "month",
+			timing: "fixed",
+			max: 6,
+			type: "restaurant_id"
+		}
+	},
 
 	funnels: [
 		{
@@ -194,7 +207,7 @@ const config = {
 			conversionRate: 55,
 			timeToConvert: 1,
 			weight: 5,
-			props: { "restaurant_id": u.pickAWinner(restaurantIds) },
+			props: { "restaurant_id": restaurantIds },
 		},
 		{
 			// Search-driven ordering
@@ -209,7 +222,7 @@ const config = {
 			conversionRate: 65,
 			timeToConvert: 2,
 			weight: 4,
-			props: { "order_id": u.pickAWinner(orderIds) },
+			props: { "order_id": orderIds },
 		},
 		{
 			// Post-order: rate and reorder
@@ -248,7 +261,8 @@ const config = {
 			isFirstEvent: true,
 			properties: {
 				"signup_method": ["email", "google", "apple", "facebook"],
-				"referral_code": u.pickAWinner([true, false], 0.3),
+				"referral_code": [false, false, true],
+				"trial_retained": [false],
 			}
 		},
 		{
@@ -266,14 +280,15 @@ const config = {
 					"Mediterranean"
 				],
 				"sort_by": ["recommended", "distance", "rating", "price"],
-				"filter_applied": u.pickAWinner([true, false], 0.4),
+				"filter_applied": [false, false, false, true, true],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "restaurant viewed",
 			weight: 15,
 			properties: {
-				"restaurant_id": u.pickAWinner(restaurantIds),
+				"restaurant_id": restaurantIds,
 				"cuisine_type": [
 					"American",
 					"Italian",
@@ -287,33 +302,40 @@ const config = {
 				"avg_rating": u.weighNumRange(1, 5, 0.8, 30),
 				"delivery_time_est_mins": u.weighNumRange(15, 90, 1.2, 40),
 				"price_tier": ["$", "$$", "$$$", "$$$$"],
+				"late_night_order": [false],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "item added to cart",
 			weight: 14,
 			properties: {
-				"item_id": u.pickAWinner(itemIds),
+				"item_id": itemIds,
 				"item_category": ["entree", "appetizer", "drink", "dessert", "side"],
 				"item_price": u.weighNumRange(3, 65, 1.0, 40),
 				"customization_count": u.weighNumRange(0, 5, 1.5, 20),
+				"late_night_order": [false],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "item removed from cart",
 			weight: 5,
 			properties: {
-				"item_id": u.pickAWinner(itemIds),
+				"item_id": itemIds,
 				"removal_reason": ["changed_mind", "too_expensive", "substitution"],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "coupon applied",
 			weight: 4,
 			properties: {
-				"coupon_code": u.pickAWinner(couponCodes),
+				"coupon_code": couponCodes,
 				"discount_type": ["percent", "flat", "free_delivery"],
 				"discount_value": u.weighNumRange(5, 50, 1.2, 20),
+				"coupon_injected": [false],
+				"trial_retained": [false],
 			}
 		},
 		{
@@ -322,46 +344,54 @@ const config = {
 			properties: {
 				"cart_total": u.weighNumRange(8, 150, 0.8, 40),
 				"items_count": u.weighNumRange(1, 8, 1.2, 20),
-				"delivery_address_saved": u.pickAWinner([true, false], 0.7),
+				"delivery_address_saved": [false, false, false, true, true, true, true, true, true, true],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "order placed",
 			weight: 10,
 			properties: {
-				"order_id": u.pickAWinner(orderIds),
+				"order_id": orderIds,
 				"payment_method": ["credit_card", "apple_pay", "google_pay", "paypal", "cash"],
 				"order_total": u.weighNumRange(10, 200, 0.8, 40),
 				"tip_amount": u.weighNumRange(0, 30, 1.5, 20),
 				"delivery_fee": u.weighNumRange(0, 12, 1.0, 20),
+				"surge_pricing": [false],
+				"rainy_week": [false],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "order tracked",
 			weight: 13,
 			properties: {
-				"order_id": u.pickAWinner(orderIds),
+				"order_id": orderIds,
 				"order_status": ["confirmed", "preparing", "picked_up", "en_route", "delivered"],
 				"eta_mins": u.weighNumRange(5, 60, 1.0, 30),
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "order delivered",
 			weight: 9,
 			properties: {
-				"order_id": u.pickAWinner(orderIds),
+				"order_id": orderIds,
 				"actual_delivery_mins": u.weighNumRange(12, 90, 1.0, 40),
-				"on_time": u.pickAWinner([true, false], 0.7),
+				"on_time": [false, false, false, true, true, true, true, true, true, true],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "order rated",
 			weight: 7,
 			properties: {
-				"order_id": u.pickAWinner(orderIds),
+				"order_id": orderIds,
 				"food_rating": u.weighNumRange(1, 5, 0.8, 30),
 				"delivery_rating": u.weighNumRange(1, 5, 0.8, 30),
-				"would_reorder": u.pickAWinner([true, false], 0.65),
+				"would_reorder": [false, true, true],
+				"referral_user": [false],
+				"trial_retained": [false],
 			}
 		},
 		{
@@ -375,6 +405,7 @@ const config = {
 				]),
 				"results_count": u.weighNumRange(0, 50, 0.8, 30),
 				"search_type": ["restaurant", "cuisine", "dish"],
+				"trial_retained": [false],
 			}
 		},
 		{
@@ -384,15 +415,17 @@ const config = {
 				"promo_id": () => `promo_${v.uid(5)}`,
 				"promo_type": ["banner", "push", "in_feed"],
 				"promo_value": ["10%", "15%", "20%", "25%", "30%", "40%", "50%"],
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "subscription started",
 			weight: 2,
 			properties: {
-				"plan": u.pickAWinner(["quickbite_plus_monthly", "quickbite_plus_monthly", "quickbite_plus_annual"]),
-				"price": u.pickAWinner([9.99, 9.99, 79.99]),
-				"trial": u.pickAWinner([true, false], 0.5),
+				"plan": ["quickbite_plus_monthly", "quickbite_plus_monthly", "quickbite_plus_annual"],
+				"price": [9.99, 9.99, 79.99],
+				"trial": [true, false],
+				"trial_retained": [false],
 			}
 		},
 		{
@@ -401,6 +434,7 @@ const config = {
 			properties: {
 				"reason": ["too_expensive", "not_ordering_enough", "found_alternative", "bad_experience"],
 				"months_subscribed": u.weighNumRange(1, 24, 1.5, 15),
+				"trial_retained": [false],
 			}
 		},
 		{
@@ -408,22 +442,25 @@ const config = {
 			weight: 3,
 			properties: {
 				"issue_type": ["missing_item", "wrong_order", "late_delivery", "quality_issue", "refund_request"],
-				"order_id": u.pickAWinner(orderIds),
+				"order_id": orderIds,
+				"trial_retained": [false],
 			}
 		},
 		{
 			event: "reorder initiated",
 			weight: 6,
 			properties: {
-				"order_id": u.pickAWinner(orderIds),
+				"order_id": orderIds,
 				"original_order_age_days": u.weighNumRange(1, 60, 1.5, 30),
+				"referral_user": [false],
+				"trial_retained": [false],
 			}
 		},
 	],
 
 	superProps: {
 		platform: ["iOS", "Android", "Web"],
-		subscription_tier: u.pickAWinner(["Free", "Free", "Free", "Free", "QuickBite+"]),
+		subscription_tier: ["Free", "Free", "Free", "Free", "QuickBite+"],
 		city: ["New York", "Los Angeles", "Chicago", "Houston", "Phoenix", "San Francisco"],
 	},
 
@@ -441,6 +478,8 @@ const config = {
 		"avg_order_value": u.weighNumRange(15, 80, 0.8, 40),
 		"orders_per_month": u.weighNumRange(1, 20, 1.5, 10),
 		"favorite_restaurant_count": u.weighNumRange(1, 10),
+		"is_high_risk": [false],
+		"churn_risk_score": [0],
 	},
 
 	groupKeys: [
@@ -559,7 +598,9 @@ const config = {
 						'milliseconds'
 					).toISOString();
 
+					const couponTemplate = record.find(e => e.event === "coupon applied");
 					const couponEvent = {
+						...(couponTemplate || firstEvent),
 						event: "coupon applied",
 						time: midTime,
 						user_id: firstEvent.user_id,
@@ -670,7 +711,7 @@ const config = {
 				if (isReferralUser && event.event === "reorder initiated" && chance.bool({ likelihood: 50 })) {
 					const eventTime = dayjs(event.time);
 					const extraReorder = {
-						event: "reorder initiated",
+						...event,
 						time: eventTime.add(chance.integer({ min: 1, max: 7 }), 'days').toISOString(),
 						user_id: event.user_id,
 						order_id: chance.pickone(orderIds),