aboutsummaryrefslogtreecommitdiff
path: root/utils/meta/pagination.go
blob: 92442948281a37ea5ddbb9946696d2b8411937c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package meta

import (
	"dove/types"
	"strconv"

	"github.com/gofiber/fiber/v2"
	"gorm.io/gorm"
)

func Paginate(context *fiber.Ctx) Pagination {
	requestData := Request(context)

	page := DEFAULT_PAGE
	perPage := DEFAULT_PER_PAGE

	if requestData != nil {
		if pageValue := requestData.Query("page"); pageValue != nil {
			parsed, _ := strconv.Atoi(pageValue.String())
			if parsed >= DEFAULT_PAGE {
				page = parsed
			}
		}

		if perPageValue := requestData.Query("per_page"); perPageValue != nil {
			parsed, _ := strconv.Atoi(perPageValue.String())
			if parsed >= DEFAULT_PAGE && parsed <= MAX_PER_PAGE {
				perPage = parsed
			}
		}
	}

	return Pagination{Page: page, PerPage: perPage}
}

func Sort(context *fiber.Ctx, allowedFields []string, fallbackField string) Sorting {
	requestData := Request(context)

	field := fallbackField
	direction := "desc"

	if requestData != nil {
		if sortValue := requestData.Query("sort"); sortValue != nil {
			for _, allowedField := range allowedFields {
				if sortValue.String() == allowedField {
					field = sortValue.String()
					break
				}
			}
		}

		if orderValue := requestData.Query("order"); orderValue != nil {
			switch orderValue.String() {
			case "asc", "desc":
				direction = orderValue.String()
			}
		}
	}

	return Sorting{Field: field, Direction: direction}
}

func (self Pagination) Apply(query *gorm.DB) *gorm.DB {
	return query.Offset((self.Page - 1) * self.PerPage).Limit(self.PerPage)
}

func (self Sorting) Apply(query *gorm.DB) *gorm.DB {
	return query.Order(self.Field + " " + self.Direction)
}

func (self Pagination) Response(items any, total int64) types.PaginatedResponse {
	totalPages := int(total) / self.PerPage
	if int(total)%self.PerPage > 0 {
		totalPages++
	}

	return types.PaginatedResponse{
		Items:      items,
		Total:      total,
		Page:       self.Page,
		PerPage:    self.PerPage,
		TotalPages: totalPages,
	}
}