tools/mailer/html2text.go - master - githubmirror/pocketbase

tools/mailer/html2text.go
118 строк3 KB
Gani Georgiev
updated API preview examples
26 ноя 2023, 15:59
26 ноя 2023, 15:593f2e38c
package mailer

import (
	"regexp"
	"strings"

	"github.com/pocketbase/pocketbase/tools/list"
	"golang.org/x/net/html"
)

var whitespaceRegex = regexp.MustCompile(`\s+`)

var tagsToSkip = []string{
	"style", "script", "iframe", "applet", "object", "svg", "img",
	"button", "form", "textarea", "input", "select", "option", "template",
}

var inlineTags = []string{
	"a", "span", "small", "strike", "strong",
	"sub", "sup", "em", "b", "u", "i",
}

// Very rudimentary auto HTML to Text mail body converter.
//
// Caveats:
// - This method doesn't check for correctness of the HTML document.
// - Links will be converted to "[text](url)" format.
// - List items (<li>) are prefixed with "- ".
// - Indentation is stripped (both tabs and spaces).
// - Trailing spaces are preserved.
// - Multiple consequence newlines are collapsed as one unless multiple <br> tags are used.
func html2Text(htmlDocument string) (string, error) {
	doc, err := html.Parse(strings.NewReader(htmlDocument))
	if err != nil {
		return "", err
	}

	var builder strings.Builder
	var canAddNewLine bool

	// see https://pkg.go.dev/golang.org/x/net/html#Parse
	var f func(*html.Node, *strings.Builder)
	f = func(n *html.Node, activeBuilder *strings.Builder) {
		isLink := n.Type == html.ElementNode && n.Data == "a"

		if isLink {
			var linkBuilder strings.Builder
			activeBuilder = &linkBuilder
		} else if activeBuilder == nil {
			activeBuilder = &builder
		}

		switch n.Type {
		case html.TextNode:
			txt := whitespaceRegex.ReplaceAllString(n.Data, " ")

			// the prev node has new line so it is safe to trim the indentation
			if !canAddNewLine {
				txt = strings.TrimLeft(txt, " ")
			}

			if txt != "" {
				activeBuilder.WriteString(txt)
				canAddNewLine = true
			}
		case html.ElementNode:
			if n.Data == "br" {
				// always write new lines when <br> tag is used
				activeBuilder.WriteString("\r\n")
				canAddNewLine = false
			} else if canAddNewLine && !list.ExistInSlice(n.Data, inlineTags) {
				activeBuilder.WriteString("\r\n")
				canAddNewLine = false
			}

			// prefix list items with dash
			if n.Data == "li" {
				activeBuilder.WriteString("- ")
			}
		}

		for c := n.FirstChild; c != nil; c = c.NextSibling {
			if c.Type != html.ElementNode || !list.ExistInSlice(c.Data, tagsToSkip) {
				f(c, activeBuilder)
			}
		}

		// format links as [label](href)
		if isLink {
			linkTxt := strings.TrimSpace(activeBuilder.String())
			if linkTxt == "" {
				linkTxt = "LINK"
			}

			builder.WriteString("[")
			builder.WriteString(linkTxt)
			builder.WriteString("]")

			// link href attr extraction
			for _, a := range n.Attr {
				if a.Key == "href" {
					if a.Val != "" {
						builder.WriteString("(")
						builder.WriteString(a.Val)
						builder.WriteString(")")
					}
					break
				}
			}

			activeBuilder.Reset()
		}
	}

	f(doc, &builder)

	return strings.TrimSpace(builder.String()), nil
}