aboutsummaryrefslogtreecommitdiff
path: root/ui/messages/html
diff options
context:
space:
mode:
Diffstat (limited to 'ui/messages/html')
-rw-r--r--ui/messages/html/base.go95
-rw-r--r--ui/messages/html/blockquote.go14
-rw-r--r--ui/messages/html/codeblock.go14
-rw-r--r--ui/messages/html/container.go144
-rw-r--r--ui/messages/html/entity.go266
-rw-r--r--ui/messages/html/list.go14
-rw-r--r--ui/messages/html/parser.go97
-rw-r--r--ui/messages/html/text.go142
8 files changed, 462 insertions, 324 deletions
diff --git a/ui/messages/html/base.go b/ui/messages/html/base.go
new file mode 100644
index 0000000..8014dc0
--- /dev/null
+++ b/ui/messages/html/base.go
@@ -0,0 +1,95 @@
+// 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"
+
+ "maunium.net/go/mauview"
+ "maunium.net/go/tcell"
+)
+
+type BaseEntity struct {
+ // The HTML tag of this entity.
+ Tag string
+ // Style for this entity.
+ Style tcell.Style
+ // Whether or not this is a block-type entity.
+ Block bool
+ // Height to use for entity if both text and children are empty.
+ DefaultHeight int
+
+ prevWidth int
+ startX int
+ height int
+}
+
+// AdjustStyle changes the style of this text entity.
+func (be *BaseEntity) AdjustStyle(fn AdjustStyleFunc) Entity {
+ be.Style = fn(be.Style)
+ return be
+}
+
+// IsBlock returns whether or not this is a block-type entity.
+func (be *BaseEntity) IsBlock() bool {
+ return be.Block
+}
+
+// GetTag returns the HTML tag of this entity.
+func (be *BaseEntity) GetTag() string {
+ return be.Tag
+}
+
+// Height returns the render height of this entity.
+func (be *BaseEntity) Height() int {
+ return be.height
+}
+
+func (be *BaseEntity) getStartX() int {
+ return be.startX
+}
+
+// Clone creates a copy of this base entity.
+func (be *BaseEntity) Clone() Entity {
+ return &BaseEntity{
+ Tag: be.Tag,
+ Style: be.Style,
+ Block: be.Block,
+ DefaultHeight: be.DefaultHeight,
+ }
+}
+
+func (be *BaseEntity) PlainText() string {
+ return ""
+}
+
+// String returns a textual representation of this BaseEntity struct.
+func (be *BaseEntity) String() string {
+ return fmt.Sprintf(`&html.BaseEntity{Tag="%s", Style=%d, Block=%t, startX=%d, height=%d},`,
+ be.Tag, be.Style, be.Block, be.startX, be.height)
+}
+
+// CalculateBuffer prepares this entity for rendering with the given parameters.
+func (be *BaseEntity) CalculateBuffer(width, startX int, bare bool) int {
+ be.height = be.DefaultHeight
+ be.startX = startX
+ if be.Block {
+ be.startX = 0
+ }
+ return be.startX
+}
+
+func (be *BaseEntity) Draw(screen mauview.Screen) {}
diff --git a/ui/messages/html/blockquote.go b/ui/messages/html/blockquote.go
index 17799a2..288d86d 100644
--- a/ui/messages/html/blockquote.go
+++ b/ui/messages/html/blockquote.go
@@ -24,26 +24,28 @@ import (
)
type BlockquoteEntity struct {
- *BaseEntity
+ *ContainerEntity
}
const BlockQuoteChar = '>'
func NewBlockquoteEntity(children []Entity) *BlockquoteEntity {
- return &BlockquoteEntity{&BaseEntity{
- Tag: "blockquote",
+ return &BlockquoteEntity{&ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "blockquote",
+ Block: true,
+ },
Children: children,
- Block: true,
Indent: 2,
}}
}
func (be *BlockquoteEntity) Clone() Entity {
- return &BlockquoteEntity{BaseEntity: be.BaseEntity.Clone().(*BaseEntity)}
+ return &BlockquoteEntity{ContainerEntity: be.ContainerEntity.Clone().(*ContainerEntity)}
}
func (be *BlockquoteEntity) Draw(screen mauview.Screen) {
- be.BaseEntity.Draw(screen)
+ be.ContainerEntity.Draw(screen)
for y := 0; y < be.height; y++ {
screen.SetContent(0, y, BlockQuoteChar, nil, be.Style)
}
diff --git a/ui/messages/html/codeblock.go b/ui/messages/html/codeblock.go
index 4a0766c..7f6b462 100644
--- a/ui/messages/html/codeblock.go
+++ b/ui/messages/html/codeblock.go
@@ -22,15 +22,17 @@ import (
)
type CodeBlockEntity struct {
- *BaseEntity
+ *ContainerEntity
Background tcell.Style
}
func NewCodeBlockEntity(children []Entity, background tcell.Style) *CodeBlockEntity {
return &CodeBlockEntity{
- BaseEntity: &BaseEntity{
- Tag: "pre",
- Block: true,
+ ContainerEntity: &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "pre",
+ Block: true,
+ },
Children: children,
},
Background: background,
@@ -39,8 +41,8 @@ func NewCodeBlockEntity(children []Entity, background tcell.Style) *CodeBlockEnt
func (ce *CodeBlockEntity) Clone() Entity {
return &CodeBlockEntity{
- BaseEntity: ce.BaseEntity.Clone().(*BaseEntity),
- Background: ce.Background,
+ ContainerEntity: ce.ContainerEntity.Clone().(*ContainerEntity),
+ Background: ce.Background,
}
}
diff --git a/ui/messages/html/container.go b/ui/messages/html/container.go
new file mode 100644
index 0000000..b17bc1f
--- /dev/null
+++ b/ui/messages/html/container.go
@@ -0,0 +1,144 @@
+// 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"
+ "strings"
+
+ "maunium.net/go/mauview"
+)
+
+type ContainerEntity struct {
+ *BaseEntity
+
+ // The children of this container entity.
+ Children []Entity
+ // Number of cells to indent children.
+ Indent int
+}
+
+// PlainText returns the plaintext content in this entity and all its children.
+func (ce *ContainerEntity) PlainText() string {
+ if len(ce.Children) == 0 {
+ return ""
+ }
+ var buf strings.Builder
+ newlined := false
+ for _, child := range ce.Children {
+ text := child.PlainText()
+ if !strings.HasPrefix(text, "\n") && child.IsBlock() && !newlined {
+ buf.WriteRune('\n')
+ }
+ newlined = false
+ buf.WriteString(text)
+ if child.IsBlock() {
+ if !strings.HasSuffix(text, "\n") {
+ buf.WriteRune('\n')
+ }
+ newlined = true
+ }
+ }
+ return strings.TrimSpace(buf.String())
+}
+
+// AdjustStyle recursively changes the style of this entity and all its children.
+func (ce *ContainerEntity) AdjustStyle(fn AdjustStyleFunc) Entity {
+ for _, child := range ce.Children {
+ child.AdjustStyle(fn)
+ }
+ ce.Style = fn(ce.Style)
+ return ce
+}
+
+// clone creates a deep copy of this base entity.
+func (ce *ContainerEntity) Clone() Entity {
+ children := make([]Entity, len(ce.Children))
+ for i, child := range ce.Children {
+ children[i] = child.Clone()
+ }
+ return &ContainerEntity{
+ BaseEntity: ce.BaseEntity.Clone().(*BaseEntity),
+ Children: children,
+ Indent: ce.Indent,
+ }
+}
+
+// String returns a textual representation of this BaseEntity struct.
+func (ce *ContainerEntity) String() string {
+ if len(ce.Children) == 0 {
+ return fmt.Sprintf(`&html.ContainerEntity{Base=%s, Indent=%d, Children=[]}`, ce.BaseEntity, ce.Indent)
+ }
+ var buf strings.Builder
+ _, _ = fmt.Fprintf(&buf, `&html.ContainerEntity{Base=%s, Indent=%d, Children=[`, ce.BaseEntity, ce.Indent)
+ for _, child := range ce.Children {
+ buf.WriteString("\n ")
+ buf.WriteString(strings.Join(strings.Split(strings.TrimRight(child.String(), "\n"), "\n"), "\n "))
+ }
+ buf.WriteString("\n]}\n,")
+ return buf.String()
+}
+
+// Draw draws this entity onto the given mauview Screen.
+func (ce *ContainerEntity) Draw(screen mauview.Screen) {
+ if len(ce.Children) == 0 {
+ return
+ }
+ width, _ := screen.Size()
+ prevBreak := false
+ proxyScreen := &mauview.ProxyScreen{Parent: screen, OffsetX: ce.Indent, Width: width - ce.Indent, Style: ce.Style}
+ for i, entity := range ce.Children {
+ if i != 0 && entity.getStartX() == 0 {
+ proxyScreen.OffsetY++
+ }
+ proxyScreen.Height = entity.Height()
+ entity.Draw(proxyScreen)
+ proxyScreen.SetStyle(ce.Style)
+ proxyScreen.OffsetY += entity.Height() - 1
+ _, isBreak := entity.(*BreakEntity)
+ if prevBreak && isBreak {
+ proxyScreen.OffsetY++
+ }
+ prevBreak = isBreak
+ }
+}
+
+// CalculateBuffer prepares this entity and all its children for rendering with the given parameters
+func (ce *ContainerEntity) CalculateBuffer(width, startX int, bare bool) int {
+ ce.BaseEntity.CalculateBuffer(width, startX, bare)
+ if len(ce.Children) > 0 {
+ ce.height = 0
+ childStartX := ce.startX
+ prevBreak := false
+ for _, entity := range ce.Children {
+ if entity.IsBlock() || childStartX == 0 || ce.height == 0 {
+ ce.height++
+ }
+ childStartX = entity.CalculateBuffer(width-ce.Indent, childStartX, bare)
+ ce.height += entity.Height() - 1
+ _, isBreak := entity.(*BreakEntity)
+ if prevBreak && isBreak {
+ ce.height++
+ }
+ prevBreak = isBreak
+ }
+ if !ce.Block {
+ return childStartX
+ }
+ }
+ return ce.startX
+}
diff --git a/ui/messages/html/entity.go b/ui/messages/html/entity.go
index be58d61..0c7785d 100644
--- a/ui/messages/html/entity.go
+++ b/ui/messages/html/entity.go
@@ -17,13 +17,6 @@
package html
import (
- "fmt"
- "regexp"
- "strings"
-
- "github.com/mattn/go-runewidth"
-
- "maunium.net/go/gomuks/ui/widget"
"maunium.net/go/mauview"
"maunium.net/go/tcell"
)
@@ -54,262 +47,3 @@ type Entity interface {
getStartX() int
}
-
-type BaseEntity struct {
- // The HTML tag of this entity.
- Tag string
- // Text in this entity.
- Text string
- // Style for this entity.
- Style tcell.Style
- // Child entities.
- Children []Entity
- // Whether or not this is a block-type entity.
- Block bool
- // Number of cells to indent children.
- Indent int
-
- // Height to use for entity if both text and children are empty.
- DefaultHeight int
-
- buffer []string
- prevWidth int
- startX int
- height int
-}
-
-// NewTextEntity creates a new text-only Entity.
-func NewTextEntity(text string) *BaseEntity {
- return &BaseEntity{
- Tag: "text",
- Text: text,
- }
-}
-
-// AdjustStyle recursively changes the style of this entity and all its children.
-func (he *BaseEntity) AdjustStyle(fn AdjustStyleFunc) Entity {
- for _, child := range he.Children {
- child.AdjustStyle(fn)
- }
- he.Style = fn(he.Style)
- return he
-}
-
-// IsBlock returns whether or not this is a block-type entity.
-func (he *BaseEntity) IsBlock() bool {
- return he.Block
-}
-
-// GetTag returns the HTML tag of this entity.
-func (he *BaseEntity) GetTag() string {
- return he.Tag
-}
-
-// Height returns the render height of this entity.
-func (he *BaseEntity) Height() int {
- return he.height
-}
-
-func (he *BaseEntity) getStartX() int {
- return he.startX
-}
-
-// Clone creates a deep copy of this entity.
-func (he *BaseEntity) Clone() Entity {
- children := make([]Entity, len(he.Children))
- for i, child := range he.Children {
- children[i] = child.Clone()
- }
- return &BaseEntity{
- Tag: he.Tag,
- Text: he.Text,
- Style: he.Style,
- Children: children,
- Block: he.Block,
- Indent: he.Indent,
- DefaultHeight: he.DefaultHeight,
- }
-}
-
-// String returns a textual representation of this BaseEntity struct.
-func (he *BaseEntity) String() string {
- var buf strings.Builder
- buf.WriteString("&html.BaseEntity{\n")
- _, _ = fmt.Fprintf(&buf, ` Tag="%s", Style=%d, Block=%t, Indent=%d, startX=%d, height=%d,`,
- he.Tag, he.Style, he.Block, he.Indent, he.startX, he.height)
- buf.WriteRune('\n')
- _, _ = fmt.Fprintf(&buf, ` Buffer=["%s"]`, strings.Join(he.buffer, "\", \""))
- if len(he.Text) > 0 {
- buf.WriteString(",\n")
- _, _ = fmt.Fprintf(&buf, ` Text="%s"`, he.Text)
- }
- if len(he.Children) > 0 {
- buf.WriteString(",\n")
- buf.WriteString(" Children={")
- for _, child := range he.Children {
- buf.WriteString("\n ")
- buf.WriteString(strings.Join(strings.Split(strings.TrimRight(child.String(), "\n"), "\n"), "\n "))
- }
- buf.WriteString("\n },")
- }
- buf.WriteString("\n},\n")
- return buf.String()
-}
-
-// PlainText returns the plaintext content in this entity and all its children.
-func (he *BaseEntity) PlainText() string {
- if len(he.Children) == 0 {
- return he.Text
- }
- var buf strings.Builder
- buf.WriteString(he.Text)
- newlined := false
- for _, child := range he.Children {
- text := child.PlainText()
- if !strings.HasPrefix(text, "\n") && child.IsBlock() && !newlined {
- buf.WriteRune('\n')
- }
- newlined = false
- buf.WriteString(text)
- if child.IsBlock() {
- if !strings.HasSuffix(text, "\n") {
- buf.WriteRune('\n')
- }
- newlined = true
- }
- }
- return strings.TrimSpace(buf.String())
-}
-
-// Draw draws this entity onto the given mauview Screen.
-func (he *BaseEntity) Draw(screen mauview.Screen) {
- width, _ := screen.Size()
- if len(he.buffer) > 0 {
- x := he.startX
- for y, line := range he.buffer {
- widget.WriteLine(screen, mauview.AlignLeft, line, x, y, width, he.Style)
- x = 0
- }
- }
- if len(he.Children) > 0 {
- prevBreak := false
- proxyScreen := &mauview.ProxyScreen{Parent: screen, OffsetX: he.Indent, Width: width - he.Indent, Style: he.Style}
- for i, entity := range he.Children {
- if i != 0 && entity.getStartX() == 0 {
- proxyScreen.OffsetY++
- }
- proxyScreen.Height = entity.Height()
- entity.Draw(proxyScreen)
- proxyScreen.SetStyle(he.Style)
- proxyScreen.OffsetY += entity.Height() - 1
- _, isBreak := entity.(*BreakEntity)
- if prevBreak && isBreak {
- proxyScreen.OffsetY++
- }
- prevBreak = isBreak
- }
- }
-}
-
-// CalculateBuffer prepares this entity and all its children for rendering with the given parameters
-func (he *BaseEntity) CalculateBuffer(width, startX int, bare bool) int {
- he.startX = startX
- if he.Block {
- he.startX = 0
- }
- he.height = 0
- if len(he.Children) > 0 {
- childStartX := he.startX
- prevBreak := false
- for _, entity := range he.Children {
- if entity.IsBlock() || childStartX == 0 || he.height == 0 {
- he.height++
- }
- childStartX = entity.CalculateBuffer(width-he.Indent, childStartX, bare)
- he.height += entity.Height() - 1
- _, isBreak := entity.(*BreakEntity)
- if prevBreak && isBreak {
- he.height++
- }
- prevBreak = isBreak
- }
- if len(he.Text) == 0 && !he.Block {
- return childStartX
- }
- }
- if len(he.Text) > 0 {
- he.prevWidth = width
- if he.buffer == nil {
- he.buffer = []string{}
- }
- bufPtr := 0
- text := he.Text
- textStartX := he.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(he.buffer) {
- he.buffer[bufPtr] = ""
- } else {
- he.buffer = append(he.buffer, "")
- }
- bufPtr++
- textStartX = 0
- continue
- }
- if bufPtr < len(he.buffer) {
- he.buffer[bufPtr] = extract
- } else {
- he.buffer = append(he.buffer, extract)
- }
- bufPtr++
- text = text[len(extract):]
- if len(text) == 0 {
- he.buffer = he.buffer[:bufPtr]
- he.height += len(he.buffer)
- // This entity is over, return the startX for the next entity
- if he.Block {
- // ...except if it's a block entity
- return 0
- }
- return textStartX + runewidth.StringWidth(extract)
- }
- textStartX = 0
- }
- }
- if len(he.Text) == 0 && len(he.Children) == 0 {
- he.height = he.DefaultHeight
- }
- return he.startX
-}
-
-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] == ' '
-}
diff --git a/ui/messages/html/list.go b/ui/messages/html/list.go
index c611e66..22b969a 100644
--- a/ui/messages/html/list.go
+++ b/ui/messages/html/list.go
@@ -26,7 +26,7 @@ import (
)
type ListEntity struct {
- *BaseEntity
+ *ContainerEntity
Ordered bool
Start int
}
@@ -40,11 +40,13 @@ func digits(num int) int {
func NewListEntity(ordered bool, start int, children []Entity) *ListEntity {
entity := &ListEntity{
- BaseEntity: &BaseEntity{
- Tag: "ul",
+ ContainerEntity: &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "ul",
+ Block: true,
+ },
+ Indent: 2,
Children: children,
- Block: true,
- Indent: 2,
},
Ordered: ordered,
Start: start,
@@ -58,7 +60,7 @@ func NewListEntity(ordered bool, start int, children []Entity) *ListEntity {
func (le *ListEntity) Clone() Entity {
return &ListEntity{
- BaseEntity: le.BaseEntity.Clone().(*BaseEntity),
+ ContainerEntity: le.ContainerEntity.Clone().(*ContainerEntity),
Ordered: le.Ordered,
Start: le.Start,
}
diff --git a/ui/messages/html/parser.go b/ui/messages/html/parser.go
index 9e08ab7..51f5cb0 100644
--- a/ui/messages/html/parser.go
+++ b/ui/messages/html/parser.go
@@ -102,8 +102,10 @@ func (parser *htmlParser) listToEntity(node *html.Node) Entity {
}
func (parser *htmlParser) basicFormatToEntity(node *html.Node) Entity {
- entity := &BaseEntity{
- Tag: node.Data,
+ entity := &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: node.Data,
+ },
Children: parser.nodeToEntities(node.FirstChild),
}
switch node.Data {
@@ -151,12 +153,14 @@ func (parser *htmlParser) parseColor(node *html.Node, mainName, altName string)
}
func (parser *htmlParser) headerToEntity(node *html.Node) Entity {
- length := int(node.Data[1] - '0')
- prefix := strings.Repeat("#", length) + " "
- return (&BaseEntity{
- Tag: node.Data,
- Text: prefix,
- Children: parser.nodeToEntities(node.FirstChild),
+ return (&ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: node.Data,
+ },
+ Children: append(
+ []Entity{NewTextEntity(strings.Repeat("#", int(node.Data[1]-'0')) + " ")},
+ parser.nodeToEntities(node.FirstChild)...
+ ),
}).AdjustStyle(AdjustStyleBold)
}
@@ -164,29 +168,36 @@ func (parser *htmlParser) blockquoteToEntity(node *html.Node) Entity {
return NewBlockquoteEntity(parser.nodeToEntities(node.FirstChild))
}
-func (parser *htmlParser) linkToEntity(node *html.Node) Entity {
- entity := &BaseEntity{
- Tag: "a",
+func (parser *htmlParser) linkToEntity(node *html.Node) (entity Entity) {
+ entity = &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "a",
+ },
Children: parser.nodeToEntities(node.FirstChild),
}
href := parser.getAttribute(node, "href")
if len(href) == 0 {
- return entity
+ return
}
match := matrixToURL.FindStringSubmatch(href)
if len(match) == 2 {
- entity.Children = nil
pillTarget := match[1]
- entity.Text = pillTarget
+ textEntity := &TextEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "a",
+ },
+ Text: pillTarget,
+ }
if pillTarget[0] == '@' {
if member := parser.room.GetMember(pillTarget); member != nil {
- entity.Text = member.Displayname
- entity.Style = entity.Style.Foreground(widget.GetHashColor(pillTarget))
+ textEntity.Text = member.Displayname
+ textEntity.Style = textEntity.Style.Foreground(widget.GetHashColor(pillTarget))
}
}
+ entity = textEntity
}
// TODO add click action and underline on hover for links
- return entity
+ return
}
func (parser *htmlParser) imageToEntity(node *html.Node) Entity {
@@ -197,8 +208,10 @@ func (parser *htmlParser) imageToEntity(node *html.Node) Entity {
alt = "[inline image]"
}
}
- entity := &BaseEntity{
- Tag: "img",
+ entity := &TextEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "img",
+ },
Text: alt,
}
// TODO add click action and underline on hover for inline images
@@ -239,12 +252,13 @@ func (parser *htmlParser) syntaxHighlight(text, language string) Entity {
if token.Value == "\n" {
children[i] = NewBreakEntity()
} else {
- children[i] = &BaseEntity{
- Tag: token.Type.String(),
- Text: token.Value,
- Style: styleEntryToStyle(style.Get(token.Type)),
-
- DefaultHeight: 1,
+ children[i] = &TextEntity{
+ BaseEntity: &BaseEntity{
+ Tag: token.Type.String(),
+ Style: styleEntryToStyle(style.Get(token.Type)),
+ DefaultHeight: 1,
+ },
+ Text: token.Value,
}
}
}
@@ -265,7 +279,7 @@ func (parser *htmlParser) codeblockToEntity(node *html.Node) Entity {
}
}
parser.keepLinebreak = true
- text := (&BaseEntity{
+ text := (&ContainerEntity{
Children: parser.nodeToEntities(node.FirstChild),
}).PlainText()
parser.keepLinebreak = false
@@ -295,10 +309,12 @@ func (parser *htmlParser) tagNodeToEntity(node *html.Node) Entity {
case "mx-reply":
return nil
default:
- return &BaseEntity{
- Tag: node.Data,
+ return &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: node.Data,
+ Block: parser.isBlockTag(node.Data),
+ },
Children: parser.nodeToEntities(node.FirstChild),
- Block: parser.isBlockTag(node.Data),
}
}
}
@@ -312,20 +328,19 @@ func (parser *htmlParser) singleNodeToEntity(node *html.Node) Entity {
if len(node.Data) == 0 {
return nil
}
- return &BaseEntity{
- Tag: "text",
- Text: node.Data,
- }
+ return NewTextEntity(node.Data)
case html.ElementNode:
return parser.tagNodeToEntity(node)
case html.DocumentNode:
if node.FirstChild.Data == "html" && node.FirstChild.NextSibling == nil {
return parser.singleNodeToEntity(node.FirstChild)
}
- return &BaseEntity{
- Tag: "html",
+ return &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "html",
+ Block: true,
+ },
Children: parser.nodeToEntities(node.FirstChild),
- Block: true,
}
default:
return nil
@@ -376,10 +391,10 @@ func Parse(room *rooms.Room, evt *mautrix.Event, senderDisplayname string) Entit
parser := htmlParser{room: room}
root := parser.Parse(htmlData)
- beRoot := root.(*BaseEntity)
+ beRoot := root.(*ContainerEntity)
beRoot.Block = false
if len(beRoot.Children) > 0 {
- beChild, ok := beRoot.Children[0].(*BaseEntity)
+ beChild, ok := beRoot.Children[0].(*ContainerEntity)
if ok && beChild.Tag == "p" {
// Hacky fix for m.emote
beChild.Block = false
@@ -387,8 +402,10 @@ func Parse(room *rooms.Room, evt *mautrix.Event, senderDisplayname string) Entit
}
if evt.Content.MsgType == mautrix.MsgEmote {
- root = &BaseEntity{
- Tag: "emote",
+ root = &ContainerEntity{
+ BaseEntity: &BaseEntity{
+ Tag: "emote",
+ },
Children: []Entity{
NewTextEntity("* "),
NewTextEntity(senderDisplayname).AdjustStyle(AdjustStyleTextColor(widget.GetHashColor(evt.Sender))),
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] == ' '
+}