from parsimonious.grammar import Grammar

from .. import document_matchers, documents


def read_document_matcher(string):
    root_node = _grammar.parse(string)
    return read_document_matcher_node(root_node)


def read_document_matcher_node(root_node):
    element_node = root_node.children[0]
    if element_node.expr_name == "paragraph":
        return _read_paragraph_node(element_node)
    elif element_node.expr_name == "run":
        return _read_run_node(element_node)
    

def _read_paragraph_node(paragraph_node):
    style_id = _read_style_node(paragraph_node.children[1])
    numbering = _read_list_node(paragraph_node.children[2])
    return document_matchers.paragraph(style_id=style_id, numbering=numbering)
    

def _read_run_node(run_node):
    style_id = _read_style_node(run_node.children[1])
    return document_matchers.run(style_id=style_id)


def _read_style_node(style_node):
    if style_node.children:
        return style_node.children[0].children[1].text
    else:
        return None


def _read_list_node(list_node):
    if list_node.children:
        return documents.numbering_level(
            int(list_node.children[0].children[3].text) - 1,
            is_ordered=list_node.children[0].children[1].text == "ordered-list",
        )
    else:
        return None

grammar_text = r"""
document_matcher = paragraph / run

paragraph = "p" style_id? list?

run = "r" style_id?

style_id = "." style_identifier

style_identifier = ~"[A-Z0-9]*"i

list = ":" list_type "(" ~"[0-9]+" ")"

list_type = "ordered-list" / "unordered-list"
"""
_grammar = Grammar(grammar_text)