1 files changed, 142 insertions, 0 deletions
diff --git a/ui/messages/filemessage.go b/ui/messages/filemessage.go
new file mode 100644
index 0000000..3878b56
--- /dev/null
+++ b/ui/messages/filemessage.go
@@ -0,0 +1,142 @@
+// gomuks - A terminal Matrix client written in Go.
+// Copyright (C) 2019 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package messages
+
+import (
+	"bytes"
+	"fmt"
+	"image"
+	"image/color"
+
+	"maunium.net/go/gomuks/matrix/event"
+	"maunium.net/go/mauview"
+	"maunium.net/go/tcell"
+
+	"maunium.net/go/gomuks/config"
+	"maunium.net/go/gomuks/debug"
+	"maunium.net/go/gomuks/interface"
+	"maunium.net/go/gomuks/lib/ansimage"
+	"maunium.net/go/gomuks/ui/messages/tstring"
+)
+
+type FileMessage struct {
+	Body       string
+	Homeserver string
+	FileID     string
+	data       []byte
+	buffer     []tstring.TString
+
+	matrix ifc.MatrixContainer
+}
+
+// NewFileMessage creates a new FileMessage object with the provided values and the default state.
+func NewFileMessage(matrix ifc.MatrixContainer, evt *event.Event, displayname string, body, homeserver, fileID string, data []byte) *UIMessage {
+	return newUIMessage(evt, displayname, &FileMessage{
+		Body:       body,
+		Homeserver: homeserver,
+		FileID:     fileID,
+		data:       data,
+		matrix:     matrix,
+	})
+}
+
+func (msg *FileMessage) Clone() MessageRenderer {
+	data := make([]byte, len(msg.data))
+	copy(data, msg.data)
+	return &FileMessage{
+		Body:       msg.Body,
+		Homeserver: msg.Homeserver,
+		FileID:     msg.FileID,
+		data:       data,
+		matrix:     msg.matrix,
+	}
+}
+
+func (msg *FileMessage) RegisterMatrix(matrix ifc.MatrixContainer) {
+	msg.matrix = matrix
+
+	if len(msg.data) == 0 {
+		go msg.updateData()
+	}
+}
+
+func (msg *FileMessage) NotificationContent() string {
+	return "Sent a file"
+}
+
+func (msg *FileMessage) PlainText() string {
+	return fmt.Sprintf("%s: %s", msg.Body, msg.matrix.GetDownloadURL(msg.Homeserver, msg.FileID))
+}
+
+func (msg *FileMessage) String() string {
+	return fmt.Sprintf(`&messages.FileMessage{Body="%s", Homeserver="%s", FileID="%s"}`, msg.Body, msg.Homeserver, msg.FileID)
+}
+
+func (msg *FileMessage) updateData() {
+	defer debug.Recover()
+	debug.Print("Loading file:", msg.Homeserver, msg.FileID)
+	data, _, _, err := msg.matrix.Download(fmt.Sprintf("mxc://%s/%s", msg.Homeserver, msg.FileID))
+	if err != nil {
+		debug.Printf("Failed to download file %s/%s: %v", msg.Homeserver, msg.FileID, err)
+		return
+	}
+	debug.Print("File", msg.Homeserver, msg.FileID, "loaded.")
+	msg.data = data
+}
+
+func (msg *FileMessage) Path() string {
+	return msg.matrix.GetCachePath(msg.Homeserver, msg.FileID)
+}
+
+func (msg *FileMessage) CalculateBuffer(prefs config.UserPreferences, width int, uiMsg *UIMessage) {
+	if width < 2 {
+		return
+	}
+
+	if prefs.BareMessageView || prefs.DisableImages || uiMsg.Type != "m.image" {
+		msg.buffer = calculateBufferWithText(prefs, tstring.NewTString(msg.PlainText()), width, uiMsg)
+		return
+	}
+
+	img, _, err := image.DecodeConfig(bytes.NewReader(msg.data))
+	if err != nil {
+		debug.Print("File could not be decoded:", err)
+	}
+	imgWidth := img.Width
+	if img.Width > width {
+		imgWidth = width / 3
+	}
+
+	ansFile, err := ansimage.NewScaledFromReader(bytes.NewReader(msg.data), 0, imgWidth, color.Black)
+	if err != nil {
+		msg.buffer = []tstring.TString{tstring.NewColorTString("Failed to display image", tcell.ColorRed)}
+		debug.Print("Failed to display image:", err)
+		return
+	}
+
+	msg.buffer = ansFile.Render()
+}
+
+func (msg *FileMessage) Height() int {
+	return len(msg.buffer)
+}
+
+func (msg *FileMessage) Draw(screen mauview.Screen) {
+	for y, line := range msg.buffer {
+		line.Draw(screen, 0, y)
+	}
+}