#!/usr/bin/env python
#
# This python file contains utility scripts to manage Python docs Ukrainian translation.
# It has to be run inside the python-docs-uk git root directory.

import os
import re
from argparse import ArgumentParser
from pathlib import Path

from transifex.api import transifex_api


transifex_api.setup(auth=os.getenv('TX_TOKEN'))

RESOURCE_NAME_MAP = {'glossary_': 'glossary'}

LANG = 'uk'

ORGANISATION_ID = 'o:python-doc'
PROJECT_ID = 'o:python-doc:p:python-newest'
LANGUAGE_ID = f'l:{LANG}'
ORGANISATION = transifex_api.Organization.get(id=ORGANISATION_ID)
PROJECT = transifex_api.Project.get(id=PROJECT_ID)
LANGUAGE = transifex_api.Language.get(id=LANGUAGE_ID)


def _slug_to_file_path(slug: str) -> Path:
    """Set of rules how to transform slug to translation file path"""
    file_path = RESOURCE_NAME_MAP.get(slug, slug)  # Legacy slug to file mapping
    file_path = file_path.replace('--', '/')
    if re.match(r'\d+_\d+', file_path):
        file_path = file_path.replace('_', '.')
    file_path = file_path + '.po'
    return Path(file_path)


def recreate_config() -> None:
    """Regenerate Transifex client config for all resources."""
    resources = transifex_api.Resource.filter(project=PROJECT).all()
    with open('.tx/config', 'w') as fo:
        fo.writelines(('[main]\n', 'host = https://api.transifex.com\n',))
        for resource in resources:
            path = _slug_to_file_path(resource.slug)
            fo.writelines((
                '\n',
                f'[{resource.id}]\n',
                f'file_filter = {path}\n',
                'type = PO\n',
                'source_lang = en\n',
                'minimum_perc = 0\n',
                f'trans.{LANG} = {path}\n',
                f'source_file = {path}\n',
                f'resource_name = {resource.name}\n'
            ))


def recreate_resource_stats() -> None:
    """Create resource stats."""
    stats = transifex_api.ResourceLanguageStats.filter(project=PROJECT, language=LANGUAGE).all()
    with open('RESOURCE.md', 'w') as fo:
        fo.writelines(('| Файл | Перекладено | Переглянуто | Вичитано |\n', '|:-----|:-----|:-----|:-----|\n'))
        for stat in stats:
            file_name = _slug_to_file_path(stat.id.split(':')[5])
            translated_pct = round(100 * stat.attributes['translated_words'] / stat.attributes['total_words'], 1)
            reviewed_pct = round(100 * stat.attributes['reviewed_words'] / stat.attributes['total_words'], 1)
            proofread_pct = round(100 * stat.attributes['proofread_words'] / stat.attributes['total_words'], 1)
            fo.writelines(f'| {file_name} | {translated_pct} % | {reviewed_pct} % | {proofread_pct} % |\n')


def recreate_team_stats() -> None:
    """Create contributor stats"""
    members = transifex_api.TeamMembership.filter(organization=ORGANISATION, language=LANGUAGE).all()

    users = {member.user.id: member.attributes['role'] for member in members}
    translators = dict.fromkeys(users.keys(), 0)
    reviewers = dict.fromkeys(users.keys(), 0)
    proofreaders = dict.fromkeys(users.keys(), 0)

    resources = transifex_api.Resource.filter(project=PROJECT).all()
    for resource in resources:
        translations = transifex_api.ResourceTranslation.filter(resource=resource, language=LANGUAGE).all()
        for translation in translations:
            if translation.relationships['translator']:
                translators[translation.relationships['translator']['data']['id']] += 1
            if translation.relationships['reviewer']:
                reviewers[translation.relationships['reviewer']['data']['id']] += 1
            if translation.relationships['proofreader']:
                proofreaders[translation.relationships['proofreader']['data']['id']] += 1

    with open('TEAM.md', 'w') as fo:
        fo.writelines(('| | Роль | Переклав | Переглянув | Вичитав |\n', '|:---|:---|:---|:---|:---|\n',))
        for user, role in users.items():
            fo.writelines(f"| {user} | {role} | {translators[user]} | {reviewers[user]} | {proofreaders[user]} |\n")


if __name__ == "__main__":
    RUNNABLE_SCRIPTS = ('recreate_config', 'recreate_resource_stats', 'recreate_team_stats')

    parser = ArgumentParser()
    parser.add_argument('cmd', nargs=1, choices=RUNNABLE_SCRIPTS)
    options = parser.parse_args()

    eval(options.cmd[0])()