summaryrefslogtreecommitdiffhomepage
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.go102
1 files changed, 0 insertions, 102 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 a28d561..0000000
--- a/vendor/github.com/a-h/templ/handler.go
+++ /dev/null
@@ -1,102 +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
-}
-
-const componentHandlerErrorMessage = "templ: failed to render template"
-
-func (ch *ComponentHandler) ServeHTTPBuffered(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)
- err := ch.Component.Render(r.Context(), buf)
- if err != nil {
- 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)
- return
- }
- 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) ServeHTTPStreamed(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", ch.ContentType)
- if ch.Status != 0 {
- w.WriteHeader(ch.Status)
- }
- if err := ch.Component.Render(r.Context(), w); err != nil {
- 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)
- }
-}
-
-// 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
- }
-}