mirror/go-proton-api
1
0
Fork 0
mirror of https://github.com/ProtonMail/go-proton-api.git synced 2025-12-23 23:57:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: Use paging in ListChildren

Browse Source
This commit is contained in:
James Houlahan
2023-02-26 16:20:20 +01:00
committed by James
parent 1bad007170
commit fdedd64a6f
1 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
link_folder.go
View File

@@ -3,6 +3,7 @@ package proton
import (
"context"
"fmt"
"strconv"
"github.com/bradenaw/juniper/xslices"
"github.com/go-resty/resty/v2"
@@ -13,13 +14,29 @@ func (c *Client) ListChildren(ctx context.Context, shareID, linkID string) ([]Li
Links []Link
}
if err := c.do(ctx, func(r *resty.Request) (*resty.Response, error) {
return r.SetResult(&res).Get("/drive/shares/" + shareID + "/folders/" + linkID + "/children")
}); err != nil {
return nil, err
var links []Link
for page := 0; ; page++ {
if err := c.do(ctx, func(r *resty.Request) (*resty.Response, error) {
return r.
SetQueryParams(map[string]string{
"Page": strconv.Itoa(page),
"PageSize": strconv.Itoa(maxPageSize),
}).
SetResult(&res).
Get("/drive/shares/" + shareID + "/folders/" + linkID + "/children")
}); err != nil {
return nil, err
}
if len(res.Links) == 0 {
break
}
links = append(links, res.Links...)
}
return res.Links, nil
return links, nil
}
func (c *Client) TrashChildren(ctx context.Context, shareID, linkID string, childIDs ...string) error {
@@ -34,7 +51,7 @@ func (c *Client) TrashChildren(ctx context.Context, shareID, linkID string, chil
}
}
for _, childIDs := range xslices.Chunk(childIDs, 150) {
for _, childIDs := range xslices.Chunk(childIDs, maxPageSize) {
req := struct {
LinkIDs []string
}{
@@ -69,7 +86,7 @@ func (c *Client) DeleteChildren(ctx context.Context, shareID, linkID string, chi
}
}
for _, childIDs := range xslices.Chunk(childIDs, 150) {
for _, childIDs := range xslices.Chunk(childIDs, maxPageSize) {
req := struct {
LinkIDs []string
}{