{"version":3,"sources":["../../src/hapi/plugin.ts"],"sourcesContent":["import type { Request, Server } from \"@hapi/hapi\";\nimport { AsyncLocalStorage } from \"node:async_hooks\";\nimport { performance } from \"node:perf_hooks\";\n\nimport { ApitallyClient } from \"../common/client.js\";\nimport { consumerFromStringOrObject } from \"../common/consumerRegistry.js\";\nimport type { LogRecord } from \"../common/requestLogger.js\";\nimport { convertHeaders } from \"../common/requestLogger.js\";\nimport type { SpanHandle } from \"../common/spanCollector.js\";\nimport { ApitallyConfig, ApitallyConsumer } from \"../common/types.js\";\nimport {\n  handleHapiRequestEvent,\n  patchConsole,\n  patchPinoLogger,\n  patchWinston,\n} from \"../loggers/index.js\";\nimport { getAppInfo, isBoom } from \"./utils.js\";\n\nconst START_TIME_SYMBOL = Symbol(\"apitally.startTime\");\nconst REQUEST_BODY_SYMBOL = Symbol(\"apitally.requestBody\");\nconst REQUEST_SIZE_SYMBOL = Symbol(\"apitally.requestSize\");\nconst RESPONSE_BODY_SYMBOL = Symbol(\"apitally.responseBody\");\nconst RESPONSE_SIZE_SYMBOL = Symbol(\"apitally.responseSize\");\nconst LOGS_SYMBOL = Symbol(\"apitally.logs\");\nconst SPAN_HANDLE_SYMBOL = Symbol(\"apitally.spanHandle\");\n\ndeclare module \"@hapi/hapi\" {\n  interface Request {\n    [START_TIME_SYMBOL]?: number;\n    [REQUEST_BODY_SYMBOL]?: Buffer;\n    [REQUEST_SIZE_SYMBOL]?: number;\n    [RESPONSE_BODY_SYMBOL]?: Buffer;\n    [RESPONSE_SIZE_SYMBOL]?: number;\n    [LOGS_SYMBOL]?: LogRecord[];\n    [SPAN_HANDLE_SYMBOL]?: SpanHandle;\n    apitallyConsumer?: ApitallyConsumer | string;\n  }\n\n  interface ResponseObject {\n    readonly contentType: string | null; // Missing in Hapi types\n  }\n}\n\nexport default function apitallyPlugin(config: ApitallyConfig) {\n  const client = new ApitallyClient(config);\n  const logsContext = new AsyncLocalStorage<LogRecord[]>();\n  let pinoLoggerPatched = false;\n\n  if (client.requestLogger.enabled && client.requestLogger.config.captureLogs) {\n    patchConsole(logsContext);\n    patchWinston(logsContext);\n  }\n\n  return {\n    name: \"apitally\",\n    register: async function (server: Server) {\n      server.ext(\"onPostStart\", async () => {\n        if (\n          \"logger\" in server &&\n          client.requestLogger.enabled &&\n          client.requestLogger.config.captureLogs\n        ) {\n          pinoLoggerPatched = await patchPinoLogger(\n            (server as any).logger,\n            logsContext,\n          );\n        }\n      });\n\n      server.ext(\"onPostStart\", () => {\n        const appInfo = getAppInfo(server, config.appVersion);\n        client.setStartupData(appInfo);\n        client.startSync();\n      });\n\n      server.ext(\"onPreStop\", async () => {\n        await client.handleShutdown();\n      });\n\n      server.events.on(\"request\", (_, event) => {\n        if (\n          !pinoLoggerPatched &&\n          client.requestLogger.enabled &&\n          client.requestLogger.config.captureLogs &&\n          event.channel === \"app\"\n        ) {\n          handleHapiRequestEvent(event, logsContext);\n        }\n      });\n\n      server.ext(\"onRequest\", async (request, h) => {\n        if (!client.isEnabled() || request.method.toUpperCase() === \"OPTIONS\") {\n          return h.continue;\n        }\n\n        request[START_TIME_SYMBOL] = performance.now();\n\n        // Patch the lifecycle function to run with span and logs context\n        const spanHandle = client.spanCollector.startSpan();\n        const logs: LogRecord[] = [];\n        request[SPAN_HANDLE_SYMBOL] = spanHandle;\n        request[LOGS_SYMBOL] = logs;\n        const lifecycle = (request as any)._lifecycle.bind(request);\n        (request as any)._lifecycle = () =>\n          spanHandle.runInContext(() => logsContext.run(logs, lifecycle));\n\n        const captureRequestBody =\n          client.requestLogger.enabled &&\n          client.requestLogger.config.logRequestBody &&\n          client.requestLogger.isSupportedContentType(\n            request.headers[\"content-type\"],\n          );\n        const chunks: Buffer[] = [];\n        let size = 0;\n        request.events.on(\"peek\", (chunk: string, encoding: string) => {\n          if (captureRequestBody) {\n            chunks.push(Buffer.from(chunk, encoding as BufferEncoding));\n          }\n          size += Buffer.byteLength(chunk, encoding as BufferEncoding);\n        });\n        request.events.once(\"finish\", () => {\n          if (chunks.length > 0) {\n            request[REQUEST_BODY_SYMBOL] = Buffer.concat(chunks);\n          }\n          request[REQUEST_SIZE_SYMBOL] = size;\n        });\n\n        return h.continue;\n      });\n\n      server.ext(\"onPreResponse\", async (request, h) => {\n        if (\n          !client.isEnabled() ||\n          request.method.toUpperCase() === \"OPTIONS\" ||\n          isBoom(request.response)\n        ) {\n          return h.continue;\n        }\n\n        const captureResponseBody =\n          client.requestLogger.enabled &&\n          client.requestLogger.config.logResponseBody &&\n          client.requestLogger.isSupportedContentType(\n            request.response.contentType,\n          );\n        const chunks: Buffer[] = [];\n        let size = 0;\n        request.response.events.on(\n          \"peek\",\n          (chunk: string, encoding: string) => {\n            if (captureResponseBody) {\n              chunks.push(Buffer.from(chunk, encoding as BufferEncoding));\n            }\n            size += Buffer.byteLength(chunk, encoding as BufferEncoding);\n          },\n        );\n        request.response.events.once(\"finish\", () => {\n          if (chunks.length > 0) {\n            request[RESPONSE_BODY_SYMBOL] = Buffer.concat(chunks);\n          }\n          request[RESPONSE_SIZE_SYMBOL] = size;\n        });\n\n        return h.continue;\n      });\n\n      server.ext(\"onPostResponse\", async (request, h) => {\n        if (!client.isEnabled() || request.method.toUpperCase() === \"OPTIONS\") {\n          return h.continue;\n        }\n\n        const startTime = request[START_TIME_SYMBOL];\n        const responseTime = startTime ? performance.now() - startTime : 0;\n        const timestamp = (Date.now() - responseTime) / 1000;\n\n        const spanHandle = request[SPAN_HANDLE_SYMBOL];\n        spanHandle?.setName(\n          `${request.method.toUpperCase()} ${request.route.path}`,\n        );\n        const spans = spanHandle?.end();\n        const traceId = spanHandle?.traceId;\n\n        const requestBody = request[REQUEST_BODY_SYMBOL];\n        const requestSize = request[REQUEST_SIZE_SYMBOL];\n        const response = request.response;\n        let statusCode: number;\n        let responseHeaders: Record<string, any>;\n        let responseBody = request[RESPONSE_BODY_SYMBOL];\n        let responseSize = request[RESPONSE_SIZE_SYMBOL];\n        const error =\n          (response as any)._error instanceof Error\n            ? (response as any)._error\n            : undefined;\n\n        if (isBoom(response)) {\n          // Handle Boom error object\n          statusCode = response.output?.statusCode ?? 500;\n          responseHeaders = response.output?.headers ?? {};\n        } else {\n          // Handle normal ResponseObject\n          statusCode = response.statusCode ?? 200;\n          responseHeaders = response.headers ?? {};\n        }\n\n        if (\n          !responseBody &&\n          (response as any)._payload &&\n          client.requestLogger.enabled &&\n          client.requestLogger.config.logResponseBody &&\n          (isBoom(response) ||\n            client.requestLogger.isSupportedContentType(response.contentType))\n        ) {\n          const responsePayload = (response as any)._payload;\n          if (\n            responsePayload &&\n            responsePayload._data &&\n            responsePayload._encoding &&\n            typeof responsePayload._data === \"string\" &&\n            typeof responsePayload._encoding === \"string\"\n          ) {\n            responseBody = Buffer.from(\n              responsePayload._data,\n              responsePayload._encoding as BufferEncoding,\n            );\n          }\n        }\n\n        if (!responseSize && responseBody) {\n          responseSize = responseBody.length;\n        }\n\n        const consumer = request.apitallyConsumer\n          ? consumerFromStringOrObject(request.apitallyConsumer)\n          : null;\n        client.consumerRegistry.addOrUpdateConsumer(consumer);\n\n        if (request.route.path) {\n          client.requestCounter.addRequest({\n            consumer: consumer?.identifier,\n            method: request.method,\n            path: request.route.path,\n            statusCode,\n            responseTime,\n            requestSize,\n            responseSize,\n          });\n\n          if (statusCode === 500 && error) {\n            client.serverErrorCounter.addServerError({\n              consumer: consumer?.identifier,\n              method: request.method,\n              path: request.route.path,\n              type: error.name,\n              msg: error.message,\n              traceback: error.stack || \"\",\n            });\n          }\n        }\n\n        if (client.requestLogger.enabled) {\n          const logs = request[LOGS_SYMBOL];\n          client.requestLogger.logRequest(\n            {\n              timestamp,\n              consumer: consumer?.identifier,\n              method: request.method,\n              path: request.route.path,\n              url: request.url?.href || \"\",\n              headers: convertHeaders(request.headers),\n              size: requestSize,\n              body: requestBody,\n            },\n            {\n              statusCode,\n              responseTime: responseTime / 1000,\n              headers: convertHeaders(responseHeaders),\n              size: responseSize,\n              body: responseBody,\n            },\n            error,\n            logs,\n            spans,\n            traceId,\n          );\n        }\n\n        return h.continue;\n      });\n    },\n  };\n}\n\nexport function setConsumer(\n  request: Request,\n  consumer: ApitallyConsumer | string | null | undefined,\n) {\n  request.apitallyConsumer = consumer || undefined;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AACA;;;;;;AAAA,8BAAkC;AAClC,6BAA4B;AAE5B,oBAA+B;AAC/B,8BAA2C;AAE3C,2BAA+B;AAG/B,qBAKO;AACP,mBAAmC;AAEnC,MAAMA,oBAAoBC,uBAAO,oBAAA;AACjC,MAAMC,sBAAsBD,uBAAO,sBAAA;AACnC,MAAME,sBAAsBF,uBAAO,sBAAA;AACnC,MAAMG,uBAAuBH,uBAAO,uBAAA;AACpC,MAAMI,uBAAuBJ,uBAAO,uBAAA;AACpC,MAAMK,cAAcL,uBAAO,eAAA;AAC3B,MAAMM,qBAAqBN,uBAAO,qBAAA;AAmBnB,SAAf,eAAuCO,QAAsB;AAC3D,QAAMC,SAAS,IAAIC,6BAAeF,MAAAA;AAClC,QAAMG,cAAc,IAAIC,0CAAAA;AACxB,MAAIC,oBAAoB;AAExB,MAAIJ,OAAOK,cAAcC,WAAWN,OAAOK,cAAcN,OAAOQ,aAAa;AAC3EC,qCAAaN,WAAAA;AACbO,qCAAaP,WAAAA;EACf;AAEA,SAAO;IACLQ,MAAM;IACNC,UAAU,sCAAgBC,QAAc;AACtCA,aAAOC,IAAI,eAAe,YAAA;AACxB,YACE,YAAYD,UACZZ,OAAOK,cAAcC,WACrBN,OAAOK,cAAcN,OAAOQ,aAC5B;AACAH,8BAAoB,UAAMU,gCACvBF,OAAeG,QAChBb,WAAAA;QAEJ;MACF,CAAA;AAEAU,aAAOC,IAAI,eAAe,MAAA;AACxB,cAAMG,cAAUC,yBAAWL,QAAQb,OAAOmB,UAAU;AACpDlB,eAAOmB,eAAeH,OAAAA;AACtBhB,eAAOoB,UAAS;MAClB,CAAA;AAEAR,aAAOC,IAAI,aAAa,YAAA;AACtB,cAAMb,OAAOqB,eAAc;MAC7B,CAAA;AAEAT,aAAOU,OAAOC,GAAG,WAAW,CAACC,GAAGC,UAAAA;AAC9B,YACE,CAACrB,qBACDJ,OAAOK,cAAcC,WACrBN,OAAOK,cAAcN,OAAOQ,eAC5BkB,MAAMC,YAAY,OAClB;AACAC,qDAAuBF,OAAOvB,WAAAA;QAChC;MACF,CAAA;AAEAU,aAAOC,IAAI,aAAa,OAAOe,SAASC,MAAAA;AACtC,YAAI,CAAC7B,OAAO8B,UAAS,KAAMF,QAAQG,OAAOC,YAAW,MAAO,WAAW;AACrE,iBAAOH,EAAEI;QACX;AAEAL,gBAAQrC,iBAAAA,IAAqB2C,mCAAYC,IAAG;AAG5C,cAAMC,aAAapC,OAAOqC,cAAcC,UAAS;AACjD,cAAMC,OAAoB,CAAA;AAC1BX,gBAAQ9B,kBAAAA,IAAsBsC;AAC9BR,gBAAQ/B,WAAAA,IAAe0C;AACvB,cAAMC,YAAaZ,QAAgBa,WAAWC,KAAKd,OAAAA;AAClDA,gBAAgBa,aAAa,MAC5BL,WAAWO,aAAa,MAAMzC,YAAY0C,IAAIL,MAAMC,SAAAA,CAAAA;AAEtD,cAAMK,qBACJ7C,OAAOK,cAAcC,WACrBN,OAAOK,cAAcN,OAAO+C,kBAC5B9C,OAAOK,cAAc0C,uBACnBnB,QAAQoB,QAAQ,cAAA,CAAe;AAEnC,cAAMC,SAAmB,CAAA;AACzB,YAAIC,OAAO;AACXtB,gBAAQN,OAAOC,GAAG,QAAQ,CAAC4B,OAAeC,aAAAA;AACxC,cAAIP,oBAAoB;AACtBI,mBAAOI,KAAKC,OAAOC,KAAKJ,OAAOC,QAAAA,CAAAA;UACjC;AACAF,kBAAQI,OAAOE,WAAWL,OAAOC,QAAAA;QACnC,CAAA;AACAxB,gBAAQN,OAAOmC,KAAK,UAAU,MAAA;AAC5B,cAAIR,OAAOS,SAAS,GAAG;AACrB9B,oBAAQnC,mBAAAA,IAAuB6D,OAAOK,OAAOV,MAAAA;UAC/C;AACArB,kBAAQlC,mBAAAA,IAAuBwD;QACjC,CAAA;AAEA,eAAOrB,EAAEI;MACX,CAAA;AAEArB,aAAOC,IAAI,iBAAiB,OAAOe,SAASC,MAAAA;AAC1C,YACE,CAAC7B,OAAO8B,UAAS,KACjBF,QAAQG,OAAOC,YAAW,MAAO,iBACjC4B,qBAAOhC,QAAQiC,QAAQ,GACvB;AACA,iBAAOhC,EAAEI;QACX;AAEA,cAAM6B,sBACJ9D,OAAOK,cAAcC,WACrBN,OAAOK,cAAcN,OAAOgE,mBAC5B/D,OAAOK,cAAc0C,uBACnBnB,QAAQiC,SAASG,WAAW;AAEhC,cAAMf,SAAmB,CAAA;AACzB,YAAIC,OAAO;AACXtB,gBAAQiC,SAASvC,OAAOC,GACtB,QACA,CAAC4B,OAAeC,aAAAA;AACd,cAAIU,qBAAqB;AACvBb,mBAAOI,KAAKC,OAAOC,KAAKJ,OAAOC,QAAAA,CAAAA;UACjC;AACAF,kBAAQI,OAAOE,WAAWL,OAAOC,QAAAA;QACnC,CAAA;AAEFxB,gBAAQiC,SAASvC,OAAOmC,KAAK,UAAU,MAAA;AACrC,cAAIR,OAAOS,SAAS,GAAG;AACrB9B,oBAAQjC,oBAAAA,IAAwB2D,OAAOK,OAAOV,MAAAA;UAChD;AACArB,kBAAQhC,oBAAAA,IAAwBsD;QAClC,CAAA;AAEA,eAAOrB,EAAEI;MACX,CAAA;AAEArB,aAAOC,IAAI,kBAAkB,OAAOe,SAASC,MAAAA;AArKnD;AAsKQ,YAAI,CAAC7B,OAAO8B,UAAS,KAAMF,QAAQG,OAAOC,YAAW,MAAO,WAAW;AACrE,iBAAOH,EAAEI;QACX;AAEA,cAAMgC,YAAYrC,QAAQrC,iBAAAA;AAC1B,cAAM2E,eAAeD,YAAY/B,mCAAYC,IAAG,IAAK8B,YAAY;AACjE,cAAME,aAAaC,KAAKjC,IAAG,IAAK+B,gBAAgB;AAEhD,cAAM9B,aAAaR,QAAQ9B,kBAAAA;AAC3BsC,iDAAYiC,QACV,GAAGzC,QAAQG,OAAOC,YAAW,CAAA,IAAMJ,QAAQ0C,MAAMC,IAAI;AAEvD,cAAMC,QAAQpC,yCAAYqC;AAC1B,cAAMC,UAAUtC,yCAAYsC;AAE5B,cAAMC,cAAc/C,QAAQnC,mBAAAA;AAC5B,cAAMmF,cAAchD,QAAQlC,mBAAAA;AAC5B,cAAMmE,WAAWjC,QAAQiC;AACzB,YAAIgB;AACJ,YAAIC;AACJ,YAAIC,eAAenD,QAAQjC,oBAAAA;AAC3B,YAAIqF,eAAepD,QAAQhC,oBAAAA;AAC3B,cAAMqF,QACHpB,SAAiBqB,kBAAkBC,QAC/BtB,SAAiBqB,SAClBE;AAEN,gBAAIxB,qBAAOC,QAAAA,GAAW;AAEpBgB,yBAAahB,cAASwB,WAATxB,mBAAiBgB,eAAc;AAC5CC,8BAAkBjB,cAASwB,WAATxB,mBAAiBb,YAAW,CAAC;QACjD,OAAO;AAEL6B,uBAAahB,SAASgB,cAAc;AACpCC,4BAAkBjB,SAASb,WAAW,CAAC;QACzC;AAEA,YACE,CAAC+B,gBACAlB,SAAiByB,YAClBtF,OAAOK,cAAcC,WACrBN,OAAOK,cAAcN,OAAOgE,wBAC3BH,qBAAOC,QAAAA,KACN7D,OAAOK,cAAc0C,uBAAuBc,SAASG,WAAW,IAClE;AACA,gBAAMuB,kBAAmB1B,SAAiByB;AAC1C,cACEC,mBACAA,gBAAgBC,SAChBD,gBAAgBE,aAChB,OAAOF,gBAAgBC,UAAU,YACjC,OAAOD,gBAAgBE,cAAc,UACrC;AACAV,2BAAezB,OAAOC,KACpBgC,gBAAgBC,OAChBD,gBAAgBE,SAAS;UAE7B;QACF;AAEA,YAAI,CAACT,gBAAgBD,cAAc;AACjCC,yBAAeD,aAAarB;QAC9B;AAEA,cAAMgC,WAAW9D,QAAQ+D,uBACrBC,oDAA2BhE,QAAQ+D,gBAAgB,IACnD;AACJ3F,eAAO6F,iBAAiBC,oBAAoBJ,QAAAA;AAE5C,YAAI9D,QAAQ0C,MAAMC,MAAM;AACtBvE,iBAAO+F,eAAeC,WAAW;YAC/BN,UAAUA,qCAAUO;YACpBlE,QAAQH,QAAQG;YAChBwC,MAAM3C,QAAQ0C,MAAMC;YACpBM;YACAX;YACAU;YACAI;UACF,CAAA;AAEA,cAAIH,eAAe,OAAOI,OAAO;AAC/BjF,mBAAOkG,mBAAmBC,eAAe;cACvCT,UAAUA,qCAAUO;cACpBlE,QAAQH,QAAQG;cAChBwC,MAAM3C,QAAQ0C,MAAMC;cACpB6B,MAAMnB,MAAMvE;cACZ2F,KAAKpB,MAAMqB;cACXC,WAAWtB,MAAMuB,SAAS;YAC5B,CAAA;UACF;QACF;AAEA,YAAIxG,OAAOK,cAAcC,SAAS;AAChC,gBAAMiC,OAAOX,QAAQ/B,WAAAA;AACrBG,iBAAOK,cAAcoG,WACnB;YACEtC;YACAuB,UAAUA,qCAAUO;YACpBlE,QAAQH,QAAQG;YAChBwC,MAAM3C,QAAQ0C,MAAMC;YACpBmC,OAAK9E,aAAQ8E,QAAR9E,mBAAa+E,SAAQ;YAC1B3D,aAAS4D,qCAAehF,QAAQoB,OAAO;YACvCE,MAAM0B;YACNiC,MAAMlC;UACR,GACA;YACEE;YACAX,cAAcA,eAAe;YAC7BlB,aAAS4D,qCAAe9B,eAAAA;YACxB5B,MAAM8B;YACN6B,MAAM9B;UACR,GACAE,OACA1C,MACAiC,OACAE,OAAAA;QAEJ;AAEA,eAAO7C,EAAEI;MACX,CAAA;IACF,GAzOU;EA0OZ;AACF;AAvPwB6E;AAyPjB,SAASC,YACdnF,SACA8D,UAAsD;AAEtD9D,UAAQ+D,mBAAmBD,YAAYN;AACzC;AALgB2B;","names":["START_TIME_SYMBOL","Symbol","REQUEST_BODY_SYMBOL","REQUEST_SIZE_SYMBOL","RESPONSE_BODY_SYMBOL","RESPONSE_SIZE_SYMBOL","LOGS_SYMBOL","SPAN_HANDLE_SYMBOL","config","client","ApitallyClient","logsContext","AsyncLocalStorage","pinoLoggerPatched","requestLogger","enabled","captureLogs","patchConsole","patchWinston","name","register","server","ext","patchPinoLogger","logger","appInfo","getAppInfo","appVersion","setStartupData","startSync","handleShutdown","events","on","_","event","channel","handleHapiRequestEvent","request","h","isEnabled","method","toUpperCase","continue","performance","now","spanHandle","spanCollector","startSpan","logs","lifecycle","_lifecycle","bind","runInContext","run","captureRequestBody","logRequestBody","isSupportedContentType","headers","chunks","size","chunk","encoding","push","Buffer","from","byteLength","once","length","concat","isBoom","response","captureResponseBody","logResponseBody","contentType","startTime","responseTime","timestamp","Date","setName","route","path","spans","end","traceId","requestBody","requestSize","statusCode","responseHeaders","responseBody","responseSize","error","_error","Error","undefined","output","_payload","responsePayload","_data","_encoding","consumer","apitallyConsumer","consumerFromStringOrObject","consumerRegistry","addOrUpdateConsumer","requestCounter","addRequest","identifier","serverErrorCounter","addServerError","type","msg","message","traceback","stack","logRequest","url","href","convertHeaders","body","apitallyPlugin","setConsumer"]}