From 40f69b354d449b727666fe85159fa627f22345ba Mon Sep 17 00:00:00 2001
From: Radek Krejci <rkrejci@cesnet.cz>
Date: Thu, 11 Feb 2016 12:53:39 +0100
Subject: [PATCH] tree printer FEATURE print augment definitions

Relates to #24
---
 src/printer_tree.c | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/printer_tree.c b/src/printer_tree.c
index 03ae0cebc..47cfb2a58 100644
--- a/src/printer_tree.c
+++ b/src/printer_tree.c
@@ -653,6 +653,7 @@ tree_print_model(struct lyout *out, const struct lys_module *module)
     unsigned int max_child_len;
     int level = 1, have_rpcs = 0, have_notifs = 0;
     char *indent = malloc((level * 4 + 1) * sizeof (char));
+    int i;
 
     if (!indent) {
         LOGMEM;
@@ -691,6 +692,16 @@ tree_print_model(struct lyout *out, const struct lys_module *module)
         }
     }
 
+    /* augment */
+    for (i = 0; i < module->augment_size; i++) {
+        ly_print(out, "augment %s:\n", module->augment[i].target_name);
+        LY_TREE_FOR(module->augment[i].child, node) {
+            tree_print_snode(out, module, level, indent, max_child_len, node,
+                             LYS_CHOICE | LYS_CONTAINER | LYS_LEAF | LYS_LEAFLIST | LYS_LIST
+                             | LYS_ANYXML | LYS_USES, 0);
+        }
+    }
+
     /* rpc */
     if (have_rpcs) {
         ly_print(out, "rpcs:\n");