//! Module graph pretty printer using S-expression syntax.

use std::fmt;

use std::mem;

use std::io;

use std::lang::sexpr;

use std::lang::alloc;

/// Format a u32 and allocate the result in the arena.

fn formatId(a: *mut alloc::Arena, id: u32) -> *[u8] {

    let mut digits: [u8; 10] = undefined;

    let text = fmt::formatU32(id, &mut digits[..]);

    let ptr = try alloc::allocSlice(a, 1, 1, text.len) catch { return "?"; };

    let slice = ptr as *mut [u8];

    try mem::copy(slice, text) catch { return "?"; };

    return slice;

}

/// Convert a module state into an S-expression symbol.

fn stateToExpr(state: super::ModuleState) -> sexpr::Expr {

    match state {

        case super::ModuleState::Vacant => return sexpr::sym("vacant"),

        case super::ModuleState::Registered => return sexpr::sym("registered"),

        case super::ModuleState::Parsing => return sexpr::sym("parsing"),

        case super::ModuleState::Parsed => return sexpr::sym("parsed"),

        case super::ModuleState::Analyzing => return sexpr::sym("analyzing"),

        case super::ModuleState::Ready => return sexpr::sym("ready"),

    }

}

/// Recursively convert a module entry and its descendants to an S-expression.

fn subtreeToExpr(

    a: *mut alloc::Arena,

    graph: *super::ModuleGraph,

    entry: *super::ModuleEntry

) -> sexpr::Expr {

    let idText = formatId(a, entry.id as u32);

    let path = super::moduleQualifiedPath(entry);

    let mut pathBuf: *[sexpr::Expr] = &[];

    if path.len > 0 {

        let buf = try! sexpr::allocExprs(a, path.len);

        for seg, i in path { buf[i] = sexpr::sym(seg); }

        pathBuf = buf;

    }

    let mut childBuf: *[sexpr::Expr] = &[];

    if entry.childrenLen > 0 {

        let buf = try! sexpr::allocExprs(a, entry.childrenLen);

        for i in 0..entry.childrenLen {

            if let child = super::get(graph, entry.children[i]) {

                buf[i] = subtreeToExpr(a, graph, child);

            }

        }

        childBuf = buf;

    }

    return sexpr::block(a, "module", &[

        sexpr::sym(entry.name),

        sexpr::list(a, "id", &[sexpr::sym(idText)]),

        sexpr::list(a, "state", &[stateToExpr(entry.state)]),

        sexpr::Expr::Str(entry.filePath),

        sexpr::Expr::List { head: "::", tail: pathBuf, multiline: false }

    ], childBuf);

}

/// Print the entire module graph in S-expression format.

pub fn printGraph(graph: *super::ModuleGraph, arena: *mut alloc::Arena) {

    // Print all root modules.

    for i in 0..graph.entriesLen {

        let entry = &graph.entries[i];

        if entry.parent == nil {

            sexpr::print(subtreeToExpr(arena, graph, entry), 0);

            io::print("\n");

        }

    }

}