Split container/text HTML entities and add support for <hr>

1 files changed, 142 insertions, 0 deletions

diff --git a/ui/messages/html/text.go b/ui/messages/html/text.go
new file mode 100644
index 0000000..b168994
--- /dev/null
+++ b/ui/messages/html/text.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 html
+
+import (
+	"fmt"
+	"regexp"
+
+	"github.com/mattn/go-runewidth"
+
+	"maunium.net/go/gomuks/ui/widget"
+	"maunium.net/go/mauview"
+)
+
+type TextEntity struct {
+	*BaseEntity
+	// Text in this entity.
+	Text string
+
+	buffer []string
+}
+
+// NewTextEntity creates a new text-only Entity.
+func NewTextEntity(text string) *TextEntity {
+	return &TextEntity{
+		BaseEntity: &BaseEntity{
+			Tag: "text",
+		},
+		Text: text,
+	}
+}
+
+func (te *TextEntity) Clone() Entity {
+	return &TextEntity{
+		BaseEntity: te.BaseEntity.Clone().(*BaseEntity),
+		Text:       te.Text,
+	}
+}
+
+func (te *TextEntity) PlainText() string {
+	return te.Text
+}
+
+func (te *TextEntity) String() string {
+	return fmt.Sprintf("&html.TextEntity{Text=%s, Base=%s},\n", te.Text, te.BaseEntity)
+}
+
+func (te *TextEntity) Draw(screen mauview.Screen) {
+	width, _ := screen.Size()
+	x := te.startX
+	for y, line := range te.buffer {
+		widget.WriteLine(screen, mauview.AlignLeft, line, x, y, width, te.Style)
+		x = 0
+	}
+}
+
+func (te *TextEntity) CalculateBuffer(width, startX int, bare bool) int {
+	te.BaseEntity.CalculateBuffer(width, startX, bare)
+	te.prevWidth = width
+	if te.buffer == nil {
+		te.buffer = []string{}
+	}
+	bufPtr := 0
+	text := te.Text
+	textStartX := te.startX
+	for {
+		// TODO add option no wrap and character wrap options
+		extract := runewidth.Truncate(text, width-textStartX, "")
+		extract, wordWrapped := trim(extract, text, bare)
+		if !wordWrapped && textStartX > 0 {
+			if bufPtr < len(te.buffer) {
+				te.buffer[bufPtr] = ""
+			} else {
+				te.buffer = append(te.buffer, "")
+			}
+			bufPtr++
+			textStartX = 0
+			continue
+		}
+		if bufPtr < len(te.buffer) {
+			te.buffer[bufPtr] = extract
+		} else {
+			te.buffer = append(te.buffer, extract)
+		}
+		bufPtr++
+		text = text[len(extract):]
+		if len(text) == 0 {
+			te.buffer = te.buffer[:bufPtr]
+			te.height += len(te.buffer)
+			// This entity is over, return the startX for the next entity
+			if te.Block {
+				// ...except if it's a block entity
+				return 0
+			}
+			return textStartX + runewidth.StringWidth(extract)
+		}
+		textStartX = 0
+	}
+}
+
+var (
+	boundaryPattern     = regexp.MustCompile(`([[:punct:]]\s*|\s+)`)
+	bareBoundaryPattern = regexp.MustCompile(`(\s+)`)
+	spacePattern        = regexp.MustCompile(`\s+`)
+)
+
+func trim(extract, full string, bare bool) (string, bool) {
+	if len(extract) == len(full) {
+		return extract, true
+	}
+	if spaces := spacePattern.FindStringIndex(full[len(extract):]); spaces != nil && spaces[0] == 0 {
+		extract = full[:len(extract)+spaces[1]]
+	}
+	regex := boundaryPattern
+	if bare {
+		regex = bareBoundaryPattern
+	}
+	matches := regex.FindAllStringIndex(extract, -1)
+	if len(matches) > 0 {
+		if match := matches[len(matches)-1]; len(match) >= 2 {
+			if until := match[1]; until < len(extract) {
+				extract = extract[:until]
+				return extract, true
+			}
+		}
+	}
+	return extract, len(extract) > 0 && extract[len(extract)-1] == ' '
+}