package koushinviewhtml import ( "bytes" "fmt" "html/template" "io/ioutil" "strings" "git.sr.ht/~emersion/koushin" koushinbase "git.sr.ht/~emersion/koushin/plugins/base" "github.com/emersion/go-message" ) const tpl = ` ` type viewer struct{} func (viewer) ViewMessagePart(ctx *koushin.Context, msg *koushinbase.IMAPMessage, part *message.Entity) (interface{}, error) { mimeType, _, err := part.Header.ContentType() if err != nil { return nil, err } if !strings.EqualFold(mimeType, "text/html") { return nil, koushinbase.ErrViewUnsupported } body, err := ioutil.ReadAll(part.Body) if err != nil { return nil, fmt.Errorf("failed to read part body: %v", err) } body, err = sanitizeHTML(body) if err != nil { return nil, fmt.Errorf("failed to sanitize HTML part: %v", err) } t := template.Must(template.New("view-html.html").Parse(tpl)) var buf bytes.Buffer err = t.Execute(&buf, string(body)) if err != nil { return nil, err } return template.HTML(buf.String()), nil } func init() { koushinbase.RegisterViewer(viewer{}) }