@wopr-network/platform-core 1.67.1 → 1.69.0

package/src/server/test-container.ts

@@ -0,0 +1,120 @@
+/**
+ * createTestContainer — builds a PlatformContainer with sensible mock
+ * defaults for unit tests. All feature sub-containers default to null.
+ * Core services get minimal stubs that satisfy their interfaces.
+ *
+ * Usage:
+ *   const c = createTestContainer();
+ *   const c2 = createTestContainer({ creditLedger: myCustomLedger });
+ */
+
+import type { IUserRoleRepository } from "../auth/user-role-repository.js";
+import type { ILedger } from "../credits/ledger.js";
+import type { DrizzleDb } from "../db/index.js";
+import type { ProductConfig } from "../product-config/repository-types.js";
+import type { IOrgMemberRepository } from "../tenancy/org-member-repository.js";
+import type { OrgService } from "../tenancy/org-service.js";
+import type { PlatformContainer } from "./container.js";
+
+// ---------------------------------------------------------------------------
+// Stub factories (satisfy interface contracts with no-op implementations)
+// ---------------------------------------------------------------------------
+
+function stubLedger(): ILedger {
+  const zero = 0 as never;
+  const emptyEntry = {} as never;
+  return {
+    post: async () => emptyEntry,
+    credit: async () => emptyEntry,
+    debit: async () => emptyEntry,
+    balance: async () => zero,
+    hasReferenceId: async () => false,
+    history: async () => [],
+    tenantsWithBalance: async () => [],
+    memberUsage: async () => [],
+    lifetimeSpend: async () => zero,
+    lifetimeSpendBatch: async () => new Map(),
+    expiredCredits: async () => [],
+    trialBalance: async () => ({ balanced: true }) as never,
+    accountBalance: async () => zero,
+    seedSystemAccounts: async () => {},
+    existsByReferenceIdLike: async () => false,
+    sumPurchasesForPeriod: async () => zero,
+    getActiveTenantIdsInWindow: async () => [],
+    debitCapped: async () => null,
+  };
+}
+
+function stubOrgMemberRepo(): IOrgMemberRepository {
+  return {
+    listMembers: async () => [],
+    addMember: async () => {},
+    updateMemberRole: async () => {},
+    removeMember: async () => {},
+    findMember: async () => null,
+    countAdminsAndOwners: async () => 0,
+    listInvites: async () => [],
+    createInvite: async () => {},
+    findInviteById: async () => null,
+    findInviteByToken: async () => null,
+    deleteInvite: async () => {},
+    deleteAllMembers: async () => {},
+    deleteAllInvites: async () => {},
+    listOrgsByUser: async () => [],
+    markInviteAccepted: async () => {},
+  };
+}
+
+function stubUserRoleRepo(): IUserRoleRepository {
+  return {
+    getTenantIdByUserId: async () => null,
+    grantRole: async () => {},
+    revokeRole: async () => false,
+    listRolesByUser: async () => [],
+    listUsersByRole: async () => [],
+    isPlatformAdmin: async () => false,
+  };
+}
+
+function stubProductConfig(): ProductConfig {
+  return {
+    product: {
+      slug: "test",
+      name: "Test Product",
+    } as never,
+    navItems: [],
+    domains: [],
+    features: null,
+    fleet: null,
+    billing: null,
+  };
+}
+
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+
+/**
+ * Create a PlatformContainer pre-filled with no-op stubs.
+ * Pass overrides for any field you need to customize in your test.
+ */
+export function createTestContainer(overrides?: Partial<PlatformContainer>): PlatformContainer {
+  const defaults: PlatformContainer = {
+    db: {} as DrizzleDb,
+    pool: { end: async () => {} } as never,
+    productConfig: stubProductConfig(),
+    creditLedger: stubLedger(),
+    orgMemberRepo: stubOrgMemberRepo(),
+    orgService: {} as OrgService,
+    userRoleRepo: stubUserRoleRepo(),
+
+    // Feature sub-containers default to null (not enabled)
+    fleet: null,
+    crypto: null,
+    stripe: null,
+    gateway: null,
+    hotPool: null,
+  };
+
+  return { ...defaults, ...overrides };
+}

package/src/trpc/auth-helpers.ts

@@ -0,0 +1,28 @@
+/**
+ * Shared auth helper factories for tRPC routers.
+ *
+ * These helpers can be constructed with explicit deps (for container-based DI)
+ * instead of relying on module-level singletons.
+ */
+
+import { TRPCError } from "@trpc/server";
+import type { IOrgMemberRepository } from "../tenancy/org-member-repository.js";
+
+/**
+ * Creates an assertOrgAdminOrOwner function closed over the given repository.
+ *
+ * Usage:
+ * ```ts
+ * const assertOrgAdmin = createAssertOrgAdminOrOwner(container.orgMemberRepo);
+ * await assertOrgAdmin(tenantId, userId);
+ * ```
+ */
+export function createAssertOrgAdminOrOwner(orgMemberRepo: IOrgMemberRepository) {
+  return async function assertOrgAdminOrOwner(tenantId: string, userId: string): Promise<void> {
+    if (tenantId === userId) return;
+    const member = await orgMemberRepo.findMember(tenantId, userId);
+    if (!member || (member.role !== "owner" && member.role !== "admin")) {
+      throw new TRPCError({ code: "FORBIDDEN", message: "Organization admin access required" });
+    }
+  };
+}

