@m5kdev/backend 0.6.0 → 0.8.0

package/dist/src/modules/workflow/workflow.utils.mjs

@@ -0,0 +1,181 @@
+import { logger } from "../../utils/logger.mjs";
+import { ServerError } from "../../utils/errors.mjs";
+import { runWithPosthogRequestState } from "../../utils/posthog.mjs";
+import { err } from "neverthrow";
+import { Worker } from "bullmq";
+//#region src/modules/workflow/workflow.utils.ts
+const errorOptions = {
+	layer: "workflow",
+	layerName: "WorkflowTrigger"
+};
+function shouldDisablePosthogFromPayload(payload) {
+	if (!payload || typeof payload !== "object") return false;
+	const session = payload.ctx?.session;
+	return Boolean(session?.impersonatedBy);
+}
+function createWorkflowTrigger(service, queues, defaultQueue) {
+	const trigger = async ({ name, payload, meta, options }) => {
+		const { queue: queueName, userId, tags, timeout = 3600 * 1e3 } = meta || {};
+		const disablePosthogCapture = meta?.disablePosthogCapture ?? shouldDisablePosthogFromPayload(payload);
+		const compiledQueueName = queueName || defaultQueue;
+		if (!compiledQueueName || !Object.keys(queues).includes(compiledQueueName)) return err(new ServerError({
+			code: "NOT_FOUND",
+			...errorOptions
+		}));
+		const queue = queues[compiledQueueName];
+		if (!queue) return err(new ServerError({
+			code: "NOT_FOUND",
+			...errorOptions
+		}));
+		try {
+			const job = await queue.add(name, {
+				payload,
+				meta: {
+					...meta,
+					timeout,
+					disablePosthogCapture
+				}
+			}, {
+				removeOnComplete: { age: 24 * 3600 },
+				removeOnFail: { age: 168 * 3600 },
+				...options
+			});
+			if (!job || !job.id) return err(new ServerError({
+				code: "INTERNAL_SERVER_ERROR",
+				...errorOptions
+			}));
+			return await service.added({
+				userId,
+				jobId: job.id,
+				jobName: name,
+				input: payload,
+				timeout,
+				queueName: compiledQueueName,
+				tags
+			});
+		} catch (error) {
+			return err(new ServerError({
+				code: "INTERNAL_SERVER_ERROR",
+				...errorOptions
+			}));
+		}
+	};
+	const triggerMany = async ({ name, payload, meta, options }) => {
+		const { queue: queueName, userId, tags, timeout = 3600 * 1e3 } = meta || {};
+		const compiledQueueName = queueName || defaultQueue;
+		if (!compiledQueueName || !Object.keys(queues).includes(compiledQueueName)) return err(new ServerError({
+			code: "NOT_FOUND",
+			...errorOptions
+		}));
+		const queue = queues[compiledQueueName];
+		if (!queue) return err(new ServerError({
+			code: "NOT_FOUND",
+			...errorOptions
+		}));
+		try {
+			const jobs = await queue.addBulk(payload.map((p) => {
+				const disablePosthogCapture = meta?.disablePosthogCapture ?? shouldDisablePosthogFromPayload(p);
+				return {
+					name,
+					data: {
+						payload: p,
+						meta: {
+							...meta,
+							timeout,
+							disablePosthogCapture
+						}
+					},
+					options: {
+						removeOnComplete: { age: 24 * 3600 },
+						removeOnFail: { age: 168 * 3600 },
+						...options
+					}
+				};
+			}));
+			if (!jobs || jobs.length === 0) return err(new ServerError({
+				code: "INTERNAL_SERVER_ERROR",
+				...errorOptions
+			}));
+			return await service.addedMany(jobs.map((j) => ({
+				userId,
+				jobId: j.id,
+				jobName: name,
+				input: j.data.payload,
+				timeout,
+				queueName: compiledQueueName,
+				tags
+			})));
+		} catch (error) {
+			return err(new ServerError({
+				code: "INTERNAL_SERVER_ERROR",
+				...errorOptions
+			}));
+		}
+	};
+	return {
+		trigger,
+		triggerMany
+	};
+}
+function createWorkflowWorker(service, queueName, workerOptions, jobs, errorHandler) {
+	const logger$1 = logger.child({ layer: "worker" });
+	const worker = new Worker(queueName, async (job) => {
+		const { meta } = job.data;
+		return await runWithPosthogRequestState({ disableCapture: Boolean(meta?.disablePosthogCapture) }, async () => {
+			const timer = meta?.timeout && setTimeout(() => {
+				const error = /* @__PURE__ */ new Error("Job timed out");
+				errorHandler?.(error);
+				throw error;
+			}, job.data.meta.timeout);
+			const handler = jobs[job.name];
+			if (!handler) {
+				const error = /* @__PURE__ */ new Error(`Unknown job: ${job.name}`);
+				errorHandler?.(error);
+				throw error;
+			}
+			try {
+				await service.started(job);
+				const result = await handler.run(job);
+				if (result.isErr()) {
+					await handler.onFailure?.(job, result.error);
+					const error = new Error(result.error.message);
+					errorHandler?.(error);
+					throw error;
+				}
+				await handler.onSuccess?.(job).catch((err) => {
+					const error = /* @__PURE__ */ new Error(`Job ${job.id}:${job.name} failed to run onSuccess: ${err}`);
+					logger$1.error(error.message);
+					errorHandler?.(error);
+				});
+				return result.value ?? null;
+			} catch (err) {
+				await handler.onFailure?.(job, err).catch((err) => {
+					const error = /* @__PURE__ */ new Error(`Job ${job.id}:${job.name} failed to run onFailure: ${err}`);
+					logger$1.error(error.message);
+					errorHandler?.(error);
+				});
+				if (err instanceof Error && err.message === "Job timed out") logger$1.error(`Job ${job.id}:${job.name} timed out`);
+				throw err;
+			} finally {
+				await handler.onComplete?.(job).catch((err) => {
+					const error = /* @__PURE__ */ new Error(`Job ${job.id}:${job.name} failed to run onComplete: ${err}`);
+					logger$1.error(error.message);
+					errorHandler?.(error);
+				});
+				if (timer) clearTimeout(timer);
+			}
+		});
+	}, workerOptions);
+	worker.on("completed", service.completed.bind(service));
+	worker.on("failed", service.failed.bind(service));
+	return worker;
+}
+function createWorkflowWorkers(service, workerSettings, jobs, errorHandler) {
+	const workers = {};
+	for (const [queueName, workerOptions] of Object.entries(workerSettings)) workers[queueName] = createWorkflowWorker(service, queueName, workerOptions, jobs, errorHandler);
+	return workers;
+}
+//#endregion
+export { createWorkflowTrigger, createWorkflowWorker, createWorkflowWorkers };
+
+//# sourceMappingURL=workflow.utils.mjs.map

