driftctl/pkg/resource/deserializer.go

package resource

import (
	"encoding/json"

	"github.com/snyk/driftctl/enumeration/resource"
	"github.com/zclconf/go-cty/cty"
	ctyjson "github.com/zclconf/go-cty/cty/json"
)

type Deserializer struct {
	factory resource.ResourceFactory
}

func NewDeserializer(factory resource.ResourceFactory) *Deserializer {
	return &Deserializer{factory}
}

func (s *Deserializer) Deserialize(ty string, rawList []cty.Value) ([]*resource.Resource, error) {
	resources := make([]*resource.Resource, 0)
	for _, rawResource := range rawList {
		rawResource := rawResource
		res, err := s.DeserializeOne(ty, rawResource)
		if err != nil {
			return nil, err
		}
		resources = append(resources, res)
	}
	return resources, nil
}

func (s *Deserializer) DeserializeOne(ty string, value cty.Value) (*resource.Resource, error) {
	if value.IsNull() {
		return nil, nil
	}

	// Marked values cannot be deserialized to JSON.
	// For example, this ensures we can deserialize sensitive values too.
	unmarkedVal, _ := value.UnmarkDeep()

	var attrs resource.Attributes
	bytes, _ := ctyjson.Marshal(unmarkedVal, unmarkedVal.Type())
	err := json.Unmarshal(bytes, &attrs)
	if err != nil {
		return nil, err
	}

	return s.factory.CreateAbstractResource(ty, value.GetAttr("id").AsString(), attrs), nil
}