add import to random_pet resource #184

Author: wakeful

docs/resources/pet.md

@@ -54,4 +54,12 @@ resource "aws_instance" "server" {
 
 - `id` (String) The random pet name.
 
+## Import
 
+Import is supported using the following syntax:
+
+```shell
+# Random Pet can be imported with format: pet_name,separator,prefix
+
+terraform import random_pet.example pet_name,separator,prefix
+```

examples/resources/random_pet/import.sh

@@ -0,0 +1,3 @@
+# Random Pet can be imported with format: pet_name,separator,prefix
+
+terraform import random_pet.example pet_name,separator,prefix

internal/provider/resource_pet.go

@@ -7,6 +7,7 @@ import (
 
 	petname "github.com/dustinkirkland/golang-petname"
 	"github.com/hashicorp/terraform-plugin-framework/diag"
+	"github.com/hashicorp/terraform-plugin-framework/path"
 	"github.com/hashicorp/terraform-plugin-framework/resource"
 	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
 	"github.com/hashicorp/terraform-plugin-framework/types"
@@ -148,6 +149,31 @@ func (r *petResource) Update(ctx context.Context, req resource.UpdateRequest, re
 func (r *petResource) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) {
 }
 
+func (r *petResource) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) {
+	idParts := strings.Split(req.ID, ",")
+	if len(idParts) != 3 {
+		resp.Diagnostics.AddError(
+			"Unexpected Import Identifier",
+			fmt.Sprintf("Expected import identifier with format: pet_name,separator,prefix. Got: %q", req.ID),
+		)
+		return
+	}
+
+	id, separator, prefix := idParts[0], idParts[1], idParts[2]
+
+	resp.Diagnostics.Append(resp.State.SetAttribute(ctx, path.Root("id"), id)...)
+	resp.Diagnostics.Append(resp.State.SetAttribute(ctx, path.Root("length"), int64(len(id)))...)
+
+	if separator == "" {
+		separator = "-"
+	}
+	resp.Diagnostics.Append(resp.State.SetAttribute(ctx, path.Root("separator"), separator)...)
+
+	if prefix != "" {
+		resp.Diagnostics.Append(resp.State.SetAttribute(ctx, path.Root("prefix"), prefix)...)
+	}
+}
+
 type petModelV0 struct {
 	ID        types.String `tfsdk:"id"`
 	Keepers   types.Map    `tfsdk:"keepers"`