Add importer for startup script

squat · Nov 30, 2017 · cc6fb30 · cc6fb30
1 parent 9fb7df0
commit cc6fb30
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/vultr/resource_startup_script.go b/vultr/resource_startup_script.go
@@ -14,6 +14,9 @@ func resourceStartupScript() *schema.Resource {
 		Read:   resourceStartupScriptRead,
 		Update: resourceStartupScriptUpdate,
 		Delete: resourceStartupScriptDelete,
+		Importer: &schema.ResourceImporter{
+			State: schema.ImportStatePassthrough,
+		},
 
 		Schema: map[string]*schema.Schema{
 			"content": {
@@ -67,6 +70,12 @@ func resourceStartupScriptRead(d *schema.ResourceData, meta interface{}) error {
 		return fmt.Errorf("Error getting startup script (%s): %v", d.Id(), err)
 	}
 
+	if script == (lib.StartupScript{}) {
+		log.Printf("[WARN] Removing startup script (%s) because it is gone", d.Id())
+		d.SetId("")
+		return nil
+	}
+
 	d.Set("content", script.Content)
 	d.Set("name", script.Name)
 	d.Set("type", script.Type)