@automateinc/fleet-types 1.0.104 → 1.0.105

package/bin/fleet-types.mjs

@@ -8,6 +8,37 @@ import path from "node:path";
 import { loadEnvFile } from "node:process";
 import { fileURLToPath } from "node:url";
 
+const SERIALIZED_OUTPUT_TYPES = `type FleetIsAny<T> = 0 extends 1 & T ? true : false;
+type FleetSimplify<T> = { [TKey in keyof T]: T[TKey] } & {};
+type FleetSerializeObject<T extends object> = {
+\t[TKey in keyof T as FleetIsAny<T[TKey]> extends true
+\t\t? TKey
+\t\t: [T[TKey]] extends [undefined]
+\t\t\t? never
+\t\t\t: T[TKey] extends (...args: never[]) => unknown
+\t\t\t\t? never
+\t\t\t\t: TKey]: FleetSerializeOutput<T[TKey]>;
+};
+type FleetSerializeOutput<T> = FleetIsAny<T> extends true
+\t? T
+\t: T extends (...args: never[]) => unknown
+\t\t? never
+\t\t: T extends Date
+\t\t\t? string
+\t\t\t: T extends readonly (infer TItem)[]
+\t\t\t\t? FleetSerializeOutput<TItem>[]
+\t\t\t\t: T extends { toJSON(): infer TJSON }
+\t\t\t\t\t? TJSON extends object
+\t\t\t\t\t\t? FleetSimplify<
+\t\t\t\t\t\t\t\tOmit<FleetSerializeObject<Omit<T, "toJSON">>, keyof TJSON> &
+\t\t\t\t\t\t\t\t\tFleetSerializeOutput<TJSON>
+\t\t\t\t\t\t\t>
+\t\t\t\t\t\t: FleetSerializeOutput<TJSON>
+\t\t\t\t\t: T extends object
+\t\t\t\t\t\t? FleetSerializeObject<T>
+\t\t\t\t\t\t: T;
+type FleetProcedureOutput<T> = { toJSON(): FleetSerializeOutput<T> };`;
+
 const command = process.argv[2];
 
 if (!command || command === "--help" || command === "-h") {
@@ -123,7 +154,9 @@ async function generateTypes() {
 
 	const emittedPath = routerSourcePath.replace(/\.ts$/, ".d.ts");
 	const sourceFile = ts.createSourceFile(emittedPath, emittedDeclaration, ts.ScriptTarget.Latest, true);
-	const statements = sourceFile.statements.filter(
+	const sanitizedSourceFile = sanitizeBackendTypes(ts, sourceFile);
+	const transformedSourceFile = transformProcedureOutputs(ts, sanitizedSourceFile);
+	const statements = transformedSourceFile.statements.filter(
 		statement =>
 			ts.isImportDeclaration(statement) ||
 			ts.isImportEqualsDeclaration(statement) ||
@@ -140,14 +173,14 @@ async function generateTypes() {
 
 	const printer = ts.createPrinter({ newLine: ts.NewLineKind.LineFeed });
 	const appRouterDeclaration = statements
-		.map(statement => printer.printNode(ts.EmitHint.Unspecified, statement, sourceFile))
+		.map(statement => printer.printNode(ts.EmitHint.Unspecified, statement, transformedSourceFile))
 		.join("\n")
 		.replace(/^(?: {4})+/gm, indentation => "\t".repeat(indentation.length / 4));
 
 	await mkdir(path.dirname(outputPath), { recursive: true });
 	await writeFile(
 		outputPath,
-		`// Generated by \`npx @automateinc/fleet-types generate\`. Do not edit manually.\n${appRouterDeclaration}\n`,
+		`// Generated by \`npx @automateinc/fleet-types generate\`. Do not edit manually.\n${SERIALIZED_OUTPUT_TYPES}\n${appRouterDeclaration}\n`,
 	);
 
 	if (existsSync(biomePath)) {
@@ -166,6 +199,100 @@ async function generateTypes() {
 	console.log(`Generated ${path.relative(callerRoot, outputPath)} from ${routerSourcePath}`);
 }
 
+function transformProcedureOutputs(ts, sourceFile) {
+	const procedureTypes = new Set(["TRPCMutationProcedure", "TRPCQueryProcedure", "TRPCSubscriptionProcedure"]);
+	const transformation = ts.transform(sourceFile, [
+		context => {
+			const visit = node => {
+				if (
+					ts.isImportTypeNode(node) &&
+					ts.isIdentifier(node.qualifier) &&
+					procedureTypes.has(node.qualifier.text) &&
+					node.typeArguments?.length
+				) {
+					const [procedureDefinition, ...remainingTypeArguments] = node.typeArguments;
+
+					if (ts.isTypeLiteralNode(procedureDefinition)) {
+						const members = procedureDefinition.members.map(member => {
+							if (
+								ts.isPropertySignature(member) &&
+								member.type &&
+								ts.isIdentifier(member.name) &&
+								member.name.text === "output"
+							) {
+								return ts.factory.updatePropertySignature(
+									member,
+									member.modifiers,
+									member.name,
+									member.questionToken,
+									ts.factory.createTypeReferenceNode("FleetProcedureOutput", [member.type]),
+								);
+							}
+
+							return member;
+						});
+						const updatedDefinition = ts.factory.updateTypeLiteralNode(procedureDefinition, members);
+
+						return ts.factory.updateImportTypeNode(
+							node,
+							node.argument,
+							node.attributes,
+							node.qualifier,
+							[updatedDefinition, ...remainingTypeArguments],
+							node.isTypeOf,
+						);
+					}
+				}
+
+				return ts.visitEachChild(node, visit, context);
+			};
+
+			return rootNode => ts.visitNode(rootNode, visit);
+		},
+	]);
+	const transformedSourceFile = transformation.transformed[0];
+
+	transformation.dispose();
+
+	return transformedSourceFile;
+}
+
+function sanitizeBackendTypes(ts, sourceFile) {
+	const transformation = ts.transform(sourceFile, [
+		context => {
+			const visit = node => {
+				if (ts.isTypeReferenceNode(node)) {
+					let typeName = node.typeName;
+
+					while (ts.isQualifiedName(typeName)) typeName = typeName.left;
+
+					if (ts.isIdentifier(typeName) && typeName.text === "PrismaJson") {
+						return ts.factory.createKeywordTypeNode(ts.SyntaxKind.UnknownKeyword);
+					}
+				}
+
+				if (
+					ts.isImportTypeNode(node) &&
+					ts.isLiteralTypeNode(node.argument) &&
+					ts.isStringLiteral(node.argument.literal) &&
+					node.argument.literal.text.startsWith("@/")
+				) {
+					return ts.factory.createKeywordTypeNode(ts.SyntaxKind.UnknownKeyword);
+				}
+
+				return ts.visitEachChild(node, visit, context);
+			};
+
+			return rootNode => ts.visitNode(rootNode, visit);
+		},
+	]);
+	const transformedSourceFile = transformation.transformed[0];
+
+	transformation.dispose();
+
+	return transformedSourceFile;
+}
+
 function formatTypeScriptDiagnostics(ts, diagnostics, currentDirectory) {
 	return ts.formatDiagnosticsWithColorAndContext(diagnostics, {
 		getCanonicalFileName: fileName => fileName,

package/package.json

@@ -57,5 +57,5 @@
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "types": "dist/types/index.d.ts",
-  "version": "1.0.104"
+  "version": "1.0.105"
 }

package/dist/types/trpc.d.ts

@@ -1,61 +0,0 @@
-// Generated by `npx @automateinc/fleet-types generate`. Do not edit manually.
-export declare const appRouter: import("@trpc/server").TRPCBuiltRouter<
-	{
-		ctx: {
-			client: "FLEET_WEB" | "FLEET_MOBILE" | "FLEET_MOBILE_USER" | undefined;
-			regionId: string | undefined;
-			request: import("express").Request<
-				import("express-serve-static-core").ParamsDictionary,
-				any,
-				any,
-				import("qs").ParsedQs,
-				Record<string, any>
-			>;
-			token: string | undefined;
-		};
-		meta: object;
-		errorShape: import("@trpc/server").TRPCDefaultErrorShape;
-		transformer: false;
-	},
-	import("@trpc/server").TRPCDecorateCreateRouterOptions<{
-		auth: import("@trpc/server").TRPCBuiltRouter<
-			{
-				ctx: {
-					client: "FLEET_WEB" | "FLEET_MOBILE" | "FLEET_MOBILE_USER" | undefined;
-					regionId: string | undefined;
-					request: import("express").Request<
-						import("express-serve-static-core").ParamsDictionary,
-						any,
-						any,
-						import("qs").ParsedQs,
-						Record<string, any>
-					>;
-					token: string | undefined;
-				};
-				meta: object;
-				errorShape: import("@trpc/server").TRPCDefaultErrorShape;
-				transformer: false;
-			},
-			import("@trpc/server").TRPCDecorateCreateRouterOptions<{
-				login: import("@trpc/server").TRPCMutationProcedure<{
-					input: {
-						email: string;
-						password: string;
-					};
-					output: {
-						token: string;
-					};
-					meta: object;
-				}>;
-				refreshToken: import("@trpc/server").TRPCMutationProcedure<{
-					input: void;
-					output: {
-						token: string;
-					};
-					meta: object;
-				}>;
-			}>
-		>;
-	}>
->;
-export type AppRouter = typeof appRouter;