summaryrefslogtreecommitdiff
path: root/vendor/github.com/a-h/templ/handler.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/a-h/templ/handler.go')
-rw-r--r--vendor/github.com/a-h/templ/handler.go163
1 files changed, 0 insertions, 163 deletions
diff --git a/vendor/github.com/a-h/templ/handler.go b/vendor/github.com/a-h/templ/handler.go
deleted file mode 100644
index ae1552b..0000000
--- a/vendor/github.com/a-h/templ/handler.go
+++ /dev/null
@@ -1,163 +0,0 @@
-package templ
-
-import (
- "net/http"
-)
-
-// ComponentHandler is a http.Handler that renders components.
-type ComponentHandler struct {
- Component Component
- Status int
- ContentType string
- ErrorHandler func(r *http.Request, err error) http.Handler
- StreamResponse bool
- FragmentIDs []any
-}
-
-const componentHandlerErrorMessage = "templ: failed to render template"
-
-func (ch *ComponentHandler) handleRenderErr(w http.ResponseWriter, r *http.Request, err error) {
- if ch.ErrorHandler != nil {
- w.Header().Set("Content-Type", ch.ContentType)
- ch.ErrorHandler(r, err).ServeHTTP(w, r)
- return
- }
- http.Error(w, componentHandlerErrorMessage, http.StatusInternalServerError)
-}
-
-func (ch *ComponentHandler) ServeHTTPBufferedFragment(w http.ResponseWriter, r *http.Request) {
- // Since the component may error, write to a buffer first.
- // This prevents partial responses from being written to the client.
- buf := GetBuffer()
- defer ReleaseBuffer(buf)
-
- // Render the component into io.Discard, but use the buffer for fragments.
- if err := RenderFragments(r.Context(), buf, ch.Component, ch.FragmentIDs...); err != nil {
- ch.handleRenderErr(w, r, err)
- return
- }
-
- // The component rendered successfully, we can write the Content-Type and Status.
- w.Header().Set("Content-Type", ch.ContentType)
- if ch.Status != 0 {
- w.WriteHeader(ch.Status)
- }
- // Ignore write error like http.Error() does, because there is
- // no way to recover at this point.
- _, _ = w.Write(buf.Bytes())
-}
-
-func (ch *ComponentHandler) ServeHTTPBufferedComplete(w http.ResponseWriter, r *http.Request) {
- // Since the component may error, write to a buffer first.
- // This prevents partial responses from being written to the client.
- buf := GetBuffer()
- defer ReleaseBuffer(buf)
-
- // Render the component into the buffer.
- if err := ch.Component.Render(r.Context(), buf); err != nil {
- ch.handleRenderErr(w, r, err)
- return
- }
-
- // The component rendered successfully, we can write the Content-Type and Status.
- w.Header().Set("Content-Type", ch.ContentType)
- if ch.Status != 0 {
- w.WriteHeader(ch.Status)
- }
- // Ignore write error like http.Error() does, because there is
- // no way to recover at this point.
- _, _ = w.Write(buf.Bytes())
-}
-
-func (ch *ComponentHandler) ServeHTTPBuffered(w http.ResponseWriter, r *http.Request) {
- // If fragments are specified, render only those.
- if len(ch.FragmentIDs) > 0 {
- ch.ServeHTTPBufferedFragment(w, r)
- return
- }
-
- // Otherwise, render the complete component.
- ch.ServeHTTPBufferedComplete(w, r)
-}
-
-func (ch *ComponentHandler) ServeHTTPStreamed(w http.ResponseWriter, r *http.Request) {
- // If streaming, we do not buffer the response, so set the headers immediately.
- w.Header().Set("Content-Type", ch.ContentType)
- if ch.Status != 0 {
- w.WriteHeader(ch.Status)
- }
-
- // Pass fragment names to the context if specified.
- if len(ch.FragmentIDs) > 0 {
-
- // Render the component into io.Discard, but use the buffer for fragments.
- if err := RenderFragments(r.Context(), w, ch.Component, ch.FragmentIDs...); err != nil {
- ch.handleRenderErr(w, r, err)
- return
- }
- return
- }
-
- // Render the component into the buffer.
- if err := ch.Component.Render(r.Context(), w); err != nil {
- ch.handleRenderErr(w, r, err)
- return
- }
-}
-
-// ServeHTTP implements the http.Handler interface.
-func (ch ComponentHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
- if ch.StreamResponse {
- ch.ServeHTTPStreamed(w, r)
- return
- }
- ch.ServeHTTPBuffered(w, r)
-}
-
-// Handler creates a http.Handler that renders the template.
-func Handler(c Component, options ...func(*ComponentHandler)) *ComponentHandler {
- ch := &ComponentHandler{
- Component: c,
- ContentType: "text/html; charset=utf-8",
- }
- for _, o := range options {
- o(ch)
- }
- return ch
-}
-
-// WithStatus sets the HTTP status code returned by the ComponentHandler.
-func WithStatus(status int) func(*ComponentHandler) {
- return func(ch *ComponentHandler) {
- ch.Status = status
- }
-}
-
-// WithContentType sets the Content-Type header returned by the ComponentHandler.
-func WithContentType(contentType string) func(*ComponentHandler) {
- return func(ch *ComponentHandler) {
- ch.ContentType = contentType
- }
-}
-
-// WithErrorHandler sets the error handler used if rendering fails.
-func WithErrorHandler(eh func(r *http.Request, err error) http.Handler) func(*ComponentHandler) {
- return func(ch *ComponentHandler) {
- ch.ErrorHandler = eh
- }
-}
-
-// WithStreaming sets the ComponentHandler to stream the response instead of buffering it.
-func WithStreaming() func(*ComponentHandler) {
- return func(ch *ComponentHandler) {
- ch.StreamResponse = true
- }
-}
-
-// WithFragments sets the ids of the fragments to render.
-// If not set, all content is rendered.
-func WithFragments(ids ...any) func(*ComponentHandler) {
- return func(ch *ComponentHandler) {
- ch.FragmentIDs = ids
- }
-}