91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
package plugins
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"github/fthvgb1/wp-go/helper"
|
|
"github/fthvgb1/wp-go/models"
|
|
"regexp"
|
|
"strings"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
var removeWpBlock = regexp.MustCompile("<!-- /?wp:.*-->")
|
|
var more = regexp.MustCompile("<!--more(.*?)?-->")
|
|
var tag = regexp.MustCompile(`<.*?>`)
|
|
var limit = 300
|
|
|
|
func ExceptRaw(str string, limit, id int) string {
|
|
if r := more.FindString(str); r != "" {
|
|
m := strings.Split(str, r)
|
|
str = m[0]
|
|
return ""
|
|
}
|
|
content := removeWpBlock.ReplaceAllString(str, "")
|
|
content = strings.Trim(content, " \t\n\r\000\x0B")
|
|
content = strings.Replace(content, "]]>", "]]>", -1)
|
|
content = helper.StripTagsX(content, "<a><b><blockquote><br><cite><code><dd><del><div><dl><dt><em><h1><h2><h3><h4><h5><h6><i><img><li><ol><p><pre><span><strong><ul>")
|
|
length := utf8.RuneCountInString(content) + 1
|
|
if length > limit {
|
|
start, l := 0, limit
|
|
end := l
|
|
ru := []rune(content)
|
|
|
|
for {
|
|
txt := string(ru[start:end])
|
|
count := 0
|
|
for _, ints := range tag.FindAllStringIndex(txt, -1) {
|
|
t := txt[ints[0]:ints[1]]
|
|
count += len(t)
|
|
l += len(t)
|
|
}
|
|
if count > 0 && length > l {
|
|
start = end
|
|
end += count
|
|
} else if count > 0 && length < l {
|
|
break
|
|
} else {
|
|
content = string(ru[:end])
|
|
closeTag := CloseHtmlTag(content)
|
|
tmp := `%s......%s<p class="read-more"><a href="/p/%d">继续阅读</a></p>`
|
|
if strings.Contains(closeTag, "pre") || strings.Contains(closeTag, "code") {
|
|
tmp = `%s%s......<p class="read-more"><a href="/p/%d">继续阅读</a></p>`
|
|
}
|
|
content = fmt.Sprintf(tmp, content, closeTag, id)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
return content
|
|
}
|
|
|
|
func Except(p *Plugin[models.WpPosts], c *gin.Context, post *models.WpPosts, scene uint) {
|
|
if scene == Detail {
|
|
return
|
|
}
|
|
post.PostContent = ExceptRaw(post.PostContent, limit, int(post.Id))
|
|
|
|
}
|
|
|
|
var tagx = regexp.MustCompile(`(</?[a-z0-9]+?)( |>)`)
|
|
var tagAllow = regexp.MustCompile(`<(a|b|blockquote|cite|code|dd|del|div|dl|dt|em|h1|h2|h3|h4|h5|h6|i|li|ol|p|pre|span|strong|ul).*?>`)
|
|
|
|
func CloseHtmlTag(str string) string {
|
|
tags := tagAllow.FindAllString(str, -1)
|
|
if len(tags) < 1 {
|
|
return str
|
|
}
|
|
var tagss = make([]string, 0, len(tags))
|
|
for _, s := range tags {
|
|
ss := strings.TrimSpace(tagx.FindString(s))
|
|
if ss[len(ss)-1] != '>' {
|
|
ss = fmt.Sprintf("%s>", ss)
|
|
}
|
|
tagss = append(tagss, ss)
|
|
}
|
|
r := helper.SliceMap[string, string](helper.ClearClosedTag(tagss), func(s string) string {
|
|
return fmt.Sprintf("</%s>", strings.Trim(s, "<>"))
|
|
})
|
|
return strings.Join(r, "")
|
|
}
|