# -*- coding: utf-8 -*-

"""Space repetition TAB format writer"""

import io

import sys

import codecs

import gadict

FINAME = None

FONAME = None

if len(sys.argv) >= 2:

    FINAME = sys.argv[1]

if len(sys.argv) >= 3:

    FONAME = sys.argv[2]

LANGS = None

if len(sys.argv) >= 4:

    LANGS = set(sys.argv[3].split(","))

FIN = io.open(FINAME, mode='r', buffering=1, encoding="utf-8")

PARSER = gadict.Parser()

try:

    DOM = PARSER.parse(FIN)

except gadict.ParseException as ex:

    sys.stdout.write("{:s}{:s}\n".format(FINAME, repr(ex)))

    if __debug__:

        import traceback

        traceback.print_exc()

    exit(1)

finally:

    FIN.close()

if FONAME is None:

    FOUT = sys.stdout

else:

    FOUT = codecs.open(FONAME, "w", "utf-8")

for (headwords, translations) in DOM[1:]:

    lines = []

    for hw in headwords:

        line = "<b>"+hw.headword+"</b>"

        if hw.pron:

            line += " ["+hw.pron+"]"

        if len(hw.attrs) > 0:

            attrs = [" «"+x+"»" for x in hw.attrs]

            attrs.sort()

            line += ",".join(attrs)

        lines.append(line)

    question = "<br>".join(lines)

    FOUT.write(question)

    FOUT.write("\t")

    for sense in translations:

        if not sense:

            raise Exception("""Empty sense for article: """ + article[0].__iter__().__next__())

        if sense.pos:

            FOUT.write('<i style="color: green;">')

            FOUT.write(sense.pos)

            FOUT.write('</i>')

            if sense.ant_list and len(sense.ant_list) > 0:

                FOUT.write(" <i>ant: ")

                FOUT.write("; ".join(sense.ant_list))

                FOUT.write("</i>")

            if sense.syn_list and len(sense.syn_list) > 0:

                FOUT.write(" <i>syn: ")

                FOUT.write("; ".join(sense.syn_list))

                FOUT.write("</i>")

            if LANGS and len(LANGS) > 1:

                FOUT.write("<br>")

            else:

                FOUT.write(" ")

        for (lang, tr) in sense.tr_list:

            tr = tr.replace('\n', ' ')

            if LANGS is None:

                FOUT.write(tr)

            elif lang in LANGS:

                if len(LANGS) == 1:

                    FOUT.write(tr)

                else:

                    FOUT.write('<i style="color: blue;">')

                    FOUT.write(lang)

                    FOUT.write("</i> ")

                    FOUT.write(tr)

            FOUT.write("<br>")

    FOUT.write("\n")