Merge pull request #252 from proto-graphql/izumin5210/emit-imported-files-option

feat: add `emit_imported_files` option

Author: izumin5210

packages/@proto-graphql/codegen-core/src/printer/options.ts

@@ -14,6 +14,7 @@ type PrinterProtobufOptions =
   | { protobuf: "ts-proto" };
 
 export type PrinterCommonOptions = {
+  emitImportedFiles: boolean;
   importPrefix: string | null;
   fileLayout: FileLayout;
   filenameSuffix: string;

packages/@proto-graphql/protoc-plugin-helpers/src/parseParams.ts

@@ -22,6 +22,7 @@ export function parseParams<DSL extends PrinterOptions["dsl"]>(
       dsl,
       protobuf: "google-protobuf",
       importPrefix: null,
+      emitImportedFiles: false,
       fileLayout: "proto_file",
       filenameSuffix: dsl === "nexus" ? "_pb_nexus.ts" : ".pb.pothos.ts",
       pothos: { builderPath: "./builder" },
@@ -61,6 +62,9 @@ export function parseParams<DSL extends PrinterOptions["dsl"]>(
       case "partial_inputs":
         params.type.partialInputs = toBool(k, v);
         break;
+      case "emit_imported_files":
+        params.printer.emitImportedFiles = toBool(k, v);
+        break;
       case "file_layout": {
         const s = toString(k, v);
         if (!checkEnum(s, fileLayouts)) {

packages/@proto-graphql/protoc-plugin-helpers/src/process.ts

@@ -32,9 +32,7 @@ export function createProcessor<DSL extends PrinterOptions["dsl"]>({
 
     const params = parseParams(req.getParameter(), dsl);
 
-    for (const fn of req.getFileToGenerateList()) {
-      const file = registry.findFile(fn);
-      if (file == null) throw new Error(`${fn} is not found`);
+    for (const file of getFilesToGenerate(req, registry)) {
       const results = generateFiles(registry, file, params);
 
       for (const result of results) {
@@ -48,3 +46,30 @@ export function createProcessor<DSL extends PrinterOptions["dsl"]>({
     return resp;
   };
 }
+
+function getFilesToGenerate(
+  req: CodeGeneratorRequest,
+  reg: ProtoRegistry
+): ProtoFile[] {
+  const filesToGenerateSet = new Set(req.getFileToGenerateList());
+  return req
+    .getProtoFileList()
+    .map((f) => f.getName())
+    .filter(isDefined)
+    .filter((n) => filesToGenerateSet.has(n))
+    .map((n) => reg.findFile(n))
+    .filter(isNotNull)
+    .filter((f) => !isWellKnownTypesFile(f));
+}
+
+function isDefined<V>(v: V | undefined): v is V {
+  return v !== undefined;
+}
+
+function isNotNull<V>(v: V | null): v is V {
+  return v !== null;
+}
+
+function isWellKnownTypesFile(f: ProtoFile): boolean {
+  return f.name.startsWith("google/protobuf/");
+}