125 lines
2.6 KiB
Go
125 lines
2.6 KiB
Go
package pagination
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/fthvgb1/wp-go/helper"
|
|
"math"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type Elements interface {
|
|
Current(page int) string
|
|
Prev(url string) string
|
|
Next(url string) string
|
|
Dots() string
|
|
Middle(page int, url string) string
|
|
}
|
|
|
|
type ParsePagination struct {
|
|
Elements
|
|
TotalPage int
|
|
TotalRaw int
|
|
PageSize int
|
|
CurrentPage int
|
|
Query string
|
|
Path string
|
|
Step int
|
|
}
|
|
|
|
func NewParsePagination(totalRaw int, pageSize int, currentPage int, query string, path string, step int) ParsePagination {
|
|
allPage := int(math.Ceil(float64(totalRaw) / float64(pageSize)))
|
|
return ParsePagination{TotalPage: allPage, TotalRaw: totalRaw, PageSize: pageSize, CurrentPage: currentPage, Query: query, Path: path, Step: step}
|
|
}
|
|
|
|
func Paginate(e Elements, p ParsePagination) string {
|
|
p.Elements = e
|
|
return p.ToHtml()
|
|
}
|
|
|
|
var complie = regexp.MustCompile(`(/page)/(\d+)`)
|
|
|
|
func (p ParsePagination) ToHtml() (html string) {
|
|
if p.TotalRaw < 2 {
|
|
return
|
|
}
|
|
pathx := p.Path
|
|
if !strings.Contains(p.Path, "/page/") {
|
|
pathx = fmt.Sprintf("%s%s", p.Path, "/page/1")
|
|
}
|
|
s := strings.Builder{}
|
|
if p.CurrentPage > p.TotalPage {
|
|
p.CurrentPage = p.TotalPage
|
|
}
|
|
r := complie
|
|
start := p.CurrentPage - p.Step
|
|
end := p.CurrentPage + p.Step
|
|
if start < 1 {
|
|
start = 1
|
|
}
|
|
if p.CurrentPage > 1 {
|
|
pp := ""
|
|
if p.CurrentPage >= 2 {
|
|
pp = replacePage(r, pathx, p.CurrentPage-1)
|
|
}
|
|
s.WriteString(p.Prev(helper.StrJoin(pp, p.Query)))
|
|
}
|
|
if p.CurrentPage >= p.Step+2 {
|
|
d := false
|
|
if p.CurrentPage > p.Step+2 {
|
|
d = true
|
|
}
|
|
e := replacePage(r, p.Path, 1)
|
|
s.WriteString(p.Middle(1, helper.StrJoin(e, p.Query)))
|
|
if d {
|
|
s.WriteString(p.Dots())
|
|
}
|
|
}
|
|
if p.TotalPage < end {
|
|
end = p.TotalPage
|
|
}
|
|
|
|
for page := start; page <= end; page++ {
|
|
h := ""
|
|
if p.CurrentPage == page {
|
|
h = p.Current(page)
|
|
} else {
|
|
d := replacePage(r, pathx, page)
|
|
h = p.Middle(page, helper.StrJoin(d, p.Query))
|
|
}
|
|
s.WriteString(h)
|
|
|
|
}
|
|
if p.TotalPage >= p.CurrentPage+p.Step+1 {
|
|
d := false
|
|
if p.TotalPage > p.CurrentPage+p.Step+1 {
|
|
d = true
|
|
}
|
|
dd := replacePage(r, pathx, p.TotalPage)
|
|
if d {
|
|
s.WriteString(p.Dots())
|
|
}
|
|
s.WriteString(p.Middle(p.TotalPage, helper.StrJoin(dd, p.Query)))
|
|
}
|
|
if p.CurrentPage < p.TotalPage {
|
|
dd := replacePage(r, pathx, p.CurrentPage+1)
|
|
s.WriteString(p.Next(helper.StrJoin(dd, p.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
|
|
}
|