from __future__ import absolute_import, division, print_function

import sys

import libfoolang


def print_title(title):
    print(title)
    print('=' * len(title))


def def_img(node):
    assert isinstance(node, libfoolang.Def)
    return '<Def {} line {}>'.format(node.f_id.text,
                                     node.sloc_range.start.line)


ctx = libfoolang.AnalysisContext()
u = ctx.get_from_file('source.txt')
if u.diagnostics:
    for d in u.diagnostics:
        print(d)
    sys.exit(1)

u.populate_lexical_env()

print_title('Testing LexicalEnv (un)wrapping')

try:
    res_none = u.root.p_env_id(None)
except TypeError as exc:
    res_none = '<TypeError: {}>'.format(exc)
print('u.root.p_env_id(None) = {}'.format(res_none))

try:
    res_int = u.root.p_env_id(42)
except TypeError as exc:
    res_int = '<TypeError: {}>'.format(exc)
print('u.root.p_env_id(42) = {}'.format(res_int))

res_field = u.root.p_env_id(u.root.children_env)
print('u.root.p_env id(u.root.children_env): result has type {}'.format(
    type(res_field)
))

print_title('Testing LexicalEnv.get()')
o = u.root.find(lambda n: isinstance(n, libfoolang.Def) and n.f_id.text == 'o')
q = o.parent.parent
foo = q.parent.parent

nodes = [foo, q, o]
symbols = ['foo', 'p', 't', 'q', 'o', 'nosymbol']
symbol_width = max(len(sym) for sym in symbols)

for node in nodes:
    print('== From {} =='.format(def_img(node)))

    for sym in symbols:
        elts = node.children_env.get(sym)
        print('  {} : {}'.format(
            sym.ljust(symbol_width),
            (', '.join(def_img(elt.el) for elt in elts)
             if elts else
             '<none>')
        ))

print('Done.')