feat: string templater

Author: roger-king

prompts/prompt.go

@@ -0,0 +1,5 @@
+package prompts
+
+type Prompt interface {
+	
+}

templater/file.go

@@ -2,7 +2,6 @@ package templater
 
 import (
 	"errors"
-	"os"
 
 	"github.com/aymerick/raymond"
 )
@@ -37,28 +36,10 @@ func (f *FileMaker) Render(vars map[string]string) error {
 	return nil
 }
 
-func (f *FileMaker) writeToFile(dest string) error {
-	file, err := os.Create(dest)
-
-    if err != nil {
-        return err
-    }
-
-    defer file.Close()
-
-    _, err = file.WriteString(f.Result)
-
-    if err != nil {
-        return err
-	}
-	
-	return nil
-}
-
 // Save - saves results to a file
 func (f *FileMaker) Save(dest string) error {
 	if len(f.Result) > 0 {
-		err := f.writeToFile(dest)
+		err := writeToFile(dest, f.Result)
 
 		if err != nil {
 			return err

templater/string.go

@@ -0,0 +1,46 @@
+package templater
+
+import (
+	"errors"
+
+	"github.com/aymerick/raymond"
+)
+
+// StringTemplate -
+type StringTemplate struct {
+	TPL string
+	Result string
+}
+
+// NewStringTemplate -
+func NewStringTemplate(tpl string) *StringTemplate {
+	return &StringTemplate {
+		TPL: tpl,
+	}
+}
+
+// Render -
+func (s *StringTemplate) Render(ctx map[string]string) error {
+	result, err := raymond.Render(s.TPL, ctx)
+
+	if err != nil {
+		return err
+	}
+
+	s.Result = result
+	return nil
+}
+
+// Save - saves results to a file
+func (s *StringTemplate) Save(dest string) error {
+	if len(s.Result) > 0 {
+		err := writeToFile(dest, s.Result)
+
+		if err != nil {
+			return err
+		}
+	} else {
+		return errors.New("no result")
+	}
+	return nil
+}

templater/templater.go

@@ -1,6 +1,27 @@
 package templater
 
+import "os"
+
 // Templater -
 type Templater interface {
-	Render() interface{}
+	Render(vars map[string]string) error
+	Save(dest string) error
+}
+
+func writeToFile(dest, result string) error {
+	file, err := os.Create(dest)
+
+    if err != nil {
+        return err
+    }
+
+    defer file.Close()
+
+    _, err = file.WriteString(result)
+
+    if err != nil {
+        return err
+	}
+	
+	return nil
 }