feat(request): trigger token generation if missing

Verify if the current client has a token, if not, try to generate one Signed-off-by: Salim Afiune Maya <[email protected]>
lacework · Mar 9, 2020 · 8cd82d6 · 8cd82d6
1 parent d61701d
commit 8cd82d6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/client/http.go b/client/http.go
@@ -31,6 +31,12 @@ func (c *client) NewRequest(method string, apiURL string, body io.Reader) (*http
 	if apiURL == apiTokens {
 		headers["X-LW-UAKS"] = c.auth.secret
 	} else {
+		// verify that the client has a token, if not, try to generate one
+		if c.auth.token == "" {
+			if _, err = c.GenerateToken(); err != nil {
+				return nil, err
+			}
+		}
 		headers["Authorization"] = c.auth.token
 	}