protoc-gen-pothos 0.1.0

package/CHANGELOG.md

@@ -0,0 +1,14 @@
+# protoc-gen-pothos
+
+## 0.1.0
+
+### Minor Changes
+
+- [#200](https://github.com/proto-graphql/proto-graphql-js/pull/200) [`515eabd`](https://github.com/proto-graphql/proto-graphql-js/commit/515eabd2f39baa0a99ae057b1b30a4ccc4149f66) Thanks [@izumin5210](https://github.com/izumin5210)! - support Pothos GraphQL
+
+### Patch Changes
+
+- Updated dependencies [[`515eabd`](https://github.com/proto-graphql/proto-graphql-js/commit/515eabd2f39baa0a99ae057b1b30a4ccc4149f66)]:
+  - @proto-graphql/codegen-core@0.1.0
+  - @proto-graphql/proto-descriptors@0.1.0
+  - @proto-graphql/protoc-plugin-helpers@0.1.0

package/README.md

@@ -0,0 +1,5 @@
+# protoc-gen-pothos
+
+Build GraphQL schema from Protobuf and [Pothos GraphQL](https://pothos-graphql.dev/).
+
+- repository: https://github.com/proto-graphql/proto-graphql-js

package/bin/protoc-gen-pothos

@@ -0,0 +1,3 @@
+#! /usr/bin/env node
+
+require("../lib/protoc-gen-pothos");

package/lib/dslgen/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./printers"), exports);

package/lib/dslgen/printers/enumType.js

@@ -0,0 +1,37 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createEnumTypeDslStmt = void 0;
+const typescript_1 = __importDefault(require("typescript"));
+const util_1 = require("./util");
+/**
+ * @example
+ * ```ts
+ * export cosnt Hello = builder.enumType("Hello", {
+ *   // ...
+ * })
+ * ```
+ */
+function createEnumTypeDslStmt(type) {
+    return (0, util_1.createDslExportConstStmt)(type.pothosRefObjectName, (0, util_1.createBuilderCallExpr)("enumType", [
+        typescript_1.default.factory.createStringLiteral(type.typeName),
+        typescript_1.default.factory.createObjectLiteralExpression([
+            (0, util_1.createDescriptionPropertyAssignment)(type),
+            typescript_1.default.factory.createPropertyAssignment("values", typescript_1.default.factory.createAsExpression(typescript_1.default.factory.createObjectLiteralExpression(type.values
+                .filter((v) => !v.isIgnored() && !v.isUnespecified())
+                .map((ev) => typescript_1.default.factory.createPropertyAssignment(ev.name, createEnumValueExpr(ev))), true // multiline
+            ), typescript_1.default.factory.createTypeReferenceNode("const"))),
+        ].filter((0, util_1.onlyNonNull)()), true),
+    ]));
+}
+exports.createEnumTypeDslStmt = createEnumTypeDslStmt;
+function createEnumValueExpr(ev) {
+    return typescript_1.default.factory.createObjectLiteralExpression([
+        (0, util_1.createDescriptionPropertyAssignment)(ev),
+        (0, util_1.createDeprecationPropertyAssignment)(ev),
+        typescript_1.default.factory.createPropertyAssignment("value", typescript_1.default.factory.createNumericLiteral(ev.number)),
+    ].filter((0, util_1.onlyNonNull)()), true // multiline
+    );
+}

package/lib/dslgen/printers/field.js

@@ -0,0 +1,111 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createNoopFieldDefinitionExpr = exports.createFieldDefinitionExpr = void 0;
+const codegen_core_1 = require("@proto-graphql/codegen-core");
+const typescript_1 = __importDefault(require("typescript"));
+const enumFieldResolver_1 = require("./fieldResolver/enumFieldResolver");
+const nonNullResolver_1 = require("./fieldResolver/nonNullResolver");
+const oneofUnionResolver_1 = require("./fieldResolver/oneofUnionResolver");
+const util_1 = require("./util");
+/**
+ * @example
+ * ```ts
+ * t.expose("name", {
+ *   type: "Boolean",
+ *   nullable: true,
+ *   description: "...",
+ * })
+ * ```
+ * ```ts
+ * t.field({
+ *   type: "Boolean",
+ *   nullable: true,
+ *   description: "...",
+ *   resolve() {
+ *     return true
+ *   }
+ * })
+ * ```
+ */
+function createFieldDefinitionExpr(field) {
+    let typeExpr = field.type instanceof codegen_core_1.ScalarType
+        ? typescript_1.default.factory.createStringLiteral(field.type.typeName)
+        : typescript_1.default.factory.createIdentifier(field.type.pothosRefObjectName);
+    const isInput = field instanceof codegen_core_1.InputObjectField;
+    const nullableToken = typescript_1.default.factory.createToken(isInput ? typescript_1.default.SyntaxKind.FalseKeyword : typescript_1.default.SyntaxKind.TrueKeyword);
+    const nonNullableToken = typescript_1.default.factory.createToken(isInput ? typescript_1.default.SyntaxKind.TrueKeyword : typescript_1.default.SyntaxKind.FalseKeyword);
+    let nullableExpr = field.isNullable() ? nullableToken : nonNullableToken;
+    // let defaultValueExpr: ts.Expression | undefined;
+    if (field.isList()) {
+        typeExpr = typescript_1.default.factory.createArrayLiteralExpression([typeExpr], false);
+        // if (isInput) {
+        //   defaultValueExpr = ts.factory.createArrayLiteralExpression([], false);
+        // }
+        nullableExpr = typescript_1.default.factory.createObjectLiteralExpression([
+            typescript_1.default.factory.createPropertyAssignment("list", nullableExpr),
+            typescript_1.default.factory.createPropertyAssignment("items", nonNullableToken),
+        ], false);
+    }
+    let createResolveStmts;
+    if (!isInput) {
+        if (field.type instanceof codegen_core_1.ObjectType && !field.isNullable()) {
+            createResolveStmts = (sourceExpr) => (0, nonNullResolver_1.createNonNullResolverStmts)(typescript_1.default.factory.createPropertyAccessExpression(sourceExpr, field.protoJsName));
+        }
+        if (field.type instanceof codegen_core_1.EnumType && field instanceof codegen_core_1.ObjectField) {
+            createResolveStmts = (sourceExpr) => (0, enumFieldResolver_1.createEnumResolverStmts)(typescript_1.default.factory.createPropertyAccessExpression(sourceExpr, field.protoJsName), field);
+        }
+        if (field instanceof codegen_core_1.ObjectOneofField) {
+            createResolveStmts = (sourceExpr) => (0, oneofUnionResolver_1.createOneofUnionResolverStmts)(sourceExpr, field);
+        }
+        if (field.type instanceof codegen_core_1.SquashedOneofUnionType && field instanceof codegen_core_1.ObjectField) {
+            createResolveStmts = (sourceExpr) => (0, oneofUnionResolver_1.createOneofUnionResolverStmts)(typescript_1.default.factory.createPropertyAccessExpression(sourceExpr, field.protoJsName), field);
+        }
+    }
+    const fieldOptionsExpr = typescript_1.default.factory.createObjectLiteralExpression([
+        typescript_1.default.factory.createPropertyAssignment("type", typeExpr),
+        typescript_1.default.factory.createPropertyAssignment(isInput ? "required" : "nullable", nullableExpr),
+        (0, util_1.createDescriptionPropertyAssignment)(field),
+        (0, util_1.createDeprecationPropertyAssignment)(field),
+        // defaultValueExpr && ts.factory.createPropertyAssignment("defaultValue", defaultValueExpr),
+        createResolveStmts && typescript_1.default.factory.createPropertyAssignment("resolve", createResolverExpr(createResolveStmts)),
+    ].filter((0, util_1.onlyNonNull)()), true);
+    const shouldUseFieldFunc = field instanceof codegen_core_1.InputObjectField || createResolveStmts != null;
+    return typescript_1.default.factory.createCallExpression(typescript_1.default.factory.createPropertyAccessExpression(typescript_1.default.factory.createIdentifier("t"), typescript_1.default.factory.createIdentifier(shouldUseFieldFunc ? "field" : "expose")), undefined, shouldUseFieldFunc ? [fieldOptionsExpr] : [typescript_1.default.factory.createStringLiteral(field.protoJsName), fieldOptionsExpr]);
+}
+exports.createFieldDefinitionExpr = createFieldDefinitionExpr;
+/**
+ * @example
+ * ```ts
+ * t.field( {
+ *   type: "Boolean",
+ *   nullable: true,
+ *   description: "noop field",
+ *   resolve() {
+ *     return true
+ *   }
+ * })
+ * ```
+ */
+function createNoopFieldDefinitionExpr(opts) {
+    return typescript_1.default.factory.createCallExpression(typescript_1.default.factory.createPropertyAccessExpression(typescript_1.default.factory.createIdentifier("t"), typescript_1.default.factory.createIdentifier("field")), undefined, [
+        typescript_1.default.factory.createObjectLiteralExpression([
+            typescript_1.default.factory.createPropertyAssignment("type", typescript_1.default.factory.createStringLiteral("Boolean")),
+            opts.input
+                ? typescript_1.default.factory.createPropertyAssignment("required", typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.FalseKeyword))
+                : typescript_1.default.factory.createPropertyAssignment("nullable", typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.TrueKeyword)),
+            typescript_1.default.factory.createPropertyAssignment("description", typescript_1.default.factory.createStringLiteral("noop field")),
+            opts.input
+                ? null
+                : typescript_1.default.factory.createMethodDeclaration(undefined, undefined, undefined, "resolve", undefined, undefined, [], undefined, typescript_1.default.factory.createBlock([
+                    typescript_1.default.factory.createReturnStatement(typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.TrueKeyword)),
+                ])),
+        ].filter((0, util_1.onlyNonNull)()), true),
+    ]);
+}
+exports.createNoopFieldDefinitionExpr = createNoopFieldDefinitionExpr;
+function createResolverExpr(createStmts) {
+    return typescript_1.default.factory.createArrowFunction(undefined, undefined, [typescript_1.default.factory.createParameterDeclaration(undefined, undefined, undefined, typescript_1.default.factory.createIdentifier("source"))], undefined, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.EqualsGreaterThanToken), typescript_1.default.factory.createBlock(createStmts(typescript_1.default.factory.createIdentifier("source")), true));
+}