package/dist/src/modules/workflow/workflow.utils.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow.utils.mjs","names":["logger","rootLogger"],"sources":["../../../../src/modules/workflow/workflow.utils.ts"],"sourcesContent":["import type { WorkflowReadOutputSchema } from \"@m5kdev/commons/modules/workflow/workflow.schema\";\r\nimport { type JobsOptions, type Queue, Worker, type WorkerOptions } from \"bullmq\";\r\nimport { err } from \"neverthrow\";\r\nimport type { ServerErrorLayer } from \"../base/base.types\";\r\nimport { ServerError } from \"../../utils/errors\";\r\nimport { logger as rootLogger } from \"../../utils/logger\";\r\nimport { runWithPosthogRequestState } from \"../../utils/posthog\";\r\nimport type { ServerResultAsync } from \"../base/base.dto\";\r\nimport type { WorkflowService } from \"./workflow.service\";\r\nimport type { WorkflowDataType, WorkflowJob, WorkflowMeta } from \"./workflow.types\";\r\n\r\nconst errorOptions = {\r\n  layer: \"workflow\" as ServerErrorLayer,\r\n  layerName: \"WorkflowTrigger\",\r\n};\r\n\r\nfunction shouldDisablePosthogFromPayload(payload: unknown): boolean {\r\n  if (!payload || typeof payload !== \"object\") return false;\r\n\r\n  const session = (\r\n    payload as {\r\n      ctx?: {\r\n        session?: {\r\n          impersonatedBy?: string | null;\r\n        };\r\n      };\r\n    }\r\n  ).ctx?.session;\r\n\r\n  return Boolean(session?.impersonatedBy);\r\n}\r\n\r\nexport function createWorkflowTrigger<\r\n  Jobs extends Record<string, WorkflowJob<WorkflowDataType<any>, any, string>>,\r\n>(service: WorkflowService, queues: Record<string, Queue>, defaultQueue: string) {\r\n  const trigger = async ({\r\n    name,\r\n    payload,\r\n    meta,\r\n    options,\r\n  }: {\r\n    name: string;\r\n    payload: Parameters<Jobs[keyof Jobs & string][\"run\"]>[0][\"data\"][\"payload\"];\r\n    meta?: WorkflowMeta;\r\n    options?: JobsOptions;\r\n  }): ServerResultAsync<WorkflowReadOutputSchema> => {\r\n    const { queue: queueName, userId, tags, timeout = 60 * 60 * 1000 } = meta || {};\r\n    const disablePosthogCapture =\r\n      meta?.disablePosthogCapture ?? shouldDisablePosthogFromPayload(payload);\r\n    const compiledQueueName = queueName || (defaultQueue as string);\r\n\r\n    if (!compiledQueueName || !Object.keys(queues).includes(compiledQueueName)) {\r\n      return err(new ServerError({ code: \"NOT_FOUND\", ...errorOptions }));\r\n    }\r\n\r\n    const queue = queues[compiledQueueName as string];\r\n\r\n    if (!queue) {\r\n      return err(new ServerError({ code: \"NOT_FOUND\", ...errorOptions }));\r\n    }\r\n    try {\r\n      const job = await queue.add(\r\n        name,\r\n        {\r\n          payload,\r\n          meta: { ...meta, timeout, disablePosthogCapture },\r\n        },\r\n        {\r\n          removeOnComplete: {\r\n            age: 24 * 3600, // keep up to 24 hours\r\n          },\r\n          removeOnFail: {\r\n            age: 7 * 24 * 3600, // keep up to week\r\n          },\r\n          ...options,\r\n        }\r\n      );\r\n\r\n      if (!job || !job.id) {\r\n        return err(new ServerError({ code: \"INTERNAL_SERVER_ERROR\", ...errorOptions }));\r\n      }\r\n\r\n      return await service.added({\r\n        userId,\r\n        jobId: job.id,\r\n        jobName: name,\r\n        input: payload,\r\n        timeout,\r\n        queueName: compiledQueueName,\r\n        tags,\r\n      });\r\n    } catch (error) {\r\n      return err(new ServerError({ code: \"INTERNAL_SERVER_ERROR\", ...errorOptions }));\r\n    }\r\n  };\r\n\r\n  const triggerMany = async ({\r\n    name,\r\n    payload,\r\n    meta,\r\n    options,\r\n  }: {\r\n    name: string;\r\n    payload: Parameters<Jobs[keyof Jobs & string][\"run\"]>[0][\"data\"][\"payload\"][];\r\n    meta?: WorkflowMeta;\r\n    options?: JobsOptions;\r\n  }): ServerResultAsync<WorkflowReadOutputSchema[]> => {\r\n    const { queue: queueName, userId, tags, timeout = 60 * 60 * 1000 } = meta || {};\r\n    const compiledQueueName = queueName || (defaultQueue as string);\r\n\r\n    if (!compiledQueueName || !Object.keys(queues).includes(compiledQueueName)) {\r\n      return err(new ServerError({ code: \"NOT_FOUND\", ...errorOptions }));\r\n    }\r\n\r\n    const queue = queues[compiledQueueName as string];\r\n\r\n    if (!queue) {\r\n      return err(new ServerError({ code: \"NOT_FOUND\", ...errorOptions }));\r\n    }\r\n    try {\r\n      const jobs = await queue.addBulk(\r\n        payload.map((p) => {\r\n          const disablePosthogCapture =\r\n            meta?.disablePosthogCapture ?? shouldDisablePosthogFromPayload(p);\r\n\r\n          return {\r\n            name,\r\n            data: { payload: p, meta: { ...meta, timeout, disablePosthogCapture } },\r\n            options: {\r\n              removeOnComplete: {\r\n                age: 24 * 3600, // keep up to 24 hours\r\n              },\r\n              removeOnFail: {\r\n                age: 7 * 24 * 3600, // keep up to week\r\n              },\r\n              ...options,\r\n            },\r\n          };\r\n        })\r\n      );\r\n\r\n      if (!jobs || jobs.length === 0) {\r\n        return err(new ServerError({ code: \"INTERNAL_SERVER_ERROR\", ...errorOptions }));\r\n      }\r\n\r\n      return await service.addedMany(\r\n        jobs.map((j) => ({\r\n          userId,\r\n          jobId: j.id as string,\r\n          jobName: name,\r\n          input: j.data.payload,\r\n          timeout,\r\n          queueName: compiledQueueName,\r\n          tags,\r\n        }))\r\n      );\r\n    } catch (error) {\r\n      return err(new ServerError({ code: \"INTERNAL_SERVER_ERROR\", ...errorOptions }));\r\n    }\r\n  };\r\n\r\n  return { trigger, triggerMany };\r\n}\r\n\r\nexport function createWorkflowWorker<\r\n  Jobs extends Record<string, WorkflowJob<WorkflowDataType<any>, any, string>>,\r\n>(\r\n  service: WorkflowService,\r\n  queueName: string,\r\n  workerOptions: WorkerOptions,\r\n  jobs: Jobs,\r\n  errorHandler?: (error: unknown) => void\r\n) {\r\n  const logger = rootLogger.child({ layer: \"worker\" });\r\n\r\n  const worker = new Worker(\r\n    queueName,\r\n    async (job) => {\r\n      const { meta } = job.data as { meta: WorkflowMeta; payload: any };\r\n      return await runWithPosthogRequestState(\r\n        { disableCapture: Boolean(meta?.disablePosthogCapture) },\r\n        async () => {\r\n          const timer =\r\n            meta?.timeout &&\r\n            setTimeout(() => {\r\n              const error = new Error(\"Job timed out\");\r\n              errorHandler?.(error);\r\n              throw error;\r\n            }, job.data.meta.timeout);\r\n          const handler = jobs[job.name as keyof Jobs];\r\n          if (!handler) {\r\n            const error = new Error(`Unknown job: ${job.name}`);\r\n            errorHandler?.(error);\r\n            throw error;\r\n          }\r\n          try {\r\n            await service.started(job);\r\n\r\n            const result = await handler.run(job);\r\n            if (result.isErr()) {\r\n              await handler.onFailure?.(job, result.error);\r\n              const error = new Error(result.error.message);\r\n              errorHandler?.(error);\r\n              throw error;\r\n            }\r\n            await handler.onSuccess?.(job).catch((err: unknown) => {\r\n              const error = new Error(`Job ${job.id}:${job.name} failed to run onSuccess: ${err}`);\r\n              logger.error(error.message);\r\n              errorHandler?.(error);\r\n            });\r\n            return result.value ?? null;\r\n          } catch (err) {\r\n            await handler.onFailure?.(job, err).catch((err: unknown) => {\r\n              const error = new Error(`Job ${job.id}:${job.name} failed to run onFailure: ${err}`);\r\n              logger.error(error.message);\r\n              errorHandler?.(error);\r\n            });\r\n            if (err instanceof Error && err.message === \"Job timed out\")\r\n              logger.error(`Job ${job.id}:${job.name} timed out`);\r\n            throw err;\r\n          } finally {\r\n            await handler.onComplete?.(job).catch((err: unknown) => {\r\n              const error = new Error(`Job ${job.id}:${job.name} failed to run onComplete: ${err}`);\r\n              logger.error(error.message);\r\n              errorHandler?.(error);\r\n            });\r\n            if (timer) clearTimeout(timer);\r\n          }\r\n        }\r\n      );\r\n    },\r\n    workerOptions as WorkerOptions\r\n  );\r\n  worker.on(\"completed\", service.completed.bind(service));\r\n  worker.on(\"failed\", service.failed.bind(service));\r\n\r\n  return worker;\r\n}\r\n\r\nexport function createWorkflowWorkers<\r\n  Jobs extends Record<string, WorkflowJob<WorkflowDataType<any>, any, string>>,\r\n>(\r\n  service: WorkflowService,\r\n  workerSettings: Record<string, WorkerOptions>,\r\n  jobs: Jobs,\r\n  errorHandler?: (error: unknown) => void\r\n) {\r\n  const workers: Record<string, Worker> = {} as Record<string, Worker>;\r\n  for (const [queueName, workerOptions] of Object.entries(workerSettings)) {\r\n    workers[queueName] = createWorkflowWorker(\r\n      service,\r\n      queueName,\r\n      workerOptions as WorkerOptions,\r\n      jobs,\r\n      errorHandler\r\n    );\r\n  }\r\n  return workers;\r\n}\r\n"],"mappings":";;;;;;AAWA,MAAM,eAAe;CACnB,OAAO;CACP,WAAW;CACZ;AAED,SAAS,gCAAgC,SAA2B;AAClE,KAAI,CAAC,WAAW,OAAO,YAAY,SAAU,QAAO;CAEpD,MAAM,UACJ,QAOA,KAAK;AAEP,QAAO,QAAQ,SAAS,eAAe;;AAGzC,SAAgB,sBAEd,SAA0B,QAA+B,cAAsB;CAC/E,MAAM,UAAU,OAAO,EACrB,MACA,SACA,MACA,cAMiD;EACjD,MAAM,EAAE,OAAO,WAAW,QAAQ,MAAM,UAAU,OAAU,QAAS,QAAQ,EAAE;EAC/E,MAAM,wBACJ,MAAM,yBAAyB,gCAAgC,QAAQ;EACzE,MAAM,oBAAoB,aAAc;AAExC,MAAI,CAAC,qBAAqB,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,kBAAkB,CACxE,QAAO,IAAI,IAAI,YAAY;GAAE,MAAM;GAAa,GAAG;GAAc,CAAC,CAAC;EAGrE,MAAM,QAAQ,OAAO;AAErB,MAAI,CAAC,MACH,QAAO,IAAI,IAAI,YAAY;GAAE,MAAM;GAAa,GAAG;GAAc,CAAC,CAAC;AAErE,MAAI;GACF,MAAM,MAAM,MAAM,MAAM,IACtB,MACA;IACE;IACA,MAAM;KAAE,GAAG;KAAM;KAAS;KAAuB;IAClD,EACD;IACE,kBAAkB,EAChB,KAAK,KAAK,MACX;IACD,cAAc,EACZ,KAAK,MAAS,MACf;IACD,GAAG;IACJ,CACF;AAED,OAAI,CAAC,OAAO,CAAC,IAAI,GACf,QAAO,IAAI,IAAI,YAAY;IAAE,MAAM;IAAyB,GAAG;IAAc,CAAC,CAAC;AAGjF,UAAO,MAAM,QAAQ,MAAM;IACzB;IACA,OAAO,IAAI;IACX,SAAS;IACT,OAAO;IACP;IACA,WAAW;IACX;IACD,CAAC;WACK,OAAO;AACd,UAAO,IAAI,IAAI,YAAY;IAAE,MAAM;IAAyB,GAAG;IAAc,CAAC,CAAC;;;CAInF,MAAM,cAAc,OAAO,EACzB,MACA,SACA,MACA,cAMmD;EACnD,MAAM,EAAE,OAAO,WAAW,QAAQ,MAAM,UAAU,OAAU,QAAS,QAAQ,EAAE;EAC/E,MAAM,oBAAoB,aAAc;AAExC,MAAI,CAAC,qBAAqB,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,kBAAkB,CACxE,QAAO,IAAI,IAAI,YAAY;GAAE,MAAM;GAAa,GAAG;GAAc,CAAC,CAAC;EAGrE,MAAM,QAAQ,OAAO;AAErB,MAAI,CAAC,MACH,QAAO,IAAI,IAAI,YAAY;GAAE,MAAM;GAAa,GAAG;GAAc,CAAC,CAAC;AAErE,MAAI;GACF,MAAM,OAAO,MAAM,MAAM,QACvB,QAAQ,KAAK,MAAM;IACjB,MAAM,wBACJ,MAAM,yBAAyB,gCAAgC,EAAE;AAEnE,WAAO;KACL;KACA,MAAM;MAAE,SAAS;MAAG,MAAM;OAAE,GAAG;OAAM;OAAS;OAAuB;MAAE;KACvE,SAAS;MACP,kBAAkB,EAChB,KAAK,KAAK,MACX;MACD,cAAc,EACZ,KAAK,MAAS,MACf;MACD,GAAG;MACJ;KACF;KACD,CACH;AAED,OAAI,CAAC,QAAQ,KAAK,WAAW,EAC3B,QAAO,IAAI,IAAI,YAAY;IAAE,MAAM;IAAyB,GAAG;IAAc,CAAC,CAAC;AAGjF,UAAO,MAAM,QAAQ,UACnB,KAAK,KAAK,OAAO;IACf;IACA,OAAO,EAAE;IACT,SAAS;IACT,OAAO,EAAE,KAAK;IACd;IACA,WAAW;IACX;IACD,EAAE,CACJ;WACM,OAAO;AACd,UAAO,IAAI,IAAI,YAAY;IAAE,MAAM;IAAyB,GAAG;IAAc,CAAC,CAAC;;;AAInF,QAAO;EAAE;EAAS;EAAa;;AAGjC,SAAgB,qBAGd,SACA,WACA,eACA,MACA,cACA;CACA,MAAMA,WAASC,OAAW,MAAM,EAAE,OAAO,UAAU,CAAC;CAEpD,MAAM,SAAS,IAAI,OACjB,WACA,OAAO,QAAQ;EACb,MAAM,EAAE,SAAS,IAAI;AACrB,SAAO,MAAM,2BACX,EAAE,gBAAgB,QAAQ,MAAM,sBAAsB,EAAE,EACxD,YAAY;GACV,MAAM,QACJ,MAAM,WACN,iBAAiB;IACf,MAAM,wBAAQ,IAAI,MAAM,gBAAgB;AACxC,mBAAe,MAAM;AACrB,UAAM;MACL,IAAI,KAAK,KAAK,QAAQ;GAC3B,MAAM,UAAU,KAAK,IAAI;AACzB,OAAI,CAAC,SAAS;IACZ,MAAM,wBAAQ,IAAI,MAAM,gBAAgB,IAAI,OAAO;AACnD,mBAAe,MAAM;AACrB,UAAM;;AAER,OAAI;AACF,UAAM,QAAQ,QAAQ,IAAI;IAE1B,MAAM,SAAS,MAAM,QAAQ,IAAI,IAAI;AACrC,QAAI,OAAO,OAAO,EAAE;AAClB,WAAM,QAAQ,YAAY,KAAK,OAAO,MAAM;KAC5C,MAAM,QAAQ,IAAI,MAAM,OAAO,MAAM,QAAQ;AAC7C,oBAAe,MAAM;AACrB,WAAM;;AAER,UAAM,QAAQ,YAAY,IAAI,CAAC,OAAO,QAAiB;KACrD,MAAM,wBAAQ,IAAI,MAAM,OAAO,IAAI,GAAG,GAAG,IAAI,KAAK,4BAA4B,MAAM;AACpF,cAAO,MAAM,MAAM,QAAQ;AAC3B,oBAAe,MAAM;MACrB;AACF,WAAO,OAAO,SAAS;YAChB,KAAK;AACZ,UAAM,QAAQ,YAAY,KAAK,IAAI,CAAC,OAAO,QAAiB;KAC1D,MAAM,wBAAQ,IAAI,MAAM,OAAO,IAAI,GAAG,GAAG,IAAI,KAAK,4BAA4B,MAAM;AACpF,cAAO,MAAM,MAAM,QAAQ;AAC3B,oBAAe,MAAM;MACrB;AACF,QAAI,eAAe,SAAS,IAAI,YAAY,gBAC1C,UAAO,MAAM,OAAO,IAAI,GAAG,GAAG,IAAI,KAAK,YAAY;AACrD,UAAM;aACE;AACR,UAAM,QAAQ,aAAa,IAAI,CAAC,OAAO,QAAiB;KACtD,MAAM,wBAAQ,IAAI,MAAM,OAAO,IAAI,GAAG,GAAG,IAAI,KAAK,6BAA6B,MAAM;AACrF,cAAO,MAAM,MAAM,QAAQ;AAC3B,oBAAe,MAAM;MACrB;AACF,QAAI,MAAO,cAAa,MAAM;;IAGnC;IAEH,cACD;AACD,QAAO,GAAG,aAAa,QAAQ,UAAU,KAAK,QAAQ,CAAC;AACvD,QAAO,GAAG,UAAU,QAAQ,OAAO,KAAK,QAAQ,CAAC;AAEjD,QAAO;;AAGT,SAAgB,sBAGd,SACA,gBACA,MACA,cACA;CACA,MAAM,UAAkC,EAAE;AAC1C,MAAK,MAAM,CAAC,WAAW,kBAAkB,OAAO,QAAQ,eAAe,CACrE,SAAQ,aAAa,qBACnB,SACA,WACA,eACA,MACA,aACD;AAEH,QAAO"}

