50 lines
2.2 KiB
TypeScript
50 lines
2.2 KiB
TypeScript
|
// ─── Audio transmit hook — captures mic, encodes to PCM, sends via WebSocket ──
|
||
|
|
import { useCallback, useRef, useState } from "react";
|
||
|
|
|
||
|
|
const SAMPLE_RATE = 24000;
|
||
|
|
|
||
|
|
export function useAudioTransmit(
|
||
|
|
socketRef: { readonly current: WebSocket | null },
|
||
|
|
) {
|
||
|
|
const [isStreaming, setIsStreaming] = useState(false);
|
||
|
|
const streamRef = useRef<MediaStream | null>(null);
|
||
|
|
const audioContextRef = useRef<AudioContext | null>(null);
|
||
|
|
const processorRef = useRef<ScriptProcessorNode | null>(null);
|
||
|
|
|
||
|
|
const stop = useCallback(() => {
|
||
|
|
setIsStreaming(false);
|
||
|
|
if (processorRef.current) { processorRef.current.disconnect(); processorRef.current = null; }
|
||
|
|
if (audioContextRef.current) { audioContextRef.current.close(); audioContextRef.current = null; }
|
||
|
|
if (streamRef.current) { for (const track of streamRef.current.getTracks()) track.stop(); streamRef.current = null; }
|
||
|
|
}, []);
|
||
|
|
|
||
|
|
const start = useCallback(async () => {
|
||
|
|
const stream = await navigator.mediaDevices.getUserMedia({ audio: true });
|
||
|
|
streamRef.current = stream;
|
||
|
|
setIsStreaming(true);
|
||
|
|
const AudioContextCtor = window.AudioContext || (window as unknown as { webkitAudioContext: typeof AudioContext }).webkitAudioContext;
|
||
|
|
const audioContext = new AudioContextCtor({ sampleRate: SAMPLE_RATE });
|
||
|
|
audioContextRef.current = audioContext;
|
||
|
|
const source = audioContext.createMediaStreamSource(stream);
|
||
|
|
const processor = audioContext.createScriptProcessor(4096, 1, 1);
|
||
|
|
processorRef.current = processor;
|
||
|
|
source.connect(processor);
|
||
|
|
processor.connect(audioContext.destination);
|
||
|
|
processor.onaudioprocess = (event) => {
|
||
|
|
if (!socketRef.current || socketRef.current.readyState !== WebSocket.OPEN) return;
|
||
|
|
const inputData = event.inputBuffer.getChannelData(0);
|
||
|
|
const pcmData = new Int16Array(inputData.length);
|
||
|
|
for (let i = 0; i < inputData.length; i++) pcmData[i] = Math.max(-1, Math.min(1, inputData[i])) * 32767;
|
||
|
|
// BUG 2 FIX: slice() to create independent copy of the ArrayBuffer
|
||
|
|
socketRef.current.send(pcmData.buffer.slice(0));
|
||
|
|
};
|
||
|
|
}, [socketRef]);
|
||
|
|
|
||
|
|
const toggle = useCallback(async () => {
|
||
|
|
if (isStreaming) stop();
|
||
|
|
else await start();
|
||
|
|
}, [isStreaming, start, stop]);
|
||
|
|
|
||
|
|
return { isStreaming, toggle, stop, start };
|
||
|
|
}
|