HoovesSocial/hippofish
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases Packages Wiki Activity Actions

chore (backend-rs): libretranslate API now supports zh-Hant translations

Browse source
This commit is contained in:
naskya 2024-07-21 12:07:59 +09:00
parent 332b6cd620
commit bf61dee538
No known key found for this signature in database
GPG key ID: 712D413B3A9FED5C
1 changed files with 24 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
packages/backend-rs/src/misc/translate.rs
View file

@ -32,6 +32,11 @@ pub struct Translation {
pub text: String,
}
#[inline]
fn is_zh_hant_tw(lang: &str) -> bool {
["zh-tw", "zh-hant", "zh-hant-tw"].contains(&lang.to_ascii_lowercase().as_str())
}
#[macros::export]
pub async fn translate(
text: &str,
@ -40,7 +45,7 @@ pub async fn translate(
) -> Result<Translation, Error> {
let config = local_server_info().await?;
let mut translation = if let Some(api_key) = config.deepl_auth_key {
let translation = if let Some(api_key) = config.deepl_auth_key {
deepl_translate::translate(
text,
source_lang,
@ -86,12 +91,6 @@ pub async fn translate(
return Err(Error::NoTranslator);
};
// DeepL translate and LibreTranslate don't provide zh-Hant-TW translations,
// so we convert zh-Hans-CN translations into zh-Hant-TW using zhconv.
if ["zh-tw", "zh-hant", "zh-hant-tw"].contains(&target_lang.to_ascii_lowercase().as_str()) {
translation.text = zhconv::zhconv(&translation.text, zhconv::Variant::ZhTW)
}
Ok(translation)
}
@ -127,6 +126,8 @@ mod deepl_translate {
"https://api-free.deepl.com/v2/translate"
};
let to_zh_hant_tw = super::is_zh_hant_tw(target_lang);
let mut target_lang = target_lang.split('-').collect::<Vec<&str>>()[0];
// DeepL API requires us to specify "en-US" or "en-GB" for English
@ -163,13 +164,21 @@ mod deepl_translate {
.ok_or(super::Error::NoResponse)?
.to_owned();
Ok(super::Translation {
let mut translation = super::Translation {
source_lang: source_lang
.map(|s| s.to_owned())
.or(result.detected_source_language)
.unwrap_or_else(|| "unknown".to_owned()),
text: result.text,
})
};
// DeepL translate don't provide zh-Hant-TW translations at this moment,
// so we convert zh-Hans-CN translations into zh-Hant-TW using zhconv.
if to_zh_hant_tw {
translation.text = zhconv::zhconv(&translation.text, zhconv::Variant::ZhTW);
}
Ok(translation)
}
}
@ -199,7 +208,12 @@ mod libre_translate {
api_key: Option<&str>,
) -> Result<super::Translation, super::Error> {
let client = http_client::client()?;
let target_lang = target_lang.split('-').collect::<Vec<&str>>()[0];
let target_lang = if super::is_zh_hant_tw(target_lang) {
"zt"
} else {
target_lang.split('-').collect::<Vec<&str>>()[0]
};
let body = if let Some(source_lang) = source_lang {
let source_lang = source_lang.split('-').collect::<Vec<&str>>()[0];