package/lib/dslgen/printers/fieldResolver/enumFieldResolver.js

@@ -0,0 +1,68 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createEnumResolverStmts = void 0;
+const typescript_1 = __importDefault(require("typescript"));
+const util_1 = require("../util");
+/**
+ * @example nullable
+ * ```ts
+ * if (!root.myEnum || root.myEnum === _$enums.myEnum.MY_ENUM_UNSPECIFIED) {
+ *   return null
+ * }
+ * return root.myEnum
+ * ```
+ * @example notNull
+ * ```ts
+ * if (!root.myEnum || root.myEnum === _$enums.myEnum.MY_ENUM_UNSPECIFIED) {
+ *   throw new Error("Message.field is required field. but got null or unspecified.")
+ * }
+ * return root.myEnum
+ * ```
+ */
+function createEnumResolverStmts(valueExpr, field) {
+    let whenNullStmt = field.isNullable() && !field.isList()
+        ? typescript_1.default.factory.createReturnStatement(typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.NullKeyword))
+        : typescript_1.default.factory.createThrowStatement(typescript_1.default.factory.createNewExpression(typescript_1.default.factory.createIdentifier("Error"), undefined, [
+            typescript_1.default.factory.createStringLiteral(`${field.name} is required field. But got unspecified.`),
+        ]));
+    whenNullStmt = typescript_1.default.factory.createBlock([whenNullStmt], true // multiline
+    );
+    if (field.isList()) {
+        const guardStmts = createGuardStmts(typescript_1.default.factory.createIdentifier("item"), whenNullStmt, field.type.unspecifiedValue, field.type.valuesWithIgnored);
+        if (guardStmts.length === 0) {
+            return [typescript_1.default.factory.createReturnStatement(valueExpr)];
+        }
+        return [
+            typescript_1.default.factory.createReturnStatement(createMapExpr(valueExpr, (itemExpr) => [...guardStmts, typescript_1.default.factory.createReturnStatement(itemExpr)])),
+        ];
+    }
+    return [
+        ...createGuardStmts(valueExpr, whenNullStmt, field.type.unspecifiedValue, field.type.valuesWithIgnored),
+        typescript_1.default.factory.createReturnStatement(valueExpr),
+    ];
+}
+exports.createEnumResolverStmts = createEnumResolverStmts;
+function createGuardStmts(valueExpr, thenStmt, unspecifiedValue, ignoredValues) {
+    return [
+        unspecifiedValue
+            ? typescript_1.default.factory.createIfStatement(typescript_1.default.factory.createBinaryExpression(valueExpr, typescript_1.default.SyntaxKind.EqualsEqualsEqualsToken, (0, util_1.createFullNameExpr)(unspecifiedValue.fullName)), thenStmt)
+            : null,
+        ...ignoredValues.map((ev) => ev.isIgnored()
+            ? typescript_1.default.factory.createIfStatement(typescript_1.default.factory.createBinaryExpression(valueExpr, typescript_1.default.SyntaxKind.EqualsEqualsEqualsToken, (0, util_1.createFullNameExpr)(ev.fullName)), typescript_1.default.factory.createBlock([
+                typescript_1.default.factory.createThrowStatement(typescript_1.default.factory.createNewExpression(typescript_1.default.factory.createIdentifier("Error"), undefined, [
+                    typescript_1.default.factory.createStringLiteral(`${ev.name} is ignored in GraphQL schema`),
+                ])),
+            ], true // multiline
+            ))
+            : null),
+    ].filter((0, util_1.onlyNonNull)());
+}
+function createMapExpr(listExpr, blockFn) {
+    return typescript_1.default.factory.createCallExpression(typescript_1.default.factory.createPropertyAccessExpression(listExpr, "map"), undefined, [
+        typescript_1.default.factory.createArrowFunction(undefined, undefined, [typescript_1.default.factory.createParameterDeclaration(undefined, undefined, undefined, "item", undefined, undefined, undefined)], undefined, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.EqualsGreaterThanToken), typescript_1.default.factory.createBlock(blockFn(typescript_1.default.factory.createIdentifier("item")), true // multiline
+        )),
+    ]);
+}

