# XPath "function" elements that transform their subpaths

import util

import xpath

##### Functions

funcs = {}

def process(path):

    if path == []: return path

    path[1:] = process(path[1:])

    name = path[0].name

    if name.startswith('_') and name in funcs:

        return funcs[name](path.pop(0).attrs, path)

    else: return path

#### XPath functions

# Function names must start with _ to avoid collisions with real tags

# Functions take arguments (args, path)

for_repl_var = xpath.XpathElem('_val')

def _forEach(args, path):

    '''Replaces "_val" in `do` with each item in the `in_` list'''

    in_, do = args

    list_ = [in_[1:]] + in_[0].other_branches

    for_path = []

    for with_ in list_: for_path += util.list_replace(do, for_repl_var, with_)

    return for_path + path

funcs['_forEach'] = _forEach