Compare commits
2 Commits
f735211a92
...
8da369b166
Author | SHA1 | Date | |
---|---|---|---|
8da369b166 | |||
9e71de8e97 |
|
@ -60,9 +60,10 @@ cacheTime:
|
||||||
commentsCacheTime: 24h
|
commentsCacheTime: 24h
|
||||||
# 随机sleep时间
|
# 随机sleep时间
|
||||||
sleepTime: [ 1s,3s ]
|
sleepTime: [ 1s,3s ]
|
||||||
# 摘要字数
|
# 摘要字数 >0截取指定字数 =0输出出空字符 <0为不截取,原样输出
|
||||||
digestWordCount: 300
|
digestWordCount: 300
|
||||||
|
# 摘要允许的标签 默认为<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>
|
||||||
|
digestTag: "<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>"
|
||||||
# 到达指定并发请求数时随机sleep
|
# 到达指定并发请求数时随机sleep
|
||||||
maxRequestSleepNum: 100
|
maxRequestSleepNum: 100
|
||||||
# 全局最大请求数,超过直接403
|
# 全局最大请求数,超过直接403
|
||||||
|
|
6
internal/pkg/cache/feed.go
vendored
6
internal/pkg/cache/feed.go
vendored
|
@ -7,6 +7,7 @@ import (
|
||||||
str "github.com/fthvgb1/wp-go/helper/strings"
|
str "github.com/fthvgb1/wp-go/helper/strings"
|
||||||
"github.com/fthvgb1/wp-go/internal/pkg/logs"
|
"github.com/fthvgb1/wp-go/internal/pkg/logs"
|
||||||
"github.com/fthvgb1/wp-go/internal/pkg/models"
|
"github.com/fthvgb1/wp-go/internal/pkg/models"
|
||||||
|
"github.com/fthvgb1/wp-go/internal/plugins"
|
||||||
"github.com/fthvgb1/wp-go/internal/plugins/wpposts"
|
"github.com/fthvgb1/wp-go/internal/plugins/wpposts"
|
||||||
"github.com/fthvgb1/wp-go/internal/wpconfig"
|
"github.com/fthvgb1/wp-go/internal/wpconfig"
|
||||||
"github.com/fthvgb1/wp-go/plugin/digest"
|
"github.com/fthvgb1/wp-go/plugin/digest"
|
||||||
|
@ -63,7 +64,7 @@ func feed(arg ...any) (xml []string, err error) {
|
||||||
wpposts.PasswordProjectTitle(&t)
|
wpposts.PasswordProjectTitle(&t)
|
||||||
wpposts.PasswdProjectContent(&t)
|
wpposts.PasswdProjectContent(&t)
|
||||||
} else {
|
} else {
|
||||||
desc = digest.Raw(t.PostContent, 55, fmt.Sprintf("/p/%d", t.Id))
|
desc = plugins.Digests(t.PostContent, t.Id, 55, nil)
|
||||||
}
|
}
|
||||||
l := ""
|
l := ""
|
||||||
if t.CommentStatus == "open" && t.CommentCount > 0 {
|
if t.CommentStatus == "open" && t.CommentCount > 0 {
|
||||||
|
@ -172,8 +173,7 @@ func commentsFeed(args ...any) (r []string, err error) {
|
||||||
wpposts.PasswdProjectContent(&post)
|
wpposts.PasswdProjectContent(&post)
|
||||||
content = post.PostContent
|
content = post.PostContent
|
||||||
} else {
|
} else {
|
||||||
desc = digest.ClearHtml(t.CommentContent)
|
content = digest.StripTags(t.CommentContent, "")
|
||||||
content = desc
|
|
||||||
}
|
}
|
||||||
return rss2.Item{
|
return rss2.Item{
|
||||||
Title: fmt.Sprintf("%s对《%s》的评论", t.CommentAuthor, post.PostTitle),
|
Title: fmt.Sprintf("%s对《%s》的评论", t.CommentAuthor, post.PostTitle),
|
||||||
|
|
|
@ -20,6 +20,7 @@ type Config struct {
|
||||||
Mail Mail `yaml:"mail" json:"mail"`
|
Mail Mail `yaml:"mail" json:"mail"`
|
||||||
CacheTime CacheTime `yaml:"cacheTime" json:"cacheTime"`
|
CacheTime CacheTime `yaml:"cacheTime" json:"cacheTime"`
|
||||||
DigestWordCount int `yaml:"digestWordCount" json:"digestWordCount,omitempty"`
|
DigestWordCount int `yaml:"digestWordCount" json:"digestWordCount,omitempty"`
|
||||||
|
DigestAllowTag string `yaml:"digestAllowTag" json:"digestAllowTag"`
|
||||||
MaxRequestSleepNum int64 `yaml:"maxRequestSleepNum" json:"maxRequestSleepNum,omitempty"`
|
MaxRequestSleepNum int64 `yaml:"maxRequestSleepNum" json:"maxRequestSleepNum,omitempty"`
|
||||||
MaxRequestNum int64 `yaml:"maxRequestNum" json:"maxRequestNum,omitempty"`
|
MaxRequestNum int64 `yaml:"maxRequestNum" json:"maxRequestNum,omitempty"`
|
||||||
SingleIpSearchNum int64 `yaml:"singleIpSearchNum" json:"singleIpSearchNum,omitempty"`
|
SingleIpSearchNum int64 `yaml:"singleIpSearchNum" json:"singleIpSearchNum,omitempty"`
|
||||||
|
|
|
@ -4,34 +4,72 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/fthvgb1/wp-go/cache"
|
"github.com/fthvgb1/wp-go/cache"
|
||||||
|
"github.com/fthvgb1/wp-go/helper"
|
||||||
"github.com/fthvgb1/wp-go/internal/cmd/cachemanager"
|
"github.com/fthvgb1/wp-go/internal/cmd/cachemanager"
|
||||||
"github.com/fthvgb1/wp-go/internal/pkg/config"
|
"github.com/fthvgb1/wp-go/internal/pkg/config"
|
||||||
"github.com/fthvgb1/wp-go/internal/pkg/models"
|
"github.com/fthvgb1/wp-go/internal/pkg/models"
|
||||||
"github.com/fthvgb1/wp-go/plugin/digest"
|
"github.com/fthvgb1/wp-go/plugin/digest"
|
||||||
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var digestCache *cache.MapCache[uint64, string]
|
var digestCache *cache.MapCache[uint64, string]
|
||||||
|
|
||||||
|
var more = regexp.MustCompile("<!--more(.*?)?-->")
|
||||||
|
|
||||||
|
var removeWpBlock = regexp.MustCompile("<!-- /?wp:.*-->")
|
||||||
|
|
||||||
func InitDigestCache() {
|
func InitDigestCache() {
|
||||||
digestCache = cachemanager.MapCacheBy[uint64](digestRaw, config.GetConfig().CacheTime.DigestCacheTime)
|
digestCache = cachemanager.MapCacheBy[uint64](digestRaw, config.GetConfig().CacheTime.DigestCacheTime)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func RemoveWpBlock(s string) string {
|
||||||
|
return removeWpBlock.ReplaceAllString(s, "")
|
||||||
|
}
|
||||||
|
|
||||||
func digestRaw(arg ...any) (string, error) {
|
func digestRaw(arg ...any) (string, error) {
|
||||||
str := arg[0].(string)
|
ctx := arg[0].(context.Context)
|
||||||
id := arg[1].(uint64)
|
s := arg[1].(string)
|
||||||
limit := arg[2].(int)
|
id := arg[2].(uint64)
|
||||||
|
limit := arg[3].(int)
|
||||||
if limit < 0 {
|
if limit < 0 {
|
||||||
return str, nil
|
return s, nil
|
||||||
} else if limit == 0 {
|
} else if limit == 0 {
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
return digest.Raw(str, limit, fmt.Sprintf("/p/%d", id)), nil
|
|
||||||
|
s = more.ReplaceAllString(s, "")
|
||||||
|
fn := helper.GetContextVal(ctx, "postMoreFn", PostsMore)
|
||||||
|
return Digests(s, id, limit, fn), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Digests(content string, id uint64, limit int, fn func(id uint64, content, closeTag string) string) string {
|
||||||
|
closeTag := ""
|
||||||
|
content = RemoveWpBlock(content)
|
||||||
|
tag := config.GetConfig().DigestAllowTag
|
||||||
|
if tag == "" {
|
||||||
|
tag = "<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>"
|
||||||
|
}
|
||||||
|
content = digest.StripTags(content, tag)
|
||||||
|
content, closeTag = digest.Html(content, limit)
|
||||||
|
if fn == nil {
|
||||||
|
return PostsMore(id, content, closeTag)
|
||||||
|
}
|
||||||
|
return fn(id, content, closeTag)
|
||||||
|
}
|
||||||
|
|
||||||
|
func PostsMore(id uint64, content, closeTag string) string {
|
||||||
|
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)
|
||||||
|
return content
|
||||||
}
|
}
|
||||||
|
|
||||||
func Digest(ctx context.Context, post *models.Posts, limit int) {
|
func Digest(ctx context.Context, post *models.Posts, limit int) {
|
||||||
content, _ := digestCache.GetCache(ctx, post.Id, time.Second, post.PostContent, post.Id, limit)
|
content, _ := digestCache.GetCache(ctx, post.Id, time.Second, ctx, post.PostContent, post.Id, limit)
|
||||||
post.PostContent = content
|
post.PostContent = content
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -42,8 +42,7 @@
|
||||||
|
|
||||||
{{if .post.TagsHtml}}
|
{{if .post.TagsHtml}}
|
||||||
<span class="tags-links">
|
<span class="tags-links">
|
||||||
<span class="screen-reader-text">标签 </span>
|
<span class="screen-reader-text">标签 </span>{{.post.TagsHtml|unescaped}}
|
||||||
{{.post.TagsHtml|unescaped}}
|
|
||||||
</span>
|
</span>
|
||||||
{{end}}
|
{{end}}
|
||||||
</footer>
|
</footer>
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
package digest
|
package digest
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"github.com/fthvgb1/wp-go/helper/html"
|
"github.com/fthvgb1/wp-go/helper/html"
|
||||||
"github.com/fthvgb1/wp-go/helper/slice"
|
"github.com/fthvgb1/wp-go/helper/slice"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
@ -9,31 +8,21 @@ import (
|
||||||
"unicode/utf8"
|
"unicode/utf8"
|
||||||
)
|
)
|
||||||
|
|
||||||
var removeWpBlock = regexp.MustCompile("<!-- /?wp:.*-->")
|
|
||||||
var more = regexp.MustCompile("<!--more(.*?)?-->")
|
|
||||||
|
|
||||||
var quto = regexp.MustCompile(`" *|& *|< *|> ?| *`)
|
var quto = regexp.MustCompile(`" *|& *|< *|> ?| *`)
|
||||||
|
|
||||||
func ClearHtml(str string) string {
|
func StripTags(content, allowTag string) string {
|
||||||
content := removeWpBlock.ReplaceAllString(str, "")
|
|
||||||
content = strings.Trim(content, " \t\n\r\000\x0B")
|
content = strings.Trim(content, " \t\n\r\000\x0B")
|
||||||
content = strings.Replace(content, "]]>", "]]>", -1)
|
content = strings.Replace(content, "]]>", "]]>", -1)
|
||||||
content = html.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>")
|
content = html.StripTagsX(content, allowTag)
|
||||||
return str
|
return content
|
||||||
}
|
}
|
||||||
|
|
||||||
func Raw(str string, limit int, u string) string {
|
func Html(content string, limit int) (string, string) {
|
||||||
if r := more.FindString(str); r != "" {
|
closeTag := ""
|
||||||
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 = html.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
|
length := utf8.RuneCountInString(content) + 1
|
||||||
if length > limit {
|
if length <= limit {
|
||||||
|
return content, ""
|
||||||
|
}
|
||||||
index := quto.FindAllStringIndex(content, -1)
|
index := quto.FindAllStringIndex(content, -1)
|
||||||
end := 0
|
end := 0
|
||||||
ru := []rune(content)
|
ru := []rune(content)
|
||||||
|
@ -74,14 +63,7 @@ func Raw(str string, limit int, u string) string {
|
||||||
if i > total {
|
if i > total {
|
||||||
i = total
|
i = total
|
||||||
}
|
}
|
||||||
|
|
||||||
content = string(ru[:i])
|
content = string(ru[:i])
|
||||||
closeTag := html.CloseTag(content)
|
closeTag = html.CloseTag(content)
|
||||||
tmp := `%s......%s<p class="read-more"><a href="%s">继续阅读</a></p>`
|
return content, closeTag
|
||||||
if strings.Contains(closeTag, "pre") || strings.Contains(closeTag, "code") {
|
|
||||||
tmp = `%s%s......<p class="read-more"><a href="%s">继续阅读</a></p>`
|
|
||||||
}
|
|
||||||
content = fmt.Sprintf(tmp, content, closeTag, u)
|
|
||||||
}
|
|
||||||
return content
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user