package/lib/dslgen/printers/fieldResolver/nonNullResolver.js

@@ -0,0 +1,11 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createNonNullResolverStmts = void 0;
+const typescript_1 = __importDefault(require("typescript"));
+function createNonNullResolverStmts(valueExpr) {
+    return [typescript_1.default.factory.createReturnStatement(typescript_1.default.factory.createNonNullExpression(valueExpr))];
+}
+exports.createNonNullResolverStmts = createNonNullResolverStmts;

package/lib/dslgen/printers/fieldResolver/oneofUnionResolver.js

@@ -0,0 +1,61 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createOneofUnionResolverStmts = void 0;
+const codegen_core_1 = require("@proto-graphql/codegen-core");
+const typescript_1 = __importDefault(require("typescript"));
+const util_1 = require("../util");
+/**
+ * @example nullable
+ * ```ts
+ * const value = source.v1 ?? source.v2 ?? source.v3;
+ * if (value != null) {
+ *   throw new Error("...");
+ * }
+ * return value
+ * ```
+ */
+function createOneofUnionResolverStmts(sourceExpr, field) {
+    const createResolverStmts = (sourceExpr, field, { nullable }) => {
+        const createFieldExpr = (memberField) => {
+            if (field instanceof codegen_core_1.ObjectOneofField) {
+                return typescript_1.default.factory.createPropertyAccessExpression(sourceExpr, memberField.protoJsName);
+            }
+            return typescript_1.default.factory.createPropertyAccessChain(sourceExpr, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.QuestionDotToken), memberField.protoJsName);
+        };
+        const valueExpr = typescript_1.default.factory.createIdentifier("value");
+        return [
+            typescript_1.default.factory.createVariableStatement(undefined, typescript_1.default.factory.createVariableDeclarationList([
+                typescript_1.default.factory.createVariableDeclaration("value", undefined, undefined, 
+                // obj.field1 ?? obj.field2 ?? obj.field3 ?? ...
+                field.type.fields.reduceRight((expr, field) => {
+                    if (expr == null) {
+                        return createFieldExpr(field);
+                    }
+                    return typescript_1.default.factory.createBinaryExpression(createFieldExpr(field), typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.QuestionQuestionToken), expr);
+                }, null)),
+            ], typescript_1.default.NodeFlags.Const)),
+            nullable
+                ? undefined
+                : typescript_1.default.factory.createIfStatement(typescript_1.default.factory.createBinaryExpression(valueExpr, typescript_1.default.SyntaxKind.EqualsEqualsToken, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.NullKeyword)), typescript_1.default.factory.createBlock([
+                    typescript_1.default.factory.createThrowStatement(typescript_1.default.factory.createNewExpression(typescript_1.default.factory.createIdentifier("Error"), undefined, [
+                        typescript_1.default.factory.createStringLiteral(`${field.name} should not be null`),
+                    ])),
+                ], true)),
+            typescript_1.default.factory.createReturnStatement(valueExpr),
+        ].filter((0, util_1.onlyNonNull)());
+    };
+    if (!(field instanceof codegen_core_1.ObjectField && field.isList())) {
+        return createResolverStmts(sourceExpr, field, { nullable: field.isNullable() });
+    }
+    return [
+        typescript_1.default.factory.createReturnStatement(typescript_1.default.factory.createCallExpression(typescript_1.default.factory.createPropertyAccessExpression(sourceExpr, "map"), undefined, [
+            typescript_1.default.factory.createArrowFunction(undefined, undefined, [
+                typescript_1.default.factory.createParameterDeclaration(undefined, undefined, undefined, "item", undefined, undefined, undefined),
+            ], undefined, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.EqualsGreaterThanToken), typescript_1.default.factory.createBlock(createResolverStmts(typescript_1.default.factory.createIdentifier("item"), field, { nullable: false }), true)),
+        ])),
+    ];
+}
+exports.createOneofUnionResolverStmts = createOneofUnionResolverStmts;

