# XPath parsing

import copy

from Parser import Parser

import xml_dom

class XpathElem:

    def __init__(self, name, value=None, is_attr=False):

        self.name = name

        self.value = value

        self.is_attr = is_attr

        self.is_ptr = False

        self.keys = []

        self.attrs = []

        self.other_branches = [] # temp implementation for split paths

    def __repr__(self):

        str_ = ''

        if self.is_attr: str_ += '@'

        str_ += self.name

        if self.keys != []: str_ += repr(self.keys)

        if self.attrs != []: str_ += ':'+repr(self.attrs)

        if self.is_ptr: str_ += '->'

        if self.other_branches != []: str_ += '{'+repr(self.other_branches)+'}'

        if self.value != None: str_ += '='+repr(self.value)

        return str_

    def __eq__(self, other): return self.__dict__ == other.__dict__

def value(path): return path[-1].value

def set_value(path, value): path[-1].value = value

def backward_id(elem):

    if len(elem.keys) >= 1 and value(elem.keys[0]) == None and\

    len(elem.keys[0]) == 1: return elem.keys[0]

    else: return None

def parse(str_):

    parser = Parser(str_)

    def _path():

        tree = []

        while True:

            # Split path

            if parser.str_('{'):

                tree[-1].other_branches = _paths()

                parser.str_('}', required=True)

                tree += tree[-1].other_branches.pop(0) # use first path for now

                break # nothing allowed after split path

            elem = XpathElem(is_attr=parser.str_('@'),

                name=parser.re(r'[\w.*]+', required=True))

            # Keys used to match nodes

            if parser.str_('['):

                elem.keys = _paths()

                parser.str_(']', required=True)

            # Attrs created when no matching node exists

            if parser.str_(':'):

                parser.str_('[', required=True)

                elem.attrs = _paths()

                parser.str_(']', required=True)

            elem.is_ptr = parser.str_('->')

            tree.append(elem)

            # Lookahead assertion

            if parser.str_('('):

                parser.str_('/', required=True) # next / is inside ()

                path = _path()

                parser.str_(')', required=True)

                elem.keys.append(path)

                tree += path

            if not parser.str_('/'): break

        if parser.str_('='):

            if parser.str_('"'):

                value = parser.re(r'[^"]*')

                parser.str_('"', required=True)

            else: value = parser.re(r'[\w.]*')

            set_value(tree, value)

        # Expand * abbrs

        for i in reversed(xrange(len(tree))):

            elem = tree[i]

            id_ = backward_id(elem)

            if id_ != None: elem = id_[0]; offset = -2

            elif elem.is_ptr: offset = 2

            else: offset = 1

            before, abbr, after = elem.name.partition('*')

            if abbr != '':

                try: elem.name = before+tree[i+offset].name+after

                except IndexError: pass # no replacement elem

        return tree

    def _paths():

        paths = []

        while True:

            paths.append(_path())

            if not parser.str_(','): break

        return paths

    parser.str_('/') # optional leading /

    path = _path()

    parser.end()

    return path

instance_level = 1

def obj(path):

    obj_path = copy.deepcopy(path[:instance_level+1])

    obj_path[-1].is_ptr = False # prevent pointer w/o target

    return obj_path

def set_id(path, id_, has_types=True):

    if has_types: id_level = instance_level

    else: id_level = 0

    path[id_level].keys.append([XpathElem('id', id_, True)])

def is_id(path): return path[0].is_attr and path[0].name == 'id'

def is_instance(elem): return elem.keys != [] and is_id(elem.keys[0]) 

def get(doc, xpath, create=False, last_only=None, parent=None):

    # Warning: The last_only optimization may put data that should be together

    # into separate nodes

    if parent == None: parent = doc.documentElement

    if last_only == None: last_only = create

    for elem_idx, elem in enumerate(xpath):

        # Find possible matches

        children = []

        if elem.is_attr:

            child = parent.getAttributeNode(elem.name)

            if child != None: children = [child]

        elif elem.name == '.': children = [parent]

        else:

            children = xml_dom.by_tag_name(parent, elem.name,

            last_only and (elem.keys == [] or is_instance(elem)))

        # Check each match

        node = None

        for child in children:

            is_match = elem.value == None or xml_dom.value(child) == elem.value

            for attr in elem.keys:

                if not is_match: break

                is_match = get(doc, attr, False, last_only, child) != None

            if is_match: node = child; break

        # Create node

        if node == None:

            if not create: return None

            if elem.is_attr:

                parent.setAttribute(elem.name, '')

                node = parent.getAttributeNode(elem.name)

            else: node = parent.appendChild(doc.createElement(elem.name))

            if elem.value != None: xml_dom.set_value(doc, node, elem.value)

            for attr in elem.keys + elem.attrs:

                get(doc, attr, create, last_only, node)

        for branch in elem.other_branches:

            branch = copy.deepcopy(branch)

            set_value(branch, value(xpath))

            get(doc, branch, create, last_only, node)

        # Follow pointer

        if elem.is_ptr:

            xpath = copy.deepcopy(xpath[elem_idx+1:]) # rest of XPath

            id_elem = backward_id(xpath[instance_level])

            if id_elem != None:

                # backward (child-to-parent) pointer with target ID attr

                set_value(id_elem, xml_dom.get_id(node))

            else: # forward (parent-to-child) pointer

                id_ = xml_dom.value(node)

                obj_xpath = obj(xpath) # target object

                if id_ == None or get(doc, obj_xpath, False, True) == None:

                    # no target or target keys don't match

                    if not create: return None

                    # Use last target object's ID + 1

                    obj_xpath[-1].keys = [] # just get by tag name

                    last = get(doc, obj_xpath, False, True)

                    if last != None: id_ = str(int(xml_dom.get_id(last)) + 1)

                    else: id_ = '0'

                    # Will append if target keys didn't match. Place ! in XPath

                    # after element to fork at to avoid this.

                    xml_dom.set_value(doc, node, id_)

                else: last_only = False

                set_id(xpath, id_)

            return get(doc, xpath, create, last_only)

        parent = node

    return parent

def put_obj(doc, xpath, id_, has_types, value=None):

    xpath = copy.deepcopy(xpath) # don't modify input!

    set_id(xpath, id_, has_types)

    if value != None: set_value(xpath, value)

    get(doc, xpath, True)