init claude-code

utils/telemetryAttributes.ts

@@ -0,0 +1,71 @@
+import type { Attributes } from '@opentelemetry/api'
+import { getSessionId } from 'src/bootstrap/state.js'
+import { getOauthAccountInfo } from './auth.js'
+import { getOrCreateUserID } from './config.js'
+import { envDynamic } from './envDynamic.js'
+import { isEnvTruthy } from './envUtils.js'
+import { toTaggedId } from './taggedId.js'
+
+// Default configuration for metrics cardinality
+const METRICS_CARDINALITY_DEFAULTS = {
+  OTEL_METRICS_INCLUDE_SESSION_ID: true,
+  OTEL_METRICS_INCLUDE_VERSION: false,
+  OTEL_METRICS_INCLUDE_ACCOUNT_UUID: true,
+}
+
+function shouldIncludeAttribute(
+  envVar: keyof typeof METRICS_CARDINALITY_DEFAULTS,
+): boolean {
+  const defaultValue = METRICS_CARDINALITY_DEFAULTS[envVar]
+  const envValue = process.env[envVar]
+
+  if (envValue === undefined) {
+    return defaultValue
+  }
+
+  return isEnvTruthy(envValue)
+}
+
+export function getTelemetryAttributes(): Attributes {
+  const userId = getOrCreateUserID()
+  const sessionId = getSessionId()
+
+  const attributes: Attributes = {
+    'user.id': userId,
+  }
+
+  if (shouldIncludeAttribute('OTEL_METRICS_INCLUDE_SESSION_ID')) {
+    attributes['session.id'] = sessionId
+  }
+  if (shouldIncludeAttribute('OTEL_METRICS_INCLUDE_VERSION')) {
+    attributes['app.version'] = MACRO.VERSION
+  }
+
+  // Only include OAuth account data when actively using OAuth authentication
+  const oauthAccount = getOauthAccountInfo()
+  if (oauthAccount) {
+    const orgId = oauthAccount.organizationUuid
+    const email = oauthAccount.emailAddress
+    const accountUuid = oauthAccount.accountUuid
+
+    if (orgId) attributes['organization.id'] = orgId
+    if (email) attributes['user.email'] = email
+
+    if (
+      accountUuid &&
+      shouldIncludeAttribute('OTEL_METRICS_INCLUDE_ACCOUNT_UUID')
+    ) {
+      attributes['user.account_uuid'] = accountUuid
+      attributes['user.account_id'] =
+        process.env.CLAUDE_CODE_ACCOUNT_TAGGED_ID ||
+        toTaggedId('user', accountUuid)
+    }
+  }
+
+  // Add terminal type if available
+  if (envDynamic.terminal) {
+    attributes['terminal.type'] = envDynamic.terminal
+  }
+
+  return attributes
+}