package/lib/dslgen/printers/index.js

@@ -0,0 +1,62 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTypeDslStmts = exports.createReExportStmts = exports.createImportDecls = void 0;
+const codegen_core_1 = require("@proto-graphql/codegen-core");
+const typescript_1 = __importDefault(require("typescript"));
+const enumType_1 = require("./enumType");
+const inputObjectType_1 = require("./inputObjectType");
+const objectType_1 = require("./objectType");
+const oneofUnionType_1 = require("./oneofUnionType");
+const util_1 = require("./util");
+function createImportDecls(types) {
+    return types
+        .flatMap((t) => t.importModules)
+        .filter((0, util_1.onlyUnique)((m) => JSON.stringify([m.alias, m.module])))
+        .sort(({ module: a }, { module: b }) => {
+        const pat = /^\.+\//;
+        const [aIsRel, bIsRel] = [a.match(pat), b.match(pat)];
+        if (aIsRel && !bIsRel)
+            return 1;
+        if (!aIsRel && bIsRel)
+            return -1;
+        if (a < b)
+            return -1;
+        if (a > b)
+            return 1;
+        return 0;
+    })
+        .map((m) => (0, util_1.createImportDecl)(m));
+}
+exports.createImportDecls = createImportDecls;
+function createReExportStmts(types) {
+    return types
+        .flatMap((t) => t.exportTypes)
+        .sort()
+        .filter((0, util_1.onlyUnique)(({ type }) => (0, util_1.fullNameString)(type)))
+        .map(({ name, type }) => typescript_1.default.factory.createTypeAliasDeclaration(undefined, [typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.ExportKeyword)], name, undefined, typescript_1.default.factory.createTypeReferenceNode((0, util_1.createQualifiedName)(type))));
+}
+exports.createReExportStmts = createReExportStmts;
+function createTypeDslStmts(types) {
+    return types
+        .flatMap((type) => {
+        if (type instanceof codegen_core_1.ObjectType) {
+            return (0, objectType_1.createObjectTypeDslStmts)(type);
+        }
+        if (type instanceof codegen_core_1.InputObjectType) {
+            return (0, inputObjectType_1.createInputObjectTypeDslStmts)(type);
+        }
+        if (type instanceof codegen_core_1.EnumType) {
+            return [(0, enumType_1.createEnumTypeDslStmt)(type)];
+        }
+        if (type instanceof codegen_core_1.OneofUnionType || type instanceof codegen_core_1.SquashedOneofUnionType) {
+            return [(0, oneofUnionType_1.createOneofUnionTypeDslStmt)(type)];
+        }
+        const _exhaustiveCheck = type;
+        throw "unreachable";
+    })
+        .filter((0, util_1.onlyNonNull)());
+}
+exports.createTypeDslStmts = createTypeDslStmts;

