import json
import os
import shutil
from collections import OrderedDict

import regex as re
from orderedset import OrderedSet

from dateparser_scripts.order_languages import language_locale_dict
from dateparser_scripts.utils import get_dict_difference, get_raw_data

APOSTROPHE_LOOK_ALIKE_CHARS = [
    '\N{RIGHT SINGLE QUOTATION MARK}',     # '\u2019'
    '\N{MODIFIER LETTER APOSTROPHE}',      # '\u02bc'
    '\N{MODIFIER LETTER TURNED COMMA}',    # '\u02bb'
    '\N{ARMENIAN APOSTROPHE}',             # '\u055a'
    '\N{LATIN SMALL LETTER SALTILLO}',     # '\ua78c'
    '\N{PRIME}',                           # '\u2032'
    '\N{REVERSED PRIME}',                  # '\u2035'
    '\N{MODIFIER LETTER PRIME}',           # '\u02b9'
    '\N{FULLWIDTH APOSTROPHE}',            # '\uff07'
]

DATE_ORDER_PATTERN = re.compile('([DMY])+\u200f*[-/. \t]*([DMY])+\u200f*[-/. \t]*([DMY])+')
RELATIVE_PATTERN = re.compile(r'(?<![\+\-]\s*)\{0\}')
DEFAULT_MONTH_PATTERN = re.compile(r'^M?\d+$', re.U)
RE_SANITIZE_APOSTROPHE = re.compile('|'.join(APOSTROPHE_LOOK_ALIKE_CHARS))
AM_PATTERN = re.compile(r'^\s*[Aa]\s*\.?\s*[Mm]\s*\.?\s*$')
PM_PATTERN = re.compile(r'^\s*[Pp]\s*\.?\s*[Mm]\s*\.?\s*$')
PARENTHESIS_PATTERN = re.compile(r'[\(\)]')

os.chdir(os.path.dirname(os.path.abspath(__file__)))
get_raw_data()

cldr_dates_full_dir = "../raw_data/cldr_dates_full/main/"


def _filter_relative_string(relative_string):
    return (isinstance(relative_string, str) and
            RELATIVE_PATTERN.search(relative_string) and
            not PARENTHESIS_PATTERN.search(relative_string))


def _filter_month_name(month_name):
    return not DEFAULT_MONTH_PATTERN.match(month_name)


