feat(build_token): worker changes for build token implementation

cmd/vela-worker/client.go

@@ -11,7 +11,7 @@ import (
 )
 
 // helper function to setup the queue from the CLI arguments.
-func setupClient(s *Server) (*vela.Client, error) {
+func setupClient(s *Server, token string) (*vela.Client, error) {
 	logrus.Debug("creating vela client from worker configuration")
 
 	// create a new Vela client from the server configuration
@@ -21,11 +21,10 @@ func setupClient(s *Server) (*vela.Client, error) {
 	if err != nil {
 		return nil, err
 	}
-
 	// set token for authentication with the server
 	//
 	// https://pkg.go.dev/github.com/go-vela/sdk-go/vela?tab=doc#AuthenticationService.SetTokenAuth
-	vela.Authentication.SetTokenAuth(s.Secret)
+	vela.Authentication.SetTokenAuth(token)
 
 	return vela, nil
 }

cmd/vela-worker/exec.go

@@ -36,6 +36,18 @@ func (w *Worker) exec(index int) error {
 		return nil
 	}
 
+	// GET build token from server to setup execBuildClient
+	bt, _, err := w.VelaClient.Build.GetBuildToken(item.Repo.GetOrg(), item.Repo.GetName(), item.Build.GetNumber())
+	if err != nil {
+		logrus.Errorf("Unable to GetBuildToken: %s", err)
+		return err
+	}
+	// set up build client with build token as auth
+	execBuildClient, err := setupClient(w.Config.Server, bt.GetToken())
+	if err != nil {
+		return err
+	}
+
 	// create logger with extra metadata
 	//
 	// https://pkg.go.dev/github.com/sirupsen/logrus?tab=doc#WithFields
@@ -79,7 +91,7 @@ func (w *Worker) exec(index int) error {
 		LogStreamingTimeout: w.Config.Executor.LogStreamingTimeout,
 		EnforceTrustedRepos: w.Config.Executor.EnforceTrustedRepos,
 		PrivilegedImages:    w.Config.Runtime.PrivilegedImages,
-		Client:              w.VelaClient,
+		Client:              execBuildClient,
 		Hostname:            w.Config.API.Address.Hostname(),
 		Runtime:             w.Runtime,
 		Build:               item.Build,

cmd/vela-worker/operate.go

@@ -22,8 +22,8 @@ import (
 func (w *Worker) operate(ctx context.Context) error {
 	var err error
 
-	// setup the client
-	w.VelaClient, err = setupClient(w.Config.Server)
+	// setup the vela client with the server
+	w.VelaClient, err = setupClient(w.Config.Server, w.Config.Server.Secret)
 	if err != nil {
 		return err
 	}

cmd/vela-worker/register.go

@@ -12,7 +12,7 @@ import (
 	"github.com/sirupsen/logrus"
 )
 
-// checkIn is a helper function to to phone home to the server.
+// checkIn is a helper function to phone home to the server.
 func (w *Worker) checkIn(config *library.Worker) error {
 	// check to see if the worker already exists in the database
 	logrus.Infof("retrieving worker %s from the server", config.GetHostname())