aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorpg9182 <96569817+pg9182@users.noreply.github.com>2022-10-11 17:33:23 -0400
committerpg9182 <96569817+pg9182@users.noreply.github.com>2022-10-11 17:33:23 -0400
commit25d1b93ee7aa562df436e3ad49afd0f5148161d6 (patch)
treecd3e340fadad2cac72ac1c0c65d5086e2d5ed1c6
parentbac8e828e955562d53b722e30332f770e62aa3fb (diff)
downloadAtlas-25d1b93ee7aa562df436e3ad49afd0f5148161d6.tar.gz
Atlas-25d1b93ee7aa562df436e3ad49afd0f5148161d6.zip
pkg/origin: Implement GetUserInfo
-rw-r--r--pkg/origin/origin.go145
-rw-r--r--pkg/origin/origin_test.go20
2 files changed, 165 insertions, 0 deletions
diff --git a/pkg/origin/origin.go b/pkg/origin/origin.go
index 57b1dbf..2638f3e 100644
--- a/pkg/origin/origin.go
+++ b/pkg/origin/origin.go
@@ -1,3 +1,148 @@
// Package origin is a client for parts of the Origin API used by Northstar for
// authentication.
package origin
+
+import (
+ "context"
+ "encoding/xml"
+ "fmt"
+ "io"
+ "net/http"
+ "strconv"
+ "strings"
+ "sync/atomic"
+)
+
+type SIDStore interface {
+ GetSID(ctx context.Context) (string, error)
+ SetSID(ctx context.Context, sid string) error
+}
+
+type MemorySIDStore struct {
+ SID atomic.Pointer[string]
+}
+
+var _ SIDStore = (*MemorySIDStore)(nil)
+
+func (s *MemorySIDStore) GetSID(ctx context.Context) (string, error) {
+ if v := s.SID.Load(); v != nil {
+ return *v, nil
+ }
+ return "", nil
+}
+
+func (s *MemorySIDStore) SetSID(ctx context.Context, sid string) error {
+ s.SID.Store(&sid)
+ return nil
+}
+
+type Client struct {
+ Endpoint string
+ Username string
+ Password string
+ SIDStore SIDStore
+ Transport http.Transport
+}
+
+func (c *Client) endpoint() string {
+ if c.Endpoint != "" {
+ return strings.TrimRight(c.Endpoint, "/")
+ }
+ return "https://api1.origin.com"
+}
+
+func (c *Client) do(req *http.Request) (*http.Response, error) {
+ return (&http.Client{
+ Transport: &c.Transport,
+ Jar: nil,
+ }).Do(req)
+}
+
+func (c *Client) Login(ctx context.Context) error {
+ panic("not implemented")
+}
+
+type UserInfo struct {
+ UserID int
+ PersonaID string
+ EAID string
+}
+
+func (c *Client) GetUserInfo(ctx context.Context, uid ...int) ([]UserInfo, error) {
+ return c.getUserInfo(true, ctx, uid...)
+}
+
+func (c *Client) getUserInfo(retry bool, ctx context.Context, uid ...int) ([]UserInfo, error) {
+ uids := make([]string, len(uid))
+ for _, x := range uid {
+ uids = append(uids, strconv.Itoa(x))
+ }
+
+ req, err := http.NewRequestWithContext(ctx, http.MethodGet, c.endpoint()+"/atom/users?userIds="+strings.Join(uids, ","), nil)
+ if err != nil {
+ return nil, err
+ }
+
+ sid, err := c.SIDStore.GetSID(ctx)
+ if err != nil {
+ return nil, err
+ }
+
+ req.Header.Set("AuthToken", sid)
+
+ resp, err := c.do(req)
+ if err != nil {
+ return nil, err
+ }
+ defer resp.Body.Close()
+
+ if needAuth, err := checkResponse(resp); err != nil {
+ if retry && needAuth {
+ if err := c.Login(ctx); err != nil {
+ return nil, err
+ }
+ return c.getUserInfo(false, ctx, uid...)
+ }
+ return nil, err
+ }
+ return parseUserInfo(resp.Body)
+}
+
+func checkResponse(resp *http.Response) (bool, error) {
+ // TODO: return true and err for auth required
+ if resp.StatusCode != http.StatusOK {
+ return false, fmt.Errorf("response status %q", resp.Status)
+ }
+ return false, nil
+}
+
+func parseUserInfo(r io.Reader) ([]UserInfo, error) {
+ var obj struct {
+ XMLName xml.Name `xml:"users"`
+ User []struct {
+ UserID string `xml:"userId"`
+ PersonaID string `xml:"personaId"`
+ EAID string `xml:"EAID"`
+ } `xml:"user"`
+ }
+ buf, err := io.ReadAll(r)
+ if err != nil {
+ return nil, err
+ }
+ if err := xml.Unmarshal(buf, &obj); err != nil {
+ return nil, err
+ }
+ res := make([]UserInfo, len(obj.User))
+ for i, x := range obj.User {
+ var v UserInfo
+ if uid, err := strconv.Atoi(x.UserID); err == nil {
+ v.UserID = uid
+ } else {
+ return nil, fmt.Errorf("parse userId %q: %w", x.UserID, err)
+ }
+ v.PersonaID = x.PersonaID
+ v.EAID = x.EAID
+ res[i] = v
+ }
+ return res, nil
+}
diff --git a/pkg/origin/origin_test.go b/pkg/origin/origin_test.go
new file mode 100644
index 0000000..ac06d82
--- /dev/null
+++ b/pkg/origin/origin_test.go
@@ -0,0 +1,20 @@
+package origin
+
+import (
+ "reflect"
+ "strings"
+ "testing"
+)
+
+func TestUserInfoResponse(t *testing.T) {
+ ui, err := parseUserInfo(strings.NewReader(`<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?><users><user><userId>1001111111111</userId><personaId>1001111111111</personaId><EAID>test</EAID></user><user><userId>1001111111112</userId><personaId>1001111111112</personaId><EAID>test1</EAID></user></users>`))
+ if err != nil {
+ t.Fatal(err)
+ }
+ if !reflect.DeepEqual(ui, []UserInfo{
+ {UserID: 1001111111111, PersonaID: "1001111111111", EAID: "test"},
+ {UserID: 1001111111112, PersonaID: "1001111111112", EAID: "test1"},
+ }) {
+ t.Errorf("unexpected result %#v", ui)
+ }
+}