[PROF-13798] [Browser Profiler] Quota check

packages/core/src/domain/configuration/configuration.ts

@@ -272,7 +272,7 @@ type GenericBeforeSendCallback = (event: any, context?: any) => unknown
  * path: /api/vX/product
  * parameters: xxx=yyy&zzz=aaa
  */
-export type ProxyFn = (options: { path: string; parameters: string }) => string
+export type ProxyFn = (options: { path: string; parameters: string; host: string }) => string
 
 /**
  * @internal

packages/core/src/domain/configuration/endpointBuilder.ts

@@ -65,10 +65,10 @@ function createEndpointUrlWithParametersBuilder(
     const normalizedProxyUrl = normalizeUrl(proxy)
     return (parameters) => `${normalizedProxyUrl}?ddforward=${encodeURIComponent(`${path}?${parameters}`)}`
   }
+  const host = buildEndpointHost(initConfiguration)
   if (typeof proxy === 'function') {
-    return (parameters) => proxy({ path, parameters })
+    return (parameters) => proxy({ path, parameters, host })
   }
-  const host = buildEndpointHost(initConfiguration)
   return (parameters) => `https://${host}${path}?${parameters}`
 }
 

packages/core/src/domain/configuration/transportConfiguration.ts

@@ -1,6 +1,6 @@
 import type { Site } from '../intakeSites'
 import { INTAKE_SITE_US1, INTAKE_URL_PARAMETERS } from '../intakeSites'
-import type { InitConfiguration, SdkSource } from './configuration'
+import type { InitConfiguration, ProxyFn, SdkSource } from './configuration'
 import type { EndpointBuilder, TransportSource } from './endpointBuilder'
 import { createEndpointBuilder } from './endpointBuilder'
 
@@ -14,6 +14,8 @@ export interface TransportConfiguration {
   debuggerEndpointBuilder: EndpointBuilder
   datacenter?: string | undefined
   replica?: ReplicaConfiguration
+  clientToken: string
+  proxy?: string | ProxyFn | undefined
   site: Site
   source: SdkSource
 }
@@ -55,6 +57,8 @@ export function computeTransportConfiguration(
   const replicaConfiguration = computeReplicaConfiguration(resolvedConfiguration)
 
   return {
+    clientToken: initConfiguration.clientToken,
+    proxy: initConfiguration.proxy,
     replica: replicaConfiguration,
     site,
     source,

packages/rum/src/domain/profiling/datadogProfiler.ts

@@ -31,6 +31,8 @@ import { assembleProfilingPayload } from './transport/assembly'
 import { createLongTaskHistory } from './longTaskHistory'
 import { createActionHistory } from './actionHistory'
 import { createVitalHistory } from './vitalHistory'
+import { checkProfilingQuota } from './quotaCheck'
+import type { QuotaReason } from './quotaCheck'
 
 export const DEFAULT_RUM_PROFILER_CONFIGURATION: RUMProfilerConfiguration = {
   sampleIntervalMs: 10, // Sample stack trace every 10ms
@@ -58,6 +60,7 @@ export function createRumProfiler(
   const vitalHistory = mockable(createVitalHistory)(lifeCycle)
 
   let instance: RumProfilerInstance = { state: 'stopped', stateReason: 'initializing' }
+  let quotaCheckGeneration = 0
 
   // Stops the profiler when session expires
   lifeCycle.subscribe(LifeCycleEventType.SESSION_EXPIRED, () => {
@@ -66,7 +69,10 @@ export function createRumProfiler(
 
   // Start the profiler again when session is renewed
   lifeCycle.subscribe(LifeCycleEventType.SESSION_RENEWED, () => {
-    if (instance.state === 'stopped' && instance.stateReason === 'session-expired') {
+    if (
+      instance.state === 'stopped' &&
+      (instance.stateReason === 'session-expired' || instance.stateReason === 'quota_ko')
+    ) {
       start()
     }
   })
@@ -97,14 +103,35 @@ export function createRumProfiler(
 
     // Start profiler instance
     startNextProfilerInstance()
+
+    // Quota check — optimistic: profiler already recording, only gates sending.
+    // Generation counter invalidates results from a prior session (incremented on each start() call).
+    // State guard handles within-session cancellation (user stop, session expiry, etc.).
+    const checkGeneration = ++quotaCheckGeneration
+    const sessionId = session.findTrackedSession()?.id
+    if (sessionId) {
+      mockable(checkProfilingQuota)(configuration, sessionId)
+        .then((result) => {
+          if (checkGeneration !== quotaCheckGeneration) {
+            return
+          }
+          if (instance.state !== 'running' && instance.state !== 'paused') {
+            return
+          }
+          if (result.decision === 'quota_ko') {
+            stopProfiling('quota_ko', result.reason)
+          }
+        })
+        .catch(monitorError)
+    }
   }
 
   // Public API to manually stop the profiler.
   function stop() {
     stopProfiling('stopped-by-user')
   }
 
-  function stopProfiling(reason: RumProfilerStoppedInstance['stateReason']) {
+  function stopProfiling(reason: RumProfilerStoppedInstance['stateReason'], quotaReason?: QuotaReason) {
     // Stop current profiler instance (data collection happens async in background)
     stopProfilerInstance(reason)
 
@@ -113,7 +140,7 @@ export function createRumProfiler(
     globalCleanupTasks.length = 0
 
     // Update Profiling status once the Profiler has been stopped.
-    profilingContextManager.set({ status: 'stopped', error_reason: undefined })
+    profilingContextManager.set({ status: 'stopped', error_reason: undefined, quota_reason: quotaReason })
   }
 
   /**
@@ -279,8 +306,15 @@ export function createRumProfiler(
     // Cleanup instance-specific tasks (e.g., view listener)
     runningInstance.cleanupTasks.forEach((cleanupTask) => cleanupTask())
 
-    // Collect and send profile data in background - doesn't block state transitions
-    collectProfilerInstance(runningInstance)
+    if (stateReason === 'quota_ko') {
+      // Discard data — quota denied means we should not send anything
+      clearTimeout(runningInstance.timeoutId)
+      runningInstance.profiler.removeEventListener('samplebufferfull', handleSampleBufferFull)
+      void runningInstance.profiler.stop().catch(monitorError)
+    } else {
+      // Collect and send profile data in background - doesn't block state transitions
+      collectProfilerInstance(runningInstance)
+    }
   }
 
   function pauseProfilerInstance() {