def _retrieve_locale_data(locale):
    ca_gregorian_file = cldr_dates_full_dir + locale + '/ca-gregorian.json'
    dateFields_file = cldr_dates_full_dir + locale + '/dateFields.json'
    with open(ca_gregorian_file) as f:
        cldr_gregorian_data = json.load(f)

    with open(dateFields_file) as g:
        cldr_datefields_data = json.load(g)

    gregorian_dict = cldr_gregorian_data["main"][locale]["dates"]["calendars"]["gregorian"]
    date_fields_dict = cldr_datefields_data["main"][locale]["dates"]["fields"]

    json_dict = OrderedDict()

    field_keys_1 = ["stand-alone", "format"]
    field_keys_2 = ["wide", "abbreviated"]    # neglecting "narrow" to avoid problems in translation
    year_keys = ["year", "year-short", "year-narrow"]
    month_keys = ["month", "month-short", "month-narrow"]
    week_keys = ["week", "week-short", "week-narrow"]
    day_keys = ["day", "day-short", "day-narrow"]
    hour_keys = ["hour", "hour-short", "hour-narrow"]
    minute_keys = ["minute", "minute-short", "minute-narrow"]
    second_keys = ["second", "second-short", "second-narrow"]
    relative_keys = ["relativeTimePattern-count-one", "relativeTimePattern-count-other"]

    json_dict["name"] = locale

    try:
        date_format_string = gregorian_dict["dateFormats"]["short"].upper()
    except:
        date_format_string = gregorian_dict["dateFormats"]["short"]["_value"].upper()

    json_dict["date_order"] = DATE_ORDER_PATTERN.sub(
            r'\1\2\3', DATE_ORDER_PATTERN.search(date_format_string).group())

    json_dict["january"] = list(filter(_filter_month_name,
                                       [gregorian_dict["months"][key1][key2]["1"]
                                        for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["february"] = list(filter(_filter_month_name,
                                        [gregorian_dict["months"][key1][key2]["2"]
                                         for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["march"] = list(filter(_filter_month_name,
                                     [gregorian_dict["months"][key1][key2]["3"]
                                      for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["april"] = list(filter(_filter_month_name,
                                     [gregorian_dict["months"][key1][key2]["4"]
                                      for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["may"] = list(filter(_filter_month_name,
                                   [gregorian_dict["months"][key1][key2]["5"]
                                    for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["june"] = list(filter(_filter_month_name,
                                    [gregorian_dict["months"][key1][key2]["6"]
                                     for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["july"] = list(filter(_filter_month_name,
                                    [gregorian_dict["months"][key1][key2]["7"]
                                     for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["august"] = list(filter(_filter_month_name,
                                      [gregorian_dict["months"][key1][key2]["8"]
                                       for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["september"] = list(filter(_filter_month_name,
                                         [gregorian_dict["months"][key1][key2]["9"]
                                          for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["october"] = list(filter(_filter_month_name,
                                       [gregorian_dict["months"][key1][key2]["10"]
                                        for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["november"] = list(filter(_filter_month_name,
                                        [gregorian_dict["months"][key1][key2]["11"]
                                         for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["december"] = list(filter(_filter_month_name,
                                        [gregorian_dict["months"][key1][key2]["12"]
                                         for key1 in field_keys_1 for key2 in field_keys_2]))

    json_dict["monday"] = [gregorian_dict["days"][key1][key2]["mon"]
                           for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["tuesday"] = [gregorian_dict["days"][key1][key2]["tue"]
                            for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["wednesday"] = [gregorian_dict["days"][key1][key2]["wed"]
                              for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["thursday"] = [gregorian_dict["days"][key1][key2]["thu"]
                             for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["friday"] = [gregorian_dict["days"][key1][key2]["fri"]
                           for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["saturday"] = [gregorian_dict["days"][key1][key2]["sat"]
                             for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["sunday"] = [gregorian_dict["days"][key1][key2]["sun"]
                           for key1 in field_keys_1 for key2 in field_keys_2]

    json_dict["am"] = [AM_PATTERN.sub("am", x) for x in
                       [gregorian_dict["dayPeriods"][key1][key2]["am"]
                        for key1 in field_keys_1 for key2 in field_keys_2]]

    json_dict["pm"] = [PM_PATTERN.sub("pm", x) for x in
                       [gregorian_dict["dayPeriods"][key1][key2]["pm"]
                        for key1 in field_keys_1 for key2 in field_keys_2]]

    json_dict["year"] = [date_fields_dict[key]["displayName"] for key in year_keys]

    json_dict["month"] = [date_fields_dict[key]["displayName"] for key in month_keys]

    json_dict["week"] = [date_fields_dict[key]["displayName"] for key in week_keys]

    json_dict["day"] = [date_fields_dict[key]["displayName"] for key in day_keys]

    json_dict["hour"] = [date_fields_dict[key]["displayName"] for key in hour_keys]

    json_dict["minute"] = [date_fields_dict[key]["displayName"] for key in minute_keys]

    json_dict["second"] = [date_fields_dict[key]["displayName"] for key in second_keys]

    json_dict["relative-type"] = OrderedDict()

    json_dict["relative-type"]["1 year ago"] = [date_fields_dict[key]["relative-type--1"]
                                                for key in year_keys]

    json_dict["relative-type"]["0 year ago"] = [date_fields_dict[key]["relative-type-0"]
                                                for key in year_keys]

    json_dict["relative-type"]["in 1 year"] = [date_fields_dict[key]["relative-type-1"]
                                               for key in year_keys]

    json_dict["relative-type"]["1 month ago"] = [date_fields_dict[key]["relative-type--1"]
                                                 for key in month_keys]

    json_dict["relative-type"]["0 month ago"] = [date_fields_dict[key]["relative-type-0"]
                                                 for key in month_keys]

    json_dict["relative-type"]["in 1 month"] = [date_fields_dict[key]["relative-type-1"]
                                                for key in month_keys]

    json_dict["relative-type"]["1 week ago"] = [date_fields_dict[key]["relative-type--1"]
                                                for key in week_keys]

    json_dict["relative-type"]["0 week ago"] = [date_fields_dict[key]["relative-type-0"]
                                                for key in week_keys]

    json_dict["relative-type"]["in 1 week"] = [date_fields_dict[key]["relative-type-1"]
                                               for key in week_keys]

    json_dict["relative-type"]["1 day ago"] = [date_fields_dict[key]["relative-type--1"]
                                               for key in day_keys]

    json_dict["relative-type"]["0 day ago"] = [date_fields_dict[key]["relative-type-0"]
                                               for key in day_keys]

    json_dict["relative-type"]["in 1 day"] = [date_fields_dict[key]["relative-type-1"]
                                              for key in day_keys]

    json_dict["relative-type"]["0 hour ago"] = [date_fields_dict[key]["relative-type-0"]
                                                for key in hour_keys]

    json_dict["relative-type"]["0 minute ago"] = [date_fields_dict[key]["relative-type-0"]
                                                  for key in minute_keys]

    json_dict["relative-type"]["0 second ago"] = [date_fields_dict[key]["relative-type-0"]
                                                  for key in second_keys]

    json_dict["relative-type-regex"] = OrderedDict()

    json_dict["relative-type-regex"]["in \\1 year"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in year_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 year ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in year_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["in \\1 month"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in month_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 month ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in month_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["in \\1 week"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in week_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 week ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in week_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["in \\1 day"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in day_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 day ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in day_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["in \\1 hour"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in hour_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 hour ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in hour_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["in \\1 minute"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in minute_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 minute ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in minute_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["in \\1 second"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-future"].get(key2)
                     for key1 in second_keys for key2 in relative_keys])))

    json_dict["relative-type-regex"]["\\1 second ago"] = (
        list(filter(_filter_relative_string,
                    [date_fields_dict[key1]["relativeTime-type-past"].get(key2)
                     for key1 in second_keys for key2 in relative_keys])))

    return json_dict


def _clean_string(given_string):
    given_string = RE_SANITIZE_APOSTROPHE.sub("'", given_string)
    given_string = given_string.replace('.', '')
    given_string = given_string.lower()
    return ' '.join(given_string.split())


def _clean_dict(json_dict):
    for key, value in json_dict.items():
        if isinstance(value, list):
            json_dict[key] = list(OrderedSet(map(_clean_string, value)))
        elif isinstance(value, dict):
            json_dict[key] = _clean_dict(value)
    return OrderedDict(filter(lambda x: x[1], json_dict.items()))


def main():
    parent_directory = "../dateparser_data/cldr_language_data"
    directory = "../dateparser_data/cldr_language_data/date_translation_data/"
    if not os.path.isdir(parent_directory):
        os.mkdir(parent_directory)
    if os.path.isdir(directory):
        shutil.rmtree(directory)
    os.mkdir(directory)

    for language in language_locale_dict:
        json_language_dict = _clean_dict(_retrieve_locale_data(language))
        locale_specific_dict = OrderedDict()
        locales_list = language_locale_dict[language]
        for locale in locales_list:
            json_locale_dict = _clean_dict(_retrieve_locale_data(locale))
            locale_specific_dict[locale] = get_dict_difference(json_language_dict, json_locale_dict)
        json_language_dict["locale_specific"] = locale_specific_dict
        filename = directory + language + ".json"
        print("writing " + filename)
        json_string = json.dumps(json_language_dict, indent=4, separators=(',', ': '),
                                 ensure_ascii=False).encode('utf-8')
        with open(filename, 'wb') as f:
            f.write(json_string)


if __name__ == '__main__':
    main()
