feat(auth): add token api

Author: duanhongyi

api/token.go

@@ -0,0 +1,11 @@
+package api
+
+// Token is the structure of the token object.
+type Token struct {
+	UUID    string `json:"uuid"`
+	Owner   string `json:"owner"`
+	Alias   string `json:"alias"`
+	Key     string `json:"fuzzy_key"`
+	Created string `json:"created"`
+	Updated string `json:"updated"`
+}

auth/auth.go

@@ -44,8 +44,8 @@ func Login(c *drycc.Client, username, password string) (string, error) {
 }
 
 // Token to the controller and get a token
-func Token(c *drycc.Client, key string) (api.AuthTokenResponse, error) {
-	path := fmt.Sprintf("/v2/auth/token/%s/", key)
+func Token(c *drycc.Client, key, alias string) (api.AuthTokenResponse, error) {
+	path := fmt.Sprintf("/v2/auth/token/%s/?alias=%s", key, alias)
 	res, reqErr := c.Request("GET", path, nil)
 	if reqErr != nil && !drycc.IsErrAPIMismatch(reqErr) {
 		return api.AuthTokenResponse{}, reqErr

auth/auth_test.go

@@ -29,7 +29,7 @@ func (f *fakeHTTPServer) ServeHTTP(res http.ResponseWriter, req *http.Request) {
 			res.Write(nil)
 		}
 		if len(body) == 0 {
-			res.Header().Add("Location", fmt.Sprintf("/v2/login/drycc/?key=%s/", keyFixture))
+			res.Header().Add("Location", fmt.Sprintf("/v2/login/drycc/?key=%s/?alias=test", keyFixture))
 			res.WriteHeader(http.StatusFound)
 			res.Write(nil)
 		} else {
@@ -66,7 +66,7 @@ func TestLogin(t *testing.T) {
 		t.Error(err)
 	}
 
-	expected := fmt.Sprintf("/v2/login/drycc/?key=%s/", keyFixture)
+	expected := fmt.Sprintf("/v2/login/drycc/?key=%s/?alias=test", keyFixture)
 	if actual != expected {
 		t.Errorf("Expected %s, Got %s", expected, actual)
 	}
@@ -97,7 +97,7 @@ func TestToken(t *testing.T) {
 		Token:    "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ.SflKxwRJSMeKKF2QT4fwpMeJf36POk6yJV_adQssw5c",
 	}
 
-	token, err := Token(drycc, keyFixture)
+	token, err := Token(drycc, keyFixture, "test")
 
 	if err != nil {
 		t.Error(err)

tokens/tokens.go

@@ -0,0 +1,35 @@
+package tokens
+
+import (
+	"encoding/json"
+	"fmt"
+
+	drycc "github.com/drycc/controller-sdk-go"
+	"github.com/drycc/controller-sdk-go/api"
+)
+
+// List tokens.
+func List(c *drycc.Client, results int) ([]api.Token, int, error) {
+	body, count, reqErr := c.LimitedRequest("/v2/tokens/", results)
+
+	if reqErr != nil && !drycc.IsErrAPIMismatch(reqErr) {
+		return []api.Token{}, -1, reqErr
+	}
+
+	var tokens []api.Token
+	if err := json.Unmarshal([]byte(body), &tokens); err != nil {
+		return []api.Token{}, -1, err
+	}
+
+	return tokens, count, reqErr
+}
+
+// Delete a token
+func Delete(c *drycc.Client, id string) error {
+	u := fmt.Sprintf("/v2/tokens/%s/", id)
+	res, err := c.Request("DELETE", u, nil)
+	if err == nil {
+		res.Body.Close()
+	}
+	return err
+}

tokens/tokens_test.go

@@ -0,0 +1,117 @@
+package tokens
+
+import (
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"reflect"
+	"testing"
+
+	drycc "github.com/drycc/controller-sdk-go"
+	"github.com/drycc/controller-sdk-go/api"
+)
+
+const tokensFixture string = `
+{
+    "count": 2,
+    "next": null,
+    "previous": null,
+    "results": [
+		{
+			"uuid": "f71e3b18-e702-409e-bd7f-8fb0a66d7b12",
+			"owner": "test",
+			"alias": "",
+			"fuzzy_key": "c8e74fa4cbf...e4954d602ec5ed19ba",
+			"created": "2023-04-18T00:00:00UTC",
+			"updated": "2023-04-19T00:00:00UTC"
+		},
+		{
+			"uuid": "f71e3b18-e702-499e-bd7f-8fb0a66d7b12",
+			"owner": "test",
+			"alias": "test",
+			"fuzzy_key": "c8e74fa4cbf...e4954d60cec5ed19ba",
+			"created": "2023-04-18T10:00:00UTC",
+			"updated": "2023-04-19T12:00:00UTC"
+		}
+    ]
+}`
+
+type fakeHTTPServer struct{}
+
+func (fakeHTTPServer) ServeHTTP(res http.ResponseWriter, req *http.Request) {
+	res.Header().Add("DRYCC_API_VERSION", drycc.APIVersion)
+
+	if req.URL.Path == "/v2/tokens/" && req.Method == "GET" {
+		res.Write([]byte(tokensFixture))
+		return
+	}
+
+	if req.URL.Path == "/v2/tokens/f71e3b18-e702-499e-bd7f-8fb0a66d7b12/" && req.Method == "DELETE" {
+		res.WriteHeader(204)
+		return
+	}
+
+	fmt.Printf("Unrecognized URL %s\n", req.URL)
+	res.WriteHeader(http.StatusNotFound)
+	res.Write(nil)
+}
+
+func TestTokensList(t *testing.T) {
+	t.Parallel()
+
+	expected := []api.Token{
+		{
+			UUID:    "f71e3b18-e702-409e-bd7f-8fb0a66d7b12",
+			Owner:   "test",
+			Alias:   "",
+			Key:     "c8e74fa4cbf...e4954d602ec5ed19ba",
+			Created: "2023-04-18T00:00:00UTC",
+			Updated: "2023-04-19T00:00:00UTC",
+		},
+		{
+			UUID:    "f71e3b18-e702-499e-bd7f-8fb0a66d7b12",
+			Owner:   "test",
+			Alias:   "test",
+			Key:     "c8e74fa4cbf...e4954d60cec5ed19ba",
+			Created: "2023-04-18T10:00:00UTC",
+			Updated: "2023-04-19T12:00:00UTC",
+		},
+	}
+	handler := fakeHTTPServer{}
+	server := httptest.NewServer(handler)
+	defer server.Close()
+
+	drycc, err := drycc.New(false, server.URL, "abc")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	actual, _, err := List(drycc, 100)
+
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !reflect.DeepEqual(expected, actual) {
+		t.Error(fmt.Errorf("Expected %v, Got %v", expected, actual))
+	}
+}
+
+func TestTokensRemove(t *testing.T) {
+	t.Parallel()
+
+	handler := fakeHTTPServer{}
+	server := httptest.NewServer(handler)
+	defer server.Close()
+
+	drycc, err := drycc.New(false, server.URL, "abc")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = Delete(drycc, "f71e3b18-e702-499e-bd7f-8fb0a66d7b12")
+
+	if err != nil {
+		t.Fatal(err)
+	}
+}