110 lines
2.7 KiB
Go
110 lines
2.7 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 {
|
|
ID string
|
|
Sender string
|
|
Text string
|
|
Timestamp string
|
|
Date string
|
|
|
|
Buffer []string
|
|
SenderColor tcell.Color
|
|
}
|
|
|
|
func NewMessage(id, sender, text, timestamp, date string, senderColor tcell.Color) *Message {
|
|
return &Message{
|
|
ID: id,
|
|
Sender: sender,
|
|
Text: text,
|
|
Timestamp: timestamp,
|
|
Date: date,
|
|
SenderColor: senderColor,
|
|
}
|
|
}
|
|
|
|
var (
|
|
boundaryPattern = regexp.MustCompile("([[:punct:]]\\s*|\\s+)")
|
|
spacePattern = regexp.MustCompile(`\s+`)
|
|
)
|
|
|
|
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):]
|
|
}
|
|
}
|
|
}
|
|
|
|
func (message *Message) GetSender() string {
|
|
return message.Sender
|
|
}
|
|
|
|
func (message *Message) GetSenderColor() tcell.Color {
|
|
return message.SenderColor
|
|
}
|
|
|
|
func (message *Message) GetTimestamp() string {
|
|
return message.Timestamp
|
|
}
|
|
|
|
func (message *Message) GetDate() string {
|
|
return message.Date
|
|
}
|
|
|
|
func (message *Message) GetTextColor() tcell.Color {
|
|
return tcell.ColorDefault
|
|
}
|
|
|
|
func (message *Message) GetTimestampColor() tcell.Color {
|
|
return tcell.ColorDefault
|
|
}
|