@webiny/handler 0.0.0-unstable.1e66d121db → 0.0.0-unstable.2696f9d9e8

package/fastify.js

@@ -1,77 +1,83 @@
 "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");
-
-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 _ModifyFastifyPlugin = require("./plugins/ModifyFastifyPlugin");
+var _HandlerOnRequestPlugin = require("./plugins/HandlerOnRequestPlugin");
+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(key => {
-      const type = key;
-
-      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 OPTIONS_HEADERS = {
-  "Access-Control-Max-Age": "86400",
-  "Cache-Control": "public, max-age=86400"
+const stringifyError = error => {
+  const {
+    name,
+    message,
+    code,
+    stack,
+    data
+  } = error;
+  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 createHandler = params => {
   const definedRoutes = {
     POST: [],
@@ -91,80 +97,88 @@ const createHandler = params => {
     TRACE: [],
     UNLOCK: []
   };
-
   const throwOnDefinedRoute = (type, path, options) => {
     if (type === "ALL") {
-      const all = Object.keys(definedRoutes).some(key => {
+      const all = Object.keys(definedRoutes).find(key => {
         const routes = definedRoutes[key];
         return routes.includes(path);
       });
-
       if (!all) {
         return;
       }
-
+      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.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 = (inputType, path) => {
-    const type = inputType.toUpperCase();
-
+  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)({
+    ...(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,
+    onUnsupportedEncoding: (encoding, _, reply) => {
+      reply.code(406);
+      return `We do not support the ${encoding} encoding.`;
+    },
+    inflateIfDeflated: true
   });
   /**
    * Route helpers - mostly for users.
    */
-
   const routes = {
     defined: definedRoutes,
     onPost: (path, handler, options) => {
@@ -200,120 +214,244 @@ const createHandler = params => {
       app.head(path, handler);
     }
   };
-  const context = new _Context.Context({
-    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.
-     */
-    WEBINY_VERSION: process.env.WEBINY_VERSION,
-    server: app,
-    routes
-  });
-  /**
-   * We are attaching our custom context to webiny variable on the fastify app so it is accessible everywhere
-   */
-
-  app.decorate("webiny", context);
+  let context;
+  const plugins = new _types.PluginsContainer([
   /**
-   * 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.
+   * 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,
+      /**
+       * Inserted via webpack at build time.
+       */
+      WEBINY_VERSION: process.env.WEBINY_VERSION,
+      routes
+    });
+  } catch (ex) {
+    console.error(`Error while constructing the Context.`);
+    console.error(stringifyError(ex));
+    throw ex;
+  }
 
-  const routePlugins = app.webiny.plugins.byType(_RoutePlugin.RoutePlugin.type);
   /**
-   * Add routes to the system.
+   * We are attaching our custom context to webiny variable on the fastify app, so it is accessible everywhere.
    */
+  app.decorate("webiny", context);
 
-  for (const plugin of routePlugins) {
-    plugin.cb((0, _objectSpread2.default)((0, _objectSpread2.default)({}, app.webiny.routes), {}, {
-      context: app.webiny
-    }));
-  }
   /**
    * 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);
-
-    if (request.method !== "OPTIONS") {
+    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.error(`Error while running the "HandlerOnRequestPlugin" ${name ? `(${name})` : ""} plugin in the onRequest hook.`);
+      console.error(stringifyError(ex));
+      throw ex;
+    }
+    /**
+     * When we receive the OPTIONS request, we end it before it goes any further as there is no need for anything to run after this - at least for our use cases.
+     *
+     * Users can prevent this by creating their own HandlerOnRequestPlugin and returning false as the result of the callable.
+     */
+    if (!isOptionsRequest) {
       return;
     }
-
-    const raw = reply.code(204).hijack().raw;
-    const headers = (0, _objectSpread2.default)((0, _objectSpread2.default)({}, defaultHeaders), OPTIONS_HEADERS);
-
-    for (const key in headers) {
-      raw.setHeader(key, headers[key]);
+    if (reply.sent) {
+      /**
+       * At this point throwing an exception will not do anything with the response. So just log it.
+       */
+      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;
     }
-
-    raw.end("");
+    reply.code(204).send("").hijack();
   });
-  app.addHook("preParsing", async request => {
+  app.addHook("preParsing", async (request, reply) => {
     app.webiny.request = request;
+    app.webiny.reply = reply;
     const plugins = app.webiny.plugins.byType(_api.ContextPlugin.type);
-
-    for (const plugin of plugins) {
-      await plugin.apply(app.webiny);
+    let name;
+    try {
+      for (const plugin of plugins) {
+        name = plugin.name;
+        await plugin.apply(app.webiny);
+      }
+    } catch (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);
-
-    for (const plugin of plugins) {
-      await plugin.apply(app.webiny);
+    let name;
+    try {
+      for (const plugin of plugins) {
+        name = plugin.name;
+        await plugin.apply(app.webiny);
+      }
+    } catch (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);
-
-    for (const plugin of plugins) {
-      await plugin.handle(app.webiny, payload);
+    let name;
+    try {
+      for (const plugin of plugins) {
+        name = plugin.name;
+        await plugin.handle(app.webiny, payload);
+      }
+    } catch (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({
+      "Cache-Control": "no-store"
+    }).send(
+    /**
+     * When we are sending the error in the response, we cannot send the whole error object, as it might contain some sensitive data.
+     */
+    JSON.stringify({
+      message: error.message,
+      code: error.code,
+      data: error.data
+    }));
+  });
   app.addHook("onError", async (_, reply, error) => {
-    const plugins = app.webiny.plugins.byType(_HandlerErrorPlugin.HandlerErrorPlugin.type); // Log error to cloud, as these can be extremely annoying to debug!
-
-    console.log("@webiny/handler");
-    console.log(JSON.stringify((0, _objectSpread2.default)((0, _objectSpread2.default)({}, error || {}), {}, {
-      message: error === null || error === void 0 ? void 0 : error.message,
-      code: error === null || error === void 0 ? void 0 : error.code
-    })));
+    const plugins = app.webiny.plugins.byType(_HandlerErrorPlugin.HandlerErrorPlugin.type);
+    /**
+     * Log error to cloud, as these can be extremely annoying to debug!
+     */
+    console.error("@webiny/handler");
+    console.error(stringifyError(error));
+    reply.status(500).headers({
+      "Cache-Control": "no-store"
+    }).send(
+    /**
+     * When we are sending the error in the response, we cannot send the whole error object, as it might contain some sensitive data.
+     */
+    JSON.stringify({
+      message: error.message,
+      code: error.code,
+      data: error.data
+    }));
     const handler = (0, _middleware.middleware)(plugins.map(pl => {
       return (context, error, next) => {
         return pl.handle(context, error, next);
       };
     }));
     await handler(app.webiny, error);
-    return reply.headers({
-      "Cache-Control": "no-store"
-    }).status(500);
+    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;
+  });
+
+  /**
+   * 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.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