deeplx 2.2.6

use std::io;

use rand::prelude::*;

use super::data::{DeepLXTranslationResult, Lang, Params, PostData, TextItem, TranslationResponse};
use super::error::{Error, LangDetectError};
use super::utils::{get_i_count, get_random_number, get_timestamp};

use reqwest::{
    Client, Response, StatusCode,
    header::{
        ACCEPT, ACCEPT_ENCODING, ACCEPT_LANGUAGE, CONNECTION, CONTENT_TYPE, COOKIE, DNT, HeaderMap,
        HeaderValue, UPGRADE_INSECURE_REQUESTS, USER_AGENT,
    },
};
#[cfg(not(target_arch = "wasm32"))]
use reqwest::{Proxy, retry};

/// Configuration settings for the `DeepLX` translation client.
///
/// # Examples
///
/// ## Using Default Configuration
///
/// ```no_run
/// use deeplx::{Config, DeepLX};
///
/// let translator = DeepLX::new(Config::default());
/// ```
///
/// ## Custom Base URL
///
/// ```no_run
/// use deeplx::{Config, DeepLX};
///
/// let translator = DeepLX::new(Config {
///     base_url: "https://custom.deepl.api/jsonrpc".to_string(),
///     ..Default::default()
/// });
/// ```
///
/// ## Configuring a Proxy (Not available on wasm32)
///
/// ```no_run
/// use deeplx::{Config, DeepLX};
///
/// let translator = DeepLX::new(Config {
///     #[cfg(not(target_arch = "wasm32"))]
///     proxy: Some("http://pro.xy".to_string()),
///     ..Default::default()
/// });
/// ```
pub struct Config {
    pub base_url: String,
    #[cfg(not(target_arch = "wasm32"))]
    pub proxy: Option<String>,
}

impl Default for Config {
    fn default() -> Self {
        Self {
            base_url: "https://www2.deepl.com/jsonrpc".to_string(),
            #[cfg(not(target_arch = "wasm32"))]
            proxy: None,
        }
    }
}

/// The main entry point for interacting with the DeepL translation service.
///
/// `DeepLX` provides methods to create a translation client and perform translation
/// requests. You can optionally specify a proxy (on non‑wasm32 targets),
/// choose source and target languages, and retrieve alternative translations.
#[derive(Clone)]
pub struct DeepLX {
    base_url: String,
    #[cfg(not(target_arch = "wasm32"))]
    proxy: Option<String>,
    headers: HeaderMap,
}

impl DeepLX {
    /// Constructs a new `DeepLX` instance.
    ///
    /// # Parameters
    ///
    /// * `Config` - Configuration settings for the translation client.
    ///
    /// # Panics
    ///
    /// This method will panic if the provided proxy string is invalid.
    ///
    /// # Examples
    ///
    /// ```no_run
    /// use deeplx::{Config, DeepLX};
    ///
    /// let translator = DeepLX::new(Config::default());
    /// let translator_with_proxy = DeepLX::new(Config {
    ///     #[cfg(not(target_arch = "wasm32"))]
    ///     proxy: Some("http://pro.xy".to_string()),
    ///     ..Default::default()
    /// });
    /// ```
    pub fn new(config: Config) -> Self {
        Self {
            base_url: config.base_url,
            #[cfg(not(target_arch = "wasm32"))]
            proxy: config.proxy,
            headers: headers(),
        }
    }

