fix: Pass user team to Docker auth endpoint (#210)

Needed to track downloads ---
cloudquery · Jan 2, 2024 · 452df35 · 452df35
1 parent 19c848f
commit 452df35
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/managedplugin/docker.go b/managedplugin/docker.go
@@ -84,7 +84,7 @@ func isDockerImageAvailable(ctx context.Context, imageName string) (bool, error)
 	return len(images) > 0, nil
 }
 
-func pullDockerImage(ctx context.Context, imageName string, authToken string) error {
+func pullDockerImage(ctx context.Context, imageName string, authToken string, teamName string) error {
 	// Pull the image
 	additionalHeaders := make(map[string]string)
 	opts := types.ImagePullOptions{}
@@ -98,6 +98,7 @@ func pullDockerImage(ctx context.Context, imageName string, authToken string) er
 			return fmt.Errorf("failed to parse Docker image tag: %v", err)
 		}
 		additionalHeaders["X-Meta-Plugin-Version"] = nameWithTag.Tag()
+		additionalHeaders["X-Meta-User-Team-Name"] = teamName
 		authConfig := registry.AuthConfig{
 			Username: "managedplugin",
 			Password: authToken,

diff --git a/managedplugin/plugin.go b/managedplugin/plugin.go
@@ -173,7 +173,7 @@ func (c *Client) downloadPlugin(ctx context.Context, typ PluginType) error {
 		if imageAvailable, err := isDockerImageAvailable(ctx, c.config.Path); err != nil {
 			return err
 		} else if !imageAvailable {
-			return pullDockerImage(ctx, c.config.Path, c.authToken)
+			return pullDockerImage(ctx, c.config.Path, c.authToken, c.teamName)
 		}
 		return nil
 	case RegistryCloudQuery: