UNPKG

@webuildbots/webuildbots-sdk

Version:
125 lines (124 loc) 6.51 kB
"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()); }); }; 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 (_) 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 }; } }; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); exports.WebhookClient = void 0; var md5_1 = __importDefault(require("md5")); var jsonwebtoken_1 = __importDefault(require("jsonwebtoken")); var http_status_codes_1 = require("http-status-codes"); var logger_1 = __importDefault(require("wbb-logger/dist/logger")); var response_builder_1 = __importDefault(require("../builder/response-builder")); var logger = new logger_1.default('webhook-client.ts'); var WebhookClient = /** @class */ (function () { function WebhookClient(secret) { this.secret = secret; this.handlerMap = {}; } WebhookClient.prototype.addHandler = function (handlerName, handler) { if (this.handlerMap[handlerName]) { throw Error("\"" + handlerName + "\" has already been added"); } this.handlerMap[handlerName] = handler; return this; }; WebhookClient.prototype.handleRequest = function (body, token) { return __awaiter(this, void 0, void 0, function () { var request, handlerName, handler, builder, err_1; return __generator(this, function (_a) { switch (_a.label) { case 0: // Verify the request if not public function if (this.secret) { try { this.verifyPayload(body, token); } catch (err) { logger.warn(err); return [2 /*return*/, { status: http_status_codes_1.StatusCodes.UNAUTHORIZED, body: http_status_codes_1.ReasonPhrases.UNAUTHORIZED }]; } } request = JSON.parse(body); handlerName = request.functionBlock.function.handlerName; handler = this.handlerMap[handlerName]; if (!handler) { return [2 /*return*/, { status: http_status_codes_1.StatusCodes.NOT_FOUND, body: "No handler for " + handlerName }]; } _a.label = 1; case 1: _a.trys.push([1, 3, , 4]); builder = new response_builder_1.default(); return [4 /*yield*/, handler(request, builder)]; case 2: _a.sent(); return [2 /*return*/, { status: http_status_codes_1.StatusCodes.OK, body: builder.build() }]; case 3: err_1 = _a.sent(); logger.error(err_1); return [2 /*return*/, { status: http_status_codes_1.StatusCodes.INTERNAL_SERVER_ERROR, body: http_status_codes_1.ReasonPhrases.INTERNAL_SERVER_ERROR }]; case 4: return [2 /*return*/]; } }); }); }; WebhookClient.prototype.verifyPayload = function (body, token) { if (!token) { throw Error('no token provided'); } var bodyMd5Hash = jsonwebtoken_1.default.verify(token, this.secret).bodyMd5Hash; // Make sure the requestId's match if (bodyMd5Hash !== md5_1.default(body)) { throw Error('bodyMd5Hash does not match'); } }; return WebhookClient; }()); exports.WebhookClient = WebhookClient;