EU token support

Author: petyosi

.changeset/fine-apples-rest.md

@@ -0,0 +1,5 @@
+---
+"logfire": minor
+---
+
+Support us/eu tokens

.gitignore

@@ -3,3 +3,4 @@ packages/*/*.tgz
 .turbo
 .env
 CLAUDE.md
+scratch/

package-lock.json

@@ -45,7 +45,8 @@
     "preview": "vite preview",
     "typecheck": "tsc",
     "prepack": "cp ../../LICENSE .",
-    "postpack": "rm LICENSE"
+    "postpack": "rm LICENSE",
+    "test": "vitest"
   },
   "dependencies": {
     "@pydantic/logfire-api": "*"
@@ -67,7 +68,7 @@
     "typescript": "^5.8.2",
     "vite": "^6.2.0",
     "vite-plugin-dts": "^4.5.3",
-    "vitest": "^3.0.8"
+    "vitest": "^3.1.1"
   },
   "peerDependencies": {
     "@opentelemetry/api": "^1.9.0",

packages/logfire/package.json

@@ -0,0 +1,18 @@
+import { expect, test } from 'vitest'
+
+import { resolveBaseUrl } from './logfireConfig'
+
+test('returns the passed url', () => {
+  const baseUrl = resolveBaseUrl('https://example.com', 'token')
+  expect(baseUrl).toBe('https://example.com')
+})
+
+test('resolves the US base url from the token', () => {
+  const baseUrl = resolveBaseUrl(undefined, 'pylf_v1_us_1234567890')
+  expect(baseUrl).toBe('https://logfire-us.pydantic.dev')
+})
+
+test('resolves the EU base url from the token', () => {
+  const baseUrl = resolveBaseUrl(undefined, 'pylf_v1_eu_mFMvBQ7BWLPJ0fHYBGLVBmJ70TpkhlskgRLng0jFsb3n')
+  expect(baseUrl).toBe('https://logfire-eu.pydantic.dev')
+})

packages/logfire/src/logfireConfig.test.ts

@@ -43,6 +43,11 @@ export interface SrubbingOptions {
   extraPatterns?: string[]
 }
 
+export interface RegionData {
+  baseUrl: string
+  gcpRegion: string
+}
+
 export interface LogfireConfigOptions {
   /**
    * Additional span processors to be added to the OpenTelemetry SDK
@@ -103,7 +108,6 @@ export interface LogfireConfigOptions {
   token?: string
 }
 
-const DEFAULT_LOGFIRE_BASE_URL = 'https://logfire-api.pydantic.dev/'
 const DEFAULT_OTEL_SCOPE = 'logfire'
 const TRACE_ENDPOINT_PATH = 'v1/traces'
 const METRIC_ENDPOINT_PATH = 'v1/metrics'
@@ -161,8 +165,8 @@ export function configure(config: LogfireConfigOptions = {}) {
     logfireApi.configureLogfireApi({ otelScope })
   }
 
-  const baseUrl = resolveBaseUrl(cnf.advanced?.baseUrl)
   const token = cnf.token ?? env.LOGFIRE_TOKEN
+  const baseUrl = resolveBaseUrl(cnf.advanced?.baseUrl, token)
 
   Object.assign(logfireConfig, {
     additionalSpanProcessors: cnf.additionalSpanProcessors ?? [],
@@ -214,8 +218,8 @@ function resolveSendToLogfire(option: LogfireConfigOptions['sendToLogfire'], tok
   }
 }
 
-function resolveBaseUrl(option: string | undefined) {
-  let url = option ?? process.env.LOGFIRE_BASE_URL ?? DEFAULT_LOGFIRE_BASE_URL
+export function resolveBaseUrl(passedUrl: string | undefined, token: string | undefined) {
+  let url = passedUrl ?? process.env.LOGFIRE_BASE_URL ?? getBaseUrlFromToken(token)
   if (url.endsWith('/')) {
     url = url.slice(0, -1)
   }
@@ -226,3 +230,30 @@ function resolveDistributedTracing(option: LogfireConfigOptions['distributedTrac
   const envDistributedTracing = process.env.LOGFIRE_DISTRIBUTED_TRACING
   return (option ?? envDistributedTracing === undefined) ? true : envDistributedTracing === 'true'
 }
+
+const PYDANTIC_LOGFIRE_TOKEN_PATTERN = /^(?<safe_part>pylf_v(?<version>[0-9]+)_(?<region>[a-z]+)_)(?<token>[a-zA-Z0-9]+)$/
+
+const REGIONS: Record<string, RegionData> = {
+  eu: {
+    baseUrl: 'https://logfire-eu.pydantic.dev',
+    gcpRegion: 'europe-west4',
+  },
+  us: {
+    baseUrl: 'https://logfire-us.pydantic.dev',
+    gcpRegion: 'us-east4',
+  },
+}
+
+function getBaseUrlFromToken(token: string | undefined): string {
+  let regionKey = 'us'
+  if (token) {
+    const match = PYDANTIC_LOGFIRE_TOKEN_PATTERN.exec(token)
+    if (match) {
+      const region = match.groups?.region
+      if (region && region in REGIONS) {
+        regionKey = region
+      }
+    }
+  }
+  return REGIONS[regionKey].baseUrl
+}