feat: Pass installation ID from env to usage report

Author: disq

go.mod

@@ -11,7 +11,7 @@ require (
 	github.com/aws/aws-sdk-go-v2/service/licensemanager v1.30.1
 	github.com/aws/aws-sdk-go-v2/service/marketplacemetering v1.26.2
 	github.com/bradleyjkemp/cupaloy/v2 v2.8.0
-	github.com/cloudquery/cloudquery-api-go v1.13.7
+	github.com/cloudquery/cloudquery-api-go v1.13.8
 	github.com/cloudquery/plugin-pb-go v1.26.8
 	github.com/cloudquery/plugin-sdk/v2 v2.7.0
 	github.com/goccy/go-json v0.10.5

go.sum

@@ -50,8 +50,8 @@ github.com/buger/jsonparser v1.1.1 h1:2PnMjfWD7wBILjqQbt530v576A/cAbQvEW9gGIpYMU
 github.com/buger/jsonparser v1.1.1/go.mod h1:6RYKKt7H4d4+iWqouImQ9R2FZql3VbhNgx27UK13J/0=
 github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
 github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
-github.com/cloudquery/cloudquery-api-go v1.13.7 h1:XWn9TVLpKswTrNQT9MY/HUiQWwhyoMWKpGWoRcQM5bE=
-github.com/cloudquery/cloudquery-api-go v1.13.7/go.mod h1:ZhEjPkDGDL2KZKlQLUnsgQ0mPz3qC7qftr37q3q+IcA=
+github.com/cloudquery/cloudquery-api-go v1.13.8 h1:8n5D0G2wynbUdexr1GS8ND8i0uOwm0gXKNipJsijPe0=
+github.com/cloudquery/cloudquery-api-go v1.13.8/go.mod h1:ZhEjPkDGDL2KZKlQLUnsgQ0mPz3qC7qftr37q3q+IcA=
 github.com/cloudquery/plugin-pb-go v1.26.8 h1:mHNyOyG/MW/rjDVR0OR8DdezXmyoD4drphyHlYmnJoY=
 github.com/cloudquery/plugin-pb-go v1.26.8/go.mod h1:orruK+wdIP3kOvmEQhCcFND2bVVH7o0fFSGrqZYMt7k=
 github.com/cloudquery/plugin-sdk/v2 v2.7.0 h1:hRXsdEiaOxJtsn/wZMFQC9/jPfU1MeMK3KF+gPGqm7U=

premium/usage.go

@@ -180,8 +180,9 @@ type BatchUpdater struct {
 	awsMarketplaceClient AWSMarketplaceClientInterface
 
 	// Plugin details
-	teamName   cqapi.TeamName
-	pluginMeta plugin.Meta
+	teamName       cqapi.TeamName
+	pluginMeta     plugin.Meta
+	installationID string
 
 	// Configuration
 	batchLimit            uint32
@@ -284,6 +285,7 @@ func NewUsageClient(meta plugin.Meta, ops ...UsageClientOptions) (UsageClient, e
 		}
 		u.teamName = teamName
 	}
+	u.installationID = maybeGetInstallationID(u.tokenClient.GetTokenType())
 
 	u.backgroundUpdater()
 
@@ -634,7 +636,9 @@ func (u *BatchUpdater) updateUsageWithRetryAndBackoff(ctx context.Context, rows
 		PluginName: u.pluginMeta.Name,
 		Rows:       int(rows),
 	}
-
+	if len(u.installationID) > 0 {
+		payload.InstallationID = &u.installationID
+	}
 	if len(tables) > 0 {
 		payload.Tables = &tables
 	}
@@ -724,6 +728,15 @@ func (u *BatchUpdater) getTeamNameByTokenType(tokenType auth.TokenType) (string,
 	}
 }
 
+func maybeGetInstallationID(tokenType auth.TokenType) string {
+	switch tokenType {
+	case auth.SyncRunAPIKey, auth.SyncTestConnectionAPIKey:
+		return os.Getenv("_CQ_INSTALLATION_ID")
+	default:
+		return ""
+	}
+}
+
 type NoOpUsageClient struct {
 	TeamNameValue string
 }