@webiny/handler 0.0.0-unstable.d7f521b032 → 0.0.0-unstable.de38392959

package/fastify.js

@@ -1,79 +1,66 @@
 "use strict";
 
 var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.createHandler = void 0;
-
-var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
-
+var _types = require("@webiny/plugins/types");
 var _fastify = _interopRequireDefault(require("fastify"));
-
 var _utils = require("@webiny/utils");
-
 var _Context = require("./Context");
-
 var _error = _interopRequireDefault(require("@webiny/error"));
-
 var _RoutePlugin = require("./plugins/RoutePlugin");
-
 var _handlerClient = require("@webiny/handler-client");
-
 var _cookie = _interopRequireDefault(require("@fastify/cookie"));
-
 var _compress = _interopRequireDefault(require("@fastify/compress"));
-
 var _middleware = require("./middleware");
-
 var _api = require("@webiny/api");
-
 var _BeforeHandlerPlugin = require("./plugins/BeforeHandlerPlugin");
-
 var _HandlerResultPlugin = require("./plugins/HandlerResultPlugin");
-
 var _HandlerErrorPlugin = require("./plugins/HandlerErrorPlugin");
-
 var _ModifyFastifyPlugin = require("./plugins/ModifyFastifyPlugin");
-
 var _HandlerOnRequestPlugin = require("./plugins/HandlerOnRequestPlugin");
-
-const DEFAULT_HEADERS = (0, _objectSpread2.default)({
-  "Cache-Control": "no-store",
-  "Content-Type": "application/json; charset=utf-8",
-  "Access-Control-Allow-Origin": "*",
-  "Access-Control-Allow-Headers": "*",
-  "Access-Control-Allow-Methods": "OPTIONS,POST,GET,DELETE,PUT,PATCH"
-}, (0, _utils.getWebinyVersionHeaders)());
-
+var _ResponseHeaders = require("./ResponseHeaders");
+var _ModifyResponseHeadersPlugin = require("./plugins/ModifyResponseHeadersPlugin");
+function createDefaultHeaders() {
+  return _ResponseHeaders.ResponseHeaders.create({
+    "content-type": "application/json; charset=utf-8",
+    "cache-control": "no-store",
+    "access-control-allow-origin": "*",
+    "access-control-allow-headers": "*",
+    "access-control-allow-methods": "OPTIONS,POST,GET,DELETE,PUT,PATCH",
+    ...(0, _utils.getWebinyVersionHeaders)()
+  });
+}
+const getDefaultOptionsHeaders = () => {
+  return _ResponseHeaders.ResponseHeaders.create({
+    "access-control-max-age": "86400",
+    "cache-control": "public, max-age=86400"
+  });
+};
 const getDefaultHeaders = routes => {
+  const headers = createDefaultHeaders();
+
   /**
    * If we are accepting all headers, just output that one.
    */
   const keys = Object.keys(routes);
   const all = keys.every(key => routes[key].length > 0);
-
   if (all) {
-    return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, DEFAULT_HEADERS), {}, {
-      "Access-Control-Allow-Methods": "*"
-    });
-  }
-
-  return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, DEFAULT_HEADERS), {}, {
-    "Access-Control-Allow-Methods": keys.filter(type => {
-      if (!routes[type] || Array.isArray(routes[type]) === false) {
+    headers.set("access-control-allow-methods", "*");
+  } else {
+    const allowedMethods = keys.filter(type => {
+      if (!routes[type] || !Array.isArray(routes[type])) {
         return false;
       }
-
       return routes[type].length > 0;
-    }).sort().join(",")
-  });
+    }).sort().join(",");
+    headers.set("access-control-allow-methods", allowedMethods);
+  }
+  return headers;
 };
