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")
})
}
|