优化代码 使用继承方式重构列表页

This commit is contained in:
xing 2023-02-11 00:32:46 +08:00
parent cd636e2dc9
commit a2cebfbb3f
11 changed files with 431 additions and 344 deletions

View File

@ -5,6 +5,7 @@ package number
import ( import (
"fmt" "fmt"
"golang.org/x/exp/constraints" "golang.org/x/exp/constraints"
"math"
"math/rand" "math/rand"
) )
@ -86,3 +87,7 @@ func Mul[T constraints.Integer | constraints.Float](i, j T) T {
func Divide[T constraints.Integer | constraints.Float](i, j T) T { func Divide[T constraints.Integer | constraints.Float](i, j T) T {
return i / j return i / j
} }
func CalTotalPage[T constraints.Integer](totalRows, size T) T {
return T(math.Ceil(float64(totalRows) / float64(size)))
}

View File

@ -219,3 +219,34 @@ func TestAbs(t *testing.T) {
}) })
} }
} }
func TestCalTotalPage(t *testing.T) {
type args[T constraints.Integer] struct {
totalRows T
size T
}
type testCase[T constraints.Integer] struct {
name string
args args[T]
want T
}
tests := []testCase[int]{
{
name: "t1",
args: args[int]{5, 2},
want: 3,
},
{
name: "t1",
args: args[int]{4, 2},
want: 2,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := CalTotalPage(tt.args.totalRows, tt.args.size); got != tt.want {
t.Errorf("CalTotalPage() = %v, want %v", got, tt.want)
}
})
}
}

View File

