118 lines
4.5 KiB
JavaScript
118 lines
4.5 KiB
JavaScript
"use strict";
|
|
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());
|
|
});
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.parseTwirpPath = exports.getRequestData = exports.validateRequest = exports.getContentType = exports.TwirpContentType = void 0;
|
|
const errors_1 = require("./errors");
|
|
/**
|
|
* Supported Twirp Content-Type
|
|
*/
|
|
var TwirpContentType;
|
|
(function (TwirpContentType) {
|
|
TwirpContentType[TwirpContentType["Protobuf"] = 0] = "Protobuf";
|
|
TwirpContentType[TwirpContentType["JSON"] = 1] = "JSON";
|
|
TwirpContentType[TwirpContentType["Unknown"] = 2] = "Unknown";
|
|
})(TwirpContentType = exports.TwirpContentType || (exports.TwirpContentType = {}));
|
|
/**
|
|
* Get supported content-type
|
|
* @param mimeType
|
|
*/
|
|
function getContentType(mimeType) {
|
|
switch (mimeType) {
|
|
case "application/protobuf":
|
|
return TwirpContentType.Protobuf;
|
|
case "application/json":
|
|
return TwirpContentType.JSON;
|
|
default:
|
|
return TwirpContentType.Unknown;
|
|
}
|
|
}
|
|
exports.getContentType = getContentType;
|
|
/**
|
|
* Validate a twirp request
|
|
* @param ctx
|
|
* @param request
|
|
* @param pathPrefix
|
|
*/
|
|
function validateRequest(ctx, request, pathPrefix) {
|
|
if (request.method !== "POST") {
|
|
const msg = `unsupported method ${request.method} (only POST is allowed)`;
|
|
throw new errors_1.BadRouteError(msg, request.method || "", request.url || "");
|
|
}
|
|
const path = parseTwirpPath(request.url || "");
|
|
if (path.pkgService !==
|
|
(ctx.packageName ? ctx.packageName + "." : "") + ctx.serviceName) {
|
|
const msg = `no handler for path ${request.url}`;
|
|
throw new errors_1.BadRouteError(msg, request.method || "", request.url || "");
|
|
}
|
|
if (path.prefix !== pathPrefix) {
|
|
const msg = `invalid path prefix ${path.prefix}, expected ${pathPrefix}, on path ${request.url}`;
|
|
throw new errors_1.BadRouteError(msg, request.method || "", request.url || "");
|
|
}
|
|
const mimeContentType = request.headers["content-type"] || "";
|
|
if (ctx.contentType === TwirpContentType.Unknown) {
|
|
const msg = `unexpected Content-Type: ${request.headers["content-type"]}`;
|
|
throw new errors_1.BadRouteError(msg, request.method || "", request.url || "");
|
|
}
|
|
return Object.assign(Object.assign({}, path), { mimeContentType, contentType: ctx.contentType });
|
|
}
|
|
exports.validateRequest = validateRequest;
|
|
/**
|
|
* Get request data from the body
|
|
* @param req
|
|
*/
|
|
function getRequestData(req) {
|
|
return new Promise((resolve, reject) => {
|
|
const reqWithRawBody = req;
|
|
if (reqWithRawBody.rawBody instanceof Buffer) {
|
|
resolve(reqWithRawBody.rawBody);
|
|
return;
|
|
}
|
|
const chunks = [];
|
|
req.on("data", (chunk) => chunks.push(chunk));
|
|
req.on("end", () => __awaiter(this, void 0, void 0, function* () {
|
|
const data = Buffer.concat(chunks);
|
|
resolve(data);
|
|
}));
|
|
req.on("error", (err) => {
|
|
if (req.aborted) {
|
|
reject(new errors_1.TwirpError(errors_1.TwirpErrorCode.DeadlineExceeded, "failed to read request: deadline exceeded"));
|
|
}
|
|
else {
|
|
reject(new errors_1.TwirpError(errors_1.TwirpErrorCode.Malformed, err.message).withCause(err));
|
|
}
|
|
});
|
|
req.on("close", () => {
|
|
reject(new errors_1.TwirpError(errors_1.TwirpErrorCode.Canceled, "failed to read request: context canceled"));
|
|
});
|
|
});
|
|
}
|
|
exports.getRequestData = getRequestData;
|
|
/**
|
|
* Parses twirp url path
|
|
* @param path
|
|
*/
|
|
function parseTwirpPath(path) {
|
|
const parts = path.split("/");
|
|
if (parts.length < 2) {
|
|
return {
|
|
pkgService: "",
|
|
method: "",
|
|
prefix: "",
|
|
};
|
|
}
|
|
return {
|
|
method: parts[parts.length - 1],
|
|
pkgService: parts[parts.length - 2],
|
|
prefix: parts.slice(0, parts.length - 2).join("/"),
|
|
};
|
|
}
|
|
exports.parseTwirpPath = parseTwirpPath;
|