forked from ProtonMail/go-proton-api
/
user.go
59 lines (49 loc) · 1.27 KB
/
user.go
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
package proton
import (
"context"
"encoding/base64"
"github.com/ProtonMail/go-srp"
"github.com/go-resty/resty/v2"
)
func (c *Client) GetUser(ctx context.Context) (User, error) {
var res struct {
User User
}
if err := c.do(ctx, func(r *resty.Request) (*resty.Response, error) {
return r.SetResult(&res).Get("/core/v4/users")
}); err != nil {
return User{}, err
}
return res.User, nil
}
func (c *Client) DeleteUser(ctx context.Context, password []byte, req DeleteUserReq) error {
user, err := c.GetUser(ctx)
if err != nil {
return err
}
info, err := c.m.AuthInfo(ctx, AuthInfoReq{Username: user.Name})
if err != nil {
return err
}
srpAuth, err := srp.NewAuth(info.Version, user.Name, password, info.Salt, info.Modulus, info.ServerEphemeral)
if err != nil {
return err
}
proofs, err := srpAuth.GenerateProofs(2048)
if err != nil {
return err
}
return c.do(ctx, func(r *resty.Request) (*resty.Response, error) {
return r.SetBody(struct {
DeleteUserReq
AuthReq
}{
DeleteUserReq: req,
AuthReq: AuthReq{
ClientProof: base64.StdEncoding.EncodeToString(proofs.ClientProof),
ClientEphemeral: base64.StdEncoding.EncodeToString(proofs.ClientEphemeral),
SRPSession: info.SRPSession,
},
}).Delete("/core/v4/users/delete")
})
}