From 9d30b586165154b0f3ad5c755a92b2f758f50f38 Mon Sep 17 00:00:00 2001
From: Roberto Jimenez Sanchez <jszroberto@gmail.com>
Date: Sat, 22 Jul 2017 14:33:33 +0200
Subject: [PATCH] Add SetKey

---
 set.go | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/set.go b/set.go
index 6e4fc86..319afaa 100644
--- a/set.go
+++ b/set.go
@@ -10,13 +10,25 @@ import (
 )
 
 func SetKeyInFile(file string, path string, key string) error {
-	bytes, err := ioutil.ReadFile(key)
+	yaml, err := ReadYamlFromFile(file)
 	if err != nil {
 		return err
 	}
-	value := string(bytes)
 
-	return SetInFile(file, path, value)
+	err = SetKey(yaml, path, key)
+	if err != nil {
+		return err
+	}
+
+	return WriteYaml(yaml, file)
+}
+
+func SetKey(yaml *simpleyaml.Yaml, path string, key string) error {
+	bytes, err := ioutil.ReadFile(key)
+	if err != nil {
+		return err
+	}
+	return Set(yaml, path, string(bytes))
 }
 
 func SetInFile(file string, path string, val interface{}) error {