From 343cdecabd46e2b505f3f92c8281753df1ee0fee Mon Sep 17 00:00:00 2001
From: tjpcc <tjp@ctrl-c.club>
Date: Sat, 26 Aug 2023 09:22:03 -0600
Subject: GmisubToAtom: output result to any io.Writer rather than return a
 string

---
 gemini/gemtext/sub.go | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

(limited to 'gemini/gemtext/sub.go')

diff --git a/gemini/gemtext/sub.go b/gemini/gemtext/sub.go
index 1247643..365d41b 100644
--- a/gemini/gemtext/sub.go
+++ b/gemini/gemtext/sub.go
@@ -3,6 +3,7 @@ package gemtext
 import (
 	"bytes"
 	"html/template"
+	"io"
 	"net/url"
 	"regexp"
 	"strconv"
@@ -14,12 +15,17 @@ import (
 //
 // It identifies feed fields and entries according to the specification at
 // gemini://gemini.circumlunar.space/docs/companion/subscription.gmi
-func GmisubToAtom(doc Document, location url.URL) string {
-	buf := &bytes.Buffer{}
-	if err := atomTmpl.Execute(buf, parseGemSub(doc, &location)); err != nil {
-		panic(err)
+func GmisubToAtom(doc Document, location url.URL, out io.Writer) error {
+	if _, err := out.Write([]byte(`<?xml version="1.0" encoding="utf-8"?>`)); err != nil {
+		return err
 	}
-	return `<?xml version="1.0" encoding="utf-8"?>` + "\n" + buf.String()
+	if _, err := out.Write([]byte{'\n'}); err != nil {
+		return err
+	}
+	if err := atomTmpl.Execute(out, parseGemSub(doc, &location)); err != nil {
+		return err
+	}
+	return nil
 }
 
 type gmiSub struct {
-- 
cgit v1.2.3