2 changed files with 0 additions and 89 deletions
@ -1,88 +0,0 @@ |
|||
// Copyright 2015 Matthew Holt and The Caddy Authors
|
|||
//
|
|||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|||
// you may not use this file except in compliance with the License.
|
|||
// You may obtain a copy of the License at
|
|||
//
|
|||
// http://www.apache.org/licenses/LICENSE-2.0
|
|||
//
|
|||
// Unless required by applicable law or agreed to in writing, software
|
|||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|||
// See the License for the specific language governing permissions and
|
|||
// limitations under the License.
|
|||
|
|||
package markdown |
|||
|
|||
import ( |
|||
"bytes" |
|||
"net/http" |
|||
"strconv" |
|||
"strings" |
|||
"sync" |
|||
|
|||
"github.com/russross/blackfriday/v2" |
|||
|
|||
"github.com/caddyserver/caddy/v2" |
|||
"github.com/caddyserver/caddy/v2/modules/caddyhttp" |
|||
) |
|||
|
|||
func init() { |
|||
caddy.RegisterModule(Markdown{}) |
|||
} |
|||
|
|||
// Markdown is a middleware for rendering a Markdown response body.
|
|||
type Markdown struct { |
|||
} |
|||
|
|||
// CaddyModule returns the Caddy module information.
|
|||
func (Markdown) CaddyModule() caddy.ModuleInfo { |
|||
return caddy.ModuleInfo{ |
|||
ID: "http.handlers.markdown", |
|||
New: func() caddy.Module { return new(Markdown) }, |
|||
} |
|||
} |
|||
|
|||
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, header http.Header) bool { |
|||
return strings.HasPrefix(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 |
|||
} |
|||
|
|||
caddyhttp.CopyHeader(w.Header(), rec.Header()) |
|||
|
|||
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) |
Loading…
Reference in new issue