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
|
package proton_api_bridge
import (
"context"
"github.com/henrybear327/go-proton-api"
)
func (protonDrive *ProtonDrive) moveToTrash(ctx context.Context, parentLinkID string, linkIDs ...string) error {
err := protonDrive.c.TrashChildren(ctx, protonDrive.MainShare.ShareID, parentLinkID, linkIDs...)
if err != nil {
return err
}
for _, link := range linkIDs {
protonDrive.removeLinkIDFromCache(link, true)
}
return nil
}
func (protonDrive *ProtonDrive) MoveFileToTrashByID(ctx context.Context, linkID string) error {
/* It's like event system, we need to get the latest information before creating the move request! */
protonDrive.removeLinkIDFromCache(linkID, false)
fileLink, err := protonDrive.getLink(ctx, linkID)
if err != nil {
return err
}
if fileLink.Type != proton.LinkTypeFile {
return ErrLinkTypeMustToBeFileType
}
return protonDrive.moveToTrash(ctx, fileLink.ParentLinkID, linkID)
}
func (protonDrive *ProtonDrive) MoveFolderToTrashByID(ctx context.Context, linkID string, onlyOnEmpty bool) error {
/* It's like event system, we need to get the latest information before creating the move request! */
protonDrive.removeLinkIDFromCache(linkID, false)
folderLink, err := protonDrive.getLink(ctx, linkID)
if err != nil {
return err
}
if folderLink.Type != proton.LinkTypeFolder {
return ErrLinkTypeMustToBeFolderType
}
childrenLinks, err := protonDrive.c.ListChildren(ctx, protonDrive.MainShare.ShareID, linkID /* false: list only active ones */, false)
if err != nil {
return err
}
if onlyOnEmpty {
if len(childrenLinks) > 0 {
return ErrFolderIsNotEmpty
}
}
return protonDrive.moveToTrash(ctx, folderLink.ParentLinkID, linkID)
}
// WARNING!!!!
// Everything in the root folder will be moved to trash
// Most likely only used for debugging when the key is messed up
func (protonDrive *ProtonDrive) EmptyRootFolder(ctx context.Context) error {
protonDrive.ClearCache()
links, err := protonDrive.c.ListChildren(ctx, protonDrive.MainShare.ShareID, protonDrive.MainShare.LinkID, true)
if err != nil {
return err
}
{
linkIDs := make([]string, 0)
for i := range links {
if links[i].State == proton.LinkStateActive /* use TrashChildren */ {
linkIDs = append(linkIDs, links[i].LinkID)
}
}
err := protonDrive.c.TrashChildren(ctx, protonDrive.MainShare.ShareID, protonDrive.MainShare.LinkID, linkIDs...)
if err != nil {
return err
}
}
{
linkIDs := make([]string, 0)
for i := range links {
if links[i].State != proton.LinkStateActive {
linkIDs = append(linkIDs, links[i].LinkID)
}
}
err := protonDrive.c.DeleteChildren(ctx, protonDrive.MainShare.ShareID, protonDrive.MainShare.LinkID, linkIDs...)
if err != nil {
return err
}
}
return nil
}
// Empty the trash
func (protonDrive *ProtonDrive) EmptyTrash(ctx context.Context) error {
protonDrive.ClearCache()
err := protonDrive.c.EmptyTrash(ctx, protonDrive.MainShare.ShareID)
if err != nil {
return err
}
return nil
}
|
