40 lines
1.0 KiB
C
40 lines
1.0 KiB
C
|
#include <bshell/ast.h>
|
||
|
|
#include <bshell/parse/token.h>
|
||
|
|
|
||
|
|
static enum bshell_status collect_children(
|
||
|
|
struct bshell_ast_node *node,
|
||
|
|
struct bshell_ast_iterator *it)
|
||
|
|
{
|
||
|
|
struct bshell_func_ast_node *func = (struct bshell_func_ast_node *)node;
|
||
|
|
|
||
|
|
fx_queue_entry *cur = fx_queue_first(&func->n_params);
|
||
|
|
while (cur) {
|
||
|
|
struct bshell_ast_node *child = fx_unbox(
|
||
|
|
struct bshell_ast_node,
|
||
|
|
cur,
|
||
|
|
n_entry);
|
||
|
|
bshell_ast_iterator_enqueue(it, child);
|
||
|
|
cur = fx_queue_next(cur);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (func->n_body) {
|
||
|
|
bshell_ast_iterator_enqueue(it, func->n_body);
|
||
|
|
}
|
||
|
|
|
||
|
|
return BSHELL_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void to_string(const struct bshell_ast_node *node, fx_bstr *out)
|
||
|
|
{
|
||
|
|
const struct bshell_func_ast_node *func = (const struct
|
||
|
|
bshell_func_ast_node *)node;
|
||
|
|
fx_bstr_write_fmt(out, NULL, "%s", func->n_name->tok_str);
|
||
|
|
}
|
||
|
|
|
||
|
|
struct bshell_ast_node_definition func_bshell_ast_node = {
|
||
|
|
.def_id = BSHELL_AST_FUNC,
|
||
|
|
.def_node_size = sizeof(struct bshell_func_ast_node),
|
||
|
|
.def_collect_children = collect_children,
|
||
|
|
.def_to_string = to_string,
|
||
|
|
};
|