chore(block): add support for identity schema

Author: remyleone

internal/services/block/snapshot.go

@@ -33,6 +33,25 @@ func ResourceSnapshot() *schema.Resource {
 			Default: schema.DefaultTimeout(defaultBlockTimeout),
 		},
 		SchemaVersion: 0,
+		Identity: &schema.ResourceIdentity{
+			Version: 0,
+			SchemaFunc: func() map[string]*schema.Schema {
+				return map[string]*schema.Schema{
+					"snapshot_id": {
+						Type:              schema.TypeString,
+						Computed:          true,
+						RequiredForImport: true,
+						Description:       "Snapshot ID",
+					},
+					"zone": {
+						Type:              schema.TypeString,
+						Computed:          true,
+						RequiredForImport: true,
+						Description:       "Zone ID",
+					},
+				}
+			},
+		},
 		Schema: map[string]*schema.Schema{
 			"name": {
 				Type:        schema.TypeString,
@@ -202,6 +221,18 @@ func ResourceBlockSnapshotRead(ctx context.Context, d *schema.ResourceData, m an
 
 	_ = d.Set("tags", snapshot.Tags)
 
+	identity, err := d.Identity()
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if err = identity.Set("snapshot_id", snapshot.ID); err != nil {
+		return diag.FromErr(err)
+	}
+	if err = identity.Set("zone", snapshot.Zone); err != nil {
+		return diag.FromErr(err)
+	}
+
 	return nil
 }
 

internal/services/block/volume.go

@@ -26,6 +26,25 @@ func ResourceVolume() *schema.Resource {
 		Importer: &schema.ResourceImporter{
 			StateContext: schema.ImportStatePassthroughContext,
 		},
+		Identity: &schema.ResourceIdentity{
+			Version: 0,
+			SchemaFunc: func() map[string]*schema.Schema {
+				return map[string]*schema.Schema{
+					"volume_id": {
+						Type:              schema.TypeString,
+						Computed:          true,
+						RequiredForImport: true,
+						Description:       "Volume ID",
+					},
+					"zone": {
+						Type:              schema.TypeString,
+						Computed:          true,
+						RequiredForImport: true,
+						Description:       "Zone",
+					},
+				}
+			},
+		},
 		Timeouts: &schema.ResourceTimeout{
 			Create:  schema.DefaultTimeout(defaultBlockTimeout),
 			Read:    schema.DefaultTimeout(defaultBlockTimeout),
@@ -186,6 +205,18 @@ func ResourceBlockVolumeRead(ctx context.Context, d *schema.ResourceData, m any)
 
 	_ = d.Set("snapshot_id", snapshotID)
 
+	identity, err := d.Identity()
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if err = identity.Set("volume_id", volume.ID); err != nil {
+		return diag.FromErr(err)
+	}
+	if err = identity.Set("zone", volume.Zone); err != nil {
+		return diag.FromErr(err)
+	}
+
 	return nil
 }