chore(core): add custom domain host to app insights

logto-io · May 13, 2024 · e479b08 · e479b08
1 parent 1fdd28b
commit e479b08
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/packages/core/src/utils/request.ts b/packages/core/src/utils/request.ts
@@ -1,4 +1,5 @@
 import { type ExceptionTelemetry } from '@logto/app-insights/node';
+import { type Context } from 'koa';
 
 // eslint-disable-next-line @typescript-eslint/ban-types
 const getRequestIdFromContext = (context: object): string | undefined => {
@@ -9,13 +10,28 @@ const getRequestIdFromContext = (context: object): string | undefined => {
   return undefined;
 };
 
+const getHostFromContext = (context: Context): string | undefined => {
+  if ('host' in context.headers && typeof context.headers.host === 'string') {
+    return context.headers.host;
+  }
+
+  return undefined;
+};
+
 // eslint-disable-next-line @typescript-eslint/ban-types
 export const buildAppInsightsTelemetry = (context: object): Partial<ExceptionTelemetry> => {
   const requestId = getRequestIdFromContext(context);
+  // eslint-disable-next-line no-restricted-syntax
+  const host = getHostFromContext(context as Context);
 
-  if (requestId) {
-    return { properties: { requestId } };
+  if (!requestId && !host) {
+    return {};
   }
 
-  return {};
+  return {
+    properties: {
+      ...(requestId ? { requestId } : {}),
+      ...(host ? { host } : {}),
+    },
+  };
 };
diff --git a/packages/core/src/utils/test-utils.ts b/packages/core/src/utils/test-utils.ts
@@ -97,6 +97,7 @@ export const createContextWithRouteParameters = (
     path: ctx.path,
     URL: ctx.URL,
     params: {},
+    headers: {},
     router: new Router(),
     _matchedRoute: undefined,
     _matchedRouteName: undefined,