mirror of
https://github.com/caddyserver/caddy.git
synced 2024-12-23 22:27:38 -05:00
d49f762f6d
- Fix static responder so it doesn't replace its own headers config, and instead replaces the actual response header values - caddyhttp.ResponseRecorder type optionally buffers response - Add interface guards to ensure regexp matchers get provisioned - Use default HTTP port if one is not explicitly set - Encode middleware writes status code 200 if not written upstream - Templates and markdown only try to execute on text responses - Static file server sets Content-Type based on file extension only (this whole thing -- MIME sniffing, etc -- needs more configurability)
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package markdown
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
|
|
"gopkg.in/russross/blackfriday.v2"
|
|
|
|
"github.com/caddyserver/caddy"
|
|
"github.com/caddyserver/caddy/modules/caddyhttp"
|
|
)
|
|
|
|
func init() {
|
|
caddy.RegisterModule(caddy.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 {
|
|
buf := bufPool.Get().(*bytes.Buffer)
|
|
buf.Reset()
|
|
defer bufPool.Put(buf)
|
|
|
|
shouldBuf := func(status int) bool {
|
|
return strings.HasPrefix(w.Header().Get("Content-Type"), "text/")
|
|
}
|
|
|
|
rec := caddyhttp.NewResponseRecorder(w, buf, shouldBuf)
|
|
|
|
err := next.ServeHTTP(rec, r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !rec.Buffered() {
|
|
return nil
|
|
}
|
|
|
|
output := blackfriday.Run(buf.Bytes())
|
|
|
|
w.Header().Set("Content-Length", strconv.Itoa(len(output)))
|
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
|
w.Header().Del("Accept-Ranges") // we don't know ranges for dynamically-created content
|
|
w.Header().Del("Etag") // don't know a way to quickly generate etag for dynamic content
|
|
w.Header().Del("Last-Modified") // useless for dynamic content since it's always changing
|
|
|
|
w.WriteHeader(rec.Status())
|
|
w.Write(output)
|
|
|
|
return nil
|
|
}
|
|
|
|
var bufPool = sync.Pool{
|
|
New: func() interface{} {
|
|
return new(bytes.Buffer)
|
|
},
|
|
}
|
|
|
|
// Interface guard
|
|
var _ caddyhttp.MiddlewareHandler = (*Markdown)(nil)
|