prehistoric-systems/omnivore
1
0
Fork 0
mirror of https://github.com/omnivore-app/omnivore.git synced 2026-03-11 08:54:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Pull token from Authorization header too

Browse source
This commit is contained in:
Hongbo Wu 2022-09-01 15:26:19 +08:00
parent 3a40dd7266
commit 7bb3d711c5
2 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
packages/text-to-speech/src/index.ts
View file

@ -81,12 +81,12 @@ export const textToSpeechHandler = Sentry.GCPFunction.wrapHttpFunction(
const audioFile = createGCSFile(bucket, audioFileName)
const writeStream = audioFile.createWriteStream({
resumable: true,
})
}) as NodeJS.WriteStream
const startTime = Date.now()
const { speechMarks } = await synthesizeTextToSpeech({
...input,
textType: 'html',
writeStream,
audioStream: writeStream,
})
console.info(
`Synthesize text to speech completed in ${Date.now() - startTime} ms`
@ -124,25 +124,28 @@ export const textToSpeechHandler = Sentry.GCPFunction.wrapHttpFunction(
export const textToSpeechStreamingHandler = Sentry.GCPFunction.wrapHttpFunction(
async (req, res) => {
console.debug('Text to speech steaming request', req)
const token = req.query.token as string
if (!process.env.JWT_SECRET) {
console.error('JWT_SECRET not exists')
return res.status(500).send('JWT_SECRET not exists')
return res.status(500).send({ errorCodes: 'JWT_SECRET_NOT_EXISTS' })
}
const token = (req.query.token || req.headers.authorization) as string
if (!token) {
return res.status(200).send({ errorCode: 'UNAUTHORIZED' })
}
try {
jwt.verify(token, process.env.JWT_SECRET)
} catch (e) {
console.error(e)
return res.status(200).send('UNAUTHENTICATED')
return res.status(200).send({ errorCode: 'UNAUTHORIZED' })
}
try {
const ssml = fs.readFileSync('./data/ssml.xml', 'utf8')
const writeStream = new PassThrough()
const audioStream = new PassThrough()
const input: TextToSpeechInput = {
text: ssml,
textType: 'ssml',
writeStream,
audioStream,
}
res.set({
'Content-Type': 'audio/mpeg',
@ -150,7 +153,7 @@ export const textToSpeechStreamingHandler = Sentry.GCPFunction.wrapHttpFunction(
})
console.info('Text to speech starts streaming')
writeStream.pipe(res)
audioStream.pipe(res)
await synthesizeTextToSpeech(input)
} catch (e) {

4
packages/text-to-speech/src/textToSpeech.ts
View file

@ -19,7 +19,7 @@ export interface TextToSpeechInput {
volume?: number
complimentaryVoice?: string
bucket?: string
writeStream: NodeJS.WritableStream
audioStream: NodeJS.ReadWriteStream
}
export interface TextToSpeechOutput {
@ -40,7 +40,7 @@ export const synthesizeTextToSpeech = async (
if (!process.env.AZURE_SPEECH_KEY || !process.env.AZURE_SPEECH_REGION) {
throw new Error('Azure Speech Key or Region not set')
}
const writeStream = input.writeStream
const writeStream = input.audioStream
const speechConfig = SpeechConfig.fromSubscription(
process.env.AZURE_SPEECH_KEY,
process.env.AZURE_SPEECH_REGION