From 34a25dd5580948ff5b83843b81e72bda1b133189 Mon Sep 17 00:00:00 2001 From: Matthew Holt Date: Thu, 23 May 2019 14:41:43 -0600 Subject: Add very simple markdown middleware for now --- modules/caddyhttp/markdown/markdown.go | 53 ++++++++++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 modules/caddyhttp/markdown/markdown.go (limited to 'modules/caddyhttp/markdown') diff --git a/modules/caddyhttp/markdown/markdown.go b/modules/caddyhttp/markdown/markdown.go new file mode 100644 index 0000000..9c23e44 --- /dev/null +++ b/modules/caddyhttp/markdown/markdown.go @@ -0,0 +1,53 @@ +package markdown + +import ( + "net/http" + "strconv" + + "gopkg.in/russross/blackfriday.v2" + + "bitbucket.org/lightcodelabs/caddy2" + "bitbucket.org/lightcodelabs/caddy2/modules/caddyhttp" +) + +func init() { + caddy2.RegisterModule(caddy2.Module{ + Name: "http.middleware.markdown", + New: func() interface{} { return new(Markdown) }, + }) +} + +// Markdown is a middleware for rendering a Markdown response body. +type Markdown struct { +} + +func (m Markdown) ServeHTTP(w http.ResponseWriter, r *http.Request, next caddyhttp.Handler) error { + mrw := &markdownResponseWriter{ + ResponseWriterWrapper: &caddyhttp.ResponseWriterWrapper{ResponseWriter: w}, + } + return next.ServeHTTP(mrw, r) +} + +type markdownResponseWriter struct { + *caddyhttp.ResponseWriterWrapper + statusCode int + wroteHeader bool +} + +func (mrw *markdownResponseWriter) WriteHeader(code int) { + mrw.statusCode = code +} + +func (mrw *markdownResponseWriter) Write(d []byte) (int, error) { + output := blackfriday.Run(d) + if !mrw.wroteHeader { + mrw.Header().Set("Content-Length", strconv.Itoa(len(output))) + mrw.Header().Set("Content-Type", "text/html; charset=utf-8") + mrw.WriteHeader(mrw.statusCode) + mrw.wroteHeader = true + } + return mrw.ResponseWriter.Write(output) +} + +// Interface guard +var _ caddyhttp.MiddlewareHandler = (*Markdown)(nil) -- cgit v1.2.3