add some options

Author: daimatz

Diff for: env.yaml.template

@@ -28,5 +28,26 @@ voice_name: ja-JP-Standard-A,ja-JP-Standard-B,ja-JP-Standard-C,ja-JP-Standard-D
 # を参照してください。
 language_code: ja-JP
 
+# Text-to-Speech API で指定する再生速度。
+# 0.25 から 4.00 までの数値。
+# 詳細は
+#   https://cloud.google.com/text-to-speech/docs/reference/rest/v1/text/synthesize#audioconfig
+# を参照してください。
+speaking_rate: "1.00"
+
+# Text-to-Speech API で指定する音程。
+# -20.0 から 20.0 までの数値。
+# 詳細は
+#   https://cloud.google.com/text-to-speech/docs/reference/rest/v1/text/synthesize#audioconfig
+# を参照してください。
+pitch: "0.0"
+
+# Text-to-Speech API で指定するボリューム。
+# -96.0 から 16.0 までの数値。
+# 詳細は
+#   https://cloud.google.com/text-to-speech/docs/reference/rest/v1/text/synthesize#audioconfig
+# を参照してください。
+volume_gain_db: "0.0"
+
 # language_code において指定した言語における区切り文字。
 delimiter: 。

Diff for: ts/Config.ts

@@ -0,0 +1,12 @@
+export type Config = {
+  bucket_name: string
+  input_pdf_path_regexp: string
+  output_path: string
+  temp_path: string
+  voice_name: string
+  language_code: string
+  speaking_rate?: number
+  pitch?: number
+  volume_gain_db?: number
+  delimiter: string
+};

Diff for: ts/TextToSpeechTask.ts

@@ -1,3 +1,4 @@
+import { Config } from './Config';
 import { GcsLib } from './GcsLib';
 const promiseRetry = require('promise-retry');
 const TextToSpeech = require('@google-cloud/text-to-speech');
@@ -6,17 +7,14 @@ export class TextToSpeechTask {
   private readonly gcs: GcsLib;
   private readonly client: typeof TextToSpeech.TextToSpeechClient;
   private readonly voiceName: string;
-  private readonly languageCode: string;
-  private readonly delimiter: string;
-  private readonly maxConcurrency = 8;
+  private readonly config: Config;
   private readonly maxLength = 5000;
 
-  constructor(gcs: GcsLib, voiceName: string, languageCode: string, delimiter: string) {
+  constructor(gcs: GcsLib, voiceName: string, config: Config) {
     this.gcs = gcs;
     this.client = new TextToSpeech.TextToSpeechClient();
     this.voiceName = voiceName;
-    this.languageCode = languageCode;
-    this.delimiter = delimiter;
+    this.config = config;
   }
 
   run(texts: Array<string>, outputPrefix: string): Promise<Array<string>> {
@@ -68,9 +66,20 @@ export class TextToSpeechTask {
   ttsRequest(text: string, outputPath: string): Promise<string> {
     console.log(`ttsRequest(..., ${outputPath})`);
     const request = {
-      input: {text: text},
-      voice: {name: this.voiceName, languageCode: this.languageCode, ssmlGender: 'NEUTRAL'},
-      audioConfig: {audioEncoding: 'MP3'},
+      input: {
+        text: text,
+      },
+      voice: {
+        name: this.voiceName,
+        languageCode: this.config.language_code,
+        ssmlGender: 'NEUTRAL',
+      },
+      audioConfig: {
+        audioEncoding: 'MP3',
+        speakingRate: Number(this.config.speaking_rate) || null,
+        pitch: Number(this.config.pitch) || null,
+        volumeGainDb: Number(this.config.volume_gain_db) || null,
+      },
     };
     return this.client.synthesizeSpeech(request).then(([response]) => {
       return new Promise((resolve, reject) => {

Diff for: ts/index.ts

@@ -1,5 +1,6 @@
-import { main, Config } from './main';
+import { Config } from './Config';
+import { main } from './main';
 
 exports.PdfAudify = async (file, context) => {
-  await main(file.name, process.env as Config);
+  await main(file.name, process.env as unknown as Config);
 };

Diff for: ts/main.ts

@@ -1,3 +1,4 @@
+import { Config } from './Config';
 import { GcsLib } from './GcsLib';
 import { OcrTask } from './OcrTask';
 import { ExtractTextTask } from './ExtractTextTask';
@@ -6,16 +7,6 @@ import { ConcatMp3Task } from './ConcatMp3Task';
 const promiseRetry = require('promise-retry');
 const path = require('path');
 
-export type Config = {
-  bucket_name: string
-  input_pdf_path_regexp: string
-  output_path: string
-  temp_path: string
-  voice_name: string
-  language_code: string
-  delimiter: string
-};
-
 export function main(gcsPath: string, config: Config): Promise<void> {
   console.log(`gcsPath: ${gcsPath}, config: ${JSON.stringify(config)}`);
   if (!gcsPath.match(new RegExp(config.input_pdf_path_regexp))) {
@@ -30,7 +21,7 @@ export function main(gcsPath: string, config: Config): Promise<void> {
   const gcs = new GcsLib(config.bucket_name);
   const ocr = new OcrTask(gcs);
   const ext = new ExtractTextTask(gcs, config.delimiter);
-  const tts = new TextToSpeechTask(gcs, voice_name, config.language_code, config.delimiter);
+  const tts = new TextToSpeechTask(gcs, voice_name, config);
   const concat = new ConcatMp3Task(gcs);
 
   const basename = path.parse(gcsPath.normalize()).name;