Merge pull request #59 from latitudesh/fix/refresh-state-404

Fix: Error when refreshing state

latitudesh/resource_project.go

@@ -3,6 +3,7 @@ package latitudesh
 import (
 	"context"
 	"errors"
+	"net/http"
 	"strings"
 	"time"
 
@@ -84,8 +85,13 @@ func resourceProjectRead(ctx context.Context, d *schema.ResourceData, m interfac
 
 	projectID := d.Id()
 
-	project, _, err := c.Projects.Get(projectID, nil)
+	project, resp, err := c.Projects.Get(projectID, nil)
 	if err != nil {
+		if resp.StatusCode == http.StatusNotFound {
+			d.SetId("")
+			return diags
+		}
+
 		return diag.FromErr(err)
 	}
 

latitudesh/resource_server.go

@@ -2,6 +2,7 @@ package latitudesh
 
 import (
 	"context"
+	"net/http"
 	"strconv"
 	"time"
 
@@ -131,8 +132,13 @@ func resourceServerRead(ctx context.Context, d *schema.ResourceData, m interface
 
 	serverID := d.Id()
 
-	server, _, err := c.Servers.Get(serverID, nil)
+	server, resp, err := c.Servers.Get(serverID, nil)
 	if err != nil {
+		if resp.StatusCode == http.StatusNotFound {
+			d.SetId("")
+			return diags
+		}
+
 		return diag.FromErr(err)
 	}
 

latitudesh/resource_ssh_key.go

@@ -2,6 +2,7 @@ package latitudesh
 
 import (
 	"context"
+	"net/http"
 	"time"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
@@ -83,8 +84,13 @@ func resourceSSHKeyRead(ctx context.Context, d *schema.ResourceData, m interface
 
 	keyID := d.Id()
 
-	key, _, err := c.SSHKeys.Get(keyID, d.Get("project").(string), nil)
+	key, resp, err := c.SSHKeys.Get(keyID, d.Get("project").(string), nil)
 	if err != nil {
+		if resp.StatusCode == http.StatusNotFound {
+			d.SetId("")
+			return diags
+		}
+
 		return diag.FromErr(err)
 	}
 

latitudesh/resource_user_data.go

@@ -2,6 +2,7 @@ package latitudesh
 
 import (
 	"context"
+	"net/http"
 	"time"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
@@ -83,8 +84,13 @@ func resourceUserDataRead(ctx context.Context, d *schema.ResourceData, m interfa
 
 	userDataID := d.Id()
 
-	userData, _, err := c.UserData.Get(userDataID, d.Get("project").(string), nil)
+	userData, resp, err := c.UserData.Get(userDataID, d.Get("project").(string), nil)
 	if err != nil {
+		if resp.StatusCode == http.StatusNotFound {
+			d.SetId("")
+			return diags
+		}
+
 		return diag.FromErr(err)
 	}
 

latitudesh/resource_virtual_network.go

@@ -2,6 +2,7 @@ package latitudesh
 
 import (
 	"context"
+	"net/http"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
@@ -84,8 +85,13 @@ func resourceVirtualNetworkRead(ctx context.Context, d *schema.ResourceData, m i
 
 	virtualNetworkID := d.Id()
 
-	virtualNetwork, _, err := c.VirtualNetworks.Get(virtualNetworkID, nil)
+	virtualNetwork, resp, err := c.VirtualNetworks.Get(virtualNetworkID, nil)
 	if err != nil {
+		if resp.StatusCode == http.StatusNotFound {
+			d.SetId("")
+			return diags
+		}
+
 		return diag.FromErr(err)
 	}
 

latitudesh/resource_vlan_assignment.go

@@ -2,6 +2,7 @@ package latitudesh
 
 import (
 	"context"
+	"net/http"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
@@ -92,8 +93,13 @@ func resourceVlanAssignmentRead(ctx context.Context, d *schema.ResourceData, m i
 
 	vlanAssignmentID := d.Id()
 
-	vlanAssignment, _, err := c.VlanAssignments.Get(vlanAssignmentID)
+	vlanAssignment, resp, err := c.VlanAssignments.Get(vlanAssignmentID)
 	if err != nil {
+		if resp.StatusCode == http.StatusNotFound {
+			d.SetId("")
+			return diags
+		}
+
 		return diag.FromErr(err)
 	}