File: user.go

package info (click to toggle)
golang-github-henrybear327-go-proton-api 1.0.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,088 kB
  • sloc: sh: 55; makefile: 26
file content (59 lines) | stat: -rw-r--r-- 1,303 bytes parent folder | download
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")
	})
}