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

"""Anki card writer from gaphrase format"""

import re

import hashlib

import os

import io

import sys

import tempfile

import shutil

import anki

from anki.exporting import AnkiPackageExporter

FINAME = None

FONAME = None

FDELNAME = None

ARG_NAME_RE = re.compile("-name=(.+)")

ARG_DELFILE_RE = re.compile("-delfile=(.+)")

look_for_files = False

for idx in range(1, len(sys.argv)):

    arg = sys.argv[idx]

    if arg == "--":

        look_for_files = True

        continue

    if not look_for_files:

        m = ARG_NAME_RE.match(arg)

        if m:

            NAME = m.group(1)

            continue

        m = ARG_DELFILE_RE.match(arg)

        if m:

            FDELNAME = m.group(1)

            continue

        if arg.startswith("-"):

            raise Exception("Unsupported option format: '{:s}'".format(arg))

    if not FINAME:

        FINAME = arg

        continue

    if not FONAME:

        FONAME = arg

        continue

    raise Exception("Unnecessary argument: '{:s}'".format(arg))

if not FINAME:

    raise Exception("Input file name is not passed...")

if FONAME is None:

    raise Exception("Output file name is not passed...")

if not NAME:

    NAME, _ = os.path.splitext(os.path.basename(FINAME))

# if FDELNAME:

#     FDEL = io.open(FDELNAME, mode='r', buffering=1, encoding="utf-8")

# else:

#     FDEL = None

################################################################

class Parser:

    COMMENT_RE = re.compile("^; ")

    NUM_RE = re.compile(u"^# ([1-9][0-9]*)$")

    PHRASE_START_RE = re.compile(u"^- (.*)")

    def __init__(self):

        pass

    def readline(self):

        while True:

            self.line = self.stream.readline()

            self.eof = len(self.line) == 0

            if self.eof:

                break

            self.lineno += 1

            if self.COMMENT_RE.search(self.line):

                continue

            self.line = self.line.strip(' \n\t')

            if len(self.line) > 0:

                break

    def parse(self, stream):

        self.lineno = 0

        self.stream = stream

        self.dom = dict()

        self.eof = False

        try:

            self.parse_prelude()

            while not self.eof:

                self.parse_article()

        except ParseException as ex:

            if sys.version_info.major == 2:

                import traceback

                traceback.print_exc()

            raise ParseException(ex.msg, self.lineno, self.line)

        return self.dom

    def parse_prelude(self):

        while True:

            self.readline()

            if self.eof:

                return

            m = self.NUM_RE.match(self.line)

            if m:

                self.num = m.group(1)

                break

    def parse_article(self):

        """Assume we are at ``# NUM`` line."""

        num = self.num

        phrase_buf = []

        phrases = []

        while True:

            self.readline()

            if self.eof:

                if len(phrase_buf) > 0:

                    phrases.append(" ".join(phrase_buf))

                break

            m = self.NUM_RE.match(self.line)

            if m:

                if len(phrase_buf) > 0:

                    phrases.append(" ".join(phrase_buf))

                self.num = m.group(1)

                break

            m = self.PHRASE_START_RE.match(self.line)

            if m:

                if len(phrase_buf) > 0:

                    phrases.append(" ".join(phrase_buf))

                phrase_buf = [m.group(1)]

            else:

                phrase_buf.append(self.line)

        if len(phrases) == 0:

            raise ParseException("""There are no any phrases...""")

        self.dom[num] = phrases

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

PARSER = Parser()

try:

    DOM = PARSER.parse(FIN)

finally:

    FIN.close()

################################################################

MODEL_CSS = """

.card {

  font-family: arial;

  font-size: 20px;

  text-align: left;

  color: black;

  background-color: white;

}

.line {

  margin-bottom: 0.5em;

}

.odd {

  color: #004000;

}

.even {

  color: #000080;

}

"""

class AnkiDbBuilder:

    def __init__(self, tmpdir, name):

        self.tmpdir = tmpdir

        self.name = name

        self.collection = collection = anki.Collection(os.path.join(self.tmpdir, 'collection.anki2'))

        deck_id = collection.decks.id(self.name)

        # It is essential to keep model['id'] unchanged between upgrades!!

        model_id = int(hashlib.sha1(self.name.encode('utf-8')).hexdigest(), 16) % (2**63)

        ################################################################

        # One face card model.

        model = collection.models.new(self.name + "_front")

        model['did'] = deck_id

        model['css'] = MODEL_CSS

        collection.models.addField(model, collection.models.newField('Front'))

        tmpl = collection.models.newTemplate('Front')

        tmpl['qfmt'] = '<div class="front">{{Front}}</div>'

        tmpl['afmt'] = '{{FrontSide}}'

        collection.models.addTemplate(model, tmpl)

        model['id'] = model_id

        collection.models.update(model)

        collection.models.save(model)

        self.model = model

    def guid(self, num):

        h = hashlib.md5(":".join((self.name, str(num))).encode('utf-8'))

        return h.hexdigest()

    def add_note(self, num, front):

        note = anki.notes.Note(self.collection, self.model)

        note['Front'] = front

        note.guid = self.guid(num)

        self.collection.addNote(note)

    def export(self, fname):

        export = AnkiPackageExporter(self.collection)

        export.exportInto(fname)

    def close(self):

        self.collection.close()

def write_lines(buf, lines):

    odd = True

    for line in lines:

        if odd:

            buf.append("<div class='line odd'>")

        else:

            buf.append("<div class='line even'>")

        buf.append("- ")

        buf.append(line)

        buf.append("</div>")

        odd = not odd

# Looks like anki libs change working directory to media directory of current deck

# Therefore absolute path should be stored before creating temporary deck

FONAME = os.path.abspath(FONAME)

TMPDIR = tempfile.mkdtemp(dir=os.path.dirname(FONAME))

try:

    BUILDER = AnkiDbBuilder(TMPDIR, NAME)

    for num, lines in DOM.items():

        buf = []

        write_lines(buf, lines)

        front = "".join(buf)

        BUILDER.add_note(num, front)

    BUILDER.export(FONAME)

finally:

    BUILDER.close()

    shutil.rmtree(TMPDIR, ignore_errors=True)