package/dist/src/types.d.mts

@@ -0,0 +1,346 @@
+import { Context, TRPCMethods } from "./utils/trpc.mjs";
+import { BillingService } from "./modules/billing/billing.service.mjs";
+import { AIService } from "./modules/ai/ai.service.mjs";
+import { AuthService } from "./modules/auth/auth.service.mjs";
+import * as _$_trpc_server0 from "@trpc/server";
+import * as _$stripe from "stripe";
+import { Mastra } from "@mastra/core";
+
+//#region src/types.d.ts
+declare const createAuthTRPCRouter: <MastraInstance extends Mastra>(trpcMethods: TRPCMethods, authService: AuthService, aiService: AIService<MastraInstance>, billingService: BillingService) => _$_trpc_server0.TRPCBuiltRouter<{
+  ctx: Context;
+  meta: any;
+  errorShape: _$_trpc_server0.TRPCDefaultErrorShape;
+  transformer: any;
+}, _$_trpc_server0.TRPCDecorateCreateRouterOptions<{
+  auth: _$_trpc_server0.TRPCBuiltRouter<{
+    ctx: Context;
+    meta: any;
+    errorShape: _$_trpc_server0.TRPCDefaultErrorShape;
+    transformer: any;
+  }, _$_trpc_server0.TRPCDecorateCreateRouterOptions<{
+    getUserWaitlistCount: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: number;
+      meta: any;
+    }>;
+    createInvitationCode: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        name?: string | undefined;
+      };
+      output: {
+        id: string;
+        name: string | null;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+        code: string | null;
+        expiresAt: Date | null;
+      };
+      meta: any;
+    }>;
+    createAccountClaimCode: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        userId: string;
+        expiresInHours?: number | undefined;
+      };
+      output: {
+        id: string;
+        claimUserId: string | null;
+        code: string | null;
+        status: string;
+        expiresAt: Date | null;
+        claimedAt: Date | null;
+        claimedEmail: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+      };
+      meta: any;
+    }>;
+    listAccountClaims: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: {
+        id: string;
+        createdAt: Date;
+        updatedAt: Date | null;
+        expiresAt: Date | null;
+        status: string;
+        claimUserId: string | null;
+        claimedAt: Date | null;
+        claimedEmail: string | null;
+      }[];
+      meta: any;
+    }>;
+    generateAccountClaimMagicLink: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        claimId: string;
+        email?: string | undefined;
+      };
+      output: {
+        id: string;
+        email: string;
+        createdAt: Date;
+        userId: string;
+        expiresAt: Date | null;
+        claimId: string;
+        url: string;
+      };
+      meta: any;
+    }>;
+    listAccountClaimMagicLinks: _$_trpc_server0.TRPCQueryProcedure<{
+      input: {
+        claimId: string;
+      };
+      output: {
+        id: string;
+        email: string;
+        createdAt: Date;
+        userId: string;
+        expiresAt: Date | null;
+        claimId: string;
+        url: string;
+      }[];
+      meta: any;
+    }>;
+    getMyAccountClaimStatus: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: {
+        id: string;
+        claimUserId: string | null;
+        code: string | null;
+        status: string;
+        expiresAt: Date | null;
+        claimedAt: Date | null;
+        claimedEmail: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+      } | null;
+      meta: any;
+    }>;
+    setMyAccountClaimEmail: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        email: string;
+      };
+      output: {
+        status: boolean;
+      };
+      meta: any;
+    }>;
+    acceptMyAccountClaim: _$_trpc_server0.TRPCMutationProcedure<{
+      input: void;
+      output: {
+        status: boolean;
+      };
+      meta: any;
+    }>;
+    listWaitlist: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: {
+        id: string;
+        name: string | null;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+        code: string | null;
+        expiresAt: Date | null;
+      }[];
+      meta: any;
+    }>;
+    listAdminWaitlist: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: {
+        name: string | null;
+        id: string;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+      }[];
+      meta: any;
+    }>;
+    addToWaitlist: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        email: string;
+      };
+      output: {
+        name: string | null;
+        id: string;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+      };
+      meta: any;
+    }>;
+    inviteToWaitlist: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        email: string;
+        name?: string | undefined;
+      };
+      output: {
+        id: string;
+        name: string | null;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+        code: string | null;
+        expiresAt: Date | null;
+      };
+      meta: any;
+    }>;
+    inviteFromWaitlist: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        id: string;
+      };
+      output: {
+        name: string | null;
+        id: string;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+      };
+      meta: any;
+    }>;
+    removeFromWaitlist: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        id: string;
+      };
+      output: {
+        name: string | null;
+        id: string;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+      };
+      meta: any;
+    }>;
+    joinWaitlist: _$_trpc_server0.TRPCMutationProcedure<{
+      input: {
+        email: string;
+      };
+      output: {
+        name: string | null;
+        id: string;
+        email: string | null;
+        createdAt: Date;
+        updatedAt: Date | null;
+        status: string;
+      };
+      meta: any;
+    }>;
+    getOnboarding: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: number;
+      meta: any;
+    }>;
+    setOnboarding: _$_trpc_server0.TRPCMutationProcedure<{
+      input: number;
+      output: number;
+      meta: any;
+    }>;
+    getPreferences: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: Record<string, unknown>;
+      meta: any;
+    }>;
+    setPreferences: _$_trpc_server0.TRPCMutationProcedure<{
+      input: Record<string, unknown>;
+      output: Record<string, unknown>;
+      meta: any;
+    }>;
+    getOrganizationPreferences: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: Record<string, unknown>;
+      meta: any;
+    }>;
+    setOrganizationPreferences: _$_trpc_server0.TRPCMutationProcedure<{
+      input: Record<string, unknown>;
+      output: Record<string, unknown>;
+      meta: any;
+    }>;
+    getMetadata: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: Record<string, unknown>;
+      meta: any;
+    }>;
+    setMetadata: _$_trpc_server0.TRPCMutationProcedure<{
+      input: Record<string, unknown>;
+      output: Record<string, unknown>;
+      meta: any;
+    }>;
+    getFlags: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: string[];
+      meta: any;
+    }>;
+    getOrganizationFlags: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: string[];
+      meta: any;
+    }>;
+    setFlags: _$_trpc_server0.TRPCMutationProcedure<{
+      input: string[];
+      output: string[];
+      meta: any;
+    }>;
+    setOrganizationFlags: _$_trpc_server0.TRPCMutationProcedure<{
+      input: string[];
+      output: string[];
+      meta: any;
+    }>;
+    validateWaitlistCode: _$_trpc_server0.TRPCQueryProcedure<{
+      input: {
+        code: string;
+      };
+      output: {
+        status: string;
+      };
+      meta: any;
+    }>;
+  }>>;
+  ai: _$_trpc_server0.TRPCBuiltRouter<{
+    ctx: Context;
+    meta: any;
+    errorShape: _$_trpc_server0.TRPCDefaultErrorShape;
+    transformer: any;
+  }, _$_trpc_server0.TRPCDecorateCreateRouterOptions<{
+    getUserUsage: _$_trpc_server0.TRPCQueryProcedure<{
+      input: {
+        userId: string;
+      };
+      output: {
+        inputTokens: number | null;
+        outputTokens: number | null;
+        totalTokens: number | null;
+        cost: number | null;
+      };
+      meta: any;
+    }>;
+  }>>;
+  billing: _$_trpc_server0.TRPCBuiltRouter<{
+    ctx: Context;
+    meta: any;
+    errorShape: _$_trpc_server0.TRPCDefaultErrorShape;
+    transformer: any;
+  }, _$_trpc_server0.TRPCDecorateCreateRouterOptions<{
+    getActiveSubscription: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: any;
+      meta: any;
+    }>;
+    listInvoices: _$_trpc_server0.TRPCQueryProcedure<{
+      input: void;
+      output: _$stripe.Stripe.Invoice[];
+      meta: any;
+    }>;
+  }>>;
+}>>;
+type BackendTRPCRouter = ReturnType<typeof createAuthTRPCRouter>;
+//#endregion
+export { BackendTRPCRouter, createAuthTRPCRouter };
+//# sourceMappingURL=types.d.mts.map

