feat: add tiktok tts

src/modules/tts-modes/tiktok.ts

@@ -0,0 +1,118 @@
+import { config } from "../../utils/config";
+import { TTSModule, TTSResponse } from "../tts"
+
+import * as https from 'https';
+import * as zlib from 'zlib';
+
+import TIKTOK_TTS_VOICES from './tiktok_voices.json';
+const TIKTOK_TTS_ENDPOINT = 'https://api16-normal-v6.tiktokv.com/media/api/text/speech/invoke'
+
+class TikTokTTS implements TTSModule {
+    public name: string = 'TikTok';
+    public defaultVoice: string = 'en_us_001';
+
+    async getVoices(): Promise<Array<string> | undefined> {
+        return TIKTOK_TTS_VOICES.voices;
+    }
+
+    async generate(voice: string, text: string): Promise<TTSResponse> {
+        const reqText = encodeURIComponent(text);
+        const path = `/?text_speaker=${voice}&req_text=${reqText}&speaker_map_type=0&aid=1233`;
+
+        const endpoint = new URL(TIKTOK_TTS_ENDPOINT);
+
+        const options: https.RequestOptions = {
+            hostname: endpoint.hostname,
+            path: endpoint.pathname + path,
+            method: 'POST',
+            headers: {
+                'User-Agent': 'com.zhiliaoapp.musically/2022600030 (Linux; U; Android 7.1.2; es_ES; SM-G988N; Build/NRD90M;tt-ok/3.12.13.1)',
+                'Cookie': `sessionid=${config.tiktok_session_id}`,
+                'Accept-Encoding': 'gzip,deflate,compress',
+                'Content-Type': 'application/x-www-form-urlencoded'
+            }
+        }
+
+        return new Promise((resolve) => {
+            const req = https.request(options, (res) => {
+                const chunks: Buffer[] = [];
+
+                const encoding = res.headers['content-encoding'];
+
+                res.on('data', (chunk) => chunks.push(chunk));
+                res.on('end', () => {
+                    try {
+                        const buffer = Buffer.concat(chunks);
+
+                        // Handle decompression if needed
+                        const decompressBuffer = (buf: Buffer): Promise<Buffer> => {
+                            return new Promise((decompressResolve, decompressReject) => {
+                                if (encoding === 'gzip' || encoding === 'deflate') {
+                                    zlib.unzip(buf, (err: any, decompressed: Buffer<ArrayBufferLike> | PromiseLike<Buffer<ArrayBufferLike>>) => {
+                                        if (err) decompressReject(err);
+                                        else decompressResolve(decompressed);
+                                    });
+                                } else {
+                                    decompressResolve(buf);
+                                }
+                            });
+                        };
+
+                        decompressBuffer(buffer)
+                            .then((decompressed) => {
+                                const result = JSON.parse(decompressed.toString());
+                                const statusCode = result?.status_code;
+
+                                if (statusCode !== 0) {
+                                    const errorMsg = this.handleStatusError(statusCode);
+                                    return resolve({ error: errorMsg });
+                                }
+
+                                const voiceStr = result?.data?.v_str;
+                                if (!voiceStr) {
+                                    return resolve({ error: 'No audio data received' });
+                                }
+
+                                resolve({ data: Buffer.from(voiceStr, 'base64') });
+                            })
+                            .catch((err) => {
+                                resolve({ error: `Decompression/Parse error: ${err.message}` });
+                            });
+
+                    } catch (err) {
+                        resolve({ error: `Parse error: ${err}` });
+                    }
+                });
+            });
+
+            req.on('error', (err) => resolve({ error: err.message }));
+            req.on('timeout', () => {
+                req.destroy();
+                resolve({ error: 'timed out' });
+            });
+
+            req.write('');
+            req.end();
+        });
+    }
+
+    canBeUsed(): boolean {
+        return config.tiktok_session_id != undefined;
+    }
+
+    handleStatusError(code: number): string {
+        switch (code) {
+            case 1:
+                return "Session ID may be invalid or expired";
+            case 2:
+                return "Text is too long";
+            case 4:
+                return "Invalid voice";
+            case 5:
+                return "No session id.";
+        }
+        return `Unknown error code: ${code}`;
+    }
+}
+
+export default new TikTokTTS();

