feat: add tiktok tts
This commit is contained in:
@@ -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();
|
||||
@@ -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"
|
||||
]
|
||||
}
|
||||
Reference in New Issue
Block a user