package/dist/src/types.mjs

@@ -0,0 +1,13 @@
+import { createAITRPC } from "./modules/ai/ai.trpc.mjs";
+import { createAuthTRPC } from "./modules/auth/auth.trpc.mjs";
+import { createBillingTRPC } from "./modules/billing/billing.trpc.mjs";
+//#region src/types.ts
+const createAuthTRPCRouter = (trpcMethods, authService, aiService, billingService) => trpcMethods.router({
+	auth: createAuthTRPC(trpcMethods, authService),
+	ai: createAITRPC(trpcMethods, aiService),
+	billing: createBillingTRPC(trpcMethods, billingService)
+});
+//#endregion
+export { createAuthTRPCRouter };
+
+//# sourceMappingURL=types.mjs.map

package/dist/src/types.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.mjs","names":[],"sources":["../../src/types.ts"],"sourcesContent":["// Create a faux router that includes the auth router for use in the web-ui types\r\nimport type { Mastra } from \"@mastra/core\";\r\nimport type { AIService } from \"./modules/ai/ai.service\";\r\nimport { createAITRPC } from \"./modules/ai/ai.trpc\";\r\nimport type { AuthService } from \"./modules/auth/auth.service\";\r\nimport { createAuthTRPC } from \"./modules/auth/auth.trpc\";\r\nimport type { BillingService } from \"./modules/billing/billing.service\";\r\nimport { createBillingTRPC } from \"./modules/billing/billing.trpc\";\r\nimport type { TRPCMethods } from \"./utils/trpc\";\r\n\r\nexport const createAuthTRPCRouter = <MastraInstance extends Mastra>(\r\n  trpcMethods: TRPCMethods,\r\n  authService: AuthService,\r\n  aiService: AIService<MastraInstance>,\r\n  billingService: BillingService\r\n) =>\r\n  trpcMethods.router({\r\n    auth: createAuthTRPC(trpcMethods, authService),\r\n    ai: createAITRPC(trpcMethods, aiService),\r\n    billing: createBillingTRPC(trpcMethods, billingService),\r\n  });\r\n\r\nexport type BackendTRPCRouter = ReturnType<typeof createAuthTRPCRouter>;\r\n"],"mappings":";;;;AAUA,MAAa,wBACX,aACA,aACA,WACA,mBAEA,YAAY,OAAO;CACjB,MAAM,eAAe,aAAa,YAAY;CAC9C,IAAI,aAAa,aAAa,UAAU;CACxC,SAAS,kBAAkB,aAAa,eAAe;CACxD,CAAC"}

