@whisklabs/grpc
Version:
gRPC generator and http library for typescript
209 lines • 12.9 kB
JavaScript
/* eslint-disable no-console */
var __assign = (this && this.__assign) || function () {
__assign = Object.assign || function(t) {
for (var s, i = 1, n = arguments.length; i < n; i++) {
s = arguments[i];
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
t[p] = s[p];
}
return t;
};
return __assign.apply(this, arguments);
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator = (this && this.__generator) || function (thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (g && (g = 0, op[0] && (_ = 0)), _) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
};
import { isBoolean, isText } from '@whisklabs/typeguards';
import { promises as fs } from 'fs';
import { extname, isAbsolute, join, parse as pathParse, relative } from 'path';
import { ModuleKind, ModuleResolutionKind, ScriptTarget, transpileModule } from 'typescript';
import { parser } from '../parser';
import { collectEmuns, collectMessages, collectServices } from './collect';
import { OPTION_MESSAGES_REQUIRED } from './constants';
import { enums } from './enum';
import { messages } from './message';
import { services } from './service';
import { walk } from './utils';
export function generator(_a) {
var dir = _a.dir, out = _a.out, _b = _a.exclude, exclude = _b === void 0 ? /^$/ : _b, _c = _a.version, version = _c === void 0 ? 'unknown' : _c, _d = _a.debug, debug = _d === void 0 ? false : _d, messageRequired = _a.messageRequired, packageName = _a.packageName, packageVersion = _a.packageVersion, _e = _a.packageUrl, packageUrl = _e === void 0 ? 'git@github.com:whisklabs/npm.git' : _e, _f = _a.packageRegistry, packageRegistry = _f === void 0 ? 'https://npm.pkg.github.com/' : _f;
return __awaiter(this, void 0, void 0, function () {
var IN_DIR, OUT_DIR, DEBUG_DIR, lic, lib, lastFile, schemas, _g, js, dts, errors, fields, names, _i, fields_1, _h, type, field, dtsString, jsString, hasError, e_1;
var _this = this;
return __generator(this, function (_j) {
switch (_j.label) {
case 0:
IN_DIR = isAbsolute(dir) ? dir : join(process.cwd(), dir);
OUT_DIR = isAbsolute(out) ? out : join(process.cwd(), out);
DEBUG_DIR = join(OUT_DIR, 'debug');
lic = "// Code created by generator @whisklabs/grpc\n// https://github.com/whisklabs/grpc-ts\n// Version: ".concat(version);
lib = "import {\n FieldMap,\n FieldRepeated,\n FieldType,\n FieldItem,\n Field,\n FieldEmpty,\n FieldGet,\n Service,\n ServiceRequest,\n ServiceResponse,\n Values\n} from '".concat(version === 'test' ? '../../src' : '@whisklabs/grpc', "';");
lastFile = '';
_j.label = 1;
case 1:
_j.trys.push([1, 10, , 11]);
return [4 /*yield*/, walk({
filename: IN_DIR,
result: { filepaths: [], schemas: [] },
each: function (_a) {
var result = _a.result, filename = _a.filename;
return __awaiter(_this, void 0, void 0, function () {
var parsed, _b, outFile;
return __generator(this, function (_c) {
switch (_c.label) {
case 0:
if (!(extname(filename) === '.proto' && !exclude.test(filename))) return [3 /*break*/, 5];
lastFile = filename;
result.filepaths.push(filename);
_b = parser;
return [4 /*yield*/, fs.readFile(filename, 'utf8')];
case 1:
parsed = _b.apply(void 0, [_c.sent()]);
if (!debug) return [3 /*break*/, 4];
outFile = pathParse(join(DEBUG_DIR, relative(dir !== null && dir !== void 0 ? dir : '', filename)));
return [4 /*yield*/, fs.mkdir(outFile.dir, { recursive: true })];
case 2:
_c.sent();
return [4 /*yield*/, fs.writeFile(join(outFile.dir, "".concat(outFile.name, ".json")), JSON.stringify(parsed, null, 2))];
case 3:
_c.sent();
_c.label = 4;
case 4:
result.schemas.push(parsed);
_c.label = 5;
case 5: return [2 /*return*/];
}
});
});
},
})];
case 2:
schemas = (_j.sent()).schemas;
_g = make(schemas, messageRequired), js = _g.js, dts = _g.dts, errors = _g.errors, fields = _g.fields, names = _g.names;
for (_i = 0, fields_1 = fields; _i < fields_1.length; _i++) {
_h = fields_1[_i], type = _h[0], field = _h[1];
if (!names.has(type)) {
errors.push("\u001B[31mNo found message or enum:\u001B[0m ".concat(type, " [").concat(field, "]"));
}
}
dts.unshift(lic, '', lib, '');
js.unshift(lic, '', '"use strict";', '');
dtsString = dts.join('\n');
jsString = js.join('\n');
checkTypes(dtsString, errors);
checkTypes(jsString, errors);
hasError = errors.length > 0;
if (!hasError) return [3 /*break*/, 3];
errors.forEach(function (i) { return console.error(i); });
return [3 /*break*/, 9];
case 3: return [4 /*yield*/, fs.mkdir(OUT_DIR, { recursive: true })];
case 4:
_j.sent();
return [4 /*yield*/, fs.writeFile(join(OUT_DIR, 'index.d.ts'), dtsString)];
case 5:
_j.sent();
return [4 /*yield*/, fs.writeFile(join(OUT_DIR, 'index.js'), transpile(jsString))];
case 6:
_j.sent();
return [4 /*yield*/, fs.writeFile(join(OUT_DIR, 'esm.js'), transpile(jsString, { module: ModuleKind.ES2015 }))];
case 7:
_j.sent();
if (!(isText(packageName) && isText(packageVersion))) return [3 /*break*/, 9];
return [4 /*yield*/, fs.writeFile(join(OUT_DIR, 'package.json'), packageTemplate({ packageName: packageName, packageVersion: packageVersion, packageUrl: packageUrl, packageRegistry: packageRegistry }))];
case 8:
_j.sent();
_j.label = 9;
case 9: return [2 /*return*/, hasError];
case 10:
e_1 = _j.sent();
console.info("Last processing file was \"\u001B[31m".concat(lastFile, "\u001B[0m\""));
console.error(e_1);
return [2 /*return*/, true];
case 11: return [2 /*return*/];
}
});
});
}
export var transpile = function (source, options) {
return transpileModule(source, {
compilerOptions: __assign({ allowJs: true, module: ModuleKind.CommonJS, target: ScriptTarget.ES5, moduleResolution: ModuleResolutionKind.NodeJs }, options),
}).outputText;
};
export function checkTypes(file, errors) {
var diagnostics = transpileModule(file, { reportDiagnostics: true }).diagnostics;
for (var _i = 0, _a = diagnostics !== null && diagnostics !== void 0 ? diagnostics : []; _i < _a.length; _i++) {
var diagnostic = _a[_i];
errors.push("\u001B[31mTS error:\u001B[0m ".concat(JSON.stringify(diagnostic.messageText)));
}
}
export function make(schemas, messageRequired) {
var _a, _b, _c;
var out = {
js: [],
dts: [],
names: new Set(),
errors: [],
fields: [],
packagesList: new Set(),
enumsList: new Set(),
};
for (var _i = 0, schemas_1 = schemas; _i < schemas_1.length; _i++) {
var schema = schemas_1[_i];
var path = (_a = schema.package) !== null && _a !== void 0 ? _a : '';
out.packagesList.add(path);
collectEmuns(path, out, schema.enums);
collectMessages(path, out, schema.messages);
collectServices(path, out, schema.services);
}
for (var _d = 0, schemas_2 = schemas; _d < schemas_2.length; _d++) {
var schema = schemas_2[_d];
var path = (_b = schema.package) !== null && _b !== void 0 ? _b : '';
enums(path, out, schema.enums);
}
for (var _e = 0, schemas_3 = schemas; _e < schemas_3.length; _e++) {
var schema = schemas_3[_e];
var path = (_c = schema.package) !== null && _c !== void 0 ? _c : '';
var option = schema.options[OPTION_MESSAGES_REQUIRED];
var isMessageRequired = isBoolean(option) ? option : messageRequired;
messages(path, out, schema.messages, isMessageRequired, []);
services(path, out, schema.services);
}
return out;
}
export var packageTemplate = function (_a) {
var packageName = _a.packageName, packageVersion = _a.packageVersion, packageUrl = _a.packageUrl, packageRegistry = _a.packageRegistry;
return "{\n \"name\": \"".concat(packageName, "\",\n \"version\": \"").concat(packageVersion, "\",\n \"repository\": {\n \"type\": \"git\",\n \"url\": \"").concat(packageUrl, "\"\n },\n \"publishConfig\": {\n \"registry\": \"").concat(packageRegistry, "\"\n },\n \"main\": \"./index.js\",\n \"module\": \"./esm.js\",\n \"types\": \"./index.d.ts\",\n \"sideEffects\": false,\n \"peerDependencies\": {\n \"@whisklabs/grpc\": \"*\"\n }\n}");
};
//# sourceMappingURL=main.js.map