Terraform Provider: organize imports of generated code

Updates code generation to sort and group imports in the same
manner as dictated by the gci rules in the golanci-lint config.
This also removes the redundant `tfschema` import alias for packages
which do not have custom schema.

terraform/gen/main.go

@@ -373,6 +373,7 @@ var (
 		Kind:                  "device",
 		ID:                    "trustedDevice.Metadata.Name",
 		HasStaticID:           true,
+		SchemaPackage:         "schemav1",
 		SchemaPackagePath:     "github.com/gravitational/teleport-plugins/terraform/tfschema/devicetrust/v1",
 		IsPlainStruct:         true,
 		UUIDMetadataName:      true,
@@ -477,7 +478,24 @@ func generate(p payload, tpl, outFile string) {
 		log.Fatal(err)
 	}
 
-	t, err := template.ParseFiles(path.Join("gen", tpl))
+	funcs := template.FuncMap{
+		"schemaImport": func(p payload) string {
+			if p.SchemaPackage == "tfschema" {
+				return `"` + p.SchemaPackagePath + `"`
+			}
+
+			return p.SchemaPackage + ` "` + p.SchemaPackagePath + `"`
+		},
+		"protoImport": func(p payload) string {
+			if p.ConvertPackagePath != "" {
+				return "convert" + ` "` + p.ConvertPackagePath + `"`
+			}
+
+			return p.ProtoPackage + ` "` + p.ProtoPackagePath + `"`
+		},
+	}
+
+	t, err := template.New(p.Name).Funcs(funcs).ParseFiles(path.Join("gen", tpl))
 	if err != nil {
 		log.Fatal(err)
 	}

terraform/gen/plural_data_source.go.tpl

@@ -20,20 +20,16 @@ package provider
 import (
 	"context"
 
+	{{ if not .IsPlainStruct }}
+    {{- protoImport . }}
+    {{ end }}
+	"github.com/gravitational/trace"
 	"github.com/hashicorp/terraform-plugin-framework/diag"
+	"github.com/hashicorp/terraform-plugin-framework/path"
 	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
 	"github.com/hashicorp/terraform-plugin-framework/types"
-	"github.com/hashicorp/terraform-plugin-framework/path"
 
-	{{.SchemaPackage}} "{{.SchemaPackagePath}}"
-	{{if not .IsPlainStruct -}}
-	{{- if .ConvertPackagePath}}
-	convert "{{.ConvertPackagePath}}"
-	{{ else }}
-	{{.ProtoPackage}} "{{.ProtoPackagePath}}"
-	{{- end}}
-	{{end -}}
-	"github.com/gravitational/trace"
+	{{ schemaImport . }}
 )
 
 // dataSourceTeleport{{.Name}}Type is the data source metadata type

terraform/gen/plural_resource.go.tpl

@@ -19,31 +19,30 @@ package provider
 
 import (
 	"context"
+{{- if .RandomMetadataName }}
+    "crypto/rand"
+    "encoding/hex"
+{{- end }}
 	"fmt"
-{{- if .RandomMetadataName}}
-	"crypto/rand"
-	"encoding/hex"
-{{- end}}
-{{- if .UUIDMetadataName}}
-	"github.com/google/uuid"
-{{- end}}
-{{- range $i, $a := .ExtraImports}}
+{{- range $i, $a := .ExtraImports }}
 	"{{$a}}"
-{{- end}}
-
-	"github.com/hashicorp/terraform-plugin-framework/diag"
-	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
-	"github.com/hashicorp/terraform-plugin-framework/types"
-	"github.com/hashicorp/terraform-plugin-framework/path"
-
+{{- end }}
+{{ if .UUIDMetadataName }}
+	"github.com/google/uuid"
+{{- end }}
 	{{.ProtoPackage}} "{{.ProtoPackagePath}}"
-	{{.SchemaPackage}} "{{.SchemaPackagePath}}"
-{{- if .ConvertPackagePath}}
-	convert "{{.ConvertPackagePath}}"
-{{- end}}
+	{{ if .ConvertPackagePath -}}
+    convert "{{.ConvertPackagePath}}"
+    {{- end}}
 	"github.com/gravitational/teleport/integrations/lib/backoff"
 	"github.com/gravitational/trace"
+	"github.com/hashicorp/terraform-plugin-framework/diag"
+	"github.com/hashicorp/terraform-plugin-framework/path"
+	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
+	"github.com/hashicorp/terraform-plugin-framework/types"
 	"github.com/jonboulle/clockwork"
+
+	{{ schemaImport . }}
 )
 
 // resourceTeleport{{.Name}}Type is the resource metadata type

terraform/gen/singular_data_source.go.tpl

@@ -20,15 +20,15 @@ package provider
 import (
 	"context"
 
+	{{ if not .IsPlainStruct }}
+    {{- protoImport . }}
+    {{- end }}
+	"github.com/gravitational/trace"
 	"github.com/hashicorp/terraform-plugin-framework/diag"
 	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
 	"github.com/hashicorp/terraform-plugin-framework/types"
 
-	{{.SchemaPackage}} "{{.SchemaPackagePath}}"
-	{{if not .IsPlainStruct -}}
-	{{.ProtoPackage}} "{{.ProtoPackagePath}}"
-	{{end -}}
-	"github.com/gravitational/trace"
+	{{ schemaImport . }}
 )
 
 // dataSourceTeleport{{.Name}}Type is the data source metadata type

terraform/gen/singular_resource.go.tpl

@@ -24,15 +24,15 @@ import (
 	 "math"
 {{- end}}
 
-	"github.com/hashicorp/terraform-plugin-framework/diag"
-	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
-	"github.com/hashicorp/terraform-plugin-framework/types"
-
 	{{.ProtoPackage}} "{{.ProtoPackagePath}}"
-	{{.SchemaPackage}} "{{.SchemaPackagePath}}"
 	"github.com/gravitational/teleport/integrations/lib/backoff"
 	"github.com/gravitational/trace"
+	"github.com/hashicorp/terraform-plugin-framework/diag"
+	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
+	"github.com/hashicorp/terraform-plugin-framework/types"
 	"github.com/jonboulle/clockwork"
+
+	{{ schemaImport . }}
 )
 
 // resourceTeleport{{.Name}}Type is the resource metadata type