68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package terraform
|
|
|
|
import (
|
|
"github.com/cloudskiff/driftctl/pkg/resource"
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/zclconf/go-cty/cty"
|
|
"github.com/zclconf/go-cty/cty/gocty"
|
|
)
|
|
|
|
type TerraformResourceFactory struct {
|
|
providerLibrary *ProviderLibrary
|
|
resourceSchemaRepository resource.SchemaRepositoryInterface
|
|
}
|
|
|
|
func NewTerraformResourceFactory(providerLibrary *ProviderLibrary, resourceSchemaRepository resource.SchemaRepositoryInterface) *TerraformResourceFactory {
|
|
return &TerraformResourceFactory{
|
|
providerLibrary: providerLibrary,
|
|
resourceSchemaRepository: resourceSchemaRepository,
|
|
}
|
|
}
|
|
|
|
func (r *TerraformResourceFactory) resolveType(ty string) (cty.Type, error) {
|
|
provider, err := r.providerLibrary.GetProviderForResourceType(ty)
|
|
if err != nil {
|
|
return cty.NilType, err
|
|
}
|
|
if schemas, exist := provider.Schema()[ty]; exist {
|
|
return schemas.Block.ImpliedType(), nil
|
|
}
|
|
|
|
return cty.NilType, errors.New("Unable to find ")
|
|
}
|
|
|
|
func (r *TerraformResourceFactory) CreateResource(data interface{}, ty string) (*cty.Value, error) {
|
|
ctyType, err := r.resolveType(ty)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
logrus.WithFields(logrus.Fields{
|
|
"type": ty,
|
|
}).Debug("Found cty type for resource")
|
|
|
|
val, err := gocty.ToCtyValue(data, ctyType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &val, nil
|
|
}
|
|
|
|
func (r *TerraformResourceFactory) CreateAbstractResource(ty, id string, data map[string]interface{}) *resource.AbstractResource {
|
|
attributes := resource.Attributes(data)
|
|
attributes.SanitizeDefaultsV3()
|
|
|
|
schema, exist := r.resourceSchemaRepository.(*resource.SchemaRepository).GetSchema(ty)
|
|
if exist && schema.NormalizeFunc != nil {
|
|
schema.NormalizeFunc(&attributes)
|
|
}
|
|
|
|
return &resource.AbstractResource{
|
|
Id: id,
|
|
Type: ty,
|
|
Attrs: &attributes,
|
|
}
|
|
}
|