# SPDX-License-Identifier: AGPL-3.0-or-later
"""Mozhi (alternative frontend for popular translation engines)"""

import random
import re
from urllib.parse import urlencode

about = {
    "website": 'https://codeberg.org/aryak/mozhi',
    "wikidata_id": None,
    "official_api_documentation": 'https://mozhi.aryak.me/api/swagger/index.html',
    "use_official_api": True,
    "require_api_key": False,
    "results": 'JSON',
}

engine_type = 'online_dictionary'
categories = ['general', 'translate']

base_url = "https://mozhi.aryak.me"
mozhi_engine = "google"

re_transliteration_unsupported = "Direction '.*' is not supported"


def request(_query, params):
    request_url = random.choice(base_url) if isinstance(base_url, list) else base_url

    args = {'from': params['from_lang'][1], 'to': params['to_lang'][1], 'text': params['query'], 'engine': mozhi_engine}
    params['url'] = f"{request_url}/api/translate?{urlencode(args)}"
    return params


def response(resp):
    translation = resp.json()

    data = {'text': translation['translated-text'], 'definitions': [], 'examples': []}

    if translation['target_transliteration'] and not re.match(
        re_transliteration_unsupported, translation['target_transliteration']
    ):
        data['transliteration'] = translation['target_transliteration']

    if translation['word_choices']:
        for word in translation['word_choices']:
            if word.get('definition'):
                data['definitions'].append(word['definition'])

            for example in word.get('examples_target', []):
                data['examples'].append(re.sub(r"<|>", "", example).lstrip('- '))

    data['synonyms'] = translation.get('source_synonyms', [])

    result = {
        'answer': translation['translated-text'],
        'answer_type': 'translations',
        'translations': [data],
    }

    return [result]