aboutsummaryrefslogtreecommitdiff
path: root/ui/messages
diff options
context:
space:
mode:
authorSimon Magnin-Feysot <simon.magninfeysot@iguanesolutions.com>2020-04-07 18:06:20 +0200
committerSimon Magnin-Feysot <simon.magninfeysot@iguanesolutions.com>2020-04-07 18:20:06 +0200
commit8b68140dc9e4bc6d51f49a4a38a09e1a0b0ffd67 (patch)
tree6c4c0559986a96cd0807e1fe5d02e2b4b1c1871a /ui/messages
parentf0e5e0022f565034393a8d5f0afcbd21470284c3 (diff)
refactor to allow every file message to be downloaded
Diffstat (limited to 'ui/messages')
-rw-r--r--ui/messages/filemessage.go (renamed from ui/messages/imagemessage.go)70
-rw-r--r--ui/messages/parser.go7
-rw-r--r--ui/messages/videomessage.go92
3 files changed, 35 insertions, 134 deletions
diff --git a/ui/messages/imagemessage.go b/ui/messages/filemessage.go
index 5e72c88..3878b56 100644
--- a/ui/messages/imagemessage.go
+++ b/ui/messages/filemessage.go
@@ -33,7 +33,7 @@ import (
"maunium.net/go/gomuks/ui/messages/tstring"
)
-type ImageMessage struct {
+type FileMessage struct {
Body string
Homeserver string
FileID string
@@ -43,30 +43,30 @@ type ImageMessage struct {
matrix ifc.MatrixContainer
}
-// NewImageMessage creates a new ImageMessage object with the provided values and the default state.
-func NewImageMessage(matrix ifc.MatrixContainer, evt *event.Event, displayname string, body, homeserver, fileID string, data []byte) *UIMessage {
- return newUIMessage(evt, displayname, &ImageMessage{
- Body: body,
+// 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,
+ FileID: fileID,
+ data: data,
+ matrix: matrix,
})
}
-func (msg *ImageMessage) Clone() MessageRenderer {
+func (msg *FileMessage) Clone() MessageRenderer {
data := make([]byte, len(msg.data))
copy(data, msg.data)
- return &ImageMessage{
- Body: msg.Body,
- Homeserver: msg.Homeserver,
- FileID: msg.FileID,
- data: data,
- matrix: msg.matrix,
+ return &FileMessage{
+ Body: msg.Body,
+ Homeserver: msg.Homeserver,
+ FileID: msg.FileID,
+ data: data,
+ matrix: msg.matrix,
}
}
-func (msg *ImageMessage) RegisterMatrix(matrix ifc.MatrixContainer) {
+func (msg *FileMessage) RegisterMatrix(matrix ifc.MatrixContainer) {
msg.matrix = matrix
if len(msg.data) == 0 {
@@ -74,72 +74,68 @@ func (msg *ImageMessage) RegisterMatrix(matrix ifc.MatrixContainer) {
}
}
-func (msg *ImageMessage) NotificationContent() string {
- return "Sent an image"
+func (msg *FileMessage) NotificationContent() string {
+ return "Sent a file"
}
-func (msg *ImageMessage) PlainText() string {
+func (msg *FileMessage) PlainText() string {
return fmt.Sprintf("%s: %s", msg.Body, msg.matrix.GetDownloadURL(msg.Homeserver, msg.FileID))
}
-func (msg *ImageMessage) String() string {
- return fmt.Sprintf(`&messages.ImageMessage{Body="%s", Homeserver="%s", FileID="%s"}`, msg.Body, 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 *ImageMessage) updateData() {
+func (msg *FileMessage) updateData() {
defer debug.Recover()
- debug.Print("Loading image:", msg.Homeserver, msg.FileID)
+ 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 image %s/%s: %v", msg.Homeserver, msg.FileID, err)
+ debug.Printf("Failed to download file %s/%s: %v", msg.Homeserver, msg.FileID, err)
return
}
- debug.Print("Image", msg.Homeserver, msg.FileID, "loaded.")
+ debug.Print("File", msg.Homeserver, msg.FileID, "loaded.")
msg.data = data
}
-func (msg *ImageMessage) Path() string {
+func (msg *FileMessage) Path() string {
return msg.matrix.GetCachePath(msg.Homeserver, msg.FileID)
}
-// CalculateBuffer generates the internal buffer for this message that consists
-// of the text of this message split into lines at most as wide as the width
-// parameter. If the message width is larger than the width of the buffer
-// the message is scaled to one third the buffer width.
-func (msg *ImageMessage) CalculateBuffer(prefs config.UserPreferences, width int, uiMsg *UIMessage) {
+func (msg *FileMessage) CalculateBuffer(prefs config.UserPreferences, width int, uiMsg *UIMessage) {
if width < 2 {
return
}
- if prefs.BareMessageView || prefs.DisableImages {
+ 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("Image could not be decoded:", err)
+ debug.Print("File could not be decoded:", err)
}
imgWidth := img.Width
if img.Width > width {
imgWidth = width / 3
}
- ansImage, err := ansimage.NewScaledFromReader(bytes.NewReader(msg.data), 0, imgWidth, color.Black)
+ 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 = ansImage.Render()
+ msg.buffer = ansFile.Render()
}
-func (msg *ImageMessage) Height() int {
+func (msg *FileMessage) Height() int {
return len(msg.buffer)
}
-func (msg *ImageMessage) Draw(screen mauview.Screen) {
+func (msg *FileMessage) Draw(screen mauview.Screen) {
for y, line := range msg.buffer {
line.Draw(screen, 0, y)
}
diff --git a/ui/messages/parser.go b/ui/messages/parser.go
index ce5827a..1c145e5 100644
--- a/ui/messages/parser.go
+++ b/ui/messages/parser.go
@@ -138,15 +138,12 @@ func ParseMessage(matrix ifc.MatrixContainer, room *rooms.Room, evt *event.Event
}
evt.Content.Body = strings.Replace(evt.Content.Body, "\t", " ", -1)
return NewTextMessage(evt, displayname, evt.Content.Body)
- case "m.image":
+ case "m.file", "m.video", "m.audio", "m.image":
data, hs, id, err := matrix.Download(evt.Content.URL)
if err != nil {
debug.Printf("Failed to download %s: %v", evt.Content.URL, err)
}
- return NewImageMessage(matrix, evt, displayname, evt.Content.Body, hs, id, data)
- case "m.video":
- _, hs, id, _ := matrix.Download(evt.Content.URL)
- return NewVideoMessage(matrix, evt, displayname, evt.Content.Body, hs, id)
+ return NewFileMessage(matrix, evt, displayname, evt.Content.Body, hs, id, data)
}
return nil
}
diff --git a/ui/messages/videomessage.go b/ui/messages/videomessage.go
deleted file mode 100644
index d5b87f8..0000000
--- a/ui/messages/videomessage.go
+++ /dev/null
@@ -1,92 +0,0 @@
-// 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 (
- "fmt"
-
- "maunium.net/go/gomuks/matrix/event"
- "maunium.net/go/mauview"
-
- "maunium.net/go/gomuks/config"
- "maunium.net/go/gomuks/interface"
- "maunium.net/go/gomuks/ui/messages/tstring"
-)
-
-type VideoMessage struct {
- Body string
- Homeserver string
- FileID string
- buffer []tstring.TString
-
- matrix ifc.MatrixContainer
-}
-
-// NewVideoMessage creates a new VideoMessage object with the provided values and the default state.
-func NewVideoMessage(matrix ifc.MatrixContainer, evt *event.Event, displayname string, body, homeserver, fileID string) *UIMessage {
- return newUIMessage(evt, displayname, &VideoMessage{
- Body: body,
- Homeserver: homeserver,
- FileID: fileID,
- matrix: matrix,
- })
-}
-
-func (msg *VideoMessage) Clone() MessageRenderer {
- return &VideoMessage{
- Body: msg.Body,
- Homeserver: msg.Homeserver,
- FileID: msg.FileID,
- matrix: msg.matrix,
- }
-}
-
-func (msg *VideoMessage) NotificationContent() string {
- return "Sent a video"
-}
-
-func (msg *VideoMessage) PlainText() string {
- return fmt.Sprintf("%s: %s", msg.Body, msg.matrix.GetDownloadURL(msg.Homeserver, msg.FileID))
-}
-
-func (msg *VideoMessage) String() string {
- return fmt.Sprintf(`&messages.VideoMessage{Body="%s", Homeserver="%s", FileID="%s"}`, msg.Body, msg.Homeserver, msg.FileID)
-}
-
-func (msg *VideoMessage) Path() string {
- return msg.matrix.GetCachePath(msg.Homeserver, msg.FileID)
-}
-
-func (msg *VideoMessage) RegisterMatrix(matrix ifc.MatrixContainer) {
- msg.matrix = matrix
-}
-
-// Print only Plain Text
-func (msg *VideoMessage) CalculateBuffer(prefs config.UserPreferences, width int, uiMsg *UIMessage) {
- msg.buffer = calculateBufferWithText(prefs, tstring.NewTString(msg.PlainText()), width, uiMsg)
- return
-}
-
-func (msg *VideoMessage) Height() int {
- return len(msg.buffer)
-}
-
-func (msg *VideoMessage) Draw(screen mauview.Screen) {
- for y, line := range msg.buffer {
- line.Draw(screen, 0, y)
- }
-}