102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package wp
|
|
|
|
import (
|
|
"github.com/fthvgb1/wp-go/app/cmd/reload"
|
|
"github.com/fthvgb1/wp-go/app/pkg/config"
|
|
"github.com/fthvgb1/wp-go/app/pkg/models"
|
|
"github.com/fthvgb1/wp-go/app/plugins"
|
|
"github.com/fthvgb1/wp-go/app/plugins/wpposts"
|
|
"github.com/fthvgb1/wp-go/helper/maps"
|
|
"github.com/fthvgb1/wp-go/helper/slice"
|
|
)
|
|
|
|
type PostsPlugin func(*Handle, *models.Posts)
|
|
|
|
func PostsPlugins(initial PostsPlugin, calls ...func(PostsPlugin, *Handle, *models.Posts)) PostsPlugin {
|
|
return slice.ReverseReduce(calls, func(t func(PostsPlugin, *Handle, *models.Posts), r PostsPlugin) PostsPlugin {
|
|
return func(handle *Handle, posts *models.Posts) {
|
|
t(r, handle, posts)
|
|
}
|
|
}, initial)
|
|
}
|
|
|
|
var pluginFns = reload.Vars(map[string]func(PostsPlugin, *Handle, *models.Posts){
|
|
"passwordProject": PasswordProject,
|
|
"digest": Digest,
|
|
})
|
|
|
|
func (h *Handle) PushPostsPlugin(name string, fn func(PostsPlugin, *Handle, *models.Posts)) {
|
|
m := pluginFns.Load()
|
|
m[name] = fn
|
|
}
|
|
|
|
// PasswordProject 标题和内容密码保护
|
|
func PasswordProject(next PostsPlugin, h *Handle, post *models.Posts) {
|
|
r := post
|
|
if post.PostPassword != "" {
|
|
wpposts.PasswordProjectTitle(r)
|
|
if h.GetPassword() != post.PostPassword {
|
|
wpposts.PasswdProjectContent(r)
|
|
return
|
|
}
|
|
}
|
|
next(h, r)
|
|
}
|
|
|
|
// Digest 生成摘要
|
|
func Digest(next PostsPlugin, h *Handle, post *models.Posts) {
|
|
if post.PostExcerpt != "" {
|
|
plugins.PostExcerpt(post)
|
|
} else {
|
|
plugins.Digest(h.C, post, config.GetConfig().DigestWordCount)
|
|
}
|
|
next(h, post)
|
|
}
|
|
|
|
var ordinaryPlugin = reload.Vars([]PostsPlugin{})
|
|
|
|
func (h *Handle) PushPostPlugin(plugin ...PostsPlugin) {
|
|
p := ordinaryPlugin.Load()
|
|
p = append(p, plugin...)
|
|
ordinaryPlugin.Store(p)
|
|
}
|
|
|
|
func PostPlugin(calls ...PostsPlugin) PostsPlugin {
|
|
return func(h *Handle, posts *models.Posts) {
|
|
for _, call := range calls {
|
|
call(h, posts)
|
|
}
|
|
}
|
|
}
|
|
|
|
func UsePostsPlugins() PostsPlugin {
|
|
m := pluginFns.Load()
|
|
pluginss := slice.FilterAndMap(config.GetConfig().ListPagePlugins, func(t string) (func(PostsPlugin, *Handle, *models.Posts), bool) {
|
|
f, ok := m[t]
|
|
return f, ok
|
|
})
|
|
slice.Unshift(&pluginss, PasswordProject)
|
|
return PostsPlugins(PostPlugin(ordinaryPlugin.Load()...), pluginss...)
|
|
}
|
|
|
|
func ListPostPlugins() map[string]func(PostsPlugin, *Handle, *models.Posts) {
|
|
return maps.Copy(pluginFns.Load())
|
|
}
|
|
|
|
func ProjectTitle(t models.Posts) models.Posts {
|
|
if t.PostPassword != "" {
|
|
wpposts.PasswordProjectTitle(&t)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func GetListPostPlugins(name []string, m map[string]func(PostsPlugin, *Handle, *models.Posts)) []func(PostsPlugin, *Handle, *models.Posts) {
|
|
return slice.FilterAndMap(name, func(t string) (func(PostsPlugin, *Handle, *models.Posts), bool) {
|
|
v, ok := m[t]
|
|
if ok {
|
|
return v, true
|
|
}
|
|
return nil, false
|
|
})
|
|
}
|