@ -5,6 +5,7 @@ import (
"github.com/fthvgb1/wp-go/helper/slice" "github.com/fthvgb1/wp-go/helper/slice"
str "github.com/fthvgb1/wp-go/helper/strings" str "github.com/fthvgb1/wp-go/helper/strings"
"github.com/fthvgb1/wp-go/internal/pkg/cache" "github.com/fthvgb1/wp-go/internal/pkg/cache"
"github.com/fthvgb1/wp-go/internal/pkg/constraints"
"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"
@ -21,7 +22,7 @@ func Detail(c *gin.Context) {
var post models.Posts var post models.Posts
recent := slice.Map(cache.RecentPosts(c, 5), common.ProjectTitle) recent := slice.Map(cache.RecentPosts(c, 5), common.ProjectTitle)
archive := cache.Archives(c) archive := cache.Archives(c)
categoryItems := cache.CategoriesTags(c, plugins.Category) categoryItems := cache.CategoriesTags(c, constraints.Category)
recentComments := cache.RecentComments(c, 5) recentComments := cache.RecentComments(c, 5)
var ginH = gin.H{ var ginH = gin.H{
"title": wpconfig.Options.Value("blogname"), "title": wpconfig.Options.Value("blogname"),
@ -32,7 +33,7 @@ func Detail(c *gin.Context) {
"post": post, "post": post,
} }
isApproveComment := false isApproveComment := false
stats := plugins.Ok stats := constraints.Ok
pw := sessions.Default(c).Get("post_password") pw := sessions.Default(c).Get("post_password")
defer func() { defer func() {
@ -40,7 +41,7 @@ func Detail(c *gin.Context) {
if err != nil { if err != nil {
code = http.StatusNotFound code = http.StatusNotFound
c.Error(err) c.Error(err)
stats = plugins.Error stats = constraints.ParamError
return return
} }
if isApproveComment == true { if isApproveComment == true {
@ -52,7 +53,7 @@ func Detail(c *gin.Context) {
C: c, C: c,
GinH: ginH, GinH: ginH,
Password: "", Password: "",
Scene: plugins.Detail, Scene: constraints.Detail,
Code: code, Code: code,
Stats: stats, Stats: stats,
}) })
@ -62,12 +63,12 @@ func Detail(c *gin.Context) {
maxId, err := cache.GetMaxPostId(c) maxId, err := cache.GetMaxPostId(c)
logs.ErrPrintln(err, "get max post id") logs.ErrPrintln(err, "get max post id")
if id > maxId || id <= 0 || err != nil { if id > maxId || id <= 0 || err != nil {
stats = plugins.Empty404 stats = constraints.Error404
return return
} }
post, err = cache.GetPostById(c, id) post, err = cache.GetPostById(c, id)
if post.Id == 0 || err != nil || post.PostStatus != "publish" { if post.Id == 0 || err != nil || post.PostStatus != "publish" {
stats = plugins.Empty404 stats = constraints.Error404
return return
} }
showComment := false showComment := false
@ -102,5 +103,5 @@ func Detail(c *gin.Context) {
ginH["maxDep"] = d ginH["maxDep"] = d
ginH["next"] = next ginH["next"] = next
ginH["user"] = user ginH["user"] = user
ginH["scene"] = plugins.Detail ginH["scene"] = constraints.Detail
} }

View File

@ -1,302 +1,21 @@
package actions package actions
import ( import (
"errors" "github.com/fthvgb1/wp-go/internal/pkg/constraints"
"fmt"
"github.com/fthvgb1/wp-go/helper/maps"
"github.com/fthvgb1/wp-go/helper/number"
"github.com/fthvgb1/wp-go/helper/slice"
str "github.com/fthvgb1/wp-go/helper/strings"
"github.com/fthvgb1/wp-go/internal/pkg/cache"
"github.com/fthvgb1/wp-go/internal/pkg/config"
"github.com/fthvgb1/wp-go/internal/pkg/dao"
"github.com/fthvgb1/wp-go/internal/pkg/logs"
"github.com/fthvgb1/wp-go/internal/pkg/models"
"github.com/fthvgb1/wp-go/internal/plugins"
"github.com/fthvgb1/wp-go/internal/theme" "github.com/fthvgb1/wp-go/internal/theme"
"github.com/fthvgb1/wp-go/internal/theme/common" "github.com/fthvgb1/wp-go/internal/theme/common"
"github.com/fthvgb1/wp-go/internal/wpconfig"
"github.com/fthvgb1/wp-go/model"
"github.com/fthvgb1/wp-go/plugin/pagination"
"github.com/gin-contrib/sessions"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"math"
"net/http" "net/http"
"strconv"
"strings"
"sync/atomic"
"time"
) )
type indexHandle struct { func Index(scene int) func(*gin.Context) {
c *gin.Context return func(ctx *gin.Context) {
session sessions.Session theme.Hook(theme.GetTemplateName(), common.Handle{
page int C: ctx,
pageSize int GinH: gin.H{},
title string Scene: scene,
titleL string Code: http.StatusOK,
titleR string Stats: constraints.Ok,
search string
author string
totalPage int
category string
categoryType string
where model.SqlBuilder
orderBy string
order string
join model.SqlBuilder
postType []any
postStatus []any
header string
paginationStep int
scene int
stats int
}
func newIndexHandle(ctx *gin.Context) *indexHandle {
size := str.ToInteger(wpconfig.Options.Value("posts_per_page"), 10)
return &indexHandle{
c: ctx,
session: sessions.Default(ctx),
page: 1,
pageSize: size,
paginationStep: 1,
titleL: wpconfig.Options.Value("blogname"),
titleR: wpconfig.Options.Value("blogdescription"),
where: model.SqlBuilder{
{"post_type", "in", ""},
{"post_status", "in", ""},
},
orderBy: "post_date",
join: model.SqlBuilder{},
postType: []any{"post"},
postStatus: []any{"publish"},
scene: plugins.Home,
stats: plugins.Ok,
}
}
var months = slice.SimpleToMap(number.Range(1, 12, 1), func(v int) int {
return v
})
func (h *indexHandle) setTitleLR(l, r string) {
h.titleL = l
h.titleR = r
}
func (h *indexHandle) getTitle() string {
h.title = fmt.Sprintf("%s-%s", h.titleL, h.titleR)
return h.title
}
func (h *indexHandle) getSearchKey() string {
return fmt.Sprintf("action:%s|%s|%s|%s|%s|%s|%d|%d", h.author, h.search, h.orderBy, h.order, h.category, h.categoryType, h.page, h.pageSize)
}
var orders = map[string]struct{}{"asc": {}, "desc": {}}
func (h *indexHandle) parseParams() (err error) {
h.order = h.c.Query("order")
if !maps.IsExists(orders, h.order) {
order := config.GetConfig().PostOrder
h.order = "asc"
if order != "" && maps.IsExists(orders, order) {
h.order = order
}
}
year := h.c.Param("year")
if year != "" {
y := str.ToInteger(year, -1)
if y > time.Now().Year() || y <= 1970 {
return errors.New(str.Join("year err : ", year))
}
h.where = append(h.where, []string{
"year(post_date)", year,
}) })
} }
month := h.c.Param("month")
if month != "" {
m := str.ToInteger(month, -1)
if !maps.IsExists(months, m) {
return errors.New(str.Join("months err ", month))
}
h.where = append(h.where, []string{
"month(post_date)", month,
})
ss := fmt.Sprintf("%s年%s月", year, strings.TrimLeft(month, "0"))
h.header = fmt.Sprintf("月度归档: <span>%s</span>", ss)
h.setTitleLR(ss, wpconfig.Options.Value("blogname"))
h.scene = plugins.Archive
}
category := h.c.Param("category")
if category != "" {
h.scene = plugins.Category
if !maps.IsExists(cache.AllCategoryTagsNames(h.c, plugins.Category), category) {
return errors.New(str.Join("not exists category ", category))
}
h.categoryType = "category"
h.header = fmt.Sprintf("分类: <span>%s</span>", category)
h.category = category
}
tag := h.c.Param("tag")
if tag != "" {
h.scene = plugins.Tag
if !maps.IsExists(cache.AllCategoryTagsNames(h.c, plugins.Tag), tag) {
return errors.New(str.Join("not exists tag ", tag))
}
h.categoryType = "post_tag"
h.header = fmt.Sprintf("标签: <span>%s</span>", tag)
h.category = tag
}
username := h.c.Param("author")
if username != "" {
allUsername, er := cache.GetAllUsername(h.c)
if er != nil {
err = er
return
}
if !maps.IsExists(allUsername, username) {
err = errors.New(str.Join("user ", username, " is not exists"))
return
}
user, er := cache.GetUserByName(h.c, username)
if er != nil {
err = er
return
}
h.author = username
h.where = append(h.where, []string{
"post_author", "=", strconv.FormatUint(user.Id, 10), "int",
})
}
if h.category != "" {
h.where = append(h.where, []string{
"d.name", h.category,
}, []string{"taxonomy", h.categoryType})
h.join = append(h.join, []string{
"a", "left join", "wp_term_relationships b", "a.Id=b.object_id",
}, []string{
"left join", "wp_term_taxonomy c", "b.term_taxonomy_id=c.term_taxonomy_id",
}, []string{
"left join", "wp_terms d", "c.term_id=d.term_id",
})
h.setTitleLR(h.category, wpconfig.Options.Value("blogname"))
}
s, ok := h.c.GetQuery("s")
if ok {
h.scene = plugins.Search
}
if s != "" && strings.Replace(s, " ", "", -1) != "" {
q := str.Join("%", s, "%")
h.where = append(h.where, []string{
"and", "post_title", "like", q, "",
"or", "post_content", "like", q, "",
"or", "post_excerpt", "like", q, "",
}, []string{"post_password", ""})
h.postType = append(h.postType, "page", "attachment")
h.header = fmt.Sprintf("<span>%s</span>的搜索结果", s)
h.setTitleLR(str.Join(`"`, s, `"`, "的搜索结果"), wpconfig.Options.Value("blogname"))
h.search = s
h.scene = plugins.Search
}
h.page = str.ToInteger(h.c.Param("page"), 1)
total := int(atomic.LoadInt64(&dao.TotalRaw))
if total > 0 && total < (h.page-1)*h.pageSize {
h.page = 1
}
if h.page > 1 && (h.category != "" || h.search != "" || month != "") {
h.setTitleLR(fmt.Sprintf("%s-第%d页", h.titleL, h.page), wpconfig.Options.Value("blogname"))
}
return
}
func (h *indexHandle) getTotalPage(totalRaws int) int {
h.totalPage = int(math.Ceil(float64(totalRaws) / float64(h.pageSize)))
return h.totalPage
}
func Index(c *gin.Context) {
h := newIndexHandle(c)
var posts []models.Posts
var totalRaw int
var err error
archive := cache.Archives(c)
recent := slice.Map(cache.RecentPosts(c, 5), common.ProjectTitle)
categoryItems := cache.CategoriesTags(c, plugins.Category)
recentComments := cache.RecentComments(c, 5)
ginH := gin.H{
"err": err,
"recentPosts": recent,
"archives": archive,
"categories": categoryItems,
"recentComments": recentComments,
"posts": posts,
}
pw := h.session.Get("post_password")
pws, ok := pw.(string)
if !ok {
pws = ""
}
defer func() {
code := http.StatusOK
if err != nil {
code = http.StatusNotFound
if h.stats == plugins.InternalErr {
code = http.StatusInternalServerError
c.Error(err)
return
}
c.Error(err)
h.stats = plugins.Error
}
t := theme.GetTemplateName()
theme.Hook(t, common.Handle{
C: c,
GinH: ginH,
Password: pws,
Scene: h.scene,
Code: code,
Stats: h.stats,
})
}()
err = h.parseParams()
if err != nil {
return
}
ginH["title"] = h.getTitle()
ginH["search"] = h.search
ginH["header"] = h.header
if c.Param("month") != "" {
posts, totalRaw, err = cache.GetMonthPostIds(c, c.Param("year"), c.Param("month"), h.page, h.pageSize, h.order)
if err != nil {
return
}
} else if h.search != "" {
posts, totalRaw, err = cache.SearchPost(c, h.getSearchKey(), c, h.where, h.page, h.pageSize, model.SqlBuilder{{h.orderBy, h.order}}, h.join, h.postType, h.postStatus)
} else {
posts, totalRaw, err = cache.PostLists(c, h.getSearchKey(), c, h.where, h.page, h.pageSize, model.SqlBuilder{{h.orderBy, h.order}}, h.join, h.postType, h.postStatus)
}
if err != nil {
h.stats = plugins.Error
logs.ErrPrintln(err, "获取数据错误")
return
}
if len(posts) < 1 && h.category != "" {
h.titleL = "未找到页面"
h.stats = plugins.Empty404
}
q := c.Request.URL.Query().Encode()
if q != "" {
q = fmt.Sprintf("?%s", q)
}
ginH["posts"] = posts
ginH["totalPage"] = h.getTotalPage(totalRaw)
ginH["currentPage"] = h.page
ginH["title"] = h.getTitle()
ginH["scene"] = h.scene
ginH["pagination"] = pagination.NewParsePagination(totalRaw, h.pageSize, h.page, h.paginationStep, q, c.Request.URL.Path)
} }

View File

@ -4,6 +4,7 @@ import (
"github.com/fthvgb1/wp-go/internal/actions" "github.com/fthvgb1/wp-go/internal/actions"
"github.com/fthvgb1/wp-go/internal/middleware" "github.com/fthvgb1/wp-go/internal/middleware"
"github.com/fthvgb1/wp-go/internal/pkg/config" "github.com/fthvgb1/wp-go/internal/pkg/config"
"github.com/fthvgb1/wp-go/internal/pkg/constraints"
"github.com/fthvgb1/wp-go/internal/static" "github.com/fthvgb1/wp-go/internal/static"
"github.com/fthvgb1/wp-go/internal/theme" "github.com/fthvgb1/wp-go/internal/theme"
"github.com/gin-contrib/gzip" "github.com/gin-contrib/gzip"
@ -61,16 +62,16 @@ func SetupRouter() (*gin.Engine, func()) {
store := cookie.NewStore([]byte("secret")) store := cookie.NewStore([]byte("secret"))
r.Use(sessions.Sessions("go-wp", store)) r.Use(sessions.Sessions("go-wp", store))
sl, slRload := middleware.SearchLimit(c.SingleIpSearchNum) sl, slRload := middleware.SearchLimit(c.SingleIpSearchNum)
r.GET("/", sl, actions.Index) r.GET("/", sl, actions.Index(constraints.Home))
r.GET("/page/:page", actions.Index) r.GET("/page/:page", actions.Index(constraints.Home))
r.GET("/p/category/:category", actions.Index) r.GET("/p/category/:category", actions.Index(constraints.Category))
r.GET("/p/category/:category/page/:page", actions.Index) r.GET("/p/category/:category/page/:page", actions.Index(constraints.Category))
r.GET("/p/tag/:tag", actions.Index) r.GET("/p/tag/:tag", actions.Index(constraints.Tag))
r.GET("/p/tag/:tag/page/:page", actions.Index) r.GET("/p/tag/:tag/page/:page", actions.Index(constraints.Tag))
r.GET("/p/date/:year/:month", actions.Index) r.GET("/p/date/:year/:month", actions.Index(constraints.Archive))
r.GET("/p/date/:year/:month/page/:page", actions.Index) r.GET("/p/date/:year/:month/page/:page", actions.Index(constraints.Archive))
r.GET("/p/author/:author", actions.Index) r.GET("/p/author/:author", actions.Index(constraints.Author))
r.GET("/p/author/:author/page/:page", actions.Index) r.GET("/p/author/:author/page/:page", actions.Index(constraints.Author))
r.POST("/login", actions.Login) r.POST("/login", actions.Login)
r.GET("/p/:id", actions.Detail) r.GET("/p/:id", actions.Detail)
r.GET("/p/:id/feed", actions.PostFeed) r.GET("/p/:id/feed", actions.PostFeed)

View File

@ -10,7 +10,7 @@ const (
Detail Detail
Ok Ok
Empty404 Error404
ParamError ParamError
InternalErr InternalErr
) )

View File

@ -24,15 +24,15 @@ type Handle struct {
Stats int Stats int
} }
func (h Handle) Detail() { func (h *Handle) Detail() {
} }
func (h Handle) Index() { func (h *Handle) Index() {
} }
func (h Handle) ExecListPagePlugin(m map[string]Plugin[models.Posts], calls ...func(*models.Posts)) { func (h *Handle) ExecListPagePlugin(m map[string]Plugin[models.Posts], calls ...func(*models.Posts)) {
pluginConf := config.GetConfig().ListPagePlugins pluginConf := config.GetConfig().ListPagePlugins
@ -45,14 +45,14 @@ func (h Handle) ExecListPagePlugin(m map[string]Plugin[models.Posts], calls ...f
} }
} }
/*func (h Handle) Pagination(paginate pagination) { /*func (h *Handle) Pagination(paginate pagination) {
}*/ }*/
type Fn[T any] func(T) T type Fn[T any] func(T) T
type Plugin[T any] func(next Fn[T], h Handle, t T) T type Plugin[T any] func(next Fn[T], h *Handle, t T) T
func PluginFn[T any](a []Plugin[T], h Handle, fn Fn[T]) Fn[T] { func PluginFn[T any](a []Plugin[T], h *Handle, fn Fn[T]) Fn[T] {
return slice.ReverseReduce(a, func(next Plugin[T], forward Fn[T]) Fn[T] { return slice.ReverseReduce(a, func(next Plugin[T], forward Fn[T]) Fn[T] {
return func(t T) T { return func(t T) T {
return next(forward, h, t) return next(forward, h, t)
@ -94,7 +94,7 @@ func GetListPostPlugins(name []string, m map[string]Plugin[models.Posts]) []Plug
} }
// PasswordProject 标题和内容密码保护 // PasswordProject 标题和内容密码保护
func PasswordProject(next Fn[models.Posts], h Handle, post models.Posts) (r models.Posts) { func PasswordProject(next Fn[models.Posts], h *Handle, post models.Posts) (r models.Posts) {
r = post r = post
if post.PostPassword != "" { if post.PostPassword != "" {
plugins.PasswordProjectTitle(&r) plugins.PasswordProjectTitle(&r)
@ -115,7 +115,7 @@ func ProjectTitle(t models.Posts) models.Posts {
} }
// Digest 生成摘要 // Digest 生成摘要
func Digest(next Fn[models.Posts], h Handle, post models.Posts) models.Posts { func Digest(next Fn[models.Posts], h *Handle, post models.Posts) models.Posts {
if post.PostExcerpt != "" { if post.PostExcerpt != "" {
plugins.PostExcerpt(&post) plugins.PostExcerpt(&post)
} else { } else {

View File

@ -0,0 +1,308 @@
package common
import (
"database/sql"
"errors"
"fmt"
"github.com/fthvgb1/wp-go/helper/maps"
"github.com/fthvgb1/wp-go/helper/number"
"github.com/fthvgb1/wp-go/helper/slice"
str "github.com/fthvgb1/wp-go/helper/strings"
"github.com/fthvgb1/wp-go/internal/pkg/cache"
"github.com/fthvgb1/wp-go/internal/pkg/config"
"github.com/fthvgb1/wp-go/internal/pkg/constraints"
"github.com/fthvgb1/wp-go/internal/pkg/dao"
"github.com/fthvgb1/wp-go/internal/pkg/models"
"github.com/fthvgb1/wp-go/internal/wpconfig"
"github.com/fthvgb1/wp-go/model"
"github.com/fthvgb1/wp-go/plugin/pagination"
"github.com/gin-gonic/gin"
"net/http"
"strconv"
"strings"
"sync/atomic"
"time"
)
type IndexParams struct {
c *gin.Context
Page int
PageSize int
Title string
titleL string
titleR string
search string
author string
totalPage int
category string
categoryType string
Where model.SqlBuilder
OrderBy string
Order string
Month string
Year string
Join model.SqlBuilder
PostType []any
PostStatus []any
header string
PaginationStep int
scene int
stats int
CacheKey string
blogName string
}
var months = slice.SimpleToMap(number.Range(1, 12, 1), func(v int) int {
return v
})
var orders = map[string]struct{}{"asc": {}, "desc": {}}
func (i *IndexParams) setTitleLR(l, r string) {
i.titleL = l
i.titleR = r
}
func (i *IndexParams) getTitle() string {
i.Title = fmt.Sprintf("%s-%s", i.titleL, i.titleR)
return i.Title
}
func (i *IndexParams) getSearchKey() string {
return fmt.Sprintf("action:%s|%s|%s|%s|%s|%s|%d|%d", i.author, i.search, i.OrderBy, i.Order, i.category, i.categoryType, i.Page, i.PageSize)
}
func newIndexHandle(ctx *gin.Context) *IndexParams {
blogName := wpconfig.Options.Value("blogname")
size := str.ToInteger(wpconfig.Options.Value("posts_per_page"), 10)
return &IndexParams{
c: ctx,
Page: 1,
PageSize: size,
PaginationStep: number.Max(1, config.GetConfig().PaginationStep),
titleL: blogName,
titleR: wpconfig.Options.Value("blogdescription"),
Where: model.SqlBuilder{
{"post_type", "in", ""},
{"post_status", "in", ""},
},
OrderBy: "post_date",
Join: model.SqlBuilder{},
PostType: []any{"post"},
PostStatus: []any{"publish"},
scene: constraints.Home,
stats: constraints.Ok,
blogName: wpconfig.Options.Value("blogname"),
}
}
func (i *IndexParams) ParseSearch() {
s := i.c.Query("s")
if s != "" && strings.Replace(s, " ", "", -1) != "" {
q := str.Join("%", s, "%")
i.Where = append(i.Where, []string{
"and", "post_title", "like", q, "",
"or", "post_content", "like", q, "",
"or", "post_excerpt", "like", q, "",
}, []string{"post_password", ""})
i.PostType = append(i.PostType, "Page", "attachment")
i.header = fmt.Sprintf("<span>%s</span>的搜索结果", s)
i.setTitleLR(str.Join(`"`, s, `"`, "的搜索结果"), i.blogName)
i.search = s
i.scene = constraints.Search
}
}
func (i *IndexParams) ParseArchive() error {
year := i.c.Param("year")
if year != "" {
y := str.ToInteger(year, -1)
if y > time.Now().Year() || y <= 1970 {
return errors.New(str.Join("year err : ", year))
}
i.Where = append(i.Where, []string{
"year(post_date)", year,
})
i.Year = year
}
month := i.c.Param("month")
if month != "" {
m := str.ToInteger(month, -1)
if !maps.IsExists(months, m) {
return errors.New(str.Join("months err ", month))
}
i.Where = append(i.Where, []string{
"month(post_date)", month,
})
ss := fmt.Sprintf("%s年%s月", year, strings.TrimLeft(month, "0"))
i.header = fmt.Sprintf("月度归档: <span>%s</span>", ss)
i.setTitleLR(ss, i.blogName)
i.scene = constraints.Archive
i.Month = month
}
return nil
}
func (i *IndexParams) ParseCategory() error {
category := i.c.Param("category")
if category != "" {
i.scene = constraints.Category
if !maps.IsExists(cache.AllCategoryTagsNames(i.c, constraints.Category), category) {
return errors.New(str.Join("not exists category ", category))
}
i.categoryType = "category"
i.header = fmt.Sprintf("分类: <span>%s</span>", category)
i.category = category
i.CategoryCondition()
}
return nil
}
func (i *IndexParams) ParseTag() error {
tag := i.c.Param("tag")
if tag != "" {
i.scene = constraints.Tag
if !maps.IsExists(cache.AllCategoryTagsNames(i.c, constraints.Tag), tag) {
return errors.New(str.Join("not exists tag ", tag))
}
i.categoryType = "post_tag"
i.header = fmt.Sprintf("标签: <span>%s</span>", tag)
i.category = tag
i.CategoryCondition()
}
return nil
}
func (i *IndexParams) CategoryCondition() {
if i.category != "" {
i.Where = append(i.Where, []string{
"d.name", i.category,
}, []string{"taxonomy", i.categoryType})
i.Join = append(i.Join, []string{
"a", "left Join", "wp_term_relationships b", "a.Id=b.object_id",
}, []string{
"left Join", "wp_term_taxonomy c", "b.term_taxonomy_id=c.term_taxonomy_id",
}, []string{
"left Join", "wp_terms d", "c.term_id=d.term_id",
})
i.setTitleLR(i.category, i.blogName)
}
}
func (i *IndexParams) ParseAuthor() (err error) {
username := i.c.Param("author")
if username != "" {
allUsername, er := cache.GetAllUsername(i.c)
if err != nil {
err = er
return
}
if !maps.IsExists(allUsername, username) {
err = errors.New(str.Join("user ", username, " is not exists"))
return
}
user, er := cache.GetUserByName(i.c, username)
if er != nil {
return
}
i.author = username
i.Where = append(i.Where, []string{
"post_author", "=", strconv.FormatUint(user.Id, 10), "int",
})
}
return
}
func (i *IndexParams) parseParams() {
i.Order = i.c.Query("Order")
if !maps.IsExists(orders, i.Order) {
order := config.GetConfig().PostOrder
i.Order = "asc"
if order != "" && maps.IsExists(orders, order) {
i.Order = order
}
}
i.Page = str.ToInteger(i.c.Param("page"), 1)
total := int(atomic.LoadInt64(&dao.TotalRaw))
if total > 0 && total < (i.Page-1)*i.PageSize {
i.Page = 1
}
if i.Page > 1 && (i.category != "" || i.search != "" || i.Month != "") {
i.setTitleLR(fmt.Sprintf("%s-第%d页", i.titleL, i.Page), i.blogName)
}
return
}
func (h *Handle) ParseIndex() (i *IndexParams, err error) {
i = newIndexHandle(h.C)
switch h.Scene {
case constraints.Home, constraints.Search:
i.ParseSearch()
case constraints.Category:
err = i.ParseCategory()
case constraints.Tag:
err = i.ParseTag()
case constraints.Archive:
err = i.ParseArchive()
case constraints.Author:
err = i.ParseAuthor()
}
if err != nil {
h.Stats = constraints.ParamError
return
}
i.CacheKey = i.getSearchKey()
i.parseParams()
h.GinH["title"] = i.getTitle()
h.GinH["search"] = i.search
h.GinH["header"] = i.header
return
}
func (h *Handle) GetIndexData(i *IndexParams) (posts []models.Posts, totalRaw int, err error) {
switch h.Scene {
case constraints.Home, constraints.Category, constraints.Tag, constraints.Author:
posts, totalRaw, err = cache.PostLists(h.C, i.CacheKey, h.C, i.Where, i.Page, i.PageSize,
model.SqlBuilder{{i.OrderBy, i.Order}}, i.Join, i.PostType, i.PostStatus)
case constraints.Search:
posts, totalRaw, err = cache.SearchPost(h.C, i.CacheKey, h.C, i.Where, i.Page, i.PageSize,
model.SqlBuilder{{i.OrderBy, i.Order}}, i.Join, i.PostType, i.PostStatus)
case constraints.Archive:
posts, totalRaw, err = cache.GetMonthPostIds(h.C, i.Year, i.Month, i.Page, i.PageSize, i.Order)
}
return
}
func (h *Handle) Indexs() (err error) {
i, err := h.ParseIndex()
if err != nil {
h.Stats = constraints.ParamError
h.Code = http.StatusNotFound
return
}
posts, totalRows, err := h.GetIndexData(i)
if err != nil && err != sql.ErrNoRows {
h.Scene = constraints.InternalErr
h.Code = http.StatusInternalServerError
return
}
pw := h.Session.Get("post_password")
if pw != nil {
h.Password = pw.(string)
}
h.GinH["posts"] = posts
h.GinH["totalPage"] = number.CalTotalPage(totalRows, i.PageSize)
q := h.C.Request.URL.Query().Encode()
if q != "" {
q = fmt.Sprintf("?%s", q)
}
h.GinH["pagination"] = pagination.NewParsePagination(totalRows, i.PageSize, i.Page, i.PaginationStep, q, h.C.Request.URL.Path)
return
}

View File

@ -0,0 +1,14 @@
package common
import (
"github.com/fthvgb1/wp-go/helper/slice"
"github.com/fthvgb1/wp-go/internal/pkg/cache"
"github.com/fthvgb1/wp-go/internal/pkg/constraints"
)
func (h *Handle) WidgetAreaData() {
h.GinH["archives"] = cache.Archives(h.C)
h.GinH["recentPosts"] = slice.Map(cache.RecentPosts(h.C, 5), ProjectTitle)
h.GinH["categories"] = cache.CategoriesTags(h.C, constraints.Category)
h.GinH["recentComments"] = cache.RecentComments(h.C, 5)
}

View File

@ -7,6 +7,7 @@ import (
"github.com/fthvgb1/wp-go/internal/plugins" "github.com/fthvgb1/wp-go/internal/plugins"
"github.com/fthvgb1/wp-go/internal/theme/common" "github.com/fthvgb1/wp-go/internal/theme/common"
"github.com/fthvgb1/wp-go/plugin/pagination" "github.com/fthvgb1/wp-go/plugin/pagination"
"github.com/gin-contrib/sessions"
) )
const ThemeName = "twentyfifteen" const ThemeName = "twentyfifteen"
@ -16,13 +17,15 @@ type handle struct {
templ string templ string
} }
func Hook(h2 common.Handle) { func Hook(cHandle common.Handle) {
h := handle{ h := handle{
Handle: h2, Handle: cHandle,
templ: "twentyfifteen/posts/index.gohtml", templ: "twentyfifteen/posts/index.gohtml",
} }
//h.GinH["HeaderImage"] = h.getHeaderImage(h.C) h.WidgetAreaData()
if h.Stats == constraints.Empty404 { h.Session = sessions.Default(h.C)
if h.Stats == constraints.Error404 {
h.C.HTML(h.Code, h.templ, h.GinH) h.C.HTML(h.Code, h.templ, h.GinH)
return return
} }
@ -36,17 +39,17 @@ func Hook(h2 common.Handle) {
var plugin = common.ListPostPlugins() var plugin = common.ListPostPlugins()
func (h handle) Index() { func (h handle) Index() {
if h.Stats != constraints.Empty404 { err := h.Indexs()
if err != nil {
h.C.HTML(h.Code, h.templ, h.GinH)
return
}
h.ExecListPagePlugin(plugin) h.ExecListPagePlugin(plugin)
page, ok := maps.GetStrMapAnyVal[pagination.ParsePagination](h.GinH, "pagination") page, ok := maps.GetStrMapAnyVal[pagination.ParsePagination](h.GinH, "pagination")
if ok { if ok {
h.GinH["pagination"] = pagination.Paginate(plugins.TwentyFifteenPagination(), page) h.GinH["pagination"] = pagination.Paginate(plugins.TwentyFifteenPagination(), page)
} }
}
//h.GinH["bodyClass"] = h.bodyClass()
h.C.HTML(h.Code, h.templ, h.GinH) h.C.HTML(h.Code, h.templ, h.GinH)
} }

View File

@ -12,6 +12,7 @@ import (
"github.com/fthvgb1/wp-go/internal/plugins" "github.com/fthvgb1/wp-go/internal/plugins"
"github.com/fthvgb1/wp-go/internal/theme/common" "github.com/fthvgb1/wp-go/internal/theme/common"
"github.com/fthvgb1/wp-go/plugin/pagination" "github.com/fthvgb1/wp-go/plugin/pagination"
"github.com/gin-contrib/sessions"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"strings" "strings"
) )
@ -38,6 +39,8 @@ func Hook(cHandle common.Handle) {
Handle: cHandle, Handle: cHandle,
templ: "twentyseventeen/posts/index.gohtml", templ: "twentyseventeen/posts/index.gohtml",
} }
h.WidgetAreaData()
h.Session = sessions.Default(h.C)
h.GinH["HeaderImage"] = h.getHeaderImage(h.C) h.GinH["HeaderImage"] = h.getHeaderImage(h.C)
if h.Scene == constraints.Detail { if h.Scene == constraints.Detail {
h.Detail() h.Detail()
@ -52,23 +55,25 @@ var pluginFns = func() map[string]common.Plugin[models.Posts] {
}) })
}() }()
func (h handle) Index() { func (h *handle) Index() {
if h.Stats != constraints.Empty404 { err := h.Indexs()
h.GinH["bodyClass"] = h.bodyClass()
if err != nil {
h.C.HTML(h.Code, h.templ, h.GinH)
return
}
h.ExecListPagePlugin(pluginFns) h.ExecListPagePlugin(pluginFns)
page, ok := maps.GetStrMapAnyVal[pagination.ParsePagination](h.GinH, "pagination") page, ok := maps.GetStrMapAnyVal[pagination.ParsePagination](h.GinH, "pagination")
if ok { if ok {
h.GinH["pagination"] = pagination.Paginate(paginate, page) h.GinH["pagination"] = pagination.Paginate(paginate, page)
} }
}
h.GinH["bodyClass"] = h.bodyClass()
h.C.HTML(h.Code, h.templ, h.GinH) h.C.HTML(h.Code, h.templ, h.GinH)
} }
func (h handle) Detail() { func (h *handle) Detail() {
post := h.GinH["post"].(models.Posts) post := h.GinH["post"].(models.Posts)
h.GinH["bodyClass"] = h.bodyClass() h.GinH["bodyClass"] = h.bodyClass()
if h.Stats == constraints.Empty404 { if h.Stats == constraints.Error404 {
h.C.HTML(h.Code, h.templ, h.GinH) h.C.HTML(h.Code, h.templ, h.GinH)
return return
} }
@ -110,7 +115,7 @@ func (c comment) FormatLi(ctx *gin.Context, m models.Comments, depth int, isTls
return plugins.FormatLi(templ, ctx, m, depth, isTls, eo, parent) return plugins.FormatLi(templ, ctx, m, depth, isTls, eo, parent)
} }
func postThumbnail(next common.Fn[models.Posts], h common.Handle, t models.Posts) models.Posts { func postThumbnail(next common.Fn[models.Posts], h *common.Handle, t models.Posts) models.Posts {
if t.Thumbnail.Path != "" { if t.Thumbnail.Path != "" {
t.Thumbnail.Sizes = "(max-width: 767px) 89vw, (max-width: 1000px) 54vw, (max-width: 1071px) 543px, 580px" t.Thumbnail.Sizes = "(max-width: 767px) 89vw, (max-width: 1000px) 54vw, (max-width: 1071px) 543px, 580px"
if h.Scene == constraints.Detail { if h.Scene == constraints.Detail {
@ -120,7 +125,7 @@ func postThumbnail(next common.Fn[models.Posts], h common.Handle, t models.Posts
return next(t) return next(t)
} }
func (h handle) getHeaderImage(c *gin.Context) (r models.PostThumbnail) { func (h *handle) getHeaderImage(c *gin.Context) (r models.PostThumbnail) {
r.Path = "/wp-content/themes/twentyseventeen/assets/images/header.jpg" r.Path = "/wp-content/themes/twentyseventeen/assets/images/header.jpg"
r.Width = 2000 r.Width = 2000
r.Height = 1200 r.Height = 1200
@ -135,9 +140,9 @@ func (h handle) getHeaderImage(c *gin.Context) (r models.PostThumbnail) {
return return
} }
func (h handle) bodyClass() string { func (h *handle) bodyClass() string {
s := "" s := ""
if h.Stats == constraints.Empty404 { if constraints.Ok != h.Stats {
return "error404" return "error404"
} }
switch h.Scene { switch h.Scene {