package/lib/dslgen/printers/inputObjectType.js

@@ -0,0 +1,79 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createInputObjectTypeDslStmts = void 0;
+const codegen_core_1 = require("@proto-graphql/codegen-core");
+const typescript_1 = __importDefault(require("typescript"));
+const field_1 = require("./field");
+const util_1 = require("./util");
+/**
+ * @example
+ * ```ts
+ * export const HelloInput = builder.inputRef<Omit<_$hello$hello_pb.Hello, "$type">>("HelloInput");
+ * HelloInput.implement({
+ *   description: "...",
+ *   fields: (t) => ({
+ *     // ...
+ *   }),
+ * })
+ * ```
+ */
+function createInputObjectTypeDslStmts(type) {
+    return [
+        createInputObjectTypeShapeDecl(type),
+        (0, util_1.createDslExportConstStmt)(type.pothosRefObjectName, typescript_1.default.factory.createCallExpression((0, util_1.createBuilderPropExpr)("inputRef"), [typescript_1.default.factory.createTypeReferenceNode(createInputObjectTypeShapeIdent(type))], [typescript_1.default.factory.createStringLiteral(type.typeName)])),
+        typescript_1.default.factory.createExpressionStatement(typescript_1.default.factory.createCallExpression(typescript_1.default.factory.createPropertyAccessExpression(typescript_1.default.factory.createIdentifier(type.pothosRefObjectName), "implement"), undefined, [
+            typescript_1.default.factory.createObjectLiteralExpression([
+                (0, util_1.createDescriptionPropertyAssignment)(type),
+                typescript_1.default.factory.createPropertyAssignment("fields", createInputObjectTypeFieldsMethodExpr(type)),
+                // ts.factory.createPropertyAssignment("extensions", createExtensionsObjectLiteralExpr(type)),
+            ].filter((0, util_1.onlyNonNull)()), true),
+        ])),
+    ];
+}
+exports.createInputObjectTypeDslStmts = createInputObjectTypeDslStmts;
+/**
+ * @example
+ * ```ts
+ * definition(t) {
+ *   // ...
+ * }
+ * ```
+ */
+function createInputObjectTypeFieldsMethodExpr(type) {
+    return typescript_1.default.factory.createArrowFunction(undefined, undefined, [typescript_1.default.factory.createParameterDeclaration(undefined, undefined, undefined, "t", undefined, undefined, undefined)], undefined, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.EqualsGreaterThanToken), typescript_1.default.factory.createParenthesizedExpression(typescript_1.default.factory.createObjectLiteralExpression(type.fields.length > 0
+        ? type.fields.map((f) => typescript_1.default.factory.createPropertyAssignment(f.name, (0, field_1.createFieldDefinitionExpr)(f)))
+        : [typescript_1.default.factory.createPropertyAssignment("_", (0, field_1.createNoopFieldDefinitionExpr)({ input: true }))], true)));
+}
+/**
+ * @example
+ * ```ts
+ * {
+ *   message?: _$hello$hello_pb$Hello["message"] | null,
+ *   // ...
+ * }
+ * ```
+ */
+function createInputObjectTypeShapeDecl(type) {
+    return typescript_1.default.factory.createTypeAliasDeclaration(undefined, [typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.ExportKeyword)], createInputObjectTypeShapeIdent(type), undefined, typescript_1.default.factory.createTypeLiteralNode(type.fields.map((field) => {
+        let typeNode = typescript_1.default.factory.createIndexedAccessTypeNode(typescript_1.default.factory.createTypeReferenceNode((0, util_1.createQualifiedName)(type.protoTypeFullName)), typescript_1.default.factory.createLiteralTypeNode(typescript_1.default.factory.createStringLiteral(field.protoJsName)));
+        if (field.type instanceof codegen_core_1.InputObjectType) {
+            typeNode = typescript_1.default.factory.createTypeReferenceNode(createInputObjectTypeShapeIdent(field.type));
+            if (field.isList()) {
+                typeNode = typescript_1.default.factory.createTypeReferenceNode("Array", [typeNode]);
+            }
+        }
+        if (field.isNullable()) {
+            typeNode = typescript_1.default.factory.createUnionTypeNode([
+                typeNode,
+                typescript_1.default.factory.createLiteralTypeNode(typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.NullKeyword)),
+            ]);
+        }
+        return typescript_1.default.factory.createPropertySignature(undefined, field.name, field.isNullable() ? typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.QuestionToken) : undefined, typeNode);
+    })));
+}
+function createInputObjectTypeShapeIdent(type) {
+    return typescript_1.default.factory.createIdentifier(`${type.typeName}$Shape`);
+}

