chore: Example analytics support for resource creation and update

Author: EspenAlbert

internal/config/client.go

@@ -111,7 +111,7 @@ func (c *Config) NewClient(ctx context.Context) (any, error) {
 	tfLoggingTransport := logging.NewTransport("Atlas", digestTransport)
 	// Add UserAgentExtra fields to the User-Agent header, see wrapper_provider_server.go
 	userAgentTransport := NewUserAgentTransport(tfLoggingTransport, c.AnalyticsEnabled)
-	client := &http.Client{Transport: userAgentTransport.Transport}
+	client := &http.Client{Transport: userAgentTransport}
 
 	optsAtlas := []matlasClient.ClientOpt{matlasClient.SetUserAgent(userAgent(c))}
 	if c.BaseURL != "" {

internal/config/resource_base.go

@@ -36,6 +36,30 @@ func (r *RSCommon) ReadProviderMetaCreate(ctx context.Context, req *resource.Cre
 	return meta
 }
 
+func (r *RSCommon) ReadProviderMetaUpdate(ctx context.Context, req *resource.UpdateRequest, diags *diag.Diagnostics) ProviderMeta {
+	var meta ProviderMeta
+	diags.Append(req.ProviderMeta.Get(ctx, &meta)...)
+	return meta
+}
+
+func (r *RSCommon) AddAnalyticsCreate(ctx context.Context, req *resource.CreateRequest, diags *diag.Diagnostics) context.Context {
+	meta := r.ReadProviderMetaCreate(ctx, req, diags)
+	return AddUserAgentExtra(ctx, UserAgentExtra{
+		ScriptLocation: meta.ScriptLocation.ValueString(),
+		Name:           r.ResourceName,
+		Operation:      "create",
+	})
+}
+
+func (r *RSCommon) AddAnalyticsUpdate(ctx context.Context, req *resource.UpdateRequest, diags *diag.Diagnostics) context.Context {
+	meta := r.ReadProviderMetaUpdate(ctx, req, diags)
+	return AddUserAgentExtra(ctx, UserAgentExtra{
+		ScriptLocation: meta.ScriptLocation.ValueString(),
+		Name:           r.ResourceName,
+		Operation:      "create",
+	})
+}
+
 func (r *RSCommon) Metadata(ctx context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) {
 	resp.TypeName = fmt.Sprintf("%s_%s", req.ProviderTypeName, r.ResourceName)
 }

internal/service/project/resource_project.go

@@ -60,10 +60,7 @@ func (r *projectRS) Create(ctx context.Context, req resource.CreateRequest, resp
 
 	connV2 := r.Client.AtlasV2
 	diags := &resp.Diagnostics
-	meta := r.ReadProviderMetaCreate(ctx, &req, diags)
-	scriptLocation := meta.ScriptLocation
-	fmt.Println("found script location: " + scriptLocation.ValueString())
-
+	ctx = r.AddAnalyticsCreate(ctx, &req, diags)
 	diags.Append(req.Plan.Get(ctx, &projectPlan)...)
 	if diags.HasError() {
 		return
@@ -246,6 +243,7 @@ func (r *projectRS) Update(ctx context.Context, req resource.UpdateRequest, resp
 	var projectState TFProjectRSModel
 	var projectPlan TFProjectRSModel
 	connV2 := r.Client.AtlasV2
+	ctx = r.AddAnalyticsUpdate(ctx, &req, &resp.Diagnostics)
 
 	// get current state
 	resp.Diagnostics.Append(req.State.Get(ctx, &projectState)...)