Initial pbts CLI for generating TypeScript definitions, see #550

Author: dcodeIO

README.md

@@ -336,6 +336,20 @@ protobuf.load("bundle.json", function(err, root) {
 });
 ```
 
+#### Generating TypeScript definitions for static modules
+
+Likewise, the `pbts` command line utility can be used to generate TypeScript definitions from `pbjs`-generated static modules.
+
+```
+Generates TypeScript definitions from annotated JavaScript files.
+
+  -n, --name      Specifies the module name.
+
+  -o, --out       Saves to a file instead of writing to stdout.
+
+usage: pbts [options] file1.js file2.js ...
+```
+
 Building
 --------
 

bin/pbts

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+var path = require("path"),
+    cli  = require(path.join(__dirname, "..", "cli", "pbts.js"));
+var ret  = cli.main(process.argv);
+if (typeof ret === 'number')
+    process.exit(ret);

cli/pbjs.js

@@ -8,8 +8,7 @@ var minimist = util.require("minimist", pkg.devDependencies.minimist),
     glob     = util.require("glob", pkg.devDependencies.glob);
 
 var protobuf = require(".."),
-    targets  = util.requireAll("./targets"),
-    pkg      = require("../package.json");
+    targets  = util.requireAll("./targets");
 
 exports.main = function(args) {
     var argv = minimist(args.slice(2), {

cli/pbts.js

@@ -0,0 +1,85 @@
+var path     = require("path"),
+    fs       = require("fs"),
+    pkg      = require(path.join(__dirname, "..", "package.json")),
+    util     = require("./util");
+var child_process = require("child_process");
+
+var minimist = util.require("minimist", pkg.devDependencies.minimist),
+    chalk    = util.require("chalk", pkg.devDependencies.chalk),
+    glob     = util.require("glob", pkg.devDependencies.glob);
+
+var jsdoc    = util.require("jsdoc/package.json", pkg.devDependencies.jsdoc),
+    tsdjsdoc = util.require("tsd-jsdoc/package.json", pkg.devDependencies['tsd-jsdoc']);
+
+var protobuf = require("..");
+
+exports.main = function(args) {
+    var argv = minimist(args.slice(2), {
+        alias: {
+            out : "o"
+        },
+        string: [ "out" ]
+    });
+
+    var files  = argv._;
+
+    if (!files.length) {
+        console.log([
+            "protobuf.js v" + pkg.version + " cli for TypeScript",
+            "",
+            "Generates TypeScript definitions from annotated JavaScript files.",
+            "",
+            "  -n, --name      Specifies the module name.",
+            "",
+            "  -o, --out       Saves to a file instead of writing to stdout.",
+            "",
+            "usage: " + chalk.bold.green(path.basename(process.argv[1])) + " [options] file1.js file2.js ..."
+        ].join("\n"));
+        return 1;
+    }
+
+    // Resolve glob expressions
+    for (var i = 0; i < files.length;) {
+        if (glob.hasMagic(files[i])) {
+            var matches = glob.sync(files[i]);
+            Array.prototype.splice.apply(files, [i, 1].concat(matches));
+            i += matches.length;
+        } else
+            ++i;
+    }
+
+    // There is no proper API for jsdoc, so this executes the CLI and writes to types/types.d.ts
+    var child = child_process.exec("node node_modules/jsdoc/jsdoc.js -c jsdoc.types.json " + files.join(' '), {
+        cwd: path.join(__dirname, ".."),
+        argv0: "node",
+        stdio: "pipe"
+    });
+    child.stdout.pipe(process.stdout);
+    child.stderr.pipe(process.stderr);
+    child.on("close", function(code) {
+        if (code)
+            throw Error("exited with " + code);
+        
+        var dir = path.join(__dirname, "..", "types");
+        var dts = fs.readFileSync(path.join(dir, "types.d.ts"), "utf8");
+        fs.unlinkSync(path.join(dir, "types.d.ts"));
+
+        var header = [
+            "// pbts " + process.argv.slice(2).join(' '),
+            "// Generated " + (new Date()).toUTCString().replace(/GMT/, "UTC"),
+            ""
+        ];
+
+        // Remove declare statements and wrap everything in a module
+        dts = dts.replace(/\bdeclare\s/g, "");
+        dts = dts.replace(/^/mg, "   ");
+        dts = header.join('\n')+"\ndeclare module " + JSON.stringify(argv.name || "mymodule") + " {\n\n" + dts + "\n}\n";
+
+        if (argv.out)
+            fs.writeFileSync(argv.out, dts);
+        else
+            process.stdout.write(dts, "utf8");
+    });
+
+    return undefined;
+};

cli/util.js

@@ -70,6 +70,12 @@ exports.inspect = function inspect(object, indent) {
 };
 
 exports.require = function(name, version) {
+    var sub = "";
+    var p = name.indexOf("/");
+    if (p > -1) {
+        sub = name.substring(p);
+        name = name.substring(0, p);
+    }
     var cwd = path.join(__dirname, "..");
     var dir = path.join(cwd, "node_modules", name);
     try {
@@ -81,7 +87,7 @@ exports.require = function(name, version) {
             cwd: cwd
         });
     }
-    return require(name);
+    return require(name + sub);
 };
 
 exports.wrap = function(name, OUTPUT, ROOT) {

dist/protobuf.js

@@ -3,9 +3,6 @@
         "allowUnknownTags": false
     },
     "source": {
-        "include": [
-            "./src/"
-        ],
         "exclude": [],
         "includePattern": ".+\\.js(doc)?$",
         "excludePattern": "(^|\\/|\\\\)_"

dist/protobuf.min.js

@@ -20,7 +20,8 @@
   ],
   "main": "src/index",
   "bin": {
-    "pbjs": "bin/pbjs"
+    "pbjs": "bin/pbjs",
+    "pbts": "bin/pbts"
   },
   "types": "types/protobuf.js.d.ts",
   "scripts": {
@@ -32,7 +33,7 @@
     "prepublish": "node scripts/prepublish",
     "prof": "node bench/prof",
     "test": "tape tests/*.js | tap-spec",
-    "types": "jsdoc -c jsdoc.types.json && node scripts/types.js && tsc types/test.ts --lib es2015 --noEmit",
+    "types": "node bin/pbts --name protobufjs --out types/protobuf.js.d.ts ./src && tsc types/test.ts --lib es2015 --noEmit",
     "zuul": "zuul --ui tape --no-coverage --concurrency 4 -- tests/*.js",
     "zuul-local": "zuul --ui tape --no-coverage --concurrency 1 --local 8080 --disable-tunnel -- tests/*.js"
   },