Keep extension data source state durable

Author: IlyaasK

docs/data-sources/extension.md

@@ -24,5 +24,4 @@ Lookup durable Kernel extension metadata.
 ### Read-Only
 
 - `created_at` (String) Extension creation timestamp.
-- `last_used_at` (String) Timestamp the extension was last used, if ever.
 - `size_bytes` (Number) Extension archive size in bytes.

internal/datasources/extension/datasource.go

@@ -31,12 +31,11 @@ type extensionDataSource struct {
 }
 
 type extensionModel struct {
-	ID         types.String `tfsdk:"id"`
-	Name       types.String `tfsdk:"name"`
-	ProjectID  types.String `tfsdk:"project_id"`
-	CreatedAt  types.String `tfsdk:"created_at"`
-	SizeBytes  types.Int64  `tfsdk:"size_bytes"`
-	LastUsedAt types.String `tfsdk:"last_used_at"`
+	ID        types.String `tfsdk:"id"`
+	Name      types.String `tfsdk:"name"`
+	ProjectID types.String `tfsdk:"project_id"`
+	CreatedAt types.String `tfsdk:"created_at"`
+	SizeBytes types.Int64  `tfsdk:"size_bytes"`
 }
 
 func NewDataSource() datasource.DataSource {
@@ -80,10 +79,6 @@ func (d *extensionDataSource) Schema(ctx context.Context, req datasource.SchemaR
 				Computed:            true,
 				MarkdownDescription: "Extension archive size in bytes.",
 			},
-			"last_used_at": dschema.StringAttribute{
-				Computed:            true,
-				MarkdownDescription: "Timestamp the extension was last used, if ever.",
-			},
 		},
 	}
 }
@@ -209,26 +204,15 @@ func flattenExtension(extension kernel.ExtensionGetResponse) (extensionModel, di
 		}
 	}
 
-	// last_used_at is nullable (an extension that has never been used); a null
-	// reads as not-present via FieldPresent, leaving the attribute null.
-	lastUsedAt := types.StringNull()
-	if datasources.FieldPresent(extension.JSON.LastUsedAt.Raw()) {
-		if !datasources.ValidResponseTime(extension.JSON.LastUsedAt.Raw(), extension.JSON.LastUsedAt.Valid(), extension.LastUsedAt) {
-			datasources.AddInvalidResponseField(&diags, "Extension", "last_used_at")
-		} else {
-			lastUsedAt = types.StringValue(extension.LastUsedAt.Format(time.RFC3339Nano))
-		}
-	}
 	if diags.HasError() {
 		return extensionModel{}, diags
 	}
 
 	return extensionModel{
-		ID:         types.StringValue(extension.ID),
-		Name:       name,
-		CreatedAt:  types.StringValue(extension.CreatedAt.Format(time.RFC3339Nano)),
-		SizeBytes:  types.Int64Value(extension.SizeBytes),
-		LastUsedAt: lastUsedAt,
+		ID:        types.StringValue(extension.ID),
+		Name:      name,
+		CreatedAt: types.StringValue(extension.CreatedAt.Format(time.RFC3339Nano)),
+		SizeBytes: types.Int64Value(extension.SizeBytes),
 	}, diags
 }
 

internal/datasources/extension/datasource_test.go

