From 927c29de4ad8ac3f8642b6468fdd59d0cf587361 Mon Sep 17 00:00:00 2001
From: Taegeun Moon <taegeun.moon@devsisters.com>
Date: Thu, 1 Dec 2022 12:37:48 +0900
Subject: [PATCH] support both input and output

update readme

update readme
---
 packages/proto-loader/README.md                     |  8 ++++++--
 packages/proto-loader/bin/proto-loader-gen-types.ts | 13 +++++++------
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/packages/proto-loader/README.md b/packages/proto-loader/README.md
index 74065878..99b34a05 100644
--- a/packages/proto-loader/README.md
+++ b/packages/proto-loader/README.md
@@ -92,8 +92,12 @@ Options:
                                                         [string] [default: "%s"]
       --outputTemplate   Template for mapping output or "restricted" type names
                                                 [string] [default: "%s__Output"]
-      --branded          Emit property for branded type whose value is fullName
-                         of the Message               [boolean] [default: false]
+      --inputBranded     Output property for branded type for  "permissive"
+                         types with fullName of the Message as its value
+                                                                       [boolean]
+      --outputBranded    Output property for branded type for  "restricted"
+                         types with fullName of the Message as its value
+                                                                       [boolean]
 ```
 
 ### Example Usage
diff --git a/packages/proto-loader/bin/proto-loader-gen-types.ts b/packages/proto-loader/bin/proto-loader-gen-types.ts
index 6327da95..dc9da1a9 100644
--- a/packages/proto-loader/bin/proto-loader-gen-types.ts
+++ b/packages/proto-loader/bin/proto-loader-gen-types.ts
@@ -48,7 +48,8 @@ type GeneratorOptions = Protobuf.IParseOptions & Protobuf.IConversionOptions & {
   includeComments?: boolean;
   inputTemplate: string;
   outputTemplate: string;
-  outputBranded: boolean;
+  inputBranded?: boolean;
+  outputBranded?: boolean;
 }
 
 class TextFormatter {
@@ -272,9 +273,9 @@ function generatePermissiveMessageInterface(formatter: TextFormatter, messageTyp
     }
     formatter.writeLine(`'${oneof.name}'?: ${typeString};`);
   }
-  // if (options.inputBranded) {
-  //   formatTypeBrand(formatter, messageType);
-  // }
+  if (options.inputBranded) {
+    formatTypeBrand(formatter, messageType);
+  }
   formatter.unindent();
   formatter.writeLine('}');
 }
@@ -830,7 +831,7 @@ async function runScript() {
     .string(['includeDirs', 'grpcLib'])
     .normalize(['includeDirs', 'outDir'])
     .array('includeDirs')
-    .boolean(['keepCase', 'defaults', 'arrays', 'objects', 'oneofs', 'json', 'verbose', 'includeComments', 'outputBranded'])
+    .boolean(['keepCase', 'defaults', 'arrays', 'objects', 'oneofs', 'json', 'verbose', 'includeComments', 'inputBranded', 'outputBranded'])
     .string(['longs', 'enums', 'bytes', 'inputTemplate', 'outputTemplate'])
     .default('keepCase', false)
     .default('defaults', false)
@@ -844,7 +845,6 @@ async function runScript() {
     .default('bytes', 'Buffer')
     .default('inputTemplate', `${templateStr}`)
     .default('outputTemplate', `${templateStr}__Output`)
-    .default('outputBranded', false)
     .coerce('longs', value => {
       switch (value) {
         case 'String': return String;
@@ -884,6 +884,7 @@ async function runScript() {
       grpcLib: 'The gRPC implementation library that these types will be used with',
       inputTemplate: 'Template for mapping input or "permissive" type names',
       outputTemplate: 'Template for mapping output or "restricted" type names',
+      inputBranded: 'Output property for branded type for  "permissive" types with fullName of the Message as its value',
       outputBranded: 'Output property for branded type for  "restricted" types with fullName of the Message as its value',
     }).demandOption(['outDir', 'grpcLib'])
     .demand(1)