51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
import { AsyncLocalStorage } from "node:async_hooks";
|
|
import type { AuthenticatedEnvironment } from "./apiAuth.server";
|
|
|
|
// All fields are optional. The middleware establishes an empty scope per
|
|
// request; entry points fill what they know:
|
|
// - URL-matching paths get the slug trio from the Express middleware (zero IO).
|
|
// - The `_app` layout adds `userId` for any authenticated request.
|
|
// - The env layout adds tenant IDs / env type after its own existing DB query.
|
|
// - API routes get the full set up-front from `authenticationResult.environment`.
|
|
export type TenantContext = {
|
|
userId?: string;
|
|
orgSlug?: string;
|
|
projectSlug?: string;
|
|
envSlug?: string;
|
|
orgId?: string;
|
|
projectId?: string;
|
|
projectRef?: string;
|
|
envId?: string;
|
|
envType?: "DEVELOPMENT" | "PREVIEW" | "STAGING" | "PRODUCTION";
|
|
impersonating?: boolean;
|
|
};
|
|
|
|
const storage = new AsyncLocalStorage<TenantContext>();
|
|
|
|
export const tenantContext = {
|
|
run<T>(ctx: TenantContext, fn: () => T): T {
|
|
return storage.run(ctx, fn);
|
|
},
|
|
get(): TenantContext | undefined {
|
|
return storage.getStore();
|
|
},
|
|
enrich(patch: Partial<TenantContext>): void {
|
|
const current = storage.getStore();
|
|
if (current) Object.assign(current, patch);
|
|
},
|
|
};
|
|
|
|
export function tenantContextFromAuthEnvironment(env: AuthenticatedEnvironment): TenantContext {
|
|
return {
|
|
userId: env.orgMember?.userId,
|
|
orgSlug: env.organization.slug,
|
|
projectSlug: env.project.slug,
|
|
envSlug: env.slug,
|
|
orgId: env.organization.id,
|
|
projectId: env.project.id,
|
|
projectRef: env.project.externalRef,
|
|
envId: env.id,
|
|
envType: env.type,
|
|
};
|
|
}
|