From d544c46b09c620cb4e96a5d0e342a7297535c1ae Mon Sep 17 00:00:00 2001
From: JU4N98 <juanpablocabana2@gmail.com>
Date: Mon, 4 Dec 2023 17:29:34 -0300
Subject: [PATCH] Re use jwtSource, solve identation, add blank line.

Signed-off-by: JU4N98 <juanpablocabana2@gmail.com>
---
 README.md                       |  2 +-
 pkg/sidecar/sidecar.go          | 35 ++++++++++++++++-----------------
 test/fixture/config/helper.conf |  2 +-
 3 files changed, 19 insertions(+), 20 deletions(-)

diff --git a/README.md b/README.md
index a9d80fad..d0fbf3cc 100644
--- a/README.md
+++ b/README.md
@@ -28,7 +28,7 @@ The configuration file is an [HCL](https://github.com/hashicorp/hcl) formatted f
  |`svid_key_file_name`      | File name to be used to store the X.509 SVID private key and public certificate in PEM format.                 | `"svid_key.pem"`     |
  |`svid_bundle_file_name`   | File name to be used to store the X.509 SVID Bundle in PEM format.                                             | `"svid_bundle.pem"`  |
  |`jwt_audience`            | JWT SVID audience.                                                                                             | `"your-audience"`    |
- |`jwt_svid_file_name`      | File name to be used to store JWT SVID in Base64-encoded string.                                                         | `"jwt_svid.token"`         |
+ |`jwt_svid_file_name`      | File name to be used to store JWT SVID in Base64-encoded string.                                               | `"jwt_svid.token"`   |
  |`jwt_bundle_file_name`    | File name to be used to store JWT Bundle in JSON format.                                                       | `"jwt_bundle.json"`  |
 
 ### Configuration example
diff --git a/pkg/sidecar/sidecar.go b/pkg/sidecar/sidecar.go
index b11355d8..a9a31a49 100644
--- a/pkg/sidecar/sidecar.go
+++ b/pkg/sidecar/sidecar.go
@@ -34,6 +34,7 @@ const (
 // implements the interface Sidecar
 type Sidecar struct {
 	config         *Config
+	jwtSource      *workloadapi.JWTSource
 	processRunning int32
 	process        *os.Process
 	certReadyChan  chan struct{}
@@ -103,10 +104,17 @@ func (s *Sidecar) RunDaemon(ctx context.Context) error {
 	}
 
 	if s.config.JWTSvidFilename != "" && s.config.JWTAudience != "" {
+		jwtSource, err := workloadapi.NewJWTSource(ctx, workloadapi.WithClientOptions(s.getWorkloadAPIAdress()))
+		if err != nil {
+			s.config.Log.Fatalf("Error watching JWT svid updates: %v", err)
+		}
+		s.jwtSource = jwtSource
+		defer s.jwtSource.Close()
+
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
-			s.updateJWTSVID(ctx, s.getWorkloadAPIAdress())
+			s.updateJWTSVID(ctx)
 		}()
 	}
 
@@ -266,23 +274,14 @@ func (s *Sidecar) updateJWTBundle(jwkSet *jwtbundle.Set) {
 	}
 }
 
-func (s *Sidecar) fetchJWTSVID(ctx context.Context, options ...workloadapi.ClientOption) (*jwtsvid.SVID, error) {
-	clientOptions := workloadapi.WithClientOptions(options...)
-
-	jwtSource, err := workloadapi.NewJWTSource(ctx, clientOptions)
-	if err != nil {
-		s.config.Log.Errorf("Unable to create JWTSource: %v", err)
-		return nil, err
-	}
-	defer jwtSource.Close()
-
-	jwtSVID, err := jwtSource.FetchJWTSVID(ctx, jwtsvid.Params{Audience: s.config.JWTAudience})
+func (s *Sidecar) fetchJWTSVID(ctx context.Context) (*jwtsvid.SVID, error) {
+	jwtSVID, err := s.jwtSource.FetchJWTSVID(ctx, jwtsvid.Params{Audience: s.config.JWTAudience})
 	if err != nil {
 		s.config.Log.Errorf("Unable to fetch JWT SVID: %v", err)
 		return nil, err
 	}
 
-	_, err = jwtsvid.ParseAndValidate(jwtSVID.Marshal(), jwtSource, []string{s.config.JWTAudience})
+	_, err = jwtsvid.ParseAndValidate(jwtSVID.Marshal(), s.jwtSource, []string{s.config.JWTAudience})
 	if err != nil {
 		s.config.Log.Errorf("Unable to parse or validate token: %v", err)
 		return nil, err
@@ -313,10 +312,10 @@ func getRefreshInterval(svid *jwtsvid.SVID) time.Duration {
 	return time.Until(svid.Expiry)/2 + time.Second
 }
 
-func (s *Sidecar) performJWTSVIDUpdate(ctx context.Context, options ...workloadapi.ClientOption) (*jwtsvid.SVID, error) {
+func (s *Sidecar) performJWTSVIDUpdate(ctx context.Context) (*jwtsvid.SVID, error) {
 	s.config.Log.Debug("Updating JWT SVID")
 
-	jwtSVID, err := s.fetchJWTSVID(ctx, options...)
+	jwtSVID, err := s.fetchJWTSVID(ctx)
 	if err != nil {
 		s.config.Log.Errorf("Unable to update JWT SVID: %v", err)
 		return nil, err
@@ -332,10 +331,10 @@ func (s *Sidecar) performJWTSVIDUpdate(ctx context.Context, options ...workloada
 	return jwtSVID, nil
 }
 
-func (s *Sidecar) updateJWTSVID(ctx context.Context, options ...workloadapi.ClientOption) {
+func (s *Sidecar) updateJWTSVID(ctx context.Context) {
 	retryInterval := createRetryIntervalFunc()
 	var initialInterval time.Duration
-	jwtSVID, err := s.performJWTSVIDUpdate(ctx, options...)
+	jwtSVID, err := s.performJWTSVIDUpdate(ctx)
 	if err != nil {
 		// If the first update fails, use the retry interval
 		initialInterval = retryInterval()
@@ -351,7 +350,7 @@ func (s *Sidecar) updateJWTSVID(ctx context.Context, options ...workloadapi.Clie
 		case <-ctx.Done():
 			return
 		case <-ticker.C:
-			jwtSVID, err = s.performJWTSVIDUpdate(ctx, options...)
+			jwtSVID, err = s.performJWTSVIDUpdate(ctx)
 			if err == nil {
 				retryInterval = createRetryIntervalFunc()
 				ticker.Reset(getRefreshInterval(jwtSVID))
diff --git a/test/fixture/config/helper.conf b/test/fixture/config/helper.conf
index 0800115c..0ab57b78 100644
--- a/test/fixture/config/helper.conf
+++ b/test/fixture/config/helper.conf
@@ -10,4 +10,4 @@ jwt_svid_file_name = "jwt_svid.token"
 jwt_bundle_file_name = "jwt_bundle.json"
 jwt_audience = "your-audience"
 timeout = "10s"
-add_intermediates_to_bundle = true
\ No newline at end of file
+add_intermediates_to_bundle = true