feat(planning): ship holiday-aware planning and assistant upgrades

packages/db/src/seed-holiday-demo-resources.ts

@@ -0,0 +1,125 @@
+import { PrismaClient, type Prisma } from "@prisma/client";
+import { getHolidayDemoCityNamesByCountry, getHolidayDemoProfileForIndex } from "./holiday-demo-profiles.js";
+import { loadWorkspaceEnv } from "./load-workspace-env.js";
+
+loadWorkspaceEnv();
+
+const prisma = new PrismaClient();
+
+type CountryRecord = {
+  id: string;
+  code: string;
+};
+
+function asJsonObject(value: Prisma.JsonValue | null | undefined): Record<string, Prisma.JsonValue> {
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    return {};
+  }
+
+  return value as Record<string, Prisma.JsonValue>;
+}
+
+async function ensureCities(countryByCode: Map<string, CountryRecord>) {
+  const cityNamesByCountry = getHolidayDemoCityNamesByCountry();
+  const cityMap = new Map<string, { id: string }>();
+
+  for (const [countryCode, cityNames] of Object.entries(cityNamesByCountry)) {
+    const country = countryByCode.get(countryCode);
+    if (!country) {
+      continue;
+    }
+
+    for (const cityName of cityNames) {
+      const city = await prisma.metroCity.upsert({
+        where: {
+          countryId_name: {
+            countryId: country.id,
+            name: cityName,
+          },
+        },
+        update: {},
+        create: {
+          countryId: country.id,
+          name: cityName,
+        },
+        select: { id: true },
+      });
+      cityMap.set(`${countryCode}:${cityName}`, city);
+    }
+  }
+
+  return cityMap;
+}
+
+async function main() {
+  console.log("Normalizing active resources for holiday demo profiles...");
+
+  const countrySeeds = [
+    { code: "DE", name: "Germany", dailyWorkingHours: 8 },
+    { code: "ES", name: "Spain", dailyWorkingHours: 8 },
+    { code: "IN", name: "India", dailyWorkingHours: 9 },
+    { code: "US", name: "United States", dailyWorkingHours: 8 },
+  ] as const;
+  const countries = [];
+
+  for (const countrySeed of countrySeeds) {
+    countries.push(await prisma.country.upsert({
+      where: { code: countrySeed.code },
+      update: { name: countrySeed.name, dailyWorkingHours: countrySeed.dailyWorkingHours },
+      create: countrySeed,
+      select: { id: true, code: true },
+    }));
+  }
+
+  const countryByCode = new Map(countries.map((country) => [country.code, country] as const));
+
+  const cityByProfile = await ensureCities(countryByCode);
+  const resources = await prisma.resource.findMany({
+    where: { isActive: true },
+    select: {
+      id: true,
+      eid: true,
+      dynamicFields: true,
+    },
+    orderBy: { eid: "asc" },
+  });
+
+  let updated = 0;
+
+  for (const [index, resource] of resources.entries()) {
+    const profile = getHolidayDemoProfileForIndex(index);
+    const country = countryByCode.get(profile.countryCode)!;
+    const metroCity = cityByProfile.get(`${profile.countryCode}:${profile.cityName}`);
+
+    if (!metroCity) {
+      throw new Error(`Missing metro city for profile ${profile.countryCode}/${profile.cityName}`);
+    }
+
+    const dynamicFields = asJsonObject(resource.dynamicFields);
+
+    await prisma.resource.update({
+      where: { id: resource.id },
+      data: {
+        countryId: country.id,
+        metroCityId: metroCity.id,
+        federalState: profile.stateCode,
+        dynamicFields: {
+          ...dynamicFields,
+          city: profile.cityName,
+          holidayCountryCode: profile.countryCode,
+          holidayStateCode: profile.stateCode,
+        } as Prisma.InputJsonValue,
+      },
+    });
+    updated += 1;
+  }
+
+  console.log(`Updated ${updated} active resources.`);
+}
+
+main()
+  .catch((error) => {
+    console.error(error);
+    process.exit(1);
+  })
+  .finally(() => void prisma.$disconnect());