src/printer/tree.c

/**
 * @file printer/tree.c
 * @author Radek Krejci <rkrejci@cesnet.cz>
 * @brief TREE printer for libyang data model structure
 *
 * Copyright (c) 2015 CESNET, z.s.p.o.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 3. Neither the name of the Company nor the names of its contributors
 *    may be used to endorse or promote products derived from this
 *    software without specific prior written permission.
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>

#include "../common.h"
#include "../tree.h"

/* spec_config = 0 (no special config status), 1 (read-only - rpc output, notification), 2 (write-only - rpc input) */
static void tree_print_mnode_choice(FILE *f, struct ly_module* module, int level, char *indent, unsigned int max_name_len,
                                    struct ly_mnode *mnode, int mask, int spec_config, struct ly_submodule *main_submod);
static void tree_print_mnode(FILE *f, struct ly_module *module, int level, char *indent, unsigned int max_name_len, struct ly_mnode *mnode,
                             int mask, int spec_config, struct ly_submodule *main_submod);

static int
is_enabled(struct ly_mnode *mnode)
{
    int i;

    for (i = 0; i < mnode->features_size; i++) {
        if (!(mnode->features[i]->flags & LYS_FENABLED)) {
            return 0;
        }
    }

    if (mnode->parent && (mnode->parent->nodetype == LY_NODE_AUGMENT)) {
        for (i = 0; i < mnode->parent->features_size; i++) {
            if (!(mnode->parent->features[i]->flags & LYS_FENABLED)) {
                return 0;
            }
        }
    }

    return 1;
}

static int
sibling_is_valid_child(const struct ly_mnode *mnode, int including)
{
    struct ly_mnode *cur;

    if (mnode == NULL) {
        return 0;
    }

    /* has a following printed child */
    LY_TREE_FOR((struct ly_mnode *)(including ? mnode : mnode->next), cur) {
        if (is_enabled(cur) && (cur->nodetype &
                (LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML | LY_NODE_CHOICE |
                 LY_NODE_RPC | LY_NODE_INPUT | LY_NODE_OUTPUT | LY_NODE_NOTIF | LY_NODE_CASE))) {
            return 1;
        }
    }

    /* if in uses, the following printed child can actually be in the parent node :-/ */
    if (mnode->parent && mnode->parent->nodetype == LY_NODE_USES) {
        return sibling_is_valid_child(mnode->parent, 0);
    }

    return 0;
}

