Fix fleet resources not having ID set on import (#447)

* Fix fleet objects not having ID set on import

- Fixed fleet objects not having an ID set when importing.
- Update fleet documentation to remove kibana space id

* Update changelog

Author: taylor-swanson

CHANGELOG.md

@@ -3,6 +3,7 @@
 ### Added
 - Introduce `elasticstack_kibana_import_saved_objects` resource as an additive only way to manage Kibana saved objects ([#343](https://github.com/elastic/terraform-provider-elasticstack/pull/343)).
 - Add support for Terraform Plugin Framework ([#343](https://github.com/elastic/terraform-provider-elasticstack/pull/343)).
+- Fix fleet resources not having ID set on import ([#447](https://github.com/elastic/terraform-provider-elasticstack/pull/447))
 
 ## [0.9.0] - 2023-10-09
 

docs/resources/fleet_agent_policy.md

@@ -57,5 +57,5 @@ resource "elasticstack_fleet_agent_policy" "test_policy" {
 Import is supported using the following syntax:
 
 ```shell
-terraform import elasticstack_kibana_fleet_agent_policy.my_policy <space id>/<policy id>
+terraform import elasticstack_kibana_fleet_agent_policy.my_policy <fleet_agent_policy_id>
 ```

docs/resources/fleet_output.md

@@ -58,5 +58,5 @@ resource "elasticstack_fleet_output" "test_output" {
 Import is supported using the following syntax:
 
 ```shell
-terraform import elasticstack_fleet_output.my_output <space id>/<output id>
+terraform import elasticstack_fleet_output.my_output <fleet_output_id>
 ```

docs/resources/fleet_server_host.md

@@ -48,5 +48,5 @@ resource "elasticstack_fleet_server_host" "test_host" {
 Import is supported using the following syntax:
 
 ```shell
-terraform import elasticstack_fleet_server_host.my_host <space id>/<host id>
+terraform import elasticstack_fleet_server_host.my_host <fleet_server_host_id>
 ```

examples/resources/elasticstack_fleet_agent_policy/import.sh

@@ -1 +1 @@
-terraform import elasticstack_kibana_fleet_agent_policy.my_policy <space id>/<policy id>
+terraform import elasticstack_kibana_fleet_agent_policy.my_policy <fleet_agent_policy_id>

examples/resources/elasticstack_fleet_output/import.sh

@@ -1 +1 @@
-terraform import elasticstack_fleet_output.my_output <space id>/<output id>
+terraform import elasticstack_fleet_output.my_output <fleet_output_id>

examples/resources/elasticstack_fleet_server_host/import.sh

@@ -1 +1 @@
-terraform import elasticstack_fleet_server_host.my_host <space id>/<host id>
+terraform import elasticstack_fleet_server_host.my_host <fleet_server_host_id>

internal/fleet/agent_policy_resource.go

@@ -103,6 +103,10 @@ func resourceAgentPolicyCreate(ctx context.Context, d *schema.ResourceData, meta
 		return diags
 	}
 
+	if id := d.Get("policy_id").(string); id != "" {
+		d.SetId(id)
+	}
+
 	req := fleetapi.AgentPolicyCreateRequest{
 		Name:      d.Get("name").(string),
 		Namespace: d.Get("namespace").(string),
@@ -157,9 +161,6 @@ func resourceAgentPolicyUpdate(ctx context.Context, d *schema.ResourceData, meta
 		return diags
 	}
 
-	id := d.Get("policy_id").(string)
-	d.SetId(id)
-
 	req := fleetapi.AgentPolicyUpdateRequest{
 		Name:      d.Get("name").(string),
 		Namespace: d.Get("namespace").(string),
@@ -191,7 +192,7 @@ func resourceAgentPolicyUpdate(ctx context.Context, d *schema.ResourceData, meta
 	if len(monitoringValues) > 0 {
 		req.MonitoringEnabled = &monitoringValues
 	}
-	_, diags = fleet.UpdateAgentPolicy(ctx, fleetClient, id, req)
+	_, diags = fleet.UpdateAgentPolicy(ctx, fleetClient, d.Id(), req)
 	if diags.HasError() {
 		return diags
 	}
@@ -205,10 +206,7 @@ func resourceAgentPolicyRead(ctx context.Context, d *schema.ResourceData, meta i
 		return diags
 	}
 
-	id := d.Get("policy_id").(string)
-	d.SetId(id)
-
-	agentPolicy, diags := fleet.ReadAgentPolicy(ctx, fleetClient, id)
+	agentPolicy, diags := fleet.ReadAgentPolicy(ctx, fleetClient, d.Id())
 	if diags.HasError() {
 		return diags
 	}
@@ -280,10 +278,7 @@ func resourceAgentPolicyDelete(ctx context.Context, d *schema.ResourceData, meta
 		return diags
 	}
 
-	id := d.Get("policy_id").(string)
-	d.SetId(id)
-
-	if diags = fleet.DeleteAgentPolicy(ctx, fleetClient, id); diags.HasError() {
+	if diags = fleet.DeleteAgentPolicy(ctx, fleetClient, d.Id()); diags.HasError() {
 		return diags
 	}
 	d.SetId("")

internal/fleet/enrollment_tokens_data_source.go

@@ -78,7 +78,10 @@ func dataSourceEnrollmentTokensRead(ctx context.Context, d *schema.ResourceData,
 		return diags
 	}
 
-	policyID := d.Get("policy_id").(string)
+	if d.Id() == "" {
+		d.SetId(d.Get("policy_id").(string))
+	}
+	policyID := d.Id()
 
 	allTokens, diags := fleet.AllEnrollmentTokens(ctx, fleetClient)
 	if diags.HasError() {

internal/fleet/fleet_server_host_resource.go

@@ -60,6 +60,10 @@ func resourceFleetServerHostCreate(ctx context.Context, d *schema.ResourceData,
 		return diags
 	}
 
+	if id := d.Get("host_id").(string); id != "" {
+		d.SetId(id)
+	}
+
 	req := fleetapi.PostFleetServerHostsJSONRequestBody{
 		Name: d.Get("name").(string),
 	}
@@ -94,9 +98,6 @@ func resourceFleetServerHostUpdate(ctx context.Context, d *schema.ResourceData,
 		return diags
 	}
 
-	id := d.Get("host_id").(string)
-	d.SetId(id)
-
 	req := fleetapi.UpdateFleetServerHostsJSONRequestBody{}
 
 	if value, ok := d.Get("name").(string); ok && value != "" {
@@ -117,7 +118,7 @@ func resourceFleetServerHostUpdate(ctx context.Context, d *schema.ResourceData,
 		req.IsDefault = &value
 	}
 
-	_, diags = fleet.UpdateFleetServerHost(ctx, fleetClient, id, req)
+	_, diags = fleet.UpdateFleetServerHost(ctx, fleetClient, d.Id(), req)
 	if diags.HasError() {
 		return diags
 	}
@@ -131,10 +132,7 @@ func resourceFleetServerHostRead(ctx context.Context, d *schema.ResourceData, me
 		return diags
 	}
 
-	id := d.Get("host_id").(string)
-	d.SetId(id)
-
-	host, diags := fleet.ReadFleetServerHost(ctx, fleetClient, id)
+	host, diags := fleet.ReadFleetServerHost(ctx, fleetClient, d.Id())
 	if diags.HasError() {
 		return diags
 	}
@@ -166,10 +164,7 @@ func resourceFleetServerHostDelete(ctx context.Context, d *schema.ResourceData,
 		return diags
 	}
 
-	id := d.Get("host_id").(string)
-	d.SetId(id)
-
-	if diags = fleet.DeleteFleetServerHost(ctx, fleetClient, id); diags.HasError() {
+	if diags = fleet.DeleteFleetServerHost(ctx, fleetClient, d.Id()); diags.HasError() {
 		return diags
 	}
 	d.SetId("")