347 lines
15 KiB
JavaScript
347 lines
15 KiB
JavaScript
|
; (function (window, voice) {
|
|||
|
|
"use strict";
|
|||
|
|
if (typeof define === 'function' && define.amd) {
|
|||
|
|
define(voice);
|
|||
|
|
} else if (typeof exports === 'object') {
|
|||
|
|
module.exports = voice();
|
|||
|
|
} else {
|
|||
|
|
window.Voice = voice();
|
|||
|
|
};
|
|||
|
|
}(typeof window !== "undefined" ? window : this, () => {
|
|||
|
|
"use strict";
|
|||
|
|
return class IatRecorder {
|
|||
|
|
constructor(opts = {}) {
|
|||
|
|
// 服务接口认证信息(语音听写（流式版）WebAPI)
|
|||
|
|
this.appId = opts.appId || '';
|
|||
|
|
this.apiKey = opts.apiKey || '';
|
|||
|
|
this.apiSecret = opts.apiSecret || '';
|
|||
|
|
// 识别监听方法
|
|||
|
|
this.onTextChange = opts.onTextChange || Function();
|
|||
|
|
this.onWillStatusChange = opts.onWillStatusChange || Function();
|
|||
|
|
// 方言/语种
|
|||
|
|
this.status = 'null'
|
|||
|
|
this.language = opts.language || 'zh_cn'
|
|||
|
|
this.accent = opts.accent || 'mandarin';
|
|||
|
|
// 流媒体
|
|||
|
|
this.streamRef = [];
|
|||
|
|
// 记录音频数据
|
|||
|
|
this.audioData = [];
|
|||
|
|
// 记录听写结果
|
|||
|
|
this.resultText = '';
|
|||
|
|
// wpgs下的听写结果需要中间状态辅助记录
|
|||
|
|
this.resultTextTemp = '';
|
|||
|
|
// 音频数据多线程
|
|||
|
|
this.init();
|
|||
|
|
};
|
|||
|
|
// WebSocket请求地址鉴权
|
|||
|
|
getWebSocketUrl() {
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
// 请求地址根据语种不同变化
|
|||
|
|
try {
|
|||
|
|
const CryptoJS = require('crypto-js');
|
|||
|
|
let url = 'wss://iat-api.xfyun.cn/v2/iat',
|
|||
|
|
host = 'iat-api.xfyun.cn',
|
|||
|
|
date = new Date().toGMTString(),
|
|||
|
|
algorithm = 'hmac-sha256',
|
|||
|
|
headers = 'host date request-line',
|
|||
|
|
signatureOrigin = `host: ${host}\ndate: ${date}\nGET /v2/iat HTTP/1.1`,
|
|||
|
|
signatureSha = CryptoJS.HmacSHA256(signatureOrigin, this.apiSecret),
|
|||
|
|
signature = CryptoJS.enc.Base64.stringify(signatureSha),
|
|||
|
|
authorizationOrigin = `api_key="${this.apiKey}", algorithm="${algorithm}", headers="${headers}", signature="${signature}"`,
|
|||
|
|
authorization = btoa(authorizationOrigin);
|
|||
|
|
resolve(`${url}?authorization=${authorization}&date=${date}&host=${host}`);
|
|||
|
|
} catch (error) {
|
|||
|
|
let url = 'wss://iat-api.xfyun.cn/v2/iat',
|
|||
|
|
host = 'iat-api.xfyun.cn',
|
|||
|
|
date = new Date().toGMTString(),
|
|||
|
|
algorithm = 'hmac-sha256',
|
|||
|
|
headers = 'host date request-line',
|
|||
|
|
signatureOrigin = `host: ${host}\ndate: ${date}\nGET /v2/iat HTTP/1.1`,
|
|||
|
|
signatureSha = CryptoJS.HmacSHA256(signatureOrigin, this.apiSecret),
|
|||
|
|
signature = CryptoJS.enc.Base64.stringify(signatureSha),
|
|||
|
|
authorizationOrigin = `api_key="${this.apiKey}", algorithm="${algorithm}", headers="${headers}", signature="${signature}"`,
|
|||
|
|
authorization = btoa(authorizationOrigin);
|
|||
|
|
resolve(`${url}?authorization=${authorization}&date=${date}&host=${host}`);
|
|||
|
|
};
|
|||
|
|
});
|
|||
|
|
};
|
|||
|
|
// 操作初始化
|
|||
|
|
init() {
|
|||
|
|
const self = this;
|
|||
|
|
try {
|
|||
|
|
if (!self.appId || !self.apiKey || !self.apiSecret) {
|
|||
|
|
alert('请正确配置【迅飞语音听写（流式版）WebAPI】服务接口认证信息！');
|
|||
|
|
} else {
|
|||
|
|
self.webWorker = new Worker('./js/transcode.worker.js');
|
|||
|
|
self.webWorker.onmessage = function (event) {
|
|||
|
|
self.audioData.push(...event.data);
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
alert('对不起：请在服务器环境下运行！');
|
|||
|
|
console.error('请在服务器如：WAMP、XAMPP、Phpstudy、http-server、WebServer等环境中运行！', error);
|
|||
|
|
};
|
|||
|
|
console.log("%c ❤️使用说明：http://www.muguilin.com/blog/info/609bafc50d572b3fd79b058f", "font-size:32px; color:blue; font-weight: bold;");
|
|||
|
|
};
|
|||
|
|
// 修改录音听写状态
|
|||
|
|
setStatus(status) {
|
|||
|
|
this.onWillStatusChange && this.status !== status && this.onWillStatusChange(this.status, status);
|
|||
|
|
this.status = status;
|
|||
|
|
};
|
|||
|
|
// 设置识别结果内容
|
|||
|
|
setResultText({ resultText, resultTextTemp } = {}) {
|
|||
|
|
this.onTextChange && this.onTextChange(resultTextTemp || resultText || '');
|
|||
|
|
resultText !== undefined && (this.resultText = resultText);
|
|||
|
|
resultTextTemp !== undefined && (this.resultTextTemp = resultTextTemp);
|
|||
|
|
};
|
|||
|
|
// 修改听写参数
|
|||
|
|
setParams({ language, accent } = {}) {
|
|||
|
|
language && (this.language = language)
|
|||
|
|
accent && (this.accent = accent)
|
|||
|
|
};
|
|||
|
|
// 对处理后的音频数据进行base64编码，
|
|||
|
|
toBase64(buffer) {
|
|||
|
|
let binary = '';
|
|||
|
|
let bytes = new Uint8Array(buffer);
|
|||
|
|
for (let i = 0; i < bytes.byteLength; i++) {
|
|||
|
|
binary += String.fromCharCode(bytes[i]);
|
|||
|
|
}
|
|||
|
|
return window.btoa(binary);
|
|||
|
|
};
|
|||
|
|
// 连接WebSocket
|
|||
|
|
connectWebSocket() {
|
|||
|
|
return this.getWebSocketUrl().then(url => {
|
|||
|
|
let iatWS;
|
|||
|
|
if ('WebSocket' in window) {
|
|||
|
|
iatWS = new WebSocket(url);
|
|||
|
|
} else if ('MozWebSocket' in window) {
|
|||
|
|
iatWS = new MozWebSocket(url);
|
|||
|
|
} else {
|
|||
|
|
alert('浏览器不支持WebSocket!');
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
this.webSocket = iatWS;
|
|||
|
|
this.setStatus('init');
|
|||
|
|
iatWS.onopen = e => {
|
|||
|
|
this.setStatus('ing');
|
|||
|
|
// 重新开始录音
|
|||
|
|
setTimeout(() => {
|
|||
|
|
this.webSocketSend();
|
|||
|
|
}, 500);
|
|||
|
|
};
|
|||
|
|
iatWS.onmessage = e => {
|
|||
|
|
this.webSocketRes(e.data);
|
|||
|
|
};
|
|||
|
|
iatWS.onerror = e => {
|
|||
|
|
this.recorderStop(e);
|
|||
|
|
};
|
|||
|
|
iatWS.onclose = e => {
|
|||
|
|
this.recorderStop(e);
|
|||
|
|
};
|
|||
|
|
})
|
|||
|
|
};
|
|||
|
|
// 初始化浏览器录音
|
|||
|
|
recorderInit() {
|
|||
|
|
// 创建音频环境
|
|||
|
|
try {
|
|||
|
|
this.audioContext = this.audioContext ? this.audioContext : new (window.AudioContext || window.webkitAudioContext)();
|
|||
|
|
this.audioContext.resume();
|
|||
|
|
if (!this.audioContext) {
|
|||
|
|
alert('浏览器不支持webAudioApi相关接口');
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
} catch (e) {
|
|||
|
|
if (!this.audioContext) {
|
|||
|
|
alert('浏览器不支持webAudioApi相关接口');
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 获取浏览器录音权限成功时回调
|
|||
|
|
let getMediaSuccess = _ => {
|
|||
|
|
// 创建一个用于通过JavaScript直接处理音频
|
|||
|
|
this.scriptProcessor = this.audioContext.createScriptProcessor(0, 1, 1);
|
|||
|
|
this.scriptProcessor.onaudioprocess = e => {
|
|||
|
|
if (this.status === 'ing') {
|
|||
|
|
// 多线程音频数据处理
|
|||
|
|
try {
|
|||
|
|
this.webWorker.postMessage(e.inputBuffer.getChannelData(0));
|
|||
|
|
} catch (error) { }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 创建一个新的MediaStreamAudioSourceNode 对象，使来自MediaStream的音频可以被播放和操作
|
|||
|
|
this.mediaSource = this.audioContext.createMediaStreamSource(this.streamRef);
|
|||
|
|
this.mediaSource.connect(this.scriptProcessor);
|
|||
|
|
this.scriptProcessor.connect(this.audioContext.destination);
|
|||
|
|
this.connectWebSocket();
|
|||
|
|
};
|
|||
|
|
// 获取浏览器录音权限失败时回调
|
|||
|
|
let getMediaFail = (e) => {
|
|||
|
|
alert('对不起：录音权限获取失败!');
|
|||
|
|
this.audioContext && this.audioContext.close();
|
|||
|
|
this.audioContext = undefined;
|
|||
|
|
// 关闭websocket
|
|||
|
|
if (this.webSocket && this.webSocket.readyState === 1) {
|
|||
|
|
this.webSocket.close();
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
navigator.getUserMedia = navigator.getUserMedia || navigator.webkitGetUserMedia || navigator.mozGetUserMedia || navigator.msGetUserMedia;
|
|||
|
|
// 获取浏览器录音权限
|
|||
|
|
if (navigator.mediaDevices && navigator.mediaDevices.getUserMedia) {
|
|||
|
|
navigator.mediaDevices.getUserMedia({
|
|||
|
|
audio: true
|
|||
|
|
}).then(stream => {
|
|||
|
|
this.streamRef = stream;
|
|||
|
|
getMediaSuccess();
|
|||
|
|
}).catch(e => {
|
|||
|
|
getMediaFail(e);
|
|||
|
|
})
|
|||
|
|
} else if (navigator.getUserMedia) {
|
|||
|
|
navigator.getUserMedia({
|
|||
|
|
audio: true
|
|||
|
|
}, (stream) => {
|
|||
|
|
this.streamRef = stream;
|
|||
|
|
getMediaSuccess();
|
|||
|
|
}, function (e) {
|
|||
|
|
getMediaFail(e);
|
|||
|
|
})
|
|||
|
|
} else {
|
|||
|
|
if (navigator.userAgent.toLowerCase().match(/chrome/) && location.origin.indexOf('https://') < 0) {
|
|||
|
|
console.error('获取浏览器录音功能，因安全性问题，需要在localhost 或 127.0.0.1 或 https 下才能获取权限！');
|
|||
|
|
} else {
|
|||
|
|
alert('对不起：未识别到录音设备!');
|
|||
|
|
}
|
|||
|
|
this.audioContext && this.audioContext.close();
|
|||
|
|
return false;
|
|||
|
|
};
|
|||
|
|
};
|
|||
|
|
// 向webSocket发送数据(音频二进制数据经过Base64处理)
|
|||
|
|
webSocketSend() {
|
|||
|
|
if (this.webSocket.readyState !== 1) return false;
|
|||
|
|
// 音频数据
|
|||
|
|
const audioData = this.audioData.splice(0, 1280);
|
|||
|
|
const params = {
|
|||
|
|
common: {
|
|||
|
|
app_id: this.appId,
|
|||
|
|
},
|
|||
|
|
business: {
|
|||
|
|
language: this.language, //小语种可在控制台--语音听写（流式）--方言/语种处添加试用
|
|||
|
|
domain: 'iat',
|
|||
|
|
accent: this.accent, //中文方言可在控制台--语音听写（流式）--方言/语种处添加试用
|
|||
|
|
vad_eos: 5000,
|
|||
|
|
dwa: 'wpgs' //为使该功能生效，需到控制台开通动态修正功能（该功能免费）
|
|||
|
|
},
|
|||
|
|
data: {
|
|||
|
|
status: 0,
|
|||
|
|
format: 'audio/L16;rate=16000',
|
|||
|
|
encoding: 'raw',
|
|||
|
|
audio: this.toBase64(audioData)
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 发送数据
|
|||
|
|
this.webSocket.send(JSON.stringify(params));
|
|||
|
|
this.handlerInterval = setInterval(() => {
|
|||
|
|
// websocket未连接
|
|||
|
|
if (this.webSocket.readyState !== 1) {
|
|||
|
|
this.audioData = [];
|
|||
|
|
clearInterval(this.handlerInterval);
|
|||
|
|
return false;
|
|||
|
|
};
|
|||
|
|
if (this.audioData.length === 0) {
|
|||
|
|
if (this.status === 'end') {
|
|||
|
|
this.webSocket.send(
|
|||
|
|
JSON.stringify({
|
|||
|
|
data: {
|
|||
|
|
status: 2,
|
|||
|
|
format: 'audio/L16;rate=16000',
|
|||
|
|
encoding: 'raw',
|
|||
|
|
audio: ''
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
);
|
|||
|
|
this.audioData = [];
|
|||
|
|
clearInterval(this.handlerInterval);
|
|||
|
|
}
|
|||
|
|
return false;
|
|||
|
|
};
|
|||
|
|
// 中间帧
|
|||
|
|
this.webSocket.send(
|
|||
|
|
JSON.stringify({
|
|||
|
|
data: {
|
|||
|
|
status: 1,
|
|||
|
|
format: 'audio/L16;rate=16000',
|
|||
|
|
encoding: 'raw',
|
|||
|
|
audio: this.toBase64(this.audioData.splice(0, 1280))
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
);
|
|||
|
|
}, 40);
|
|||
|
|
};
|
|||
|
|
// 识别结束 webSocket返回数据
|
|||
|
|
webSocketRes(resultData) {
|
|||
|
|
let jsonData = JSON.parse(resultData);
|
|||
|
|
if (jsonData.data && jsonData.data.result) {
|
|||
|
|
let data = jsonData.data.result;
|
|||
|
|
let str = '';
|
|||
|
|
let ws = data.ws;
|
|||
|
|
for (let i = 0; i < ws.length; i++) {
|
|||
|
|
str = str + ws[i].cw[0].w;
|
|||
|
|
}
|
|||
|
|
// 开启wpgs会有此字段(前提：在控制台开通动态修正功能)
|
|||
|
|
// 取值为 "apd"时表示该片结果是追加到前面的最终结果；取值为"rpl" 时表示替换前面的部分结果，替换范围为rg字段
|
|||
|
|
if (data.pgs) {
|
|||
|
|
if (data.pgs === 'apd') {
|
|||
|
|
// 将resultTextTemp同步给resultText
|
|||
|
|
this.setResultText({
|
|||
|
|
resultText: this.resultTextTemp
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
// 将结果存储在resultTextTemp中
|
|||
|
|
this.setResultText({
|
|||
|
|
resultTextTemp: this.resultText + str
|
|||
|
|
});
|
|||
|
|
} else {
|
|||
|
|
this.setResultText({
|
|||
|
|
resultText: this.resultText + str
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (jsonData.code === 0 && jsonData.data.status === 2) {
|
|||
|
|
this.webSocket.close();
|
|||
|
|
}
|
|||
|
|
if (jsonData.code !== 0) {
|
|||
|
|
this.webSocket.close();
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 启动录音
|
|||
|
|
recorderStart() {
|
|||
|
|
if (!this.audioContext) {
|
|||
|
|
this.recorderInit();
|
|||
|
|
} else {
|
|||
|
|
this.audioContext.resume();
|
|||
|
|
this.connectWebSocket();
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 停止录音
|
|||
|
|
recorderStop() {
|
|||
|
|
if (!(/Safari/.test(navigator.userAgent) && !/Chrome/.test(navigator.userAgen))) {
|
|||
|
|
// safari下suspend后再次resume录音内容将是空白，设置safari下不做suspend
|
|||
|
|
this.audioContext && this.audioContext.suspend();
|
|||
|
|
}
|
|||
|
|
this.setStatus('end');
|
|||
|
|
try {
|
|||
|
|
// this.streamRef.getTracks().map(track => track.stop()) || his.streamRef.getAudioTracks()[0].stop();
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('暂停失败!');
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
// 开始
|
|||
|
|
start() {
|
|||
|
|
this.recorderStart();
|
|||
|
|
this.setResultText({ resultText: '', resultTextTemp: '' });
|
|||
|
|
};
|
|||
|
|
// 停止
|
|||
|
|
stop() {
|
|||
|
|
this.recorderStop();
|
|||
|
|
};
|
|||
|
|
};
|
|||
|
|
}));
|