From 7e7756de9b8d094d816eb30c5aa5bf73fcdc635e Mon Sep 17 00:00:00 2001
From: Calvin McLean <calvinlmc@gmail.com>
Date: Sat, 10 Aug 2024 16:00:09 -0700
Subject: [PATCH] Add ResponseWriter to remaining handlers

---
 babyapi_test.go             |  8 ++++----
 examples/event-rsvp/main.go | 12 ++++++------
 examples/todo-htmx/main.go  | 10 +++++-----
 helpers.go                  |  8 ++++----
 html/templates.go           |  2 +-
 router.go                   |  4 ++--
 6 files changed, 22 insertions(+), 22 deletions(-)

diff --git a/babyapi_test.go b/babyapi_test.go
index 1f0487b..231d292 100644
--- a/babyapi_test.go
+++ b/babyapi_test.go
@@ -51,7 +51,7 @@ func TestBabyAPI(t *testing.T) {
 		w.WriteHeader(http.StatusTeapot)
 	}))
 
-	api.AddCustomIDRoute(http.MethodGet, "/teapot", api.GetRequestedResourceAndDo(func(r *http.Request, album *Album) (render.Renderer, *babyapi.ErrResponse) {
+	api.AddCustomIDRoute(http.MethodGet, "/teapot", api.GetRequestedResourceAndDo(func(_ http.ResponseWriter, r *http.Request, album *Album) (render.Renderer, *babyapi.ErrResponse) {
 		render.Status(r, http.StatusTeapot)
 		return album, nil
 	}))
@@ -634,7 +634,7 @@ func (ul *UnorderedList) Render(w http.ResponseWriter, r *http.Request) error {
 	return nil
 }
 
-func (ul *UnorderedList) HTML(r *http.Request) string {
+func (ul *UnorderedList) HTML(_ http.ResponseWriter, r *http.Request) string {
 	templates := map[string]string{
 		"ul": `<ul>
 {{- range .Items }}
@@ -652,7 +652,7 @@ type ListItem struct {
 	Content string
 }
 
-func (d *ListItem) HTML(*http.Request) string {
+func (d *ListItem) HTML(http.ResponseWriter, *http.Request) string {
 	tmpl := template.Must(template.New("li").Parse(`<li>{{ .Content }}</li>`))
 	return babyapi.MustRenderHTML(tmpl, d)
 }
@@ -810,7 +810,7 @@ func TestAPIModifiers(t *testing.T) {
 		})
 
 	api.AddIDMiddleware(api.GetRequestedResourceAndDoMiddleware(
-		func(r *http.Request, a *Album) (*http.Request, *babyapi.ErrResponse) {
+		func(_ http.ResponseWriter, r *http.Request, a *Album) (*http.Request, *babyapi.ErrResponse) {
 			require.NotNil(t, a)
 			idMiddlewareWithRequestResource++
 			return r, nil
diff --git a/examples/event-rsvp/main.go b/examples/event-rsvp/main.go
index 3748bcb..45045bc 100644
--- a/examples/event-rsvp/main.go
+++ b/examples/event-rsvp/main.go
@@ -96,7 +96,7 @@ func (api *API) export(w http.ResponseWriter, r *http.Request) render.Renderer {
 }
 
 // Use a custom route to set RSVP so rsvpResponse can be used to return HTML buttons
-func (api *API) rsvp(r *http.Request, invite *Invite) (render.Renderer, *babyapi.ErrResponse) {
+func (api *API) rsvp(_ http.ResponseWriter, r *http.Request, invite *Invite) (render.Renderer, *babyapi.ErrResponse) {
 	if err := r.ParseForm(); err != nil {
 		return nil, babyapi.ErrInvalidRequest(fmt.Errorf("error parsing form data: %w", err))
 	}
@@ -116,7 +116,7 @@ func (api *API) rsvp(r *http.Request, invite *Invite) (render.Renderer, *babyapi
 }
 
 // Allow adding bulk invites with a single request
-func (api *API) addBulkInvites(r *http.Request, event *Event) (render.Renderer, *babyapi.ErrResponse) {
+func (api *API) addBulkInvites(_ http.ResponseWriter, r *http.Request, event *Event) (render.Renderer, *babyapi.ErrResponse) {
 	if err := r.ParseForm(); err != nil {
 		return nil, babyapi.ErrInvalidRequest(fmt.Errorf("error parsing form data: %w", err))
 	}
@@ -154,7 +154,7 @@ func (api *API) addBulkInvites(r *http.Request, event *Event) (render.Renderer,
 // authenticationMiddleware enforces access to Events and Invites. Admin access to an Event requires a password query parameter.
 // Access to Invites is allowed by the invite ID and requires no extra auth. The invite ID in the path or query parameter allows
 // read-only access to the Event
-func (api *API) authenticationMiddleware(r *http.Request, event *Event) (*http.Request, *babyapi.ErrResponse) {
+func (api *API) authenticationMiddleware(_ http.ResponseWriter, r *http.Request, event *Event) (*http.Request, *babyapi.ErrResponse) {
 	password := r.URL.Query().Get("password")
 	inviteID := r.URL.Query().Get("invite")
 	if inviteID == "" {
@@ -184,7 +184,7 @@ func (api *API) authenticationMiddleware(r *http.Request, event *Event) (*http.R
 }
 
 // getAllInvitesMiddleware will get all invites when rendering HTML so it is accessible to the endpoint
-func (api *API) getAllInvitesMiddleware(r *http.Request, event *Event) (*http.Request, *babyapi.ErrResponse) {
+func (api *API) getAllInvitesMiddleware(_ http.ResponseWriter, r *http.Request, event *Event) (*http.Request, *babyapi.ErrResponse) {
 	if render.GetAcceptedContentType(r) != render.ContentTypeHTML {
 		return r, nil
 	}
@@ -265,7 +265,7 @@ func (e *Event) Bind(r *http.Request) error {
 	return e.DefaultResource.Bind(r)
 }
 
-func (e *Event) HTML(r *http.Request) string {
+func (e *Event) HTML(_ http.ResponseWriter, r *http.Request) string {
 	return eventPage.Render(r, struct {
 		Password string
 		*Event
@@ -300,7 +300,7 @@ func (i *Invite) Bind(r *http.Request) error {
 	return i.DefaultResource.Bind(r)
 }
 
-func (i *Invite) HTML(r *http.Request) string {
+func (i *Invite) HTML(_ http.ResponseWriter, r *http.Request) string {
 	event, _ := babyapi.GetResourceFromContext[*Event](r.Context(), babyapi.ContextKey("Event"))
 
 	return invitePage.Render(r, struct {
diff --git a/examples/todo-htmx/main.go b/examples/todo-htmx/main.go
index bc7c067..e017f41 100644
--- a/examples/todo-htmx/main.go
+++ b/examples/todo-htmx/main.go
@@ -108,7 +108,7 @@ type TODO struct {
 	Completed   *bool
 }
 
-func (t *TODO) HTML(r *http.Request) string {
+func (t *TODO) HTML(_ http.ResponseWriter, r *http.Request) string {
 	return todoRow.Render(r, t)
 }
 
@@ -120,7 +120,7 @@ func (at AllTODOs) Render(w http.ResponseWriter, r *http.Request) error {
 	return nil
 }
 
-func (at AllTODOs) HTML(r *http.Request) string {
+func (at AllTODOs) HTML(_ http.ResponseWriter, r *http.Request) string {
 	return allTODOs.Render(r, at.Items)
 }
 
@@ -131,7 +131,7 @@ func createAPI() *babyapi.API[*TODO] {
 
 	// Use AllTODOs in the GetAll response since it implements HTMLer
 	api.SetGetAllResponseWrapper(func(todos []*TODO) render.Renderer {
-		return AllTODOs{ResourceList: babyapi.ResourceList[*TODO]{todos}}
+		return AllTODOs{ResourceList: babyapi.ResourceList[*TODO]{Items: todos}}
 	})
 
 	api.ApplyExtension(extensions.HTMX[*TODO]{})
@@ -140,13 +140,13 @@ func createAPI() *babyapi.API[*TODO] {
 	todoChan := api.AddServerSentEventHandler("/listen")
 
 	// Push events onto the SSE channel when new TODOs are created
-	api.SetOnCreateOrUpdate(func(_ http.ResponseWriter, r *http.Request, t *TODO) *babyapi.ErrResponse {
+	api.SetOnCreateOrUpdate(func(w http.ResponseWriter, r *http.Request, t *TODO) *babyapi.ErrResponse {
 		if r.Method != http.MethodPost {
 			return nil
 		}
 
 		select {
-		case todoChan <- &babyapi.ServerSentEvent{Event: "newTODO", Data: t.HTML(r)}:
+		case todoChan <- &babyapi.ServerSentEvent{Event: "newTODO", Data: t.HTML(w, r)}:
 		default:
 			logger := babyapi.GetLoggerFromContext(r.Context())
 			logger.Info("no listeners for server-sent event")
diff --git a/helpers.go b/helpers.go
index 549337d..720a697 100644
--- a/helpers.go
+++ b/helpers.go
@@ -72,7 +72,7 @@ func (a *API[T]) findIDParam(r *http.Request) string {
 
 // GetRequestedResourceAndDo is a wrapper that handles getting a resource from storage based on the ID in the request URL
 // and rendering the response. This is useful for imlementing a CustomIDRoute
-func (a *API[T]) GetRequestedResourceAndDo(do func(*http.Request, T) (render.Renderer, *ErrResponse)) http.HandlerFunc {
+func (a *API[T]) GetRequestedResourceAndDo(do func(http.ResponseWriter, *http.Request, T) (render.Renderer, *ErrResponse)) http.HandlerFunc {
 	return Handler(func(w http.ResponseWriter, r *http.Request) render.Renderer {
 		logger := GetLoggerFromContext(r.Context())
 
@@ -82,7 +82,7 @@ func (a *API[T]) GetRequestedResourceAndDo(do func(*http.Request, T) (render.Ren
 			return httpErr
 		}
 
-		resp, httpErr := do(r, resource)
+		resp, httpErr := do(w, r, resource)
 		if httpErr != nil {
 			return httpErr
 		}
@@ -99,7 +99,7 @@ func (a *API[T]) GetRequestedResourceAndDo(do func(*http.Request, T) (render.Ren
 // GetRequestedResourceAndDoMiddleware is a shortcut for creating an ID-scoped middleware that gets the requested resource from storage,
 // calls the provided 'do' function, and calls next.ServeHTTP. If the resource is not found for a PUT request, the error is ignored
 // The 'do' function returns *http.Request so the request context can be modified by middleware
-func (a *API[T]) GetRequestedResourceAndDoMiddleware(do func(*http.Request, T) (*http.Request, *ErrResponse)) func(next http.Handler) http.Handler {
+func (a *API[T]) GetRequestedResourceAndDoMiddleware(do func(http.ResponseWriter, *http.Request, T) (*http.Request, *ErrResponse)) func(next http.Handler) http.Handler {
 	return func(next http.Handler) http.Handler {
 		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 
@@ -119,7 +119,7 @@ func (a *API[T]) GetRequestedResourceAndDoMiddleware(do func(*http.Request, T) (
 				return
 			}
 
-			r, httpErr = do(r, resource)
+			r, httpErr = do(w, r, resource)
 			if httpErr != nil {
 				_ = render.Render(w, r, httpErr)
 				return
diff --git a/html/templates.go b/html/templates.go
index daf92bd..73b6102 100644
--- a/html/templates.go
+++ b/html/templates.go
@@ -68,6 +68,6 @@ func (h htmlRenderer) Render(_ http.ResponseWriter, _ *http.Request) error {
 	return nil
 }
 
-func (h htmlRenderer) HTML(r *http.Request) string {
+func (h htmlRenderer) HTML(_ http.ResponseWriter, r *http.Request) string {
 	return h.t.Render(r, h.data)
 }
diff --git a/router.go b/router.go
index dee1784..b47488f 100644
--- a/router.go
+++ b/router.go
@@ -45,7 +45,7 @@ var _ error = BuilderError{}
 // HTMLer allows for easily represending reponses as HTML strings when accepted content
 // type is text/html
 type HTMLer interface {
-	HTML(*http.Request) string
+	HTML(http.ResponseWriter, *http.Request) string
 }
 
 // EnableHTMLRender overrides the default render.Respond function to add support for the
@@ -56,7 +56,7 @@ func EnableHTMLRender() {
 			if render.GetAcceptedContentType(r) == render.ContentTypeHTML {
 				htmler, ok := v.(HTMLer)
 				if ok {
-					render.HTML(w, r, htmler.HTML(r))
+					render.HTML(w, r, htmler.HTML(w, r))
 					return
 				}
 			}