add strings builder, customize ALL outputs (#6)

1 files changed, 47 insertions, 0 deletions

diff --git a/pkg/writer/writer.go b/pkg/writer/writer.go
new file mode 100644
index 0000000..6b66e75
--- /dev/null
+++ b/pkg/writer/writer.go
@@ -0,0 +1,47 @@
+package writer
+
+import (
+        "fmt"
+        "strings"
+)
+
+var output strings.Builder
+
+var temp strings.Builder
+
+func Append(str string, a ...any) {
+        _, err := output.WriteString(fmt.Sprintf(str, a...))
+        if err != nil {
+                output.WriteString(err.Error())
+        }
+}
+
+func AppendLine(str string, a ...any) {
+        Append(str, a...)
+        output.WriteString("\n")
+}
+
+func GetString() string {
+        return output.String()
+}
+
+func TempAppend(str string, a ...any) {
+        _, err := temp.WriteString(fmt.Sprintf(str, a...))
+        if err != nil {
+                temp.WriteString(err.Error())
+        }
+}
+
+func TempAppendLine(str string, a ...any) {
+        TempAppend(str, a...)
+        temp.WriteString("\n")
+}
+
+func TempGetString() string {
+        return temp.String()
+}
+
+func AppendOutputFromTemp() {
+        output.WriteString(temp.String())
+        temp.Reset()
+}