"""
This script calculates translation coverage for various languages
"""

import os
import json
import sys


def calculate_coverage(filename):
    """
    Calculate translation coverage for a .po file
    """

    with open(filename, 'r') as f:
        lines = f.readlines()

    lines_count = 0
    lines_covered = 0
    lines_uncovered = 0

    for line in lines:

        if line.startswith("msgid "):
            lines_count += 1

        elif line.startswith("msgstr"):
            if line.startswith('msgstr ""') or line.startswith("msgstr ''"):
                lines_uncovered += 1
            else:
                lines_covered += 1

    # Return stats for the file
    return (lines_count, lines_covered, lines_uncovered)


if __name__ == '__main__':

    MY_DIR = os.path.dirname(os.path.realpath(__file__))
    LC_DIR = os.path.abspath(os.path.join(MY_DIR, '..', 'locale'))
    STAT_FILE = os.path.abspath(os.path.join(MY_DIR, '..', 'InvenTree/locale_stats.json'))

    locales = {}
    locales_perc = {}

    verbose = '-v' in sys.argv

    for locale in os.listdir(LC_DIR):
        path = os.path.join(LC_DIR, locale)
        if os.path.exists(path) and os.path.isdir(path):

            locale_file = os.path.join(path, 'LC_MESSAGES', 'django.po')

            if os.path.exists(locale_file) and os.path.isfile(locale_file):
                locales[locale] = locale_file

    if verbose:
        print("-" * 16)

    percentages = []

    for locale in locales.keys():
        locale_file = locales[locale]
        stats = calculate_coverage(locale_file)

        (total, covered, uncovered) = stats

        if total > 0:
            percentage = int(covered / total * 100)
        else:
            percentage = 0

        if verbose:
            print(f"| {locale.ljust(4, ' ')} : {str(percentage).rjust(4, ' ')}% |")

        locales_perc[locale] = percentage

        percentages.append(percentage)

    if verbose:
        print("-" * 16)

    # write locale stats
    with open(STAT_FILE, 'w') as target:
        json.dump(locales_perc, target)

    if len(percentages) > 0:
        avg = int(sum(percentages) / len(percentages))
    else:
        avg = 0

    print(f"InvenTree translation coverage: {avg}%")