-
 const stringifyError = error => {
-  var _error$constructor;
-
   const {
     name,
     message,
@@ -81,21 +68,16 @@ const stringifyError = error => {
     stack,
     data
   } = error;
-  return JSON.stringify((0, _objectSpread2.default)((0, _objectSpread2.default)({}, error), {}, {
-    constructorName: ((_error$constructor = error.constructor) === null || _error$constructor === void 0 ? void 0 : _error$constructor.name) || "UnknownError",
+  return JSON.stringify({
+    ...error,
+    constructorName: error.constructor?.name || "UnknownError",
     name: name || "No error name",
     message: message || "No error message",
     code: code || "NO_CODE",
     data,
     stack: process.env.DEBUG === "true" ? stack : "Turn on the debug flag to see the stack."
-  }));
-};
-
-const OPTIONS_HEADERS = {
-  "Access-Control-Max-Age": "86400",
-  "Cache-Control": "public, max-age=86400"
+  });
 };
-
 const createHandler = params => {
   const definedRoutes = {
     POST: [],
@@ -115,90 +97,78 @@ const createHandler = params => {
     TRACE: [],
     UNLOCK: []
   };
-
   const throwOnDefinedRoute = (type, path, options) => {
     if (type === "ALL") {
       const all = Object.keys(definedRoutes).find(key => {
         const routes = definedRoutes[key];
         return routes.includes(path);
       });
-
       if (!all) {
         return;
       }
-
-      console.log(`Error while registering onAll route. One of the routes is already defined.`);
-      console.log(JSON.stringify(all));
+      console.error(`Error while registering onAll route. One of the routes is already defined.`);
+      console.error(JSON.stringify(all));
       throw new _error.default(`You cannot override a route with onAll() method, please remove unnecessary route from the system.`, "OVERRIDE_ROUTE_ERROR", {
         type,
         path
       });
     } else if (definedRoutes[type].includes(path) === false) {
       return;
-    } else if ((options === null || options === void 0 ? void 0 : options.override) === true) {
+    } else if (options?.override === true) {
       return;
     }
-
-    console.log(`Error while trying to override route: [${type}] ${path}`);
+    console.error(`Error while trying to override route: [${type}] ${path}`);
     throw new _error.default(`When you are trying to override existing route, you must send "override" parameter when adding that route.`, "OVERRIDE_ROUTE_ERROR", {
       type,
       path
     });
   };
-
   const addDefinedRoute = (type, path) => {
     if (!definedRoutes[type]) {
       return;
     } else if (definedRoutes[type].includes(path)) {
       return;
     }
-
     definedRoutes[type].push(path);
   };
   /**
    * We must attach the server to our internal context if we want to have it accessible.
    */
-
-
-  const app = (0, _fastify.default)((0, _objectSpread2.default)({}, params.options || {}));
+  const app = (0, _fastify.default)({
+    bodyLimit: 10485760,
+    // 10MB
+    ...(params.options || {})
+  });
   /**
    * We need to register routes in our system so we can output headers later on and dissallow overriding routes.
    */
-
   app.addHook("onRoute", route => {
     const method = route.method;
-
     if (Array.isArray(method)) {
       for (const m of method) {
         addDefinedRoute(m, route.path);
       }
-
       return;
     }
-
     addDefinedRoute(method, route.path);
   });
   /**
    * ############################
    * Register the Fastify plugins.
    */
-
   /**
    * Package @fastify/cookie
    *
    * https://github.com/fastify/fastify-cookie
    */
-
   app.register(_cookie.default, {
     parseOptions: {} // options for parsing cookies
-
   });
   /**
    * Package @fastify/compress
    *
    * https://github.com/fastify/fastify-compress
    */
-
   app.register(_compress.default, {
     global: true,
     threshold: 1024,
@@ -211,7 +181,6 @@ const createHandler = params => {
   /**
    * Route helpers - mostly for users.
    */
-
   const routes = {
     defined: definedRoutes,
     onPost: (path, handler, options) => {
@@ -248,92 +217,61 @@ const createHandler = params => {
     }
   };
   let context;
-
+  const plugins = new _types.PluginsContainer([
+  /**
+   * We must have handlerClient by default.
+   * And it must be one of the first context plugins applied.
+   */
+  (0, _handlerClient.createHandlerClient)()]);
+  plugins.merge(params.plugins || []);
   try {
     context = new _Context.Context({
-      plugins: [
+      plugins,
       /**
-       * We must have handlerClient by default.
-       * And it must be one of the first context plugins applied.
-       */
-      (0, _handlerClient.createHandlerClient)(), ...(params.plugins || [])],
-
-      /**
-       * Inserted via webpack on build time.
+       * Inserted via webpack at build time.
        */
       WEBINY_VERSION: process.env.WEBINY_VERSION,
-      server: app,
       routes
     });
   } catch (ex) {
-    console.log(`Error while constructing the Context.`);
-    console.log(stringifyError(ex));
+    console.error(`Error while constructing the Context.`);
+    console.error(stringifyError(ex));
     throw ex;
   }
+
   /**
    * We are attaching our custom context to webiny variable on the fastify app, so it is accessible everywhere.
    */
-
-
   app.decorate("webiny", context);
-  /**
-   * We have few types of triggers:
-   *  * Events - EventPlugin
-   *  * Routes - RoutePlugin
-   *
-   * Routes are registered in fastify but events must be handled in package which implements cloud specific methods.
-   */
 
-  const routePlugins = app.webiny.plugins.byType(_RoutePlugin.RoutePlugin.type);
-  /**
-   * Add routes to the system.
-   */
-
-  let routePluginName;
-
-  try {
-    for (const plugin of routePlugins) {
-      routePluginName = plugin.name;
-      plugin.cb((0, _objectSpread2.default)((0, _objectSpread2.default)({}, app.webiny.routes), {}, {
-        context: app.webiny
-      }));
-    }
-  } catch (ex) {
-    console.log(`Error while running the "RoutePlugin" ${routePluginName ? `(${routePluginName})` : ""} plugin in the beginning of the "createHandler" callable.`);
-    console.log(stringifyError(ex));
-    throw ex;
-  }
   /**
    * On every request we add default headers, which can be changed later.
    * Also, if it is an options request, we skip everything after this hook and output options headers.
    */
-
-
   app.addHook("onRequest", async (request, reply) => {
+    const isOptionsRequest = request.method === "OPTIONS";
     /**
      * Our default headers are always set. Users can override them.
      */
     const defaultHeaders = getDefaultHeaders(definedRoutes);
-    reply.headers(defaultHeaders);
+    const initialHeaders = isOptionsRequest ? defaultHeaders.merge(getDefaultOptionsHeaders()) : defaultHeaders;
+    reply.headers(initialHeaders.getHeaders());
     /**
      * Users can define their own custom handlers for the onRequest event - so let's run them first.
      */
-
     const plugins = app.webiny.plugins.byType(_HandlerOnRequestPlugin.HandlerOnRequestPlugin.type);
     let name;
-
     try {
       for (const plugin of plugins) {
         name = plugin.name;
         const result = await plugin.exec(request, reply);
-
         if (result === false) {
           return;
         }
       }
     } catch (ex) {
-      console.log(`Error while running the "HandlerOnRequestPlugin" ${name ? `(${name})` : ""} plugin in the onRequest hook.`);
-      console.log(stringifyError(ex));
+      console.error(`Error while running the "HandlerOnRequestPlugin" ${name ? `(${name})` : ""} plugin in the onRequest hook.`);
+      console.error(stringifyError(ex));
       throw ex;
     }
     /**
@@ -341,83 +279,73 @@ const createHandler = params => {
      *
      * Users can prevent this by creating their own HandlerOnRequestPlugin and returning false as the result of the callable.
      */
-
-
-    if (request.method !== "OPTIONS") {
+    if (!isOptionsRequest) {
       return;
     }
-
     if (reply.sent) {
       /**
        * At this point throwing an exception will not do anything with the response. So just log it.
        */
-      console.log(JSON.stringify({
+      console.error(JSON.stringify({
         message: `Output was already sent. Please check custom plugins of type "HandlerOnRequestPlugin".`,
         explanation: "This error can happen if the user plugin ended the reply, but did not return false as response."
       }));
       return;
     }
-
-    reply.headers((0, _objectSpread2.default)((0, _objectSpread2.default)({}, defaultHeaders), OPTIONS_HEADERS)).code(204).send("").hijack();
+    reply.code(204).send("").hijack();
   });
   app.addHook("preParsing", async (request, reply) => {
     app.webiny.request = request;
     app.webiny.reply = reply;
     const plugins = app.webiny.plugins.byType(_api.ContextPlugin.type);
     let name;
-
     try {
       for (const plugin of plugins) {
         name = plugin.name;
         await plugin.apply(app.webiny);
       }
     } catch (ex) {
-      console.log(`Error while running the "ContextPlugin" ${name ? `(${name})` : ""} plugin in the preParsing hook.`);
-      console.log(stringifyError(ex));
+      console.error(`Error while running the "ContextPlugin" ${name ? `(${name})` : ""} plugin in the preParsing hook.`);
+      console.error(stringifyError(ex));
       throw ex;
     }
   });
   /**
    *
    */
-
   app.addHook("preHandler", async () => {
     const plugins = app.webiny.plugins.byType(_BeforeHandlerPlugin.BeforeHandlerPlugin.type);
     let name;
-
     try {
       for (const plugin of plugins) {
         name = plugin.name;
         await plugin.apply(app.webiny);
       }
     } catch (ex) {
-      console.log(`Error while running the "BeforeHandlerPlugin" ${name ? `(${name})` : ""} plugin in the preHandler hook.`);
-      console.log(stringifyError(ex));
+      console.error(`Error while running the "BeforeHandlerPlugin" ${name ? `(${name})` : ""} plugin in the preHandler hook.`);
+      console.error(stringifyError(ex));
       throw ex;
     }
   });
+
   /**
    *
    */
-
   const preSerialization = async (_, __, payload) => {
     const plugins = app.webiny.plugins.byType(_HandlerResultPlugin.HandlerResultPlugin.type);
     let name;
-
     try {
       for (const plugin of plugins) {
         name = plugin.name;
         await plugin.handle(app.webiny, payload);
       }
     } catch (ex) {
-      console.log(`Error while running the "HandlerResultPlugin" ${name ? `(${name})` : ""} plugin in the preSerialization hook.`);
-      console.log(stringifyError(ex));
+      console.error(`Error while running the "HandlerResultPlugin" ${name ? `(${name})` : ""} plugin in the preSerialization hook.`);
+      console.error(stringifyError(ex));
       throw ex;
     }
-
     return payload;
   };
-
   app.addHook("preSerialization", preSerialization);
   app.setErrorHandler(async (error, request, reply) => {
     return reply.status(500).headers({
@@ -437,9 +365,8 @@ const createHandler = params => {
     /**
      * Log error to cloud, as these can be extremely annoying to debug!
      */
-
-    console.log("@webiny/handler");
-    console.log(stringifyError(error));
+    console.error("@webiny/handler");
+    console.error(stringifyError(error));
     reply.status(500).headers({
       "Cache-Control": "no-store"
     }).send(
@@ -459,25 +386,74 @@ const createHandler = params => {
     await handler(app.webiny, error);
     return reply;
   });
+
+  /**
+   * Apply response headers modifier plugins.
+   */
+  app.addHook("onSend", async (request, reply, payload) => {
+    const modifyHeaders = app.webiny.plugins.byType(_ModifyResponseHeadersPlugin.ModifyResponseHeadersPlugin.type);
+    const headers = _ResponseHeaders.ResponseHeaders.create(reply.getHeaders());
+    modifyHeaders.forEach(plugin => {
+      plugin.modify(request, headers);
+    });
+    reply.headers(headers.getHeaders());
+    return payload;
+  });
+
   /**
-   * With these plugins we give users possibility to do anything they want on our fastify instance.
+   * We need to output the benchmark results at the end of the request in both response and timeout cases
    */
+  app.addHook("onResponse", async () => {
+    await context.benchmark.output();
+  });
+  app.addHook("onTimeout", async () => {
+    await context.benchmark.output();
+  });
 
+  /**
+   * With these plugins we give users possibility to do anything they want on our fastify instance.
+   */
   const modifyPlugins = app.webiny.plugins.byType(_ModifyFastifyPlugin.ModifyFastifyPlugin.type);
   let modifyFastifyPluginName;
-
   try {
     for (const plugin of modifyPlugins) {
       modifyFastifyPluginName = plugin.name;
       plugin.modify(app);
     }
   } catch (ex) {
-    console.log(`Error while running the "ModifyFastifyPlugin" ${modifyFastifyPluginName ? `(${modifyFastifyPluginName})` : ""} plugin in the end of the "createHandler" callable.`);
-    console.log(stringifyError(ex));
+    console.error(`Error while running the "ModifyFastifyPlugin" ${modifyFastifyPluginName ? `(${modifyFastifyPluginName})` : ""} plugin in the end of the "createHandler" callable.`);
+    console.error(stringifyError(ex));
     throw ex;
   }
 
+  /**
+   * We have few types of triggers:
+   *  * Events - EventPlugin
+   *  * Routes - RoutePlugin
+   *
+   * Routes are registered in fastify but events must be handled in package which implements cloud specific methods.
+   */
+  const routePlugins = app.webiny.plugins.byType(_RoutePlugin.RoutePlugin.type);
+
+  /**
+   * Add routes to the system.
+   */
+  let routePluginName;
+  try {
+    for (const plugin of routePlugins) {
+      routePluginName = plugin.name;
+      plugin.cb({
+        ...app.webiny.routes,
+        context: app.webiny
+      });
+    }
+  } catch (ex) {
+    console.error(`Error while running the "RoutePlugin" ${routePluginName ? `(${routePluginName})` : ""} plugin in the beginning of the "createHandler" callable.`);
+    console.error(stringifyError(ex));
+    throw ex;
+  }
   return app;
 };
+exports.createHandler = createHandler;
 
-exports.createHandler = createHandler;
+//# sourceMappingURL=fastify.js.map