static char *
create_indent(int level, const char *old_indent, const struct ly_mnode *mnode, int shorthand, struct ly_submodule *main_submod)
{
    int next_is_case = 0, is_case = 0, has_next = 0, i, found;
    char *new_indent = malloc((level * 4 + 1) * sizeof (char));

    strcpy(new_indent, old_indent);

    /* this is the indent of a case (standard or shorthand) */
    if ((mnode->nodetype == LY_NODE_CASE) || shorthand) {
        is_case = 1;
    }

    /* this is the direct child of a case */
    if (!is_case && mnode->parent && (mnode->parent->nodetype & (LY_NODE_CASE | LY_NODE_CHOICE))) {
        /* it is not the only child */
        if (mnode->next && mnode->next->parent && (mnode->next->parent->nodetype == LY_NODE_CHOICE)) {
            next_is_case = 1;
        }
    }

    /* next is a node that will actually be printed */
    has_next = sibling_is_valid_child(mnode, 0);

    /* there is no next, but we are in top-level of a submodule */
    if (!has_next && (mnode->module->type == 1) && !mnode->parent) {
        struct ly_submodule *submod = (struct ly_submodule *)mnode->module;
        struct ly_module *mod = submod->belongsto;

        /* a special case when we check the includes of a submodule */
        if (main_submod) {
            if (submod != main_submod) {
                found = 0;
                for (i = 0; i < main_submod->inc_size; i++) {
                    if (found) {
                        if (mnode->nodetype == LY_NODE_RPC) {
                            has_next = sibling_is_valid_child(main_submod->inc[i].submodule->rpc, 1);
                        } else if (mnode->nodetype == LY_NODE_NOTIF) {
                            has_next = sibling_is_valid_child(main_submod->inc[i].submodule->notif, 1);
                        } else {
                            has_next = sibling_is_valid_child(main_submod->inc[i].submodule->data, 1);
                        }
                        if (has_next) {
                            break;
                        }
                    }
                    if (!found && (submod == main_submod->inc[i].submodule)) {
                        found = 1;
                    }
                }

                if (!has_next) {
                    if (mnode->nodetype == LY_NODE_RPC) {
                        has_next = sibling_is_valid_child(main_submod->rpc, 1);
                    } else if (mnode->nodetype == LY_NODE_NOTIF) {
                        has_next = sibling_is_valid_child(main_submod->notif, 1);
                    } else {
                        has_next = sibling_is_valid_child(main_submod->data, 1);
                    }
                }
            }

            goto strcat_indent;
        }

        /* find this submodule, check all the next ones for valid printed nodes */
        found = 0;
        for (i = 0; i < mod->inc_size; i++) {
            /* we found ours, check all the following submodules and the module */
            if (found) {
                if (mnode->nodetype == LY_NODE_RPC) {
                    has_next = sibling_is_valid_child(mod->inc[i].submodule->rpc, 1);
                } else if (mnode->nodetype == LY_NODE_NOTIF) {
                    has_next = sibling_is_valid_child(mod->inc[i].submodule->notif, 1);
                } else {
                    has_next = sibling_is_valid_child(mod->inc[i].submodule->data, 1);
                }
                if (has_next) {
                    break;
                }
            }
            if (!found && (submod == mod->inc[i].submodule)) {
                found = 1;
            }
        }

        /* there is nothing in submodules, check module */
        if (!has_next) {
            if (mnode->nodetype == LY_NODE_RPC) {
                has_next = sibling_is_valid_child(mod->rpc, 1);
            } else if (mnode->nodetype == LY_NODE_NOTIF) {
                has_next = sibling_is_valid_child(mod->notif, 1);
            } else {
                has_next = sibling_is_valid_child(mod->data, 1);
            }
        }
    }

strcat_indent:
    if (has_next && !next_is_case) {
        strcat(new_indent, "|  ");
    } else {
        strcat(new_indent, "   ");
    }

    return new_indent;
}

static unsigned int
get_max_name_len(struct ly_module *module, struct ly_mnode *mnode)
{
    struct ly_mnode *sub;
    unsigned int max_name_len = 0, uses_max_name_len, name_len;

    LY_TREE_FOR(mnode, sub) {
        if (sub->nodetype == LY_NODE_USES) {
            uses_max_name_len = get_max_name_len(module, sub->child);
            if (uses_max_name_len > max_name_len) {
                max_name_len = uses_max_name_len;
            }
        } else if (sub->nodetype &
                (LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST
                | LY_NODE_ANYXML | LY_NODE_CASE)) {
            name_len = strlen(sub->name) + (module == sub->module ? 0 : strlen(sub->module->prefix)+1);
            if (name_len > max_name_len) {
                max_name_len = name_len;
            }
        }
    }

    return max_name_len;
}

static void
tree_print_type(FILE *f, struct lys_type *type)
{
    if (type->base == LY_TYPE_LEAFREF) {
        fprintf(f, "-> %s", type->info.lref.path);
    } else if (type->prefix) {
        fprintf(f, "%s:%s", type->prefix, type->der->name);
    } else {
        fprintf(f, "%s", type->der->name);
    }
}

