100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
// gomuks - A terminal Matrix client written in Go.
|
|
// Copyright (C) 2018 Tulir Asokan
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU 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 General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package types
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/gdamore/tcell"
|
|
"github.com/mattn/go-runewidth"
|
|
)
|
|
|
|
type Message struct {
|
|
BasicMeta
|
|
ID string
|
|
Text string
|
|
Buffer []string
|
|
prevBufferWidth int
|
|
}
|
|
|
|
func NewMessage(id, sender, text, timestamp, date string, senderColor tcell.Color) *Message {
|
|
return &Message{
|
|
BasicMeta: BasicMeta{
|
|
Sender: sender,
|
|
Timestamp: timestamp,
|
|
Date: date,
|
|
SenderColor: senderColor,
|
|
TextColor: tcell.ColorDefault,
|
|
TimestampColor: tcell.ColorDefault,
|
|
},
|
|
Text: text,
|
|
ID: id,
|
|
prevBufferWidth: 0,
|
|
}
|
|
}
|
|
|
|
var (
|
|
boundaryPattern = regexp.MustCompile("([[:punct:]]\\s*|\\s+)")
|
|
spacePattern = regexp.MustCompile(`\s+`)
|
|
)
|
|
|
|
func (message *Message) CopyTo(to *Message) {
|
|
to.BasicMeta = message.BasicMeta
|
|
to.ID = message.ID
|
|
to.Text = message.Text
|
|
to.RecalculateBuffer()
|
|
}
|
|
|
|
func (message *Message) CalculateBuffer(width int) {
|
|
if width < 1 {
|
|
return
|
|
}
|
|
message.Buffer = []string{}
|
|
forcedLinebreaks := strings.Split(message.Text, "\n")
|
|
newlines := 0
|
|
for _, str := range forcedLinebreaks {
|
|
if len(str) == 0 && newlines < 1 {
|
|
message.Buffer = append(message.Buffer, "")
|
|
newlines++
|
|
} else {
|
|
newlines = 0
|
|
}
|
|
// From tview/textview.go#reindexBuffer()
|
|
for len(str) > 0 {
|
|
extract := runewidth.Truncate(str, width, "")
|
|
if len(extract) < len(str) {
|
|
if spaces := spacePattern.FindStringIndex(str[len(extract):]); spaces != nil && spaces[0] == 0 {
|
|
extract = str[:len(extract)+spaces[1]]
|
|
}
|
|
|
|
matches := boundaryPattern.FindAllStringIndex(extract, -1)
|
|
if len(matches) > 0 {
|
|
extract = extract[:matches[len(matches)-1][1]]
|
|
}
|
|
}
|
|
message.Buffer = append(message.Buffer, extract)
|
|
str = str[len(extract):]
|
|
}
|
|
}
|
|
message.prevBufferWidth = width
|
|
}
|
|
|
|
func (message *Message) RecalculateBuffer() {
|
|
message.CalculateBuffer(message.prevBufferWidth)
|
|
}
|