move common logics to core package

izumin5210 · izumin5210 · commit c8b21a712b22 · 2022-12-04T23:07:25.000+09:00
diff --git a/packages/@proto-graphql/codegen-core/package.json b/packages/@proto-graphql/codegen-core/package.json
@@ -21,7 +21,8 @@
   },
   "dependencies": {
     "@proto-graphql/proto-descriptors": "^0.2.0",
-    "change-case": "^4.1.2"
+    "change-case": "^4.1.2",
+    "ts-poet": "^6.3.0"
   },
   "peerDependencies": {
     "google-protobuf": "^3.12.2"
diff --git a/packages/@proto-graphql/codegen-core/src/index.ts b/packages/@proto-graphql/codegen-core/src/index.ts
@@ -1 +1,2 @@
 export * from "./types";
+export * from "./printer";
diff --git a/packages/@proto-graphql/codegen-core/src/printer/graphqlExtensions.ts b/packages/@proto-graphql/codegen-core/src/printer/graphqlExtensions.ts
@@ -0,0 +1,86 @@
+import {
+  EnumType,
+  EnumTypeValue,
+  InputObjectField,
+  InputObjectType,
+  ObjectField,
+  ObjectOneofField,
+  ObjectType,
+  OneofUnionType,
+  SquashedOneofUnionType,
+} from "../types";
+import { compact } from "./util";
+
+export function protobufGraphQLExtensions(
+  type:
+    | ObjectType
+    | InputObjectType
+    | EnumType
+    | OneofUnionType
+    | SquashedOneofUnionType
+    | ObjectField<any>
+    | ObjectOneofField
+    | InputObjectField<any>
+    | EnumTypeValue
+): Record<string, Record<string, string>> {
+  if (type instanceof ObjectType || type instanceof InputObjectType) {
+    return {
+      protobufMessage: {
+        fullName: type.proto.fullName.toString(),
+        name: type.proto.name,
+        package: type.proto.file.package,
+      },
+    };
+  }
+  if (type instanceof EnumType) {
+    return {
+      protobufEnum: {
+        name: type.proto.name,
+        fullName: type.proto.fullName.toString(),
+        package: type.proto.file.package,
+      },
+    };
+  }
+  if (type instanceof OneofUnionType || type instanceof SquashedOneofUnionType) {
+    return {
+      protobufOneof: compact({
+        fullName: type.proto.fullName.toString(),
+        name: type.proto.name,
+        messageName: type.proto.kind === "Oneof" ? type.proto.parent.name : undefined,
+        package: (type.proto.kind === "Message" ? type.proto : type.proto.parent).file.package,
+        fields: type.fields.map((f) => ({
+          name: f.proto.name,
+          type: protoFieldTypeFullName(f),
+        })),
+      }),
+    };
+  }
+  if (type instanceof ObjectField || type instanceof ObjectOneofField || type instanceof InputObjectField) {
+    return {
+      protobufField: compact({ name: type.proto.name, typeFullName: protoFieldTypeFullName(type) }),
+    };
+  }
+  if (type instanceof EnumTypeValue) {
+    return {
+      protobufEnumValue: {
+        name: type.proto.name,
+      },
+    };
+  }
+
+  /* istanbul ignore next */
+  const _exhaustiveCheck: never = type;
+  return {};
+}
+
+function protoFieldTypeFullName(
+  field: ObjectField<any> | ObjectOneofField | InputObjectField<any>
+): string | undefined {
+  if ((field instanceof ObjectField || field instanceof InputObjectField) && field.proto.type !== null) {
+    if (field.proto.type.kind === "Scalar") {
+      return field.proto.type.type;
+    }
+    return field.proto.type.fullName.toString();
+  }
+  return undefined;
+}
diff --git a/packages/@proto-graphql/codegen-core/src/printer/index.ts b/packages/@proto-graphql/codegen-core/src/printer/index.ts
@@ -0,0 +1,3 @@
+export * from "./util";
+export * from "./graphqlExtensions";
+export * from "./print";
diff --git a/packages/@proto-graphql/codegen-core/src/printer/print.ts b/packages/@proto-graphql/codegen-core/src/printer/print.ts
@@ -0,0 +1,30 @@
+import { ProtoFile } from "@proto-graphql/proto-descriptors";
+import { code, Code } from "ts-poet";
+
+export function printCodes(codes: Code[], programName: string, file: ProtoFile): string {
+  return (codes.length === 0 ? code`export {};` : code`${codes}`).toString({
+    prefix: `
+      // Code generated by ${programName}. DO NOT EDIT.
+      // source: ${file.descriptor.getName()}
+
+      /* eslint-disable */
+    `,
+    dprintOptions: {
+      lineWidth: 80,
+      indentWidth: 2,
+      useTabs: false,
+      semiColons: "always",
+      quoteStyle: "alwaysDouble",
+      quoteProps: "asNeeded",
+      newLineKind: "lf",
+      useBraces: "whenNotSingleLine",
+      bracePosition: "sameLineUnlessHanging",
+      singleBodyPosition: "maintain",
+      nextControlFlowPosition: "sameLine",
+      trailingCommas: "onlyMultiLine",
+      operatorPosition: "nextLine",
+      preferHanging: false,
+      "arrowFunction.useParentheses": "force",
+    },
+  });
+}
diff --git a/packages/@proto-graphql/codegen-core/src/printer/util.ts b/packages/@proto-graphql/codegen-core/src/printer/util.ts
@@ -0,0 +1,69 @@
+import { ProtoEnum, ProtoMessage } from "@proto-graphql/proto-descriptors";
+import * as path from "path";
+import { code, Code, imp } from "ts-poet";
+import {
+  EnumType,
+  InputObjectType,
+  InterfaceType,
+  ObjectType,
+  OneofUnionType,
+  PrinterOptions,
+  protoImportPath,
+  SquashedOneofUnionType,
+} from "../types";
+
+export function filename(
+  type: ObjectType | InputObjectType | EnumType | OneofUnionType | SquashedOneofUnionType | InterfaceType,
+  opts: Pick<PrinterOptions, "fileLayout">
+): string {
+  switch (opts.fileLayout) {
+    case "proto_file":
+      return type.file.filename;
+    case "graphql_type": {
+      return path.join(path.dirname(type.file.filename), `${type.typeName}${type.file.extname}`);
+    }
+    /* istanbul ignore next */
+    default: {
+      const _exhaustiveCheck: never = opts.fileLayout;
+      throw "unreachable";
+    }
+  }
+}
+
+/** Remove nullish values recursively. */
+export function compact(v: any): any {
+  if (typeof v !== "object") return v;
+  if (Array.isArray(v)) return v.map(compact);
+  if (v == null) return v;
+  if ("toCodeString" in v) return v; // ignore nodes of ts-poet
+  return compactObj(v);
+}
+
+function compactObj<In extends Out, Out extends Record<string, unknown>>(obj: In): Out {
+  return Object.keys(obj).reduce((newObj, key) => {
+    const v = obj[key];
+    return v == null ? newObj : { ...newObj, [key]: compact(v) };
+  }, {} as Out);
+}
+
+export function protoType(origProto: ProtoMessage | ProtoEnum, opts: PrinterOptions): Code {
+  switch (opts.protobuf) {
+    case "google-protobuf":
+    case "protobufjs":
+      throw new Error(`not implemented: ${opts.protobuf}`);
+    case "ts-proto": {
+      let proto = origProto;
+      let name = proto.name;
+      while (proto.parent.kind !== "File") {
+        proto = proto.parent;
+        name = `${proto.name}_${name}`;
+      }
+      return code`${imp(`${name}@${protoImportPath(proto, opts)}`)}`;
+    }
+    /* istanbul ignore next */
+    default: {
+      const _exhaustiveCheck: never = opts;
+      throw "unreachable";
+    }
+  }
+}
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/enumType.ts b/packages/protoc-gen-pothos/src/dslgen/printers/enumType.ts
@@ -1,6 +1,6 @@
-import { EnumType } from "@proto-graphql/codegen-core";
+import { compact, EnumType, protobufGraphQLExtensions } from "@proto-graphql/codegen-core";
 import { code, Code, joinCode, literalOf } from "ts-poet";
-import { compact, pothosBuilder, PothosPrinterOptions, pothosRef } from "./util";
+import { pothosBuilder, PothosPrinterOptions, pothosRef } from "./util";
 
 /**
  * @example
@@ -35,13 +35,7 @@ export function createEnumTypeCode(type: EnumType, opts: PothosPrinterOptions):
             )},`
         )
     )}} as const`,
-    extensions: {
-      protobufEnum: {
-        name: type.proto.name,
-        fullName: type.proto.fullName.toString(),
-        package: type.proto.file.package,
-      },
-    },
+    extensions: protobufGraphQLExtensions(type),
   };
   return code`
     export const ${pothosRef(type)} =
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/field.ts b/packages/protoc-gen-pothos/src/dslgen/printers/field.ts
@@ -1,17 +1,19 @@
 import {
+  compact,
   EnumType,
   InputObjectField,
   ObjectField,
   ObjectOneofField,
   ObjectType,
+  protobufGraphQLExtensions,
   ScalarType,
   SquashedOneofUnionType,
 } from "@proto-graphql/codegen-core";
 import { code, Code, literalOf } from "ts-poet";
 import { createEnumResolverCode } from "./fieldResolver/enumFieldResolver";
 import { createNonNullResolverCode } from "./fieldResolver/nonNullResolver";
 import { createOneofUnionResolverCode } from "./fieldResolver/oneofUnionResolver";
-import { compact, fieldTypeRef, PothosPrinterOptions, protoFieldTypeFullName } from "./util";
+import { fieldTypeRef, PothosPrinterOptions } from "./util";
 
 /**
  * @example
@@ -70,9 +72,7 @@ export function createFieldRefCode(
     description: field.description,
     deprecationReason: field.deprecationReason,
     resolve: resolverCode ? code`${sourceExpr} => {${resolverCode}}` : null,
-    extensions: {
-      protobufField: { name: field.proto.name, typeFullName: protoFieldTypeFullName(field) },
-    },
+    extensions: protobufGraphQLExtensions(field),
   };
 
   const shouldUseFieldFunc = isInput || resolverCode != null;
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/fieldResolver/enumFieldResolver.ts b/packages/protoc-gen-pothos/src/dslgen/printers/fieldResolver/enumFieldResolver.ts
@@ -1,6 +1,5 @@
-import { EnumType, ObjectField, PrinterOptions } from "@proto-graphql/codegen-core";
+import { EnumType, ObjectField, PrinterOptions, protoType } from "@proto-graphql/codegen-core";
 import { Code, code } from "ts-poet";
-import { protoType } from "../util";
 
 /**
  * @example nullable
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/inputObjectType.ts b/packages/protoc-gen-pothos/src/dslgen/printers/inputObjectType.ts
@@ -1,7 +1,7 @@
-import { InputObjectType } from "@proto-graphql/codegen-core";
+import { compact, InputObjectType, protobufGraphQLExtensions, protoType } from "@proto-graphql/codegen-core";
 import { Code, code, imp, joinCode, literalOf } from "ts-poet";
 import { createFieldRefCode, createNoopFieldRefCode } from "./field";
-import { compact, fieldTypeShape, pothosBuilder, PothosPrinterOptions, pothosRef, protoType, shapeType } from "./util";
+import { fieldTypeShape, pothosBuilder, PothosPrinterOptions, pothosRef, shapeType } from "./util";
 
 /**
  * @example
@@ -47,13 +47,7 @@ export function createInputObjectTypeCode(type: InputObjectType, opts: PothosPri
                 ? type.fields.map((f) => code`${f.name}: ${createFieldRefCode(f, opts)},`)
                 : code`_: ${createNoopFieldRefCode({ input: true })}`
             }})`,
-            extensions: {
-              protobufMessage: {
-                fullName: type.proto.fullName.toString(),
-                name: type.proto.name,
-                package: type.proto.file.package,
-              },
-            },
+            extensions: protobufGraphQLExtensions(type),
           })
         )}
       );
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/objectType.ts b/packages/protoc-gen-pothos/src/dslgen/printers/objectType.ts
@@ -1,7 +1,7 @@
-import { InterfaceType, ObjectType } from "@proto-graphql/codegen-core";
+import { compact, InterfaceType, ObjectType, protobufGraphQLExtensions, protoType } from "@proto-graphql/codegen-core";
 import { Code, code, joinCode, literalOf } from "ts-poet";
 import { createFieldRefCode, createNoopFieldRefCode } from "./field";
-import { compact, pothosBuilder, PothosPrinterOptions, pothosRef, protoType } from "./util";
+import { pothosBuilder, PothosPrinterOptions, pothosRef } from "./util";
 
 /**
  * @example
@@ -31,13 +31,7 @@ export function createObjectTypeCode(type: ObjectType, opts: PothosPrinterOption
             === ${literalOf(type.proto.fullName.toString())};
         }
       `,
-    extensions: {
-      protobufMessage: {
-        fullName: type.proto.fullName.toString(),
-        name: type.proto.name,
-        package: type.proto.file.package,
-      },
-    },
+    extensions: protobufGraphQLExtensions(type),
   };
   const buildRefFunc = code`${pothosBuilder(type, opts)}.${isInterface ? "interface" : "object"}Ref`;
   const buildTypeFunc = code`${pothosBuilder(type, opts)}.${isInterface ? "interface" : "object"}Type`;
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/oneofUnionType.ts b/packages/protoc-gen-pothos/src/dslgen/printers/oneofUnionType.ts
@@ -1,6 +1,11 @@
-import { OneofUnionType, SquashedOneofUnionType } from "@proto-graphql/codegen-core";
+import {
+  compact,
+  OneofUnionType,
+  protobufGraphQLExtensions,
+  SquashedOneofUnionType,
+} from "@proto-graphql/codegen-core";
 import { code, Code, literalOf } from "ts-poet";
-import { compact, pothosBuilder, PothosPrinterOptions, pothosRef, protoFieldTypeFullName } from "./util";
+import { pothosBuilder, PothosPrinterOptions, pothosRef } from "./util";
 
 /**
  * @example
@@ -18,18 +23,7 @@ export function createOneofUnionTypeCode(
   const typeOpts = {
     types: type.fields.map((f) => pothosRef(f.type)),
     description: type.description,
-    extensions: {
-      protobufOneof: {
-        fullName: type.proto.fullName.toString(),
-        name: type.proto.name,
-        messageName: type.proto.kind === "Oneof" ? type.proto.parent.name : undefined,
-        package: (type.proto.kind === "Message" ? type.proto : type.proto.parent).file.package,
-        fields: type.fields.map((f) => ({
-          name: f.proto.name,
-          type: protoFieldTypeFullName(f),
-        })),
-      },
-    },
+    extensions: protobufGraphQLExtensions(type),
   };
   return code`
     export const ${pothosRef(type)} =
diff --git a/packages/protoc-gen-pothos/src/dslgen/printers/util.ts b/packages/protoc-gen-pothos/src/dslgen/printers/util.ts
diff --git a/packages/protoc-gen-pothos/src/printer.ts b/packages/protoc-gen-pothos/src/printer.ts

Original file line number	Diff line number	Diff line change
`@@ -1 +1,2 @@`
`1`	`1`	`export * from "./types";`
	`2`	`+export * from "./printer";`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+export * from "./util";`
	`2`	`+export * from "./graphqlExtensions";`
	`3`	`+export * from "./print";`