From 820ed5ae0c8cccc203db2c642af05c763eda0e87 Mon Sep 17 00:00:00 2001
From: Daniel Heule <hda@sfs.biz>
Date: Mon, 24 Feb 2014 23:27:42 +0100
Subject: [PATCH] added option comment_line to prefix key value with a comment
 line

---
 cdist/conf/type/__key_value/gencode-remote    | 52 +++++++++++++++----
 .../parameter/default/comment_line            |  1 +
 .../conf/type/__key_value/parameter/optional  |  1 +
 3 files changed, 43 insertions(+), 11 deletions(-)
 create mode 100644 cdist/conf/type/__key_value/parameter/default/comment_line

diff --git a/cdist/conf/type/__key_value/gencode-remote b/cdist/conf/type/__key_value/gencode-remote
index 0e963b7c..2fa07f5c 100755
--- a/cdist/conf/type/__key_value/gencode-remote
+++ b/cdist/conf/type/__key_value/gencode-remote
@@ -47,22 +47,26 @@ case "$state_should" in
        exit 1
 esac
 
-cat <<CDIST_HEREDOC_END_HERE_MARKER
-export state="\$(cat <<"CDIST_INPUT_END_HERE_MARKER"
+cat <<__CDIST_HEREDOC_END_HERE_MARKER
+export state="\$(cat <<"__CDIST_INPUT_END_HERE_MARKER"
 $state_should
-CDIST_INPUT_END_HERE_MARKER
+__CDIST_INPUT_END_HERE_MARKER
 )"
-export key="\$(cat <<"CDIST_INPUT_END_HERE_MARKER"
+export key="\$(cat <<"__CDIST_INPUT_END_HERE_MARKER"
 $key
-CDIST_INPUT_END_HERE_MARKER
+__CDIST_INPUT_END_HERE_MARKER
 )"
-export delimiter="\$(cat <<"CDIST_INPUT_END_HERE_MARKER"
+export delimiter="\$(cat <<"__CDIST_INPUT_END_HERE_MARKER"
 $(cat "$__object/parameter/delimiter")
-CDIST_INPUT_END_HERE_MARKER
+__CDIST_INPUT_END_HERE_MARKER
 )"
-export value="\$(cat <<"CDIST_INPUT_END_HERE_MARKER"
+export value="\$(cat <<"__CDIST_INPUT_END_HERE_MARKER"
 $(cat "$__object/parameter/value")
-CDIST_INPUT_END_HERE_MARKER
+__CDIST_INPUT_END_HERE_MARKER
+)"
+export comment="\$(cat <<"__CDIST_INPUT_END_HERE_MARKER"
+$(cat "$__object/parameter/comment_line")
+__CDIST_INPUT_END_HERE_MARKER
 )"
 
 tmpfile=\$(mktemp "${file}.cdist.XXXXXXXXXX")
@@ -75,9 +79,12 @@ BEGIN {
     key=ENVIRON["key"]
     delimiter=ENVIRON["delimiter"]
     value=ENVIRON["value"]
+    comment=ENVIRON["comment"]
     keydel=key delimiter
     line=keydel value
     inserted=0
+    ll=""
+    llpopulated=0
 }
 # enter the main loop
 {
@@ -85,25 +92,48 @@ BEGIN {
     i = index(\$0,keydel)
     if(i == 1) {
         if(state == "absent") {
+            if(ll == comment) {
+                # if comment is present, clear llpopulated flag
+                llpopulated=0
+            }
             # if absent, simple yump over this line
             next
         }
         else {
+            # if comment is present and not present in last line
+            if (llpopulated == 1) {
+                print ll
+                if( comment != "" && ll != comment) {
+                    print comment
+                }
+                llpopulated=0
+            }
             inserted=1
             # state is present, so insert correct line here
             print line
+            ll=line
             next
         }
     }
     else {
-        print \$0
+        if(llpopulated == 1) {
+            print ll
+        }
+        ll=\$0
+        llpopulated=1
     }
 }
 END {
+    if(llpopulated == 1) {
+        print ll
+    }
     if(inserted == 0 && state == "present" ) {
+        if(comment != "" && ll != comment){
+            print comment
+        }
         print line
     }
 }
 AWK_EOF
 mv -f "\$tmpfile" "$file"
-CDIST_HEREDOC_END_HERE_MARKER
+__CDIST_HEREDOC_END_HERE_MARKER
diff --git a/cdist/conf/type/__key_value/parameter/default/comment_line b/cdist/conf/type/__key_value/parameter/default/comment_line
new file mode 100644
index 00000000..8b137891
--- /dev/null
+++ b/cdist/conf/type/__key_value/parameter/default/comment_line
@@ -0,0 +1 @@
+
diff --git a/cdist/conf/type/__key_value/parameter/optional b/cdist/conf/type/__key_value/parameter/optional
index 483e3192..8eeebd0c 100644
--- a/cdist/conf/type/__key_value/parameter/optional
+++ b/cdist/conf/type/__key_value/parameter/optional
@@ -1,3 +1,4 @@
 key
 value
 state
+comment_line