@siteed/expo-audio-studio 2.4.1 → 2.6.0

package/src/WebRecorder.web.ts

@@ -7,6 +7,7 @@ import {
     EmitAudioEventFunction,
 } from './ExpoAudioStream.web'
 import { encodingToBitDepth } from './utils/encodingToBitDepth'
+import { writeWavHeader } from './utils/writeWavHeader'
 import { InlineFeaturesExtractor } from './workers/InlineFeaturesExtractor.web'
 import { InlineAudioWebWorker } from './workers/inlineAudioWebWorker.web'
 
@@ -15,6 +16,8 @@ interface AudioWorkletEvent {
         command: string
         recordedData?: Float32Array
         sampleRate?: number
+        position?: number
+        message?: string // For debug messages
     }
 }
 
@@ -33,7 +36,7 @@ const DEFAULT_WEB_NUMBER_OF_CHANNELS = 1
 const TAG = 'WebRecorder'
 
 export class WebRecorder {
-    private audioContext: AudioContext
+    public audioContext: AudioContext
     private audioWorkletNode!: AudioWorkletNode
     private featureExtractorWorker?: Worker
     private source: MediaStreamAudioSourceNode
@@ -45,14 +48,35 @@ export class WebRecorder {
     private bitDepth: number // Bit depth of the audio
     private exportBitDepth: number // Bit depth of the audio
     private audioAnalysisData: AudioAnalysis // Keep updating the full audio analysis data with latest events
-    private packetCount: number = 0
     private logger?: ConsoleLike
     private compressedMediaRecorder: MediaRecorder | null = null
     private compressedChunks: Blob[] = []
     private compressedSize: number = 0
     private pendingCompressedChunk: Blob | null = null
-    private readonly wavMimeType = 'audio/wav'
     private dataPointIdCounter: number = 0 // Add this property to track the counter
+    private deviceDisconnectionHandler: (() => void) | null = null
+    private mediaStream: MediaStream | null = null
+    private onInterruptionCallback?: (event: {
+        reason: string
+        isPaused: boolean
+        timestamp: number
+    }) => void
+    private _isDeviceDisconnected: boolean = false
+    private pcmData: Float32Array | null = null // Store original PCM data
+    private totalSampleCount: number = 0
+
+    /**
+     * Flag to indicate whether this is the first audio chunk after a device switch
+     * Used to maintain proper duration counting
+     */
+    public isFirstChunkAfterSwitch: boolean = false
+
+    /**
+     * Gets whether the recording device has been disconnected
+     */
+    get isDeviceDisconnected(): boolean {
+        return this._isDeviceDisconnected
+    }
 
     /**
      * Initializes a new WebRecorder instance for audio recording and processing
@@ -61,6 +85,7 @@ export class WebRecorder {
      * @param recordingConfig - Configuration options for the recording
      * @param emitAudioEventCallback - Callback function for audio data events
      * @param emitAudioAnalysisCallback - Callback function for audio analysis events
+     * @param onInterruption - Callback for recording interruptions
      * @param logger - Optional logger for debugging information
      */
     constructor({
@@ -69,6 +94,7 @@ export class WebRecorder {
         recordingConfig,
         emitAudioEventCallback,
         emitAudioAnalysisCallback,
+        onInterruption,
         logger,
     }: {
         audioContext: AudioContext
@@ -76,6 +102,11 @@ export class WebRecorder {
         recordingConfig: RecordingConfig
         emitAudioEventCallback: EmitAudioEventFunction
         emitAudioAnalysisCallback: EmitAudioAnalysisFunction
+        onInterruption?: (event: {
+            reason: string
+            isPaused: boolean
+            timestamp: number
+        }) => void
         logger?: ConsoleLike
     }) {
         this.audioContext = audioContext
@@ -126,6 +157,12 @@ export class WebRecorder {
         if (recordingConfig.compression?.enabled) {
             this.initializeCompressedRecorder()
         }
+
+        this.mediaStream = source.mediaStream
+        this.onInterruptionCallback = onInterruption
+
+        // Setup device disconnection detection
+        this.setupDeviceDisconnectionDetection()
     }
 
     /**
@@ -150,6 +187,11 @@ export class WebRecorder {
                 event: AudioWorkletEvent
             ) => {
                 const command = event.data.command
+                if (command === 'debug') {
+                    this.logger?.debug(`[AudioWorklet] ${event.data.message}`)
+                    return
+                }
+
                 if (command !== 'newData') return
 
                 const pcmBufferFloat = event.data.recordedData
@@ -159,18 +201,26 @@ export class WebRecorder {
                 }
 
                 // Process data in smaller chunks and emit immediately
-                const chunkSize = this.audioContext.sampleRate * 2 // Reduce to 2 seconds chunks
                 const sampleRate =
                     event.data.sampleRate ?? this.audioContext.sampleRate
+                // Use chunk size from config interval or default to 2 seconds
+                const intervalMs = this.config.interval ?? DEFAULT_WEB_INTERVAL
+                const chunkSize = Math.floor(sampleRate * (intervalMs / 1000))
                 const duration = pcmBufferFloat.length / sampleRate
 
+                // Use incoming position if provided by worklet, otherwise use our tracked position
+                const incomingPosition =
+                    typeof event.data.position === 'number'
+                        ? event.data.position
+                        : this.position
+
                 // Calculate bytes per sample based on bit depth
                 const bytesPerSample = this.bitDepth / 8
 
                 // Emit chunks without storing them
                 for (let i = 0; i < pcmBufferFloat.length; i += chunkSize) {
                     const chunk = pcmBufferFloat.slice(i, i + chunkSize)
-                    const chunkPosition = this.position + i / sampleRate
+                    const chunkPosition = incomingPosition + i / sampleRate
 
                     // Calculate byte positions and samples
                     const startPosition = Math.floor(i * bytesPerSample)
@@ -202,6 +252,17 @@ export class WebRecorder {
                         })
                     }
 
+                    // Only store PCM data if web.storeUncompressedAudio is not explicitly false
+                    const shouldStoreUncompressed =
+                        this.config.web?.storeUncompressedAudio !== false
+
+                    // Store PCM chunks when needed
+                    if (shouldStoreUncompressed) {
+                        // Store the original Float32Array data for later WAV creation
+                        this.appendPcmData(chunk)
+                        this.totalSampleCount += chunk.length
+                    }
+
                     // Emit chunk immediately
                     this.emitAudioEventCallback({
                         data: chunk,
@@ -221,24 +282,47 @@ export class WebRecorder {
                     })
                 }
 
-                this.position += duration
+                // Update our position based on the worklet's position if provided
+                this.position = incomingPosition + duration
                 this.pendingCompressedChunk = null
             }
 
-            this.logger?.debug(
-                `WebRecorder initialized -- recordSampleRate=${this.audioContext.sampleRate}`,
-                this.config
-            )
+            // Ensure we use all relevant settings from config
+            const recordSampleRate = this.audioContext.sampleRate
+            const exportSampleRate =
+                this.config.sampleRate ?? this.audioContext.sampleRate
+            const channels = this.config.channels ?? this.numberOfChannels
+            const interval = this.config.interval ?? DEFAULT_WEB_INTERVAL
+
+            this.logger?.debug(`WebRecorder initialized with config:`, {
+                recordSampleRate,
+                exportSampleRate,
+                bitDepth: this.bitDepth,
+                exportBitDepth: this.exportBitDepth,
+                channels,
+                interval,
+                position: this.position,
+                deviceId: this.config.deviceId || 'default',
+                compression: this.config.compression
+                    ? {
+                          enabled: this.config.compression.enabled,
+                          format: this.config.compression.format,
+                          bitrate: this.config.compression.bitrate,
+                      }
+                    : 'disabled',
+            })
+
+            // Initialize the worklet with all settings from config
             this.audioWorkletNode.port.postMessage({
                 command: 'init',
-                recordSampleRate: this.audioContext.sampleRate,
-                exportSampleRate:
-                    this.config.sampleRate ?? this.audioContext.sampleRate,
+                recordSampleRate,
+                exportSampleRate,
                 bitDepth: this.bitDepth,
                 exportBitDepth: this.exportBitDepth,
-                channels: this.numberOfChannels,
-                interval: this.config.interval ?? DEFAULT_WEB_INTERVAL,
-                // enableLogging: !!this.logger,
+                channels,
+                interval,
+                position: this.position, // Pass the current position to the processor
+                enableLogging: true,
             })
 
             // Connect the source to the AudioWorkletNode and start recording
@@ -249,6 +333,35 @@ export class WebRecorder {
         }
     }
 
+    /**
+     * Append new PCM data to the existing buffer
+     * @param newData New Float32Array data to append
+     */
+    private appendPcmData(newData: Float32Array): void {
+        // Clone the incoming data to ensure it's not modified
+        const dataToAdd = new Float32Array(newData)
+
+        if (!this.pcmData) {
+            // First chunk - create a copy to avoid references to original data
+            this.pcmData = new Float32Array(dataToAdd)
+            return
+        }
+
+        // Create a new buffer with increased size
+        const newBuffer = new Float32Array(
+            this.pcmData.length + dataToAdd.length
+        )
+
+        // Copy existing data
+        newBuffer.set(this.pcmData)
+
+        // Append new data
+        newBuffer.set(dataToAdd, this.pcmData.length)
+
+        // Replace existing buffer
+        this.pcmData = newBuffer
+    }
+
     /**
      * Initializes the feature extractor worker for audio analysis
      * Creates an inline worker from a blob for audio feature extraction
@@ -265,6 +378,18 @@ export class WebRecorder {
             this.featureExtractorWorker.onerror = (error) => {
                 console.error(`[${TAG}] Feature extractor worker error:`, error)
             }
+
+            // Initialize worker with counter if needed
+            if (this.dataPointIdCounter > 0) {
+                this.featureExtractorWorker.postMessage({
+                    command: 'resetCounter',
+                    value: this.dataPointIdCounter,
+                })
+                this.logger?.debug(
+                    `Initialized worker with counter value ${this.dataPointIdCounter}`
+                )
+            }
+
             this.logger?.log(
                 'Feature extractor worker initialized successfully'
             )
@@ -285,42 +410,51 @@ export class WebRecorder {
         if (event.data.command === 'features') {
             const segmentResult = event.data.result
 
-            // Update the dataPointIdCounter based on the last ID received
+            // Track existing IDs to prevent duplicates
+            const existingIds = new Set(
+                this.audioAnalysisData.dataPoints.map((dp) => dp.id)
+            )
+
+            // Filter out datapoints with duplicate IDs
+            const uniqueNewDataPoints = segmentResult.dataPoints.filter(
+                (dp) => {
+                    return !existingIds.has(dp.id)
+                }
+            )
+
+            // Log filtered duplicates if any
             if (
-                segmentResult.dataPoints &&
-                segmentResult.dataPoints.length > 0
+                uniqueNewDataPoints.length < segmentResult.dataPoints.length &&
+                this.logger?.warn
             ) {
+                this.logger.warn(
+                    `Filtered ${segmentResult.dataPoints.length - uniqueNewDataPoints.length} duplicate datapoints`
+                )
+            }
+
+            // Update counter based on the highest ID seen
+            if (uniqueNewDataPoints.length > 0) {
                 const lastDataPoint =
-                    segmentResult.dataPoints[
-                        segmentResult.dataPoints.length - 1
-                    ]
+                    uniqueNewDataPoints[uniqueNewDataPoints.length - 1]
+
                 if (lastDataPoint && typeof lastDataPoint.id === 'number') {
-                    this.dataPointIdCounter = Math.max(
-                        this.dataPointIdCounter,
-                        lastDataPoint.id + 1
-                    )
+                    const nextIdValue = lastDataPoint.id + 1
+
+                    if (nextIdValue > this.dataPointIdCounter) {
+                        this.dataPointIdCounter = nextIdValue
+                        this.logger?.debug(
+                            `Counter updated to ${this.dataPointIdCounter}`
+                        )
+                    }
                 }
             }
 
-            this.logger?.debug('[WebRecorder] Raw segment result:', {
-                dataPointsLength: segmentResult.dataPoints.length,
-                durationMs: segmentResult.durationMs,
-                sampleRate: segmentResult.sampleRate,
-                amplitudeRange: segmentResult.amplitudeRange,
-            })
-
-            // Ensure consistent sample rate in the result
-            segmentResult.sampleRate =
-                this.config.sampleRate || this.audioContext.sampleRate
-
-            // Update the full audio analysis data with proper range merging
-            this.audioAnalysisData.dataPoints.push(...segmentResult.dataPoints)
+            // Add unique data points to our analysis data
+            this.audioAnalysisData.dataPoints.push(...uniqueNewDataPoints)
             this.audioAnalysisData.durationMs += segmentResult.durationMs
-
-            // Make sure the sample rate is consistent
             this.audioAnalysisData.sampleRate = segmentResult.sampleRate
 
-            // Properly merge amplitude ranges
+            // Merge amplitude ranges
             if (segmentResult.amplitudeRange) {
                 if (!this.audioAnalysisData.amplitudeRange) {
                     this.audioAnalysisData.amplitudeRange = {
@@ -340,7 +474,7 @@ export class WebRecorder {
                 }
             }
 
-            // Properly merge RMS ranges
+            // Merge RMS ranges
             if (segmentResult.rmsRange) {
                 if (!this.audioAnalysisData.rmsRange) {
                     this.audioAnalysisData.rmsRange = {
@@ -360,77 +494,182 @@ export class WebRecorder {
                 }
             }
 
-            this.logger?.debug('features event segmentResult', segmentResult)
-            this.logger?.debug(
-                `features event audioAnalysisData duration=${this.audioAnalysisData.durationMs}`,
-                this.audioAnalysisData
-            )
-            this.emitAudioAnalysisCallback(segmentResult)
+            // Send filtered result to avoid duplicate IDs
+            const filteredSegmentResult = {
+                ...segmentResult,
+                dataPoints: uniqueNewDataPoints,
+            }
 
-            this.logger?.debug('[WebRecorder] Updated audioAnalysisData:', {
-                dataPointsLength: this.audioAnalysisData.dataPoints.length,
-                durationMs: this.audioAnalysisData.durationMs,
-                sampleRate: this.audioAnalysisData.sampleRate,
-                amplitudeRange: this.audioAnalysisData.amplitudeRange,
-            })
+            this.emitAudioAnalysisCallback(filteredSegmentResult)
         }
     }
 
     /**
-     * Resets the data point ID counter
-     * Used when starting a new recording
+     * Reset the data point counter to a specific value or zero
+     * @param startCounterFrom Optional value to start the counter from (for continuing from previous recordings)
      */
-    resetDataPointCounter() {
-        this.dataPointIdCounter = 0
+    resetDataPointCounter(startCounterFrom?: number): void {
+        // Set the counter with the passed value or 0
+        this.dataPointIdCounter =
+            startCounterFrom !== undefined ? startCounterFrom : 0
+        this.logger?.debug(
+            `Reset data point counter to ${this.dataPointIdCounter}`
+        )
 
-        // Reset the counter in the worker
+        // Update worker counter if available
         if (this.featureExtractorWorker) {
             this.featureExtractorWorker.postMessage({
                 command: 'resetCounter',
-                startCounterFrom: 0,
+                value: this.dataPointIdCounter,
             })
+        } else {
+            this.logger?.warn(
+                'No feature extractor worker available to update counter'
+            )
         }
     }
 
+    /**
+     * Get the current data point counter value
+     * @returns The current value of the data point counter
+     */
+    getDataPointCounter(): number {
+        return this.dataPointIdCounter
+    }
+
+    /**
+     * Prepares the recorder for continuity after device switch
+     * Sets up all necessary state to maintain proper recording continuity
+     */
+    prepareForDeviceSwitch(): void {
+        this.isFirstChunkAfterSwitch = true
+        this.logger?.debug(
+            `Prepared for device switch at position ${this.position}s`
+        )
+    }
+
     /**
      * Starts the audio recording process
      * Connects the audio nodes and begins capturing audio data
+     * @param preserveCounters If true, do not reset the counter (used for device switching)
      */
-    start() {
+    start(preserveCounters = false) {
         this.source.connect(this.audioWorkletNode)
         this.audioWorkletNode.connect(this.audioContext.destination)
-        this.packetCount = 0
 
-        // Reset the counter when starting a new recording
-        this.resetDataPointCounter()
+        // Only reset the counter when not preserving state (e.g., for a fresh recording)
+        if (!preserveCounters) {
+            this.logger?.debug(
+                'Starting fresh recording, resetting counter to 0'
+            )
+            this.resetDataPointCounter(0) // Explicitly reset to 0 for new recordings
+            this.isFirstChunkAfterSwitch = false
+
+            // Clear PCM data for new recording
+            this.pcmData = null
+            this.totalSampleCount = 0
+        } else {
+            this.logger?.debug(
+                `Preserving counter at ${this.dataPointIdCounter} during device switch`
+            )
+        }
 
         if (this.compressedMediaRecorder) {
             this.compressedMediaRecorder.start(this.config.interval ?? 1000)
         }
     }
 
+    /**
+     * Creates a WAV file from the stored PCM data
+     */
+    private createWavFromPcmData(): Blob | null {
+        try {
+            // Check if we have PCM data
+            if (!this.pcmData || this.pcmData.length === 0) {
+                this.logger?.warn('No PCM data available to create WAV file')
+                return null
+            }
+
+            const sampleRate =
+                this.config.sampleRate || this.audioContext.sampleRate
+            const channels = this.numberOfChannels || 1
+
+            // Convert float32 PCM data to 16-bit PCM for WAV
+            const bytesPerSample = 2 // 16-bit = 2 bytes
+            const dataLength = this.pcmData.length * bytesPerSample
+            const buffer = new ArrayBuffer(dataLength)
+            const view = new DataView(buffer)
+
+            // Convert Float32Array (-1 to 1) to Int16Array (-32768 to 32767)
+            for (let i = 0; i < this.pcmData.length; i++) {
+                const sample = Math.max(-1, Math.min(1, this.pcmData[i]))
+                const int16Value = Math.round(sample * 32767)
+                view.setInt16(i * 2, int16Value, true)
+            }
+
+            // Use the existing writeWavHeader utility to add a WAV header
+            const wavBuffer = writeWavHeader({
+                buffer,
+                sampleRate,
+                numChannels: channels,
+                bitDepth: 16,
+                isFloat: false,
+            })
+
+            return new Blob([wavBuffer], { type: 'audio/wav' })
+        } catch (error) {
+            this.logger?.error('Error creating WAV file from PCM data:', error)
+            return null
+        }
+    }
+
     /**
      * Stops the audio recording process and returns the recorded data
-     * @returns Promise resolving to an object containing PCM data and optional compressed blob
+     * @returns Promise resolving to an object containing compressed and/or uncompressed blobs
      */
-    async stop(): Promise<{ pcmData: Float32Array; compressedBlob?: Blob }> {
+    async stop(): Promise<{ compressedBlob?: Blob; uncompressedBlob?: Blob }> {
         try {
-            if (this.compressedMediaRecorder) {
+            // Stop any compressed recording first
+            if (
+                this.compressedMediaRecorder &&
+                this.compressedMediaRecorder.state !== 'inactive'
+            ) {
                 this.compressedMediaRecorder.stop()
-                return {
-                    pcmData: new Float32Array(), // Return empty array since we're streaming
-                    compressedBlob: new Blob(this.compressedChunks, {
-                        type: 'audio/webm;codecs=opus',
-                    }),
-                }
             }
-            return { pcmData: new Float32Array() }
+
+            // Wait for any pending compressed chunks to be processed
+            if (this.compressedMediaRecorder) {
+                // Small delay to ensure all data is processed
+                await new Promise((resolve) => setTimeout(resolve, 100))
+            }
+
+            // Create uncompressed WAV file from the PCM data
+            let uncompressedBlob: Blob | undefined
+
+            // Only create WAV if we have PCM data
+            if (this.pcmData && this.pcmData.length > 0) {
+                uncompressedBlob =
+                    (await this.createWavFromPcmData()) || undefined
+            }
+
+            // Return the compressed and/or uncompressed blobs if available
+            return {
+                compressedBlob:
+                    this.compressedChunks.length > 0
+                        ? new Blob(this.compressedChunks, {
+                              type: 'audio/webm;codecs=opus',
+                          })
+                        : undefined,
+                uncompressedBlob,
+            }
         } finally {
             this.cleanup()
             // Reset the chunks array
             this.compressedChunks = []
             this.compressedSize = 0
             this.pendingCompressedChunk = null
+            this.pcmData = null
+            this.totalSampleCount = 0
         }
     }
 
@@ -438,17 +677,45 @@ export class WebRecorder {
      * Cleans up resources when recording is stopped
      * Closes audio context and disconnects nodes
      */
-    private cleanup() {
-        if (this.audioContext) {
-            this.audioContext.close()
+    public cleanup() {
+        // Remove device disconnection handler
+        if (this.deviceDisconnectionHandler) {
+            this.deviceDisconnectionHandler()
+            this.deviceDisconnectionHandler = null
         }
+
+        // Check if AudioContext is already closed before attempting to close it
+        if (this.audioContext && this.audioContext.state !== 'closed') {
+            try {
+                this.audioContext.close()
+            } catch (e) {
+                // Ignore closure errors - this happens if already closed
+            }
+        }
+
+        // Safely disconnect audioWorkletNode if it exists
         if (this.audioWorkletNode) {
-            this.audioWorkletNode.disconnect()
+            try {
+                this.audioWorkletNode.disconnect()
+            } catch (e) {
+                // Ignore disconnection errors - node might be already disconnected
+            }
         }
+
+        // Safely disconnect source if it exists
         if (this.source) {
-            this.source.disconnect()
+            try {
+                this.source.disconnect()
+            } catch (e) {
+                // Ignore disconnection errors - source might be already disconnected
+            }
         }
+
+        // Always stop media stream tracks to release hardware resources
         this.stopMediaStreamTracks()
+
+        // Mark as disconnected to prevent future errors
+        this._isDeviceDisconnected = true
     }
 
     /**
@@ -456,20 +723,37 @@ export class WebRecorder {
      * Disconnects audio nodes and pauses the media recorder
      */
     pause() {
-        this.source.disconnect(this.audioWorkletNode) // Disconnect the source from the AudioWorkletNode
-        this.audioWorkletNode.disconnect(this.audioContext.destination) // Disconnect the AudioWorkletNode from the destination
-        this.audioWorkletNode.port.postMessage({ command: 'pause' })
-        this.compressedMediaRecorder?.pause()
+        try {
+            // Note: We're just pausing, not disconnecting the device
+            // Simply disconnect nodes temporarily without marking device as disconnected
+            this.source.disconnect(this.audioWorkletNode)
+            this.audioWorkletNode.disconnect(this.audioContext.destination)
+            this.audioWorkletNode.port.postMessage({ command: 'pause' })
+
+            if (this.compressedMediaRecorder?.state === 'recording') {
+                this.compressedMediaRecorder.pause()
+            }
+
+            this.logger?.debug('Recording paused successfully')
+        } catch (error) {
+            this.logger?.error('Error in pause(): ', error)
+            // Already disconnected, just ignore and continue
+        }
     }
 
     /**
      * Stops all media stream tracks to release hardware resources
      * Ensures recording indicators (like microphone icon) are turned off
      */
-    stopMediaStreamTracks() {
+    public stopMediaStreamTracks() {
         // Stop all audio tracks to stop the recording icon
-        const tracks = this.source.mediaStream.getTracks()
-        tracks.forEach((track) => track.stop())
+        if (this.mediaStream) {
+            const tracks = this.mediaStream.getTracks()
+            tracks.forEach((track) => track.stop())
+        } else if (this.source?.mediaStream) {
+            const tracks = this.source.mediaStream.getTracks()
+            tracks.forEach((track) => track.stop())
+        }
     }
 
     /**
@@ -502,10 +786,20 @@ export class WebRecorder {
      * Reconnects audio nodes and resumes the media recorder
      */
     resume() {
-        this.source.connect(this.audioWorkletNode)
-        this.audioWorkletNode.connect(this.audioContext.destination)
-        this.audioWorkletNode.port.postMessage({ command: 'resume' })
-        this.compressedMediaRecorder?.resume()
+        // If device was disconnected, we can't resume
+        if (this._isDeviceDisconnected) {
+            this.logger?.warn('Cannot resume recording: device disconnected')
+            return
+        }
+
+        try {
+            this.source.connect(this.audioWorkletNode)
+            this.audioWorkletNode.connect(this.audioContext.destination)
+            this.audioWorkletNode.port.postMessage({ command: 'resume' })
+            this.compressedMediaRecorder?.resume()
+        } catch (error) {
+            this.logger?.error('Error in resume(): ', error)
+        }
     }
 
     /**
@@ -573,8 +867,104 @@ export class WebRecorder {
                 startPosition,
                 endPosition,
                 samples,
-                startCounterFrom: this.dataPointIdCounter, // Pass the current counter value
             })
         }
     }
+
+    /**
+     * Sets up detection for device disconnection events
+     */
+    private setupDeviceDisconnectionDetection() {
+        if (!this.mediaStream) return
+
+        // Function to handle track ending (which happens on device disconnection)
+        const handleTrackEnded = () => {
+            this.logger?.warn('Audio track ended - device disconnected')
+            this._isDeviceDisconnected = true
+
+            // Use the callback to notify parent component about device disconnection
+            if (this.onInterruptionCallback) {
+                this.onInterruptionCallback({
+                    reason: 'deviceDisconnected',
+                    isPaused: true,
+                    timestamp: Date.now(),
+                })
+                this.logger?.debug('Notified about device disconnection')
+            }
+
+            // Ensure we disconnect nodes to prevent zombie recordings
+            if (this.audioWorkletNode) {
+                this.audioWorkletNode.port.postMessage({
+                    command: 'deviceDisconnected',
+                })
+
+                try {
+                    this.source.disconnect(this.audioWorkletNode)
+                    this.audioWorkletNode.disconnect()
+                } catch (e) {
+                    // Ignore disconnection errors as the track might already be gone
+                }
+            }
+        }
+
+        // Add listeners to all audio tracks
+        const tracks = this.mediaStream.getAudioTracks()
+        tracks.forEach((track) => {
+            track.addEventListener('ended', handleTrackEnded)
+        })
+
+        // Store the handler for cleanup
+        this.deviceDisconnectionHandler = () => {
+            tracks.forEach((track) => {
+                track.removeEventListener('ended', handleTrackEnded)
+            })
+        }
+    }
+
+    /**
+     * Explicitly set the position for continuous recording across device switches
+     * @param position The position in seconds to continue from
+     */
+    setPosition(position: number): void {
+        if (position >= 0) {
+            this.position = position
+            this.logger?.debug(`Position explicitly set to ${position} seconds`)
+        } else {
+            this.logger?.warn(`Invalid position value: ${position}, ignoring`)
+        }
+    }
+
+    /**
+     * Get the current position in seconds
+     * @returns The current position
+     */
+    getPosition(): number {
+        return this.position
+    }
+
+    /**
+     * Gets the current compressed chunks
+     * @returns Array of current compressed audio chunks
+     */
+    getCompressedChunks(): Blob[] {
+        return [...this.compressedChunks]
+    }
+
+    /**
+     * Sets the compressed chunks from a previous recorder
+     * @param chunks Array of compressed chunks from a previous recorder
+     */
+    setCompressedChunks(chunks: Blob[]): void {
+        if (chunks && chunks.length > 0) {
+            this.logger?.debug(
+                `Adding ${chunks.length} compressed chunks from previous device`
+            )
+            this.compressedChunks = [...chunks, ...this.compressedChunks]
+            // Update size
+            this.compressedSize = this.compressedChunks.reduce(
+                (size, chunk) => size + chunk.size,
+                0
+            )
+        }
+    }
 }