@@ -115,12 +115,12 @@ func TestDataSourceMetadataAndSchema(t *testing.T) {
 
 	var schema datasource.SchemaResponse
 	ds.Schema(context.Background(), datasource.SchemaRequest{}, &schema)
-	for _, name := range []string{"id", "name", "created_at", "size_bytes", "last_used_at"} {
+	for _, name := range []string{"id", "name", "created_at", "size_bytes"} {
 		if _, ok := schema.Schema.Attributes[name]; !ok {
 			t.Fatalf("schema missing %s attribute", name)
 		}
 	}
-	for _, name := range []string{"download_url", "zip_file"} {
+	for _, name := range []string{"download_url", "zip_file", "last_used_at"} {
 		if _, ok := schema.Schema.Attributes[name]; ok {
 			t.Fatalf("schema should not include %s", name)
 		}
@@ -168,9 +168,6 @@ func TestReadSetsTerraformState(t *testing.T) {
 	if state.SizeBytes.ValueInt64() != 1234 {
 		t.Fatalf("state size_bytes = %d, want 1234", state.SizeBytes.ValueInt64())
 	}
-	if state.LastUsedAt.ValueString() != "2026-06-05T12:00:00Z" {
-		t.Fatalf("state last_used_at = %q, want 2026-06-05T12:00:00Z", state.LastUsedAt.ValueString())
-	}
 }
 
 func TestReadExtensionByID(t *testing.T) {
@@ -237,24 +234,6 @@ func TestReadExtensionAllowsNullableName(t *testing.T) {
 	}
 }
 
-func TestReadExtensionAllowsNullableLastUsedAt(t *testing.T) {
-	t.Parallel()
-
-	ds := newDataSourceWithClient(fakeExtensionClient{
-		get: func(ctx context.Context, projectID, idOrName string) (*kernel.ExtensionGetResponse, error) {
-			return extensionWithoutLastUsedForTest("extension-1", "Extension"), nil
-		},
-	})
-
-	state, diags := ds.read(context.Background(), extensionModel{ID: types.StringValue("extension-1")})
-	if diags.HasError() {
-		t.Fatalf("unexpected diagnostics: %v", diags)
-	}
-	if !state.LastUsedAt.IsNull() {
-		t.Fatalf("state last_used_at = %q, want null", state.LastUsedAt.ValueString())
-	}
-}
-
 func TestReadRejectsExtensionNotFound(t *testing.T) {
 	t.Parallel()
 
@@ -339,10 +318,6 @@ func extensionWithoutNameForTest(id string) *kernel.ExtensionGetResponse {
 	return extensionFromJSON(`{"id":"` + id + `","name":null,"created_at":"2026-06-05T12:00:00Z","size_bytes":1234,"last_used_at":"2026-06-05T12:00:00Z"}`)
 }
 
-func extensionWithoutLastUsedForTest(id, name string) *kernel.ExtensionGetResponse {
-	return extensionFromJSON(`{"id":"` + id + `","name":"` + name + `","created_at":"2026-06-05T12:00:00Z","size_bytes":1234,"last_used_at":null}`)
-}
-
 func invalidExtension() *kernel.ExtensionGetResponse {
 	extension := extensionForTest("extension-1", "Extension")
 	extension.CreatedAt = time.Time{}
@@ -362,21 +337,19 @@ func extensionConfigValue(id, name tftypes.Value) tftypes.Value {
 	return tftypes.NewValue(
 		tftypes.Object{
 			AttributeTypes: map[string]tftypes.Type{
-				"id":           tftypes.String,
-				"name":         tftypes.String,
-				"project_id":   tftypes.String,
-				"created_at":   tftypes.String,
-				"size_bytes":   tftypes.Number,
-				"last_used_at": tftypes.String,
+				"id":         tftypes.String,
+				"name":       tftypes.String,
+				"project_id": tftypes.String,
+				"created_at": tftypes.String,
+				"size_bytes": tftypes.Number,
 			},
 		},
 		map[string]tftypes.Value{
-			"id":           id,
-			"name":         name,
-			"project_id":   tftypes.NewValue(tftypes.String, nil),
-			"created_at":   tftypes.NewValue(tftypes.String, nil),
-			"size_bytes":   tftypes.NewValue(tftypes.Number, nil),
-			"last_used_at": tftypes.NewValue(tftypes.String, nil),
+			"id":         id,
+			"name":       name,
+			"project_id": tftypes.NewValue(tftypes.String, nil),
+			"created_at": tftypes.NewValue(tftypes.String, nil),
+			"size_bytes": tftypes.NewValue(tftypes.Number, nil),
 		},
 	)
 }