From ad08a020aeb7bbc9c3cf0cb249f88c639c5c0fdd Mon Sep 17 00:00:00 2001 From: h0n9 Date: Wed, 5 Jun 2024 11:01:15 +0900 Subject: [PATCH] refactor: Update Save method to handle base64 encoded secrets The Save method in the SecretHandler struct is updated to handle base64 encoded secrets. If the decodeBase64EncodedSecret flag is set to true, the secret is decoded from base64 and then saved to the specified file. This allows for more flexibility in handling secrets that are stored in base64 format. --- handler/handler.go | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/handler/handler.go b/handler/handler.go index 4709372..02ebe22 100644 --- a/handler/handler.go +++ b/handler/handler.go @@ -55,24 +55,24 @@ func (handler *SecretHandler) Save(secretID, path string, decodeBase64EncodedSec return handler.template.Execute(file, m) } - var ( - buff bytes.Buffer - decodedSecret []byte - ) - // execute template + var buff bytes.Buffer err = handler.template.Execute(&buff, m) if err != nil { return err } // decode base64 encoded secret - decodedSecret, err = util.DecodeBase64StrToBytes(buff.String()) + decodedSecret, err := util.DecodeBase64BytesToBytes(buff.Bytes()) if err != nil { return err } // write decoded secret to file _, err = file.Write(decodedSecret) - return err + if err != nil { + return err + } + + return nil }