package/dist/src/utils/errors.d.mts

@@ -0,0 +1,62 @@
+import { ServerErrorLayer } from "../modules/base/base.types.mjs";
+import { TRPCError, TRPC_ERROR_CODE_KEY } from "@trpc/server";
+import { captureException } from "@sentry/node";
+
+//#region src/utils/errors.d.ts
+declare class ServerError extends Error {
+  readonly code: TRPC_ERROR_CODE_KEY;
+  readonly layer: ServerErrorLayer;
+  readonly layerName: string;
+  readonly clientMessage?: string;
+  readonly context?: Record<string, unknown>;
+  readonly boundaryStack?: string;
+  constructor({
+    code,
+    layer,
+    layerName,
+    message,
+    clientMessage,
+    cause,
+    context,
+    captureBoundary
+  }: {
+    code: TRPC_ERROR_CODE_KEY;
+    layer?: ServerErrorLayer;
+    layerName?: string;
+    message?: string;
+    clientMessage?: string;
+    cause?: unknown;
+    context?: Record<string, unknown>;
+    captureBoundary?: boolean;
+  });
+  is5xxError(): boolean;
+  getHTTPStatusCode(): number;
+  toTRPC(): TRPCError;
+  toJSON(): {
+    code: "PARSE_ERROR" | "BAD_REQUEST" | "INTERNAL_SERVER_ERROR" | "NOT_IMPLEMENTED" | "BAD_GATEWAY" | "SERVICE_UNAVAILABLE" | "GATEWAY_TIMEOUT" | "UNAUTHORIZED" | "PAYMENT_REQUIRED" | "FORBIDDEN" | "NOT_FOUND" | "METHOD_NOT_SUPPORTED" | "TIMEOUT" | "CONFLICT" | "PRECONDITION_FAILED" | "PAYLOAD_TOO_LARGE" | "UNSUPPORTED_MEDIA_TYPE" | "UNPROCESSABLE_CONTENT" | "TOO_MANY_REQUESTS" | "CLIENT_CLOSED_REQUEST";
+    layer: ServerErrorLayer;
+    layerName: string;
+    message: string;
+    context: Record<string, unknown> | undefined;
+    stack: string | undefined;
+    boundaryStack: string | undefined;
+    cause: unknown;
+  };
+  static fromUnknown(code: TRPC_ERROR_CODE_KEY, cause: unknown, opts?: {
+    layer?: ServerErrorLayer;
+    layerName?: string;
+    context?: Record<string, unknown>;
+  }): ServerError;
+}
+type ErrorReporter = {
+  captureException: (err: Parameters<typeof captureException>[0], hint?: Parameters<typeof captureException>[1]) => string;
+};
+declare global {
+  var m5ErrorReporter: ErrorReporter | undefined;
+}
+declare function getErrorReporter(): ErrorReporter | undefined;
+declare function setErrorReporter(reporter: ErrorReporter): void;
+declare function reportError(err: ServerError | Error | unknown, hint?: Parameters<typeof captureException>[1]): string | undefined;
+//#endregion
+export { ErrorReporter, ServerError, type ServerErrorLayer, getErrorReporter, reportError, setErrorReporter };
+//# sourceMappingURL=errors.d.mts.map