forked from vulncheck-oss/sdk
-
Notifications
You must be signed in to change notification settings - Fork 1
/
me.go
70 lines (64 loc) · 2.5 KB
/
me.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
60
61
62
63
64
65
66
67
68
69
70
package sdk
import (
"encoding/json"
"fmt"
)
type UserData struct {
ID int `json:"ID"`
Email string `json:"Email"`
Name string `json:"Name"`
Avatar string `json:"Avatar"`
Payload interface{} `json:"Payload"`
Stripe string `json:"Stripe"`
Terms bool `json:"Terms"`
Roles []string `json:"Roles"`
Settings interface{} `json:"Settings"`
Providers interface{} `json:"Providers"`
Teams interface{} `json:"Teams"`
URL string `json:"URL"`
Initials string `json:"Initials"`
TrialDays int `json:"TrialDays"`
Pivot interface{} `json:"Pivot"`
OrgRoles interface{} `json:"OrgRoles"`
CurrentToken interface{} `json:"CurrentToken"`
ActivatedAt interface{} `json:"ActivatedAt"`
OrgUsers interface{} `json:"OrgUsers"`
Orgs interface{} `json:"Orgs"`
CurrentOrg interface{} `json:"CurrentOrg"`
OrgID interface{} `json:"OrgID"`
Org interface{} `json:"Org"`
IsServiceUser bool `json:"IsServiceUser"`
HasEmployeeRole bool `json:"HasEmployeeRole"`
HasEmployeeAdminRole bool `json:"HasEmployeeAdminRole"`
HasOrgManagerRole bool `json:"HasOrgManagerRole"`
HasTrial bool `json:"HasTrial"`
HasInitial bool `json:"HasInitial"`
HasVuln bool `json:"HasVuln"`
HasAgent bool `json:"HasAgent"`
HasSbom bool `json:"HasSbom"`
OnlyCommunity bool `json:"OnlyCommunity"`
CreatedAt string `json:"created_at"`
UpdatedAt string `json:"updated_at"`
}
type UserResponse struct {
Benchmark float64 `json:"_benchmark"`
Data UserData `json:"data"`
}
// GetMe https://docs.vulncheck.com/api/me
func (c *Client) GetMe() (responseJSON *UserResponse, err error) {
resp, err := c.Request("GET", "/me")
if err != nil {
return nil, err
}
defer resp.Body.Close()
_ = json.NewDecoder(resp.Body).Decode(&responseJSON)
return responseJSON, nil
}
// Strings representation of the response
func (r UserResponse) String() string {
return fmt.Sprintf("Benchmark: %v\nData: %v\n", r.Benchmark, r.Data)
}
// GetData Returns the data from the response
func (r UserResponse) GetData() UserData {
return r.Data
}