From 60f1b629650ab04ade244857a4fa6e999345c98d Mon Sep 17 00:00:00 2001
From: Sasha Koshka <sashakoshka@tebibyte.media>
Date: Sat, 7 Dec 2024 00:59:20 -0500
Subject: [PATCH] cmd/stepd: Buffer templates

---
 cmd/stepd/handler.go | 32 +++++++++++++++++++++++++++-----
 1 file changed, 27 insertions(+), 5 deletions(-)

diff --git a/cmd/stepd/handler.go b/cmd/stepd/handler.go
index a2db3d0..3bbb120 100644
--- a/cmd/stepd/handler.go
+++ b/cmd/stepd/handler.go
@@ -5,6 +5,7 @@ import "fmt"
 import "path"
 import "io/fs"
 import "strings"
+import "strconv"
 import "net/http"
 import "path/filepath"
 import "git.tebibyte.media/sashakoshka/step"
@@ -75,22 +76,43 @@ func (this *handler) serveFile (res http.ResponseWriter, req *http.Request, file
 		return
 	}
 
-	// parse and execute
+	// parse
 	document, err := this.environment.Parse(name)
 	if err != nil {
 		this.serveError(res, req, http.StatusInternalServerError, err)
 		return
 	}
-	if contentType, ok := document.FrontMatter["content-type"]; ok {
-		res.Header().Set("Content-Type", contentType)
+
+	// set up HTTP response recorder
+	recorder := step.HTTPResponseRecorder {
+		Status: http.StatusOK,
+		Head:   res.Header(),
 	}
+	if contentType, ok := document.FrontMatter["content-type"]; ok {
+		recorder.Header().Set("Content-Type", contentType)
+	}
+	if status, ok := document.FrontMatter["status"]; ok {
+		if status, err := strconv.Atoi(status); err == nil {
+			recorder.Status = status
+		}
+	}
+
+	// execute document
 	data := step.HTTPData { }
 	data.Res.Header      = make(http.Header)
-	data.Res.WriteHeader = res.WriteHeader
+	data.Res.WriteHeader = recorder.WriteHeader
+	data.Res.Reset       = recorder.Reset
 	data.Req = req
-	err = document.Execute(res, step.ExecutionData {
+	err = document.Execute(&recorder, step.ExecutionData {
 		Data: data,
 	})
+	if err != nil {
+		this.serveError(res, req, http.StatusInternalServerError, err)
+		return
+	}
+
+	// play back recorded response
+	err = recorder.Play(res)
 	if err != nil {
 		this.logErr(name, err)
 	}