aboutsummaryrefslogtreecommitdiffhomepage
path: root/backend/graphql/resolver/pagination.go
blob: 1a14650626e0e368d9b8ddca6c854fbecf41d48c (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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
package resolver

import (
	"context"
	"fmt"
	"strconv"

	"undef.ninja/x/feedaka/db"
	"undef.ninja/x/feedaka/graphql/model"
)

const defaultPageSize = 30
const maxPageSize = 100

// articleRow is a common interface for all paginated article query rows.
type articleRow struct {
	ID               int64
	FeedID           int64
	Guid             string
	Title            string
	Url              string
	IsRead           int64
	FeedID2          int64
	FeedUrl          string
	FeedTitle        string
	FeedIsSubscribed int64
}

func toArticleRow(r any) articleRow {
	switch v := r.(type) {
	case db.GetArticlesPaginatedRow:
		return articleRow{v.ID, v.FeedID, v.Guid, v.Title, v.Url, v.IsRead, v.FeedID2, v.FeedUrl, v.FeedTitle, v.FeedIsSubscribed}
	case db.GetArticlesPaginatedAfterRow:
		return articleRow{v.ID, v.FeedID, v.Guid, v.Title, v.Url, v.IsRead, v.FeedID2, v.FeedUrl, v.FeedTitle, v.FeedIsSubscribed}
	case db.GetArticlesByFeedPaginatedRow:
		return articleRow{v.ID, v.FeedID, v.Guid, v.Title, v.Url, v.IsRead, v.FeedID2, v.FeedUrl, v.FeedTitle, v.FeedIsSubscribed}
	case db.GetArticlesByFeedPaginatedAfterRow:
		return articleRow{v.ID, v.FeedID, v.Guid, v.Title, v.Url, v.IsRead, v.FeedID2, v.FeedUrl, v.FeedTitle, v.FeedIsSubscribed}
	default:
		panic("unexpected row type")
	}
}

func rowToArticle(row articleRow) *model.Article {
	return &model.Article{
		ID:     strconv.FormatInt(row.ID, 10),
		FeedID: strconv.FormatInt(row.FeedID, 10),
		GUID:   row.Guid,
		Title:  row.Title,
		URL:    row.Url,
		IsRead: row.IsRead == 1,
		Feed: &model.Feed{
			ID:           strconv.FormatInt(row.FeedID2, 10),
			URL:          row.FeedUrl,
			Title:        row.FeedTitle,
			IsSubscribed: row.FeedIsSubscribed == 1,
		},
	}
}

func (r *queryResolver) paginatedArticles(ctx context.Context, isRead int64, feedID *string, after *string, first *int32) (*model.ArticleConnection, error) {
	userID, err := getUserIDFromContext(ctx)
	if err != nil {
		return nil, err
	}

	limit := int64(defaultPageSize)
	if first != nil {
		limit = int64(*first)
		if limit <= 0 {
			limit = int64(defaultPageSize)
		}
		if limit > maxPageSize {
			limit = maxPageSize
		}
	}

	// Fetch limit+1 to determine hasNextPage
	fetchLimit := limit + 1

	var rawRows []any

	if feedID != nil {
		parsedFeedID, err := strconv.ParseInt(*feedID, 10, 64)
		if err != nil {
			return nil, fmt.Errorf("invalid feed ID: %w", err)
		}

		if after != nil {
			cursor, err := strconv.ParseInt(*after, 10, 64)
			if err != nil {
				return nil, fmt.Errorf("invalid cursor: %w", err)
			}
			rows, err := r.Queries.GetArticlesByFeedPaginatedAfter(ctx, db.GetArticlesByFeedPaginatedAfterParams{
				IsRead: isRead,
				UserID: userID,
				FeedID: parsedFeedID,
				ID:     cursor,
				Limit:  fetchLimit,
			})
			if err != nil {
				return nil, fmt.Errorf("failed to query articles: %w", err)
			}
			for _, row := range rows {
				rawRows = append(rawRows, row)
			}
		} else {
			rows, err := r.Queries.GetArticlesByFeedPaginated(ctx, db.GetArticlesByFeedPaginatedParams{
				IsRead: isRead,
				UserID: userID,
				FeedID: parsedFeedID,
				Limit:  fetchLimit,
			})
			if err != nil {
				return nil, fmt.Errorf("failed to query articles: %w", err)
			}
			for _, row := range rows {
				rawRows = append(rawRows, row)
			}
		}
	} else {
		if after != nil {
			cursor, err := strconv.ParseInt(*after, 10, 64)
			if err != nil {
				return nil, fmt.Errorf("invalid cursor: %w", err)
			}
			rows, err := r.Queries.GetArticlesPaginatedAfter(ctx, db.GetArticlesPaginatedAfterParams{
				IsRead: isRead,
				UserID: userID,
				ID:     cursor,
				Limit:  fetchLimit,
			})
			if err != nil {
				return nil, fmt.Errorf("failed to query articles: %w", err)
			}
			for _, row := range rows {
				rawRows = append(rawRows, row)
			}
		} else {
			rows, err := r.Queries.GetArticlesPaginated(ctx, db.GetArticlesPaginatedParams{
				IsRead: isRead,
				UserID: userID,
				Limit:  fetchLimit,
			})
			if err != nil {
				return nil, fmt.Errorf("failed to query articles: %w", err)
			}
			for _, row := range rows {
				rawRows = append(rawRows, row)
			}
		}
	}

	hasNextPage := int64(len(rawRows)) > limit
	if hasNextPage {
		rawRows = rawRows[:limit]
	}

	articles := make([]*model.Article, 0, len(rawRows))
	for _, raw := range rawRows {
		articles = append(articles, rowToArticle(toArticleRow(raw)))
	}

	var endCursor *string
	if len(articles) > 0 {
		lastID := articles[len(articles)-1].ID
		endCursor = &lastID
	}

	return &model.ArticleConnection{
		Articles: articles,
		PageInfo: &model.PageInfo{
			HasNextPage: hasNextPage,
			EndCursor:   endCursor,
		},
	}, nil
}