src/modules/tts-modes/tiktok_voices.json

@@ -0,0 +1,99 @@
+{
+    "voices": [
+        "en_us_ghostface",
+        "en_us_chewbacca",
+        "en_us_c3po",
+        "en_us_stitch",
+        "en_us_stormtrooper",
+        "en_us_rocket",
+        "en_female_madam_leota",
+        "en_male_ghosthost",
+        "en_male_pirate",
+        "en_au_001",
+        "en_au_002",
+        "en_uk_001",
+        "en_uk_003",
+        "en_us_001",
+        "en_us_002",
+        "en_us_006",
+        "en_us_007",
+        "en_us_009",
+        "en_us_010",
+        "en_male_jomboy",
+        "en_male_cody",
+        "en_female_samc",
+        "en_female_makeup",
+        "en_female_richgirl",
+        "en_male_grinch",
+        "en_male_deadpool",
+        "en_male_jarvis",
+        "en_male_ashmagic",
+        "en_male_olantekkers",
+        "en_male_ukneighbor",
+        "en_male_ukbutler",
+        "en_female_shenna",
+        "en_female_pansino",
+        "en_male_trevor",
+        "en_female_betty",
+        "en_male_cupid",
+        "en_female_grandma",
+        "en_male_m2_xhxs_m03_christmas",
+        "en_male_santa_narration",
+        "en_male_sing_deep_jingle",
+        "en_male_santa_effect",
+        "en_female_ht_f08_newyear",
+        "en_male_wizard",
+        "en_female_ht_f08_halloween",
+        "fr_001",
+        "fr_002",
+        "de_001",
+        "de_002",
+        "es_002",
+        "es_mx_002",
+        "br_001",
+        "br_003",
+        "br_004",
+        "br_005",
+        "bp_female_ivete",
+        "bp_female_ludmilla",
+        "pt_female_lhays",
+        "pt_female_laizza",
+        "pt_male_bueno",
+        "id_001",
+        "jp_001",
+        "jp_003",
+        "jp_005",
+        "jp_006",
+        "kr_002",
+        "kr_003",
+        "kr_004",
+        "jp_female_fujicochan",
+        "jp_female_hasegawariona",
+        "jp_male_keiichinakano",
+        "jp_female_oomaeaika",
+        "jp_male_yujinchigusa",
+        "jp_female_shirou",
+        "jp_male_tamawakazuki",
+        "jp_female_kaorishoji",
+        "jp_female_yagishaki",
+        "jp_male_hikakin",
+        "jp_female_rei",
+        "jp_male_shuichiro",
+        "jp_male_matsudake",
+        "jp_female_machikoriiita",
+        "jp_male_matsuo",
+        "jp_male_osada",
+        "en_female_f08_salut_damour",
+        "en_male_m03_lobby",
+        "en_female_f08_warmy_breeze",
+        "en_male_m03_sunshine_soon",
+        "en_female_ht_f08_glorious",
+        "en_male_sing_funny_it_goes_up",
+        "en_male_m2_xhxs_m03_silly",
+        "en_female_ht_f08_wonderful_world",
+        "en_male_sing_funny_thanksgiving",
+        "en_male_narration",
+        "en_male_funny",
+        "en_female_emotional"
+    ]
+}

src/utils/config.ts

@@ -13,6 +13,8 @@ export interface Config {
 
 	aws_access_id: string | undefined;
 	aws_access_key: string | undefined;
+
+	tiktok_session_id: string | undefined;
 }
 
 function loadConfig(): Config {
@@ -33,7 +35,8 @@ function loadConfig(): Config {
 		tts_elevenlabs_key: process.env.TTS_ELEVENLABS_KEY,
 		steam_webapi_key: process.env.STEAM_WEBAPI_KEY,
 		aws_access_id: process.env.AWS_ACCESS_ID,
-		aws_access_key: process.env.AWS_ACCESS_KEY
+		aws_access_key: process.env.AWS_ACCESS_KEY,
+		tiktok_session_id: process.env.TIKTOK_SESSION_ID
 	};
 }