40 lines
1.0 KiB
Go
40 lines
1.0 KiB
Go
|
package theme
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/fthvgb1/wp-go/internal/plugins"
|
||
|
|
"github.com/fthvgb1/wp-go/plugin/pagination"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
var themeMap = map[string]func(int, *gin.Context, gin.H, int){}
|
||
|
|
|
||
|
|
func AddThemeHookFunc(name string, fn func(int, *gin.Context, gin.H, int)) {
|
||
|
|
if _, ok := themeMap[name]; ok {
|
||
|
|
panic("exists same name theme")
|
||
|
|
}
|
||
|
|
themeMap[name] = fn
|
||
|
|
}
|
||
|
|
|
||
|
|
func Hook(themeName string, status int, c *gin.Context, h gin.H, scene int) {
|
||
|
|
fn, ok := themeMap[themeName]
|
||
|
|
if ok && fn != nil {
|
||
|
|
fn(status, c, h, scene)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if _, ok := plugins.IndexSceneMap[scene]; ok {
|
||
|
|
p, ok := h["pagination"]
|
||
|
|
if ok {
|
||
|
|
pp, ok := p.(pagination.ParsePagination)
|
||
|
|
if ok {
|
||
|
|
h["pagination"] = pagination.Paginate(plugins.TwentyFifteenPagination(), pp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
c.HTML(status, "twentyfifteen/posts/index.gohtml", h)
|
||
|
|
return
|
||
|
|
} else if _, ok := plugins.DetailSceneMap[scene]; ok {
|
||
|
|
c.HTML(status, "twentyfifteen/posts/detail.gohtml", h)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
c.HTML(status, "twentyfifteen/posts/index.gohtml", h)
|
||
|
|
}
|