Function ConvertState

provisioner/terraform/resources.go:193–1076

ConvertState consumes Terraform state and a GraphViz representation produced by `terraform graph` to produce resources consumable by Coder. nolint:gocognit // This function makes more sense being large for now, until refactored.

(ctx context.Context, modules []*tfjson.StateModule, rawGraph string, logger slog.Logger)

// produced by `terraform graph` to produce resources consumable by Coder.
// nolint:gocognit // This function makes more sense being large for now, until refactored.
func ConvertState(ctx context.Context, modules []*tfjson.StateModule, rawGraph string, logger slog.Logger) (*State, error) {
	parsedGraph, err := gographviz.ParseString(rawGraph)
	if err != nil {
		return nil, xerrors.Errorf("parse graph: %w", err)
	}
	graph, err := gographviz.NewAnalysedGraph(parsedGraph)
	if err != nil {
		return nil, xerrors.Errorf("analyze graph: %w", err)
	}

	resources := make([]*proto.Resource, 0)
	resourceAgents := map[string][]*proto.Agent{}

	// Indexes Terraform resources by their label.
	// The label is what "terraform graph" uses to reference nodes.
	tfResourcesByLabel := map[string]map[string]*tfjson.StateResource{}

	// Extra array to preserve the order of rich parameters.
	tfResourcesRichParameters := make([]*tfjson.StateResource, 0)
	tfResourcesPresets := make([]*tfjson.StateResource, 0)
	tfResourcesAITasks := make([]*tfjson.StateResource, 0)
	var findTerraformResources func(mod *tfjson.StateModule)
	findTerraformResources = func(mod *tfjson.StateModule) {
		for _, module := range mod.ChildModules {
			findTerraformResources(module)
		}
		for _, resource := range mod.Resources {
			if resource.Type == "coder_parameter" {
				tfResourcesRichParameters = append(tfResourcesRichParameters, resource)
			}
			if resource.Type == "coder_workspace_preset" {
				tfResourcesPresets = append(tfResourcesPresets, resource)
			}
			if resource.Type == "coder_ai_task" {
				tfResourcesAITasks = append(tfResourcesAITasks, resource)
			}

			label := convertAddressToLabel(resource.Address)
			if tfResourcesByLabel[label] == nil {
				tfResourcesByLabel[label] = map[string]*tfjson.StateResource{}
			}
			tfResourcesByLabel[label][resource.Address] = resource
		}
	}
	for _, module := range modules {
		findTerraformResources(module)
	}

	// Group all resources by type in a single pass so that
	// subsequent lookups are O(1) instead of scanning the
	// full map each time.
	sortedResources := sortResourcesByType(tfResourcesByLabel)

	// Find all agents!
	agentNames := map[string]struct{}{}
	for _, tfResource := range sortedResources["coder_agent"] {
		var attrs agentAttributes
		err = mapstructure.Decode(tfResource.AttributeValues, &attrs)