feat: add import via address for ip resources

krystal · Dec 23, 2024 · a806fa6 · a806fa6
1 parent aa5b288
commit a806fa6
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/internal/v6provider/resource_ip.go b/internal/v6provider/resource_ip.go
@@ -392,7 +392,26 @@ func (r *IPResource) ImportState(
 	req resource.ImportStateRequest,
 	resp *resource.ImportStateResponse,
 ) {
-	resource.ImportStatePassthroughID(ctx, path.Root("id"), req, resp)
+	if strings.HasPrefix(req.ID, "ip_") {
+		resource.ImportStatePassthroughID(ctx, path.Root("id"), req, resp)
+		return
+	}
+
+	res, err := r.M.Core.GetIpAddressWithResponse(ctx, &core.GetIpAddressParams{
+		IpAddressAddress: &req.ID,
+	})
+	if err != nil {
+		resp.Diagnostics.AddError("IP Address Import Error", err.Error())
+		return
+	}
+
+	resp.Diagnostics.Append(
+		resp.State.SetAttribute(
+			ctx,
+			path.Root("id"),
+			*res.JSON200.IpAddress.Id,
+		)...,
+	)
 }
 
 func unflattenIPVersion(ver int64) core.IPAddressVersionEnum {