Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-19 16:46:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

GODT-35: New pmapi client and manager using resty

Browse Source
This commit is contained in:
James Houlahan
2021-02-22 18:23:51 +01:00
committed by Jakub
parent 1d538e8540
commit 2284e9ede1
163 changed files with 3333 additions and 8124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
test/liveapi/cleanup.go
View File

@ -18,6 +18,7 @@
package liveapi
import (
"context"
"time"
"github.com/ProtonMail/proton-bridge/pkg/pmapi"
@ -43,7 +44,7 @@ func cleanup(client pmapi.Client, addresses *pmapi.AddressList) error {
func cleanSystemFolders(client pmapi.Client) error {
for _, labelID := range []string{pmapi.InboxLabel, pmapi.SentLabel, pmapi.ArchiveLabel, pmapi.AllMailLabel, pmapi.DraftLabel} {
for {
messages, total, err := client.ListMessages(&pmapi.MessagesFilter{
messages, total, err := client.ListMessages(context.TODO(), &pmapi.MessagesFilter{
PageSize: 150,
LabelID: labelID,
})
@ -60,7 +61,7 @@ func cleanSystemFolders(client pmapi.Client) error {
messageIDs = append(messageIDs, message.ID)
}
if err := client.DeleteMessages(messageIDs); err != nil {
if err := client.DeleteMessages(context.TODO(), messageIDs); err != nil {
return errors.Wrap(err, "failed to delete messages")
}
@ -73,7 +74,7 @@ func cleanSystemFolders(client pmapi.Client) error {
}
func cleanCustomLables(client pmapi.Client) error {
labels, err := client.ListLabels()
labels, err := client.ListLabels(context.TODO())
if err != nil {
return errors.Wrap(err, "failed to list labels")
}
@ -82,7 +83,7 @@ func cleanCustomLables(client pmapi.Client) error {
if err := emptyFolder(client, label.ID); err != nil {
return errors.Wrap(err, "failed to empty label")
}
if err := client.DeleteLabel(label.ID); err != nil {
if err := client.DeleteLabel(context.TODO(), label.ID); err != nil {
return errors.Wrap(err, "failed to delete label")
}
}
@ -92,7 +93,7 @@ func cleanCustomLables(client pmapi.Client) error {
func cleanTrash(client pmapi.Client) error {
for {
_, total, err := client.ListMessages(&pmapi.MessagesFilter{
_, total, err := client.ListMessages(context.TODO(), &pmapi.MessagesFilter{
PageSize: 1,
LabelID: pmapi.TrashLabel,
})
@ -114,12 +115,12 @@ func cleanTrash(client pmapi.Client) error {
}
func emptyFolder(client pmapi.Client, labelID string) error {
err := client.EmptyFolder(labelID, "")
err := client.EmptyFolder(context.TODO(), labelID, "")
if err != nil {
return err
}
for {
_, total, err := client.ListMessages(&pmapi.MessagesFilter{
_, total, err := client.ListMessages(context.TODO(), &pmapi.MessagesFilter{
PageSize: 1,
LabelID: labelID,
})
@ -141,5 +142,5 @@ func reorderAddresses(client pmapi.Client, addresses *pmapi.AddressList) error {
addressIDs = append(addressIDs, address.ID)
}
return client.ReorderAddresses(addressIDs)
return client.ReorderAddresses(context.TODO(), addressIDs)
}