package/lib/dslgen/printers/objectType.js

@@ -0,0 +1,81 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createObjectTypeDslStmts = void 0;
+const codegen_core_1 = require("@proto-graphql/codegen-core");
+const typescript_1 = __importDefault(require("typescript"));
+const field_1 = require("./field");
+const util_1 = require("./util");
+/**
+ * @example
+ * ```ts
+ * export const Hello = builder.objectRef<_$hello$hello_pb.Hello>("Hello")
+ * builder.objectType(Hello, {
+ *   name: "Hello",
+ *   // ...
+ * })
+ * ```
+ */
+function createObjectTypeDslStmts(objType) {
+    const isInterface = objType instanceof codegen_core_1.InterfaceType;
+    return [
+        (0, util_1.createDslExportConstStmt)(objType.pothosRefObjectName, typescript_1.default.factory.createCallExpression((0, util_1.createBuilderPropExpr)(isInterface ? "interfaceRef" : "objectRef"), [
+            isInterface
+                ? typescript_1.default.factory.createTypeReferenceNode("Pick", [
+                    typescript_1.default.factory.createTypeReferenceNode((0, util_1.createQualifiedName)(objType.protoTypeFullName)),
+                    typescript_1.default.factory.createUnionTypeNode(objType.fields.map((f) => typescript_1.default.factory.createLiteralTypeNode(typescript_1.default.factory.createStringLiteral(f.protoJsName)))),
+                ])
+                : typescript_1.default.factory.createTypeReferenceNode((0, util_1.createQualifiedName)(objType.protoTypeFullName)),
+        ], [typescript_1.default.factory.createStringLiteral(objType.typeName)])),
+        typescript_1.default.factory.createExpressionStatement((0, util_1.createBuilderCallExpr)(isInterface ? "interfaceType" : "objectType", [
+            typescript_1.default.factory.createIdentifier(objType.pothosRefObjectName),
+            typescript_1.default.factory.createObjectLiteralExpression([
+                typescript_1.default.factory.createPropertyAssignment("name", typescript_1.default.factory.createStringLiteral(objType.typeName)),
+                (0, util_1.createDescriptionPropertyAssignment)(objType),
+                typescript_1.default.factory.createPropertyAssignment("fields", createObjectTypeFieldsFuncExpr(objType)),
+                isInterface ? null : typescript_1.default.factory.createPropertyAssignment("isTypeOf", createIsTypeOfMethodExpr(objType)),
+            ].filter((0, util_1.onlyNonNull)()), true),
+        ])),
+    ];
+}
+exports.createObjectTypeDslStmts = createObjectTypeDslStmts;
+/**
+ * @example
+ * ```ts
+ * fields: (t) => ({
+ *   // ...
+ * })
+ * ```
+ */
+function createObjectTypeFieldsFuncExpr(objType) {
+    return typescript_1.default.factory.createArrowFunction(undefined, undefined, [typescript_1.default.factory.createParameterDeclaration(undefined, undefined, undefined, "t", undefined, undefined, undefined)], undefined, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.EqualsGreaterThanToken), typescript_1.default.factory.createParenthesizedExpression(typescript_1.default.factory.createObjectLiteralExpression(objType.fields.length > 0
+        ? objType.fields.map((f) => typescript_1.default.factory.createPropertyAssignment(f.name, (0, field_1.createFieldDefinitionExpr)(f)))
+        : [typescript_1.default.factory.createPropertyAssignment("_", (0, field_1.createNoopFieldDefinitionExpr)({ input: false }))], true)));
+}
+/**
+ * @example
+ * ```ts
+ * isTypeOf(data) {
+ *   return data instanceof _$hello$hello_pb.Hello;
+ *
+ * (source) =>
+ *   // eslint-disable-next-line @typescript-eslint/ban-types
+ *   (source as _$hello$hello_pb.Hello | { $type: string & {} }).$type === "hello.Hello",
+ * }
+ * ```
+ */
+function createIsTypeOfMethodExpr(objType) {
+    return typescript_1.default.factory.createArrowFunction(undefined, undefined, [typescript_1.default.factory.createParameterDeclaration(undefined, undefined, undefined, "source", undefined, undefined, undefined)], undefined, typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.EqualsGreaterThanToken), typescript_1.default.factory.createBlock([
+        typescript_1.default.factory.createReturnStatement(typescript_1.default.factory.createBinaryExpression(typescript_1.default.factory.createPropertyAccessExpression(typescript_1.default.factory.createParenthesizedExpression(typescript_1.default.factory.createAsExpression(typescript_1.default.factory.createIdentifier("source"), typescript_1.default.factory.createUnionTypeNode([
+            typescript_1.default.factory.createTypeReferenceNode((0, util_1.createQualifiedName)(objType.protoTypeFullName)),
+            typescript_1.default.factory.createTypeLiteralNode([
+                typescript_1.default.factory.createPropertySignature(undefined, "$type", undefined, typescript_1.default.factory.createIntersectionTypeNode([
+                    typescript_1.default.factory.createToken(typescript_1.default.SyntaxKind.StringKeyword),
+                    typescript_1.default.factory.createTypeLiteralNode([]),
+                ])),
+            ]),
+        ]))), "$type"), typescript_1.default.SyntaxKind.EqualsEqualsEqualsToken, typescript_1.default.factory.createStringLiteral(objType.proto.fullName.toString()))),
+    ], true));
+}

package/lib/dslgen/printers/oneofUnionType.js

@@ -0,0 +1,27 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createOneofUnionTypeDslStmt = void 0;
+const typescript_1 = __importDefault(require("typescript"));
+const util_1 = require("./util");
+/**
+ * @example
+ * ```ts
+ * export cosnt Oneof = builder.unionType("Oneof", {
+ *   types: [...],
+ *   // ...
+ * })
+ * ```
+ */
+function createOneofUnionTypeDslStmt(type) {
+    return (0, util_1.createDslExportConstStmt)(type.pothosRefObjectName, (0, util_1.createBuilderCallExpr)("unionType", [
+        typescript_1.default.factory.createStringLiteral(type.typeName),
+        typescript_1.default.factory.createObjectLiteralExpression([
+            (0, util_1.createDescriptionPropertyAssignment)(type),
+            typescript_1.default.factory.createPropertyAssignment("types", typescript_1.default.factory.createArrayLiteralExpression(type.fields.map((f) => typescript_1.default.factory.createIdentifier(f.type.pothosRefObjectName)), true)),
+        ].filter((0, util_1.onlyNonNull)()), true),
+    ]));
+}
+exports.createOneofUnionTypeDslStmt = createOneofUnionTypeDslStmt;