static void
tree_print_features(FILE *f, const struct ly_feature **features, uint8_t features_size)
{
    int i;

    if (!features_size) {
        return;
    }

    fprintf(f, " {");
    for (i = 0; i < features_size; i++) {
        if (i > 0) {
            fprintf(f, ",");
        }
        fprintf(f, "%s", features[i]->name);
    }
    fprintf(f, "}?");
}

static void
tree_print_input_output(FILE *f, struct ly_module *module, int level, char *indent, struct ly_mnode *mnode, int spec_config, struct ly_submodule *main_submod)
{
    unsigned int max_child_len;
    char *new_indent;
    struct ly_mnode *sub;

    assert(spec_config);

    fprintf(f, "%s+--%s %s\n", indent, (spec_config == 1 ? "-w" : "ro"), (spec_config == 1 ? "input" : "output"));

    level++;
    new_indent = create_indent(level, indent, mnode, 0, main_submod);

    max_child_len = get_max_name_len(module, mnode->child);

    LY_TREE_FOR(mnode->child, sub) {
        tree_print_mnode(f, module, level, new_indent, max_child_len, sub,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML | LY_NODE_USES,
                         spec_config, main_submod);
    }

    free(new_indent);
}

static void
tree_print_container(FILE *f, struct ly_module *module, int level, char *indent, struct ly_mnode *mnode, int spec_config, struct ly_submodule *main_submod)
{
    unsigned int max_child_len;
    char *new_indent;
    struct ly_mnode_container *cont = (struct ly_mnode_container *)mnode;
    struct ly_mnode *sub;

    assert(spec_config >= 0 && spec_config <= 2);

    fprintf(f, "%s%s--", indent,
            (cont->flags & LYS_STATUS_DEPRC ? "x" : (cont->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (spec_config == 0) {
        fprintf(f, "%s ", (cont->flags & LYS_CONFIG_W ? "rw" : "ro"));
    } else if (spec_config == 1) {
        fprintf(f, "-w ");
    } else if (spec_config == 2) {
        fprintf(f, "ro ");
    }

    if (module != cont->module) {
        fprintf(f, "%s:", cont->module->prefix);
    }

    fprintf(f, "%s%s", cont->name, (cont->presence ? "!" : ""));

    tree_print_features(f, (const struct ly_feature **)cont->features, cont->features_size);

    fprintf(f, "\n");

    level++;
    new_indent = create_indent(level, indent, mnode, 0, main_submod);

    max_child_len = get_max_name_len(module, mnode->child);

    LY_TREE_FOR(cont->child, sub) {
        tree_print_mnode(f, module, level, new_indent, max_child_len, sub,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML | LY_NODE_USES,
                         spec_config, main_submod);
    }

    free(new_indent);
}

static void
tree_print_choice(FILE *f, struct ly_module *module, int level, char *indent, struct ly_mnode *mnode, int spec_config, struct ly_submodule *main_submod)
{
    unsigned int max_child_len;
    char *new_indent;
    struct ly_mnode_choice *choice = (struct ly_mnode_choice *)mnode;
    struct ly_mnode *sub;

    assert(spec_config >= 0 && spec_config <= 2);

    fprintf(f, "%s%s--", indent,
            (choice->flags & LYS_STATUS_DEPRC ? "x" : (choice->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (spec_config == 0) {
        fprintf(f, "%s ", (choice->flags & LYS_CONFIG_W ? "rw" : "ro"));
    } else if (spec_config == 1) {
        fprintf(f, "-w ");
    } else if (spec_config == 2) {
        fprintf(f, "ro ");
    }

    fprintf(f, "(");

    if (module != choice->module) {
        fprintf(f, "%s:", choice->module->prefix);
    }

    fprintf(f, "%s)%s", choice->name, (choice->flags & LYS_MAND_TRUE ? "" : "?"));

    if (choice->dflt != NULL) {
        fprintf(f, " <%s>", choice->dflt->name);
    }

    tree_print_features(f, (const struct ly_feature **)choice->features, choice->features_size);

    fprintf(f, "\n");

    level++;
    new_indent = create_indent(level, indent, mnode, 0, main_submod);

    max_child_len = get_max_name_len(module, mnode->child);

    LY_TREE_FOR(choice->child, sub) {
        tree_print_mnode_choice(f, module, level, new_indent, max_child_len, sub,
                                LY_NODE_CASE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML,
                                spec_config, main_submod);
    }

    free(new_indent);
}

static void
tree_print_case(FILE *f, struct ly_module *module, int level, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int shorthand, int spec_config, struct ly_submodule *main_submod)
{
    char *new_indent;
    struct ly_mnode_case *cas = (struct ly_mnode_case *)mnode;
    struct ly_mnode *sub;

    fprintf(f, "%s%s--:(", indent,
            (cas->flags & LYS_STATUS_DEPRC ? "x" : (cas->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (module != cas->module) {
        fprintf(f, "%s:", cas->module->prefix);
    }

    fprintf(f, "%s)", cas->name);

    tree_print_features(f, (const struct ly_feature **)cas->features, cas->features_size);

    fprintf(f, "\n");

    level++;
    new_indent = create_indent(level, indent, mnode, shorthand, main_submod);

    if (shorthand) {
        tree_print_mnode(f, module, level, new_indent, max_name_len, mnode,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML | LY_NODE_USES,
                         spec_config, main_submod);
    } else {
        LY_TREE_FOR(mnode->child, sub) {
            tree_print_mnode(f, module, level, new_indent, max_name_len, sub,
                             LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML | LY_NODE_USES,
                             spec_config, main_submod);
        }
    }

    free(new_indent);
}

static void
tree_print_anyxml(FILE *f, struct ly_module *module, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int spec_config)
{
    uint8_t prefix_len;
    struct ly_mnode_anyxml *anyxml = (struct ly_mnode_anyxml *)mnode;

    assert(spec_config >= 0 && spec_config <= 2);

    fprintf(f, "%s%s--", indent,
            (anyxml->flags & LYS_STATUS_DEPRC ? "x" : (anyxml->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (spec_config == 0) {
        fprintf(f, "%s ", (anyxml->flags & LYS_CONFIG_W ? "rw" : "ro"));
    } else if (spec_config == 1) {
        fprintf(f, "-w ");
    } else if (spec_config == 2) {
        fprintf(f, "ro ");
    }

    prefix_len = 0;
    if (module != anyxml->module) {
        fprintf(f, "%s:", anyxml->module->prefix);
        prefix_len = strlen(anyxml->module->prefix)+1;
    }

    fprintf(f, "%s%s%*sanyxml", anyxml->name, (anyxml->flags & LYS_MAND_TRUE ? " " : "?"),
            3 + (int)((max_name_len - strlen(anyxml->name)) - prefix_len), "   ");

    tree_print_features(f, (const struct ly_feature **)anyxml->features, anyxml->features_size);

    fprintf(f, "\n");
}

static void
tree_print_leaf(FILE *f, struct ly_module *module, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int spec_config)
{
    uint8_t prefix_len;
    struct ly_mnode_leaf *leaf = (struct ly_mnode_leaf *)mnode;
    struct ly_mnode *parent;
    struct ly_mnode_list *list;
    int i, is_key = 0;

    assert(spec_config >= 0 && spec_config <= 2);

    for (parent = leaf->parent; parent && parent->nodetype == LY_NODE_USES; parent = parent->parent);
    if (parent->nodetype == LY_NODE_LIST) {
        list = (struct ly_mnode_list *)parent;
        for (i = 0; i < list->keys_size; i++) {
            if (list->keys[i] == leaf) {
                is_key = 1;
                break;
            }
        }
    }

    fprintf(f, "%s%s--", indent,
            (leaf->flags & LYS_STATUS_DEPRC ? "x" : (leaf->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (spec_config == 0) {
        fprintf(f, "%s ", (leaf->flags & LYS_CONFIG_W ? "rw" : "ro"));
    } else if (spec_config == 1) {
        fprintf(f, "-w ");
    } else if (spec_config == 2) {
        fprintf(f, "ro ");
    }

    prefix_len = 0;
    if (module != leaf->module) {
        fprintf(f, "%s:", leaf->module->prefix);
        prefix_len = strlen(leaf->module->prefix)+1;
    }

    fprintf(f, "%s%s%*s", leaf->name, ((leaf->flags & LYS_MAND_TRUE) || is_key ? " " : "?"),
            3 + (int)((max_name_len - strlen(leaf->name)) - prefix_len), "   ");

    tree_print_type(f, &leaf->type);

    if (leaf->dflt != NULL) {
        fprintf(f, " <%s>", leaf->dflt);
    }

    tree_print_features(f, (const struct ly_feature **)leaf->features, leaf->features_size);

    fprintf(f, "\n");
}

static void
tree_print_leaflist(FILE *f, struct ly_module *module, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int spec_config)
{
    struct ly_mnode_leaflist *leaflist = (struct ly_mnode_leaflist *)mnode;

    assert(spec_config >= 0 && spec_config <= 2);

    fprintf(f, "%s%s--", indent,
            (leaflist->flags & LYS_STATUS_DEPRC ? "x" : (leaflist->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (spec_config == 0) {
        fprintf(f, "%s ", (leaflist->flags & LYS_CONFIG_W ? "rw" : "ro"));
    } else if (spec_config == 1) {
        fprintf(f, "-w ");
    } else if (spec_config == 2) {
        fprintf(f, "ro ");
    }

    if (module != leaflist->module) {
        fprintf(f, "%s:", leaflist->module->prefix);
    }

    fprintf(f, "%s*%*s", leaflist->name, 3 + (int)(max_name_len - strlen(leaflist->name)), "   ");

    tree_print_type(f, &leaflist->type);

    tree_print_features(f, (const struct ly_feature **)leaflist->features, leaflist->features_size);

    fprintf(f, "\n");
}

static void
tree_print_list(FILE *f, struct ly_module *module, int level, char *indent, struct ly_mnode *mnode, int spec_config, struct ly_submodule *main_submod)
{
    int i;
    unsigned int max_child_len;
    char *new_indent;
    struct ly_mnode *sub;
    struct ly_mnode_list *list = (struct ly_mnode_list *)mnode;

    fprintf(f, "%s%s--", indent,
            (list->flags & LYS_STATUS_DEPRC ? "x" : (list->flags & LYS_STATUS_OBSLT ? "o" : "+")));

    if (spec_config == 0) {
        fprintf(f, "%s ", (list->flags & LYS_CONFIG_W ? "rw" : "ro"));
    } else if (spec_config == 1) {
        fprintf(f, "-w ");
    } else if (spec_config == 2) {
        fprintf(f, "ro ");
    }

    if (module != list->module) {
        fprintf(f, "%s:", list->module->prefix);
    }

    fprintf(f, "%s*", list->name);

    for (i = 0; i < list->keys_size; i++) {
        if (i == 0) {
            fprintf(f, " [");
        }
        fprintf(f, "%s%s", list->keys[i]->name, i + 1 < list->keys_size ? "," : "]");
    }

    tree_print_features(f, (const struct ly_feature **)list->features, list->features_size);

    fprintf(f, "\n");

    level++;
    new_indent = create_indent(level, indent, mnode, 0, main_submod);

    max_child_len = get_max_name_len(module, mnode->child);

    LY_TREE_FOR(mnode->child, sub) {
        tree_print_mnode(f, module, level, new_indent, max_child_len, sub,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_USES | LY_NODE_ANYXML,
                         spec_config, main_submod);
    }

    free(new_indent);
}

static void
tree_print_uses(FILE *f, struct ly_module *module, int level, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int spec_config, struct ly_submodule *main_submod)
{
    struct ly_mnode *node;
    struct ly_mnode_uses *uses = (struct ly_mnode_uses *)mnode;

    LY_TREE_FOR(uses->child, node) {
        tree_print_mnode(f, module, level, indent, max_name_len, node,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_USES | LY_NODE_ANYXML,
                         spec_config, main_submod);
    }
}

static void
tree_print_rpc(FILE *f, struct ly_module *module, int level, char *indent, struct ly_mnode *mnode, struct ly_submodule *main_submod)
{
    char *new_indent;
    struct ly_mnode *node;
    struct ly_mnode_rpc *rpc = (struct ly_mnode_rpc *)mnode;

    if (!is_enabled(mnode)) {
        return;
    }

    fprintf(f, "%s%s---x %s", indent,
            (rpc->flags & LYS_STATUS_DEPRC ? "x" : (rpc->flags & LYS_STATUS_OBSLT ? "o" : "+")), rpc->name);

    tree_print_features(f, (const struct ly_feature **)rpc->features, rpc->features_size);

    fprintf(f, "\n");

    level++;
    new_indent = create_indent(level, indent, mnode, 0, main_submod);

    LY_TREE_FOR(rpc->child, node) {
        if (node->nodetype == LY_NODE_INPUT) {
            tree_print_input_output(f, module, level, new_indent, node, 1, main_submod);
        } else if (node->nodetype == LY_NODE_OUTPUT) {
            tree_print_input_output(f, module, level, new_indent, node, 2, main_submod);
        }
    }

    free(new_indent);
}

static void
tree_print_notif(FILE *f, struct ly_module *module, int level, char *indent, struct ly_mnode *mnode, struct ly_submodule *main_submod)
{
    unsigned int max_child_len;
    char *new_indent;
    struct ly_mnode *node;
    struct ly_mnode_notif *notif = (struct ly_mnode_notif *)mnode;

    if (!is_enabled(mnode)) {
        return;
    }

    fprintf(f, "%s%s---n %s", indent,
            (notif->flags & LYS_STATUS_DEPRC ? "x" : (notif->flags & LYS_STATUS_OBSLT ? "o" : "+")),
            notif->name);

    tree_print_features(f, (const struct ly_feature **)notif->features, notif->features_size);

    fprintf(f, "\n");

    level++;
    new_indent = create_indent(level, indent, mnode, 0, main_submod);

    max_child_len = get_max_name_len(module, mnode->child);

    LY_TREE_FOR(notif->child, node) {
        tree_print_mnode(f, module, level, new_indent, max_child_len, node,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST | LY_NODE_ANYXML | LY_NODE_USES,
                         2, main_submod);
    }

    free(new_indent);
}

static void
tree_print_mnode_choice(FILE *f, struct ly_module *module, int level, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int mask,
                        int spec_config, struct ly_submodule *main_submod)
{
    if (!is_enabled(mnode)) {
        return;
    }

    if (mnode->nodetype & mask) {
        if (mnode->nodetype == LY_NODE_CASE) {
            tree_print_case(f, module, level, indent, max_name_len, mnode, 0, spec_config, main_submod);
        } else {
            tree_print_case(f, module, level, indent, max_name_len, mnode, 1, spec_config, main_submod);
        }
    }
}

static void
tree_print_mnode(FILE *f, struct ly_module *module, int level, char *indent, unsigned int max_name_len, struct ly_mnode *mnode, int mask,
                 int spec_config, struct ly_submodule *main_submod)
{
    if (!is_enabled(mnode)) {
        return;
    }

    switch (mnode->nodetype & mask) {
    case LY_NODE_CONTAINER:
        tree_print_container(f, module, level, indent, mnode, spec_config, main_submod);
        break;
    case LY_NODE_CHOICE:
        tree_print_choice(f, module, level, indent, mnode, spec_config, main_submod);
        break;
    case LY_NODE_LEAF:
        tree_print_leaf(f, module, indent, max_name_len, mnode, spec_config);
        break;
    case LY_NODE_LEAFLIST:
        tree_print_leaflist(f, module, indent, max_name_len, mnode, spec_config);
        break;
    case LY_NODE_LIST:
        tree_print_list(f, module, level, indent, mnode, spec_config, main_submod);
        break;
    case LY_NODE_ANYXML:
        tree_print_anyxml(f, module, indent, max_name_len, mnode, spec_config);
        break;
    case LY_NODE_USES:
        tree_print_uses(f, module, level, indent, max_name_len, mnode, spec_config, main_submod);
        break;
    default:
        break;
    }
}

int
tree_print_model(FILE *f, struct ly_module *module)
{
    struct ly_mnode *mnode;
    struct ly_submodule *submod;
    unsigned int max_child_len;
    int level = 1, i, have_rpcs = 0, have_notifs = 0;
    char *indent = malloc((level * 4 + 1) * sizeof (char));
    strcpy(indent, "   ");

    if (module->type) {
        submod = (struct ly_submodule *)module;
        fprintf(f, "submodule: %s (belongs-to %s)\n", submod->name, submod->belongsto->name);
    } else {
        submod = NULL;
        fprintf(f, "module: %s\n", module->name);
    }

    /* included submodules */
    for (i = 0; i < module->inc_size; i++) {
        max_child_len = get_max_name_len((struct ly_module *)module->inc[i].submodule, module->inc[i].submodule->data);

        LY_TREE_FOR(module->inc[i].submodule->data, mnode) {
            tree_print_mnode(f, (struct ly_module *)module->inc[i].submodule, level, indent, max_child_len, mnode,
                             LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST
                             | LY_NODE_ANYXML | LY_NODE_USES, 0, submod);
        }
    }

    /* module */
    max_child_len = get_max_name_len(module, module->data);
    level++;

    LY_TREE_FOR(module->data, mnode) {
        tree_print_mnode(f, module, level, indent, max_child_len, mnode,
                         LY_NODE_CHOICE | LY_NODE_CONTAINER | LY_NODE_LEAF | LY_NODE_LEAFLIST | LY_NODE_LIST
                         | LY_NODE_ANYXML | LY_NODE_USES, 0, submod);
    }

    /* rpc */
    if (module->rpc) {
        have_rpcs = 1;
    } else {
        for (i = 0; i < module->inc_size; i++) {
            if (module->inc[i].submodule->rpc) {
                have_rpcs = 1;
                break;
            }
        }
    }
    if (have_rpcs) {
        fprintf(f, "rpcs:\n");
        for (i = 0; i < module->inc_size; i++) {
            LY_TREE_FOR(module->inc[i].submodule->rpc, mnode) {
                tree_print_rpc(f, (struct ly_module *)module->inc[i].submodule, level, indent, mnode, submod);
            }
        }
        LY_TREE_FOR(module->rpc, mnode) {
            tree_print_rpc(f, module, level, indent, mnode, submod);
        }
    }

    /* notification */
    if (module->notif) {
        have_notifs = 1;
    } else {
        for (i = 0; i < module->inc_size; i++) {
            if (module->inc[i].submodule->notif) {
                have_notifs = 1;
                break;
            }
        }
    }
    if (have_notifs) {
        fprintf(f, "notifications:\n");
        for (i = 0; i < module->inc_size; i++) {
            LY_TREE_FOR(module->inc[i].submodule->notif, mnode) {
                tree_print_notif(f, (struct ly_module *)module->inc[i].submodule, level, indent, mnode, submod);
            }
        }
        LY_TREE_FOR(module->notif, mnode) {
            tree_print_notif(f, module, level, indent, mnode, submod);
        }
    }

    free(indent);
    return EXIT_SUCCESS;
}