87 lines
1.7 KiB
Go
87 lines
1.7 KiB
Go
package digest
|
|
|
|
import (
|
|
"github.com/fthvgb1/wp-go/helper/html"
|
|
"github.com/fthvgb1/wp-go/helper/slice"
|
|
str "github.com/fthvgb1/wp-go/helper/strings"
|
|
"regexp"
|
|
"strings"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
var quto = regexp.MustCompile(`" *|& *|< *|> ?| *`)
|
|
|
|
func StripTags(content, allowTag string) string {
|
|
content = strings.Trim(content, " \t\n\r\000\x0B")
|
|
content = strings.Replace(content, "]]>", "]]>", -1)
|
|
content = html.StripTagsX(content, allowTag)
|
|
return content
|
|
}
|
|
|
|
func Html(content string, limit int, m map[string]int) (string, string) {
|
|
closeTag := ""
|
|
length := utf8.RuneCountInString(content) + 1
|
|
if length <= limit {
|
|
return content, ""
|
|
}
|
|
index := quto.FindAllStringIndex(content, -1)
|
|
var runeIndex [][]int
|
|
if len(index) > 0 {
|
|
runeIndex = slice.Map(index, func(t []int) []int {
|
|
return slice.Map(t, func(i int) int {
|
|
return utf8.RuneCountInString(content[:i])
|
|
})
|
|
})
|
|
}
|
|
end := 0
|
|
ru := []rune(content)
|
|
tagIn := false
|
|
total := len(ru)
|
|
l, r := '<', '>'
|
|
i := -1
|
|
var tag []rune
|
|
for {
|
|
i++
|
|
for len(runeIndex) > 0 && i >= runeIndex[0][0] {
|
|
ints := runeIndex[0]
|
|
if ints[0] <= i {
|
|
i = ints[1]
|
|
runeIndex = runeIndex[1:]
|
|
end++
|
|
continue
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
|
|
if end >= limit || i >= total {
|
|
break
|
|
}
|
|
if ru[i] == l {
|
|
tagIn = true
|
|
continue
|
|
} else if ru[i] == r {
|
|
tagIn = false
|
|
if len(m) > 0 {
|
|
tags := str.Join("<", strings.Split(string(tag), " ")[0], ">")
|
|
tag = tag[:0]
|
|
if n, ok := m[tags]; ok && n > 0 {
|
|
end += n
|
|
}
|
|
}
|
|
continue
|
|
}
|
|
if tagIn == false {
|
|
end++
|
|
} else if len(m) > 0 {
|
|
tag = append(tag, ru[i])
|
|
}
|
|
}
|
|
if i > total {
|
|
i = total
|
|
}
|
|
content = string(ru[:i])
|
|
closeTag = html.CloseTag(content)
|
|
return content, closeTag
|
|
}
|