package/src/trpc/container-factories.ts

@@ -0,0 +1,114 @@
+/**
+ * Container-based tRPC router factories.
+ *
+ * Each function accepts a PlatformContainer (plus any extra deps not yet on
+ * the container) and returns a tRPC router by delegating to the existing
+ * factory functions. This provides a single-call DI entry point for products
+ * that have migrated to the container pattern, while the existing factory
+ * functions and setter-based API remain fully functional for products that
+ * haven't migrated yet.
+ */
+
+import type { INotificationTemplateRepository } from "../email/notification-template-repository.js";
+import type { RolloutOrchestrator } from "../fleet/rollout-orchestrator.js";
+import type { ITenantUpdateConfigRepository } from "../fleet/tenant-update-config-repository.js";
+import type { ProductConfigService } from "../product-config/service.js";
+import type { PlatformContainer } from "../server/container.js";
+import { createAdminFleetUpdateRouter } from "./admin-fleet-update-router.js";
+import { createFleetUpdateConfigRouter } from "./fleet-update-config-router.js";
+import { setTrpcOrgMemberRepo } from "./init.js";
+import { createNotificationTemplateRouter } from "./notification-template-router.js";
+import {
+  createOrgRemovePaymentMethodRouter,
+  type OrgRemovePaymentMethodDeps,
+} from "./org-remove-payment-method-router.js";
+import { createProductConfigRouter } from "./product-config-router.js";
+
+// ---------------------------------------------------------------------------
+// Init / middleware wiring
+// ---------------------------------------------------------------------------
+
+/**
+ * Wire the PlatformContainer's orgMemberRepo into the tRPC middleware layer.
+ *
+ * This replaces the manual `setTrpcOrgMemberRepo()` call. Products using the
+ * container call this once at boot; the setter-based API remains for products
+ * that haven't migrated yet.
+ */
+export function initTrpcFromContainer(container: PlatformContainer): void {
+  setTrpcOrgMemberRepo(container.orgMemberRepo);
+}
+
+// ---------------------------------------------------------------------------
+// Router factories — thin wrappers over existing factory functions
+// ---------------------------------------------------------------------------
+
+/**
+ * Create the admin fleet-update router from a container.
+ *
+ * Requires additional fleet-specific deps (orchestrator + config repo) that
+ * are constructed at boot when fleet is enabled and are not yet on the
+ * PlatformContainer itself.
+ */
+export function createAdminFleetUpdateRouterFromContainer(
+  _container: PlatformContainer,
+  getOrchestrator: () => RolloutOrchestrator,
+  getConfigRepo: () => ITenantUpdateConfigRepository,
+) {
+  return createAdminFleetUpdateRouter(getOrchestrator, getConfigRepo);
+}
+
+/**
+ * Create the fleet-update-config router from a container.
+ *
+ * Requires the tenant-update-config repository getter (fleet-specific dep
+ * not yet on PlatformContainer).
+ */
+export function createFleetUpdateConfigRouterFromContainer(
+  _container: PlatformContainer,
+  getConfigRepo: () => ITenantUpdateConfigRepository,
+) {
+  return createFleetUpdateConfigRouter(getConfigRepo);
+}
+
+/**
+ * Create the notification-template router from a container.
+ *
+ * Requires a getter for the notification-template repository (not yet on
+ * PlatformContainer).
+ */
+export function createNotificationTemplateRouterFromContainer(
+  _container: PlatformContainer,
+  getRepo: () => INotificationTemplateRepository,
+) {
+  return createNotificationTemplateRouter(getRepo);
+}
+
+/**
+ * Create the org-remove-payment-method router from a container.
+ *
+ * The container's StripeServices.processor is typed narrowly (webhook-only),
+ * so this still requires the full OrgRemovePaymentMethodDeps to be supplied
+ * until the container's Stripe sub-container is widened to include
+ * IPaymentProcessor.
+ */
+export function createOrgRemovePaymentMethodRouterFromContainer(
+  _container: PlatformContainer,
+  getDeps: () => OrgRemovePaymentMethodDeps,
+) {
+  return createOrgRemovePaymentMethodRouter(getDeps);
+}
+
+/**
+ * Create the product-config router from a container.
+ *
+ * Requires a getter for ProductConfigService and the product slug, both of
+ * which are product-specific and not on PlatformContainer.
+ */
+export function createProductConfigRouterFromContainer(
+  _container: PlatformContainer,
+  getService: () => ProductConfigService,
+  productSlug: string,
+) {
+  return createProductConfigRouter(getService, productSlug);
+}

package/src/trpc/index.ts

@@ -1,5 +1,14 @@
 export { createAdminFleetUpdateRouter } from "./admin-fleet-update-router.js";
+export { createAssertOrgAdminOrOwner } from "./auth-helpers.js";
 export { authSocialRouter } from "./auth-social-router.js";
+export {
+  createAdminFleetUpdateRouterFromContainer,
+  createFleetUpdateConfigRouterFromContainer,
+  createNotificationTemplateRouterFromContainer,
+  createOrgRemovePaymentMethodRouterFromContainer,
+  createProductConfigRouterFromContainer,
+  initTrpcFromContainer,
+} from "./container-factories.js";
 export { createFleetUpdateConfigRouter } from "./fleet-update-config-router.js";
 export {
   adminProcedure,