    async fn make_request(
        &self,
        post_data: &PostData<'_>,
        deepl_session: Option<&str>,
    ) -> Result<(StatusCode, Response), Error> {
        let mut headers = self.headers.clone();
        if let Some(session) = deepl_session {
            headers.insert(COOKIE, session.parse().unwrap());
        }

        let data = serde_json::to_string(&post_data)?;

        let use_colon_spacing = ((post_data.id + 5) % 29 == 0) || ((post_data.id + 3) % 13 == 0);
        let replacement = if use_colon_spacing {
            r#""method" : ""#
        } else {
            r#""method": ""#
        };

        let data = data.replacen(r#""method":""#, replacement, 1);

        let builder = Client::builder();

        #[cfg(not(target_arch = "wasm32"))]
        let builder = match &self.proxy {
            Some(p) => builder.proxy(Proxy::all(p.clone())?),
            None => builder,
        };

        #[cfg(not(target_arch = "wasm32"))]
        let builder = builder.retry(
            retry::for_host(self.base_url.clone())
                .max_retries_per_request(3)
                .classify_fn(|req| match req.status() {
                    Some(status) if status == StatusCode::TOO_MANY_REQUESTS => req.retryable(),
                    _ => req.success(),
                }),
        );

        let resp = builder
            .build()?
            .post(&self.base_url)
            .headers(headers)
            .body(data)
            .send()
            .await?;

        Ok((resp.status(), resp))
    }

    /// Translates the given text from a source language to a target language.
    ///
    /// This method automatically handles splitting the text into translation jobs,
    /// detecting the source language (if set to "auto"), and returning the translated text.
    ///
    /// # Parameters
    ///
    /// * `source_lang` - The source language code, e.g. `"en"`. Use `"auto"` to let the system detect the language.
    /// * `target_lang` - The target language code, e.g. `"zh"` or `"EN-GB"` for a regional variant.
    /// * `text` - The text to translate. Cannot be empty.
    /// * `deepl_session` - An optional session string. If `None`, the "Free" method is used; otherwise "Pro".
    ///
    /// # Returns
    ///
    /// On success, returns a `DeepLXTranslationResult` containing the translated text and alternatives.
    /// On failure, returns an `Err` containing the underlying error.
    ///
    /// # Examples
    ///
    /// ```no_run
    /// use deeplx::{Config, DeepLX};
    ///
    /// async fn run() {
    ///     let translator = DeepLX::new(Config::default());
    ///     match translator
    ///         .translate("auto", "zh", "Hello, world!", None)
    ///         .await {
    ///         Ok(res) => println!("Translated: {}", res.data),
    ///         Err(e) => eprintln!("Error: {}", e),
    ///     }
    /// }
    ///
    /// #[cfg(not(target_arch = "wasm32"))]
    /// #[tokio::main]
    /// async fn main() {
    ///    run().await;
    /// }
    ///
    /// #[cfg(target_arch = "wasm32")]
    /// #[tokio::main(flavor = "current_thread")]
    /// async fn main() {
    ///    run().await;
    /// }
    /// ```
    pub async fn translate(
        &self,
        source_lang: &str,
        target_lang: &str,
        text: &str,
        deepl_session: Option<&str>,
    ) -> Result<DeepLXTranslationResult, Error> {
        // return if there's nothing to translate
        if text.is_empty() {
            return Ok(DeepLXTranslationResult {
                code: 404,
                message: Some("No text to translate".to_string()),
                ..Default::default()
            });
        }

        // determine source language
        let source_lang_detached = match source_lang {
            "auto" | "" => {
                let iso_639_3 = whatlang::detect_lang(text)
                    .ok_or(LangDetectError::from(io::Error::new(
                        io::ErrorKind::InvalidInput,
                        "Failed to detect language",
                    )))?
                    .code();

                isolang::Language::from_639_3(iso_639_3)
                    .and_then(|lang| lang.to_639_1())
                    .map(|iso_639_1| iso_639_1.to_uppercase())
                    .ok_or(LangDetectError::from(io::Error::new(
                        io::ErrorKind::InvalidInput,
                        "Could not map detected language to ISO 639-1",
                    )))?
            }
            _ => source_lang.to_uppercase(),
        };

        // check target language
        let target_lang_parts = target_lang.split('-').collect::<Vec<&str>>();
        let (target_lang_code, _) = if target_lang_parts.len() > 1 {
            (target_lang_parts[0].to_uppercase(), true)
        } else {
            (target_lang.to_uppercase(), false)
        };

        // prepare the JSON-RPC request
        let id = get_random_number();
        let i_count = get_i_count(text);
        let timestamp = get_timestamp(i_count);

        let post_data = PostData {
            json_rpc: "2.0",
            method: "LMT_handle_texts",
            id,
            params: Params {
                splitting: "newlines",
                lang: Lang {
                    source_lang_user_selected: source_lang,
                    target_lang: target_lang_code.as_str(),
                    ..Default::default()
                },
                texts: vec![TextItem {
                    text,
                    request_alternatives: 3,
                }],
                timestamp,
            },
        };

        // send request and parse response
        let (status, resp) = self.make_request(&post_data, deepl_session).await?;
        if !status.is_success() {
            return Ok(DeepLXTranslationResult {
                code: status.as_u16() as i32,
                ..Default::default()
            });
        }

        let resp: TranslationResponse = resp.json().await?;

        let texts = resp.result.texts;
        if texts.is_empty() {
            return Ok(DeepLXTranslationResult {
                code: 503,
                message: Some("Translation failed".to_string()),
                ..Default::default()
            });
        }

        let main_translation = texts[0].text.clone();

        // collect alternatives
        // let num_beams = translations[0].beams.len();
        let alternatives: Vec<String> = texts
            .iter()
            .map(|t| t.text.clone())
            .filter(|alt| !alt.is_empty())
            .collect();

        Ok(DeepLXTranslationResult {
            code: 200,
            id,
            data: main_translation,
            alternatives,
            source_lang: if resp.result.lang.is_empty() {
                source_lang_detached
            } else {
                resp.result.lang
            },
            target_lang: target_lang.to_string(),
            method: if deepl_session.is_none() {
                "Free"
            } else {
                "Pro"
            }
            .to_string(),
            ..Default::default()
        })
    }
}

const USER_AGENTS: &[&str] = &[
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:121.0) Gecko/20100101 Firefox/121.0",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:121.0) Gecko/20100101 Firefox/121.0",
];

const ACCEPT_LANGUAGES: &[&str] = &[
    "en-US,en;q=0.9",
    "en-GB,en;q=0.9",
    "en-US,en;q=0.8,es;q=0.6",
    "en-US,en;q=0.9,fr;q=0.8",
    "en-US,en;q=0.9,de;q=0.8",
];

fn headers() -> HeaderMap {
    let mut rng = rand::rng();

    let mut headers = HeaderMap::new();
    headers.insert(
        CONTENT_TYPE,
        HeaderValue::from_static("application/json; charset=utf-8"),
    );
    headers.insert(
        USER_AGENT,
        HeaderValue::from_str(USER_AGENTS.choose(&mut rng).unwrap()).unwrap(),
    );
    headers.insert(
        ACCEPT_LANGUAGE,
        HeaderValue::from_str(ACCEPT_LANGUAGES.choose(&mut rng).unwrap()).unwrap(),
    );
    headers.insert(
        ACCEPT,
        HeaderValue::from_static("application/json, text/plain, */*"),
    );
    headers.insert(
        ACCEPT_ENCODING,
        HeaderValue::from_static("gzip, deflate, br"),
    );
    headers.insert(DNT, HeaderValue::from_static("1"));
    headers.insert(CONNECTION, HeaderValue::from_static("keep-alive"));
    headers.insert(UPGRADE_INSECURE_REQUESTS, HeaderValue::from_static("1"));

    headers
}