summaryrefslogtreecommitdiffstats
path: root/vendor/layeh.com/gumble/gumbleutil/textmessage.go
blob: 1ccb62b4e052dc2fad66ffc201f29006c1b0fd00 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package gumbleutil

import (
	"bytes"
	"encoding/xml"
	"strings"

	"layeh.com/gumble/gumble"
)

// PlainText returns the Message string without HTML tags or entities.
func PlainText(tm *gumble.TextMessage) string {
	d := xml.NewDecoder(strings.NewReader(tm.Message))
	d.Strict = false
	d.AutoClose = xml.HTMLAutoClose
	d.Entity = xml.HTMLEntity

	var b bytes.Buffer
	newline := false
	for {
		t, _ := d.Token()
		if t == nil {
			break
		}
		switch node := t.(type) {
		case xml.CharData:
			if len(node) > 0 {
				b.Write(node)
				newline = false
			}
		case xml.StartElement:
			switch node.Name.Local {
			case "address", "article", "aside", "audio", "blockquote", "canvas", "dd", "div", "dl", "fieldset", "figcaption", "figure", "footer", "form", "h1", "h2", "h3", "h4", "h5", "h6", "header", "hgroup", "hr", "noscript", "ol", "output", "p", "pre", "section", "table", "tfoot", "ul", "video":
				if !newline {
					b.WriteByte('\n')
					newline = true
				}
			case "br":
				b.WriteByte('\n')
				newline = true
			}
		}
	}
	return b.String()
}