package actions import ( "context" "fmt" "github.com/gin-contrib/sessions" "github.com/gin-gonic/gin" "github/fthvgb1/wp-go/actions/common" "github/fthvgb1/wp-go/helper" "github/fthvgb1/wp-go/models" "math" "net/http" "regexp" "strconv" "strings" ) type indexHandle struct { c *gin.Context session sessions.Session page int pageSize int title string titleL string titleR string search string totalPage int category string categoryType string where models.SqlBuilder orderBy models.SqlBuilder order string join models.SqlBuilder postType []any status []any header string paginationStep int } func newIndexHandle(ctx *gin.Context) *indexHandle { return &indexHandle{ c: ctx, session: sessions.Default(ctx), page: 1, pageSize: 10, paginationStep: 1, titleL: models.Options["blogname"], titleR: models.Options["blogdescription"], where: models.SqlBuilder{ {"post_type", "in", ""}, {"post_status", "in", ""}, }, orderBy: models.SqlBuilder{}, join: models.SqlBuilder{}, postType: []any{"post"}, status: []any{"publish"}, } } 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) parseParams() { h.order = h.c.Query("order") if !helper.IsContainInArr(h.order, []string{"asc", "desc"}) { h.order = "asc" } year := h.c.Param("year") if year != "" { h.where = append(h.where, []string{ "year(post_date)", year, }) } month := h.c.Param("month") if 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("月度归档: %s", ss) h.setTitleLR(ss, models.Options["blogname"]) } category := h.c.Param("category") if category == "" { category = h.c.Param("tag") if category != "" { h.categoryType = "post_tag" h.header = fmt.Sprintf("标签: %s", category) } } else { h.categoryType = "category" h.header = fmt.Sprintf("分类: %s", category) } h.category = category if category != "" { h.where = append(h.where, []string{ "d.name", 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(category, models.Options["blogname"]) } s := h.c.Query("s") if s != "" && strings.Replace(s, " ", "", -1) != "" { q := helper.StrJoin("%", 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("%s的搜索结果", s) h.setTitleLR(helper.StrJoin(`"`, s, `"`, "的搜索结果"), models.Options["blogname"]) h.search = s } else { h.status = append(h.status, "private") } p := h.c.Query("paged") if p == "" { p = h.c.Param("page") } if p != "" { if pa, err := strconv.Atoi(p); err == nil { h.page = pa } } if h.page > 1 && (h.category != "" || h.search != "" || month != "") { h.setTitleLR(fmt.Sprintf("%s-第%d页", h.titleL, h.page), models.Options["blogname"]) } } 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) h.parseParams() ctx := context.TODO() archive := common.Archives() recent := common.RecentPosts(ctx) categoryItems := common.Categories(ctx) comments := common.RecentComments(ctx) ginH := gin.H{ "options": models.Options, "recentPosts": recent, "archives": archive, "categories": categoryItems, "search": h.search, "header": h.header, "title": h.getTitle(), "comments": comments, } postIds, totalRaw, err := models.SimplePagination[models.WpPosts](h.where, "ID", "", h.page, h.pageSize, h.orderBy, h.join, h.postType, h.status) defer func() { c.HTML(http.StatusOK, "posts/index.gohtml", ginH) if err != nil { c.Error(err) } }() if err != nil { return } if len(postIds) < 1 && h.category != "" { h.titleL = "未找到页面" } err = common.QueryAndSetPostCache(postIds) pw := h.session.Get("post_password") for i, v := range postIds { post, _ := common.PostsCache.Load(v.Id) pp := post.(*models.WpPosts) px := *pp common.PasswordProjectTitle(&px) if px.PostPassword != "" && pw != px.PostPassword { common.PasswdProjectContent(&px) } postIds[i] = px } for i, post := range recent { if post.PostPassword != "" && pw != post.PostPassword { common.PasswdProjectContent(&recent[i]) } } q := c.Request.URL.Query().Encode() ginH["posts"] = postIds ginH["totalPage"] = h.getTotalPage(totalRaw) ginH["pagination"] = pagination(h.page, h.totalPage, h.paginationStep, c.Request.URL.Path, q) ginH["title"] = h.getTitle() } func pagination(currentPage, totalPage, step int, path, query string) (html string) { if totalPage < 2 { return } pathx := path if !strings.Contains(path, "/page/") { pathx = fmt.Sprintf("%s%s", path, "/page/1") } s := strings.Builder{} if currentPage > totalPage { currentPage = totalPage } r := regexp.MustCompile(`(/page)/(\d+)`) start := currentPage - step end := currentPage + step if start < 1 { start = 1 } if currentPage > 1 { pp := "" if currentPage >= 2 { pp = replacePage(r, pathx, currentPage-1) } s.WriteString(fmt.Sprintf(``, pp, query)) } if currentPage >= step+2 { d := "" if currentPage > step+2 { d = `` } e := replacePage(r, path, 1) s.WriteString(fmt.Sprintf(` 1 %s `, e, query, d)) } if totalPage < end { end = totalPage } for page := start; page <= end; page++ { h := "" if currentPage == page { h = fmt.Sprintf(` %d `, page) } else { d := replacePage(r, pathx, page) h = fmt.Sprintf(` %d `, d, query, page) } s.WriteString(h) } if totalPage >= currentPage+step+1 { if totalPage > currentPage+step+1 { s.WriteString(``) } dd := replacePage(r, pathx, totalPage) s.WriteString(fmt.Sprintf(` %d`, dd, query, totalPage)) } if currentPage < totalPage { dd := replacePage(r, pathx, currentPage+1) s.WriteString(fmt.Sprintf(``, dd, query)) } html = s.String() return } func replacePage(r *regexp.Regexp, path string, page int) (src string) { if page == 1 { src = r.ReplaceAllString(path, "") } else { s := fmt.Sprintf("$1/%d", page) src = r.ReplaceAllString(path, s) } src = strings.Replace(src, "//", "/", -1) if src == "" { src = "/" } return }