From 8299617ebc24a4a5bd1dc03070e17713be7e1e1b Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Wed, 12 Feb 2020 14:42:51 +0100
Subject: Turn message part viewers into plugins

---
 plugins/viewhtml/viewer.go | 57 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 plugins/viewhtml/viewer.go

(limited to 'plugins/viewhtml/viewer.go')

diff --git a/plugins/viewhtml/viewer.go b/plugins/viewhtml/viewer.go
new file mode 100644
index 0000000..9734c37
--- /dev/null
+++ b/plugins/viewhtml/viewer.go
@@ -0,0 +1,57 @@
+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 = `
+<!-- allow-same-origin is required to resize the frame with its content -->
+<!-- allow-popups is required for target="_blank" links -->
+<iframe id="email-frame" srcdoc="{{.}}" sandbox="allow-same-origin allow-popups"></iframe>
+<script src="/plugins/viewhtml/assets/script.js"></script>
+<link rel="stylesheet" href="/plugins/viewhtml/assets/style.css">
+`
+
+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{})
+}
-- 
cgit v1.2.3-59-g8ed1b