chore: fork code from origin repo (#1)

Author: czy88840616

.editorconfig

@@ -0,0 +1,11 @@
+# 🎨 editorconfig.org
+
+root = true
+
+[*]
+charset = utf-8
+end_of_line = lf
+indent_style = space
+indent_size = 2
+trim_trailing_whitespace = true
+insert_final_newline = true

.eslintrc.json

@@ -0,0 +1,7 @@
+{
+  "extends": "./node_modules/mwts/",
+  "ignorePatterns": ["node_modules", "dist", "test", "jest.config.js", "interface.ts", "app.js", "agent.js"],
+  "env": {
+    "jest": true
+  }
+}

.prettierrc.js

@@ -0,0 +1,3 @@
+module.exports = {
+  ...require('mwts/.prettierrc.json')
+}

README.md

@@ -46,14 +46,14 @@ import { Metadata } from '@grpc/grpc-js';
 
 export namespace hero {
   export interface HeroesService {
-      findOne(data: HeroById, metadata?: Metadata): Promise<Hero>;
+    findOne(data: HeroById, metadata?: Metadata): Promise<Hero>;
   }
   export interface HeroById {
-      id?: number;
+    id?: number;
   }
   export interface Hero {
-      id?: number;
-      name?: string;
+    id?: number;
+    name?: string;
   }
 }
 ```
@@ -76,10 +76,6 @@ Ignore directories or files:
 ```bash
 $ tsproto --path grpc-proto --ignore grpc-proto/ignore-dir
 ```
-Custom handlebar's template for output:
-```bash
-$ tsproto --path grpc-proto --template custom-template.hbs
-```
 
 ## Options
 
@@ -90,8 +86,6 @@ The following options are available:
   --help, -h      Show help                                            [boolean]
   --path, -p      Path to root directory                      [array] [required]
   --output, -o    Path to output directory                              [string]
-  --template      Handlebar's template for output
-                                 [string] [default: "templates/nestjs-grpc.hbs"]
   --target, -t    Proto files                      [array] [default: [".proto"]]
   --ignore, -i    Ignore file or directories
                                       [array] [default: ["node_modules","dist"]]

bin/tsproto.js

@@ -1,3 +1,77 @@
 #! /usr/bin/env node
+const { Compiller } = require('./dist/compiller');
+const { usage } = require('yargs');
+const { red } = require('chalk');
+const { join } = require('path');
 
-require('../dist/cli');
+const options = {
+  path: [],
+  output: null,
+  target: ['.proto'],
+  ignore: ['node_modules', 'dist'],
+  keepCase: false,
+  comments: true,
+  verbose: true,
+};
+/** Set CLI */
+const cli = usage('Extract and merge locale files.\nUsage: $0 [options]')
+  /* eslint-disable @typescript-eslint/no-var-requires */
+  .version(require(join(__dirname, '../package.json')).version)
+  /* eslint-enable @typescript-eslint/no-var-requires */
+  .alias('version', 'v')
+  .help('help')
+  .alias('help', 'h')
+  .option('path', {
+    alias: 'p',
+    describe: 'Path to root directory',
+    type: 'array',
+    normalize: true,
+  })
+  .option('output', {
+    alias: 'o',
+    describe: 'Path to output directory',
+    type: 'string',
+    normalize: true,
+  })
+  .option('target', {
+    alias: 't',
+    describe: 'Proto files',
+    default: options.target,
+    type: 'array',
+  })
+  .option('ignore', {
+    alias: 'i',
+    describe: 'Ignore file or directories',
+    default: options.ignore,
+    type: 'array',
+  })
+  .option('comments', {
+    alias: 'c',
+    describe: 'Add comments from proto',
+    default: options.comments,
+    type: 'boolean',
+  })
+  .option('verbose', {
+    describe: 'Log all output to console',
+    default: options.verbose,
+    type: 'boolean',
+  })
+  .demandOption(
+    ['path'],
+    red.bold(
+      'Please provide both run and [path] argument to work with this tool'
+    )
+  )
+  .exitProcess(true)
+  .parse(process.argv);
+
+/**
+ * Init Compiller
+ *
+ * @type {Compiller}
+ * @param {IGenOptions}
+ */
+const compiller = new Compiller({ ...options, ...cli });
+
+/** CLI Task Run */
+compiller.compile();

package.json

@@ -5,10 +5,12 @@
   "main": "dist/index",
   "typings": "dist/index.d.ts",
   "scripts": {
-    "build": "midway-bin build -c",
+    "build": "midway-bin build -c && rm -rf ./dist/.mwcc-cache",
     "test": "midway-bin test --ts",
     "cov": "midway-bin cov --ts",
-    "ci": "npm run test"
+    "ci": "npm run test",
+    "lint": "mwts lint",
+    "lint:fix": "mwts fix"
   },
   "bin": {
     "tsproto": "bin/tsproto.js"
@@ -28,13 +30,14 @@
   "license": "MIT",
   "devDependencies": {
     "@types/node": "^14.0.11",
-    "@midwayjs/cli": "^1.2.36",
-    "fs-extra": "^8.0.1",
+    "@midwayjs/cli": "^1.0.0",
     "mwts": "^1.0.5"
   },
   "dependencies": {
-    "nestjs-proto-gen-ts": "^1.0.0",
     "chalk": "^4.0.0",
+    "fs-extra": "^9.0.1",
+    "handlebars": "^4.7.6",
+    "protobufjs": "^6.9.0",
     "yargs": "^15.3.1"
   },
   "author": "Harry Chen <[email protected]>",

src/cli.ts

@@ -0,0 +1,7 @@
+import { registerHelper } from 'handlebars';
+
+registerHelper('comment', function (this: any) {
+  if (this.comment) {
+    return `// ${this.comment.replace(/\n/g, '\n// ')}`;
+  }
+});

src/compiller/handlebars/comment-helper.ts

@@ -0,0 +1,29 @@
+import { registerHelper } from 'handlebars';
+
+import { ENumberTypes } from '../../types';
+
+registerHelper('defaultValue', field => {
+  if (field.type === 'string') {
+    return '""';
+  }
+
+  if (field.type === 'bool') {
+    return 'false';
+  }
+
+  if (field.type === 'bytes') {
+    return 'new Uint8Array(0)';
+  }
+
+  if (field.type in ENumberTypes || field.options.enum) {
+    return '0';
+  }
+
+  // Map
+  if (field.keyType) {
+    return '{}';
+  }
+
+  // Default
+  return 'null';
+});

src/compiller/handlebars/default-value-helper.ts

@@ -0,0 +1,10 @@
+import { registerHelper } from 'handlebars';
+
+registerHelper('enumComment', (conditional, options) => {
+  if (options.data.root.comments && options.data.root.comments[conditional]) {
+    return `// ${options.data.root.comments[conditional].replace(
+      '\n',
+      '\n// '
+    )}`;
+  }
+});

src/compiller/handlebars/enum-comment-helper.ts

@@ -0,0 +1,60 @@
+import { registerHelper } from 'handlebars';
+
+import { EGoogleTypes, ENumberTypes } from '../../types';
+
+const KNOWN_PREFIX = 'google.protobuf.';
+
+function jsType(protoType: string): string {
+  if (protoType === 'string') {
+    return 'string';
+  }
+
+  if (protoType === 'bool') {
+    return 'boolean';
+  }
+
+  if (protoType === 'bytes') {
+    return 'Uint8Array';
+  }
+
+  if (protoType in ENumberTypes) {
+    return 'number';
+  }
+
+  if (protoType in EGoogleTypes) {
+    return `${KNOWN_PREFIX}${protoType}`;
+  }
+
+  if (protoType.substr(0, KNOWN_PREFIX.length) === KNOWN_PREFIX) {
+    return protoType;
+  }
+
+  return null;
+}
+
+registerHelper('type', field => {
+  // Check for known JS types
+  let type = jsType(field.type);
+
+  if (!type) {
+    // If it's not a known type, default to the field type
+    type = field.type;
+
+    // Check for a parent
+    if (field.options && field.options.parent) {
+      type = `${field.options.parent}.${type}`;
+    }
+  }
+
+  // Maps
+  if (field.keyType) {
+    type = `{ [key: ${jsType(field.keyType)}]: ${type} }`;
+  }
+
+  // Repeated
+  if (field.repeated) {
+    type += '[]';
+  }
+
+  return type;
+});

src/compiller/handlebars/type-helper.ts

@@ -0,0 +1,5 @@
+import { registerHelper } from 'handlebars';
+
+registerHelper('uncapitalize', conditional => {
+  return conditional[0].toLowerCase() + conditional.slice(1);
+});

src/compiller/handlebars/uncapitalize-hepler.ts

@@ -0,0 +1,5 @@
+import { registerHelper } from 'handlebars';
+
+registerHelper('var', (varName, varValue, options) => {
+  options.data.root[varName] = varValue;
+});