feat: worker auth endpoints

vela/admin.go

@@ -5,6 +5,8 @@
 package vela
 
 import (
+	"fmt"
+
 	"github.com/go-vela/types/library"
 )
 
@@ -20,6 +22,7 @@ type (
 		Service    *AdminSvcService
 		Step       *AdminStepService
 		User       *AdminUserService
+		Worker     *AdminWorkerService
 	}
 
 	// AdminBuildService handles retrieving admin builds from
@@ -53,6 +56,10 @@ type (
 	// AdminUserService handles retrieving admin users from
 	// the server methods of the Vela API.
 	AdminUserService service
+
+	// AdminWorkerService handles managing admin worker functionality
+	// from the server methods of the Vela API.
+	AdminWorkerService service
 )
 
 // GetQueueOptions specifies the optional parameters to the
@@ -196,3 +203,17 @@ func (svc *AdminUserService) Update(u *library.User) (*library.User, *Response,
 
 	return v, resp, err
 }
+
+// RegistrationToken generates a worker registration token with the provided details.
+func (svc *AdminUserService) RegistrationToken(w *library.Worker) (*library.Token, *Response, error) {
+	// set the API endpoint path we send the request to
+	url := fmt.Sprintf("/api/v1/workers/%s/register-token", w.GetHostname())
+
+	// library Token type we want to return
+	t := new(library.Token)
+
+	// send request using client
+	resp, err := svc.client.Call("POST", url, nil, t)
+
+	return t, resp, err
+}

vela/authentication.go

@@ -34,6 +34,11 @@ type AuthenticationService struct {
 	authType            AuthenticationType
 }
 
+// SetTokenAuth sets the authentication type as a plain token.
+func (svc *AuthenticationService) GetToken() string {
+	return *svc.token
+}
+
 // SetTokenAuth sets the authentication type as a plain token.
 func (svc *AuthenticationService) SetTokenAuth(token string) {
 	svc.token = String(token)
@@ -203,3 +208,13 @@ func extractRefreshToken(cookies []*http.Cookie) string {
 
 	return c
 }
+
+// ValidateToken makes a request to validate tokens with the Vela server.
+func (svc *AuthenticationService) ValidateToken() (*Response, error) {
+	// set the API endpoint path we send the request to
+	u := "/validate-token"
+
+	// attempt to validate a server token
+	resp, err := svc.client.Call("GET", u, nil, nil)
+	return resp, err
+}

vela/client.go

@@ -133,6 +133,7 @@ func NewClient(baseURL, id string, httpClient *http.Client) (*Client, error) {
 		&AdminSvcService{client: c},
 		&AdminStepService{client: c},
 		&AdminUserService{client: c},
+		&AdminWorkerService{client: c},
 	}
 	c.Build = &BuildService{client: c}
 	c.Deployment = &DeploymentService{client: c}

vela/client_test.go

@@ -45,6 +45,7 @@ func TestVela_NewClient(t *testing.T) {
 		&AdminSvcService{client: want},
 		&AdminStepService{client: want},
 		&AdminUserService{client: want},
+		&AdminWorkerService{client: want},
 	}
 	want.Build = &BuildService{client: want}
 	want.Deployment = &DeploymentService{client: want}