117 lines
3.4 KiB
Go
117 lines
3.4 KiB
Go
package grpcwrap
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"strings"
|
|
"unicode/utf8"
|
|
|
|
"github.com/hashicorp/terraform/configs/configschema"
|
|
"github.com/hashicorp/terraform/internal/communicator/shared"
|
|
"github.com/hashicorp/terraform/internal/tfplugin5"
|
|
"github.com/hashicorp/terraform/plugin/convert"
|
|
"github.com/hashicorp/terraform/provisioners"
|
|
)
|
|
|
|
// New wraps a provisioners.Interface to implement a grpc ProviderServer.
|
|
// This is useful for creating a test binary out of an internal provider
|
|
// implementation.
|
|
func Provisioner(p provisioners.Interface) tfplugin5.ProvisionerServer {
|
|
return &provisioner{
|
|
provisioner: p,
|
|
schema: p.GetSchema().Provisioner,
|
|
}
|
|
}
|
|
|
|
type provisioner struct {
|
|
provisioner provisioners.Interface
|
|
schema *configschema.Block
|
|
}
|
|
|
|
func (p *provisioner) GetSchema(_ context.Context, req *tfplugin5.GetProvisionerSchema_Request) (*tfplugin5.GetProvisionerSchema_Response, error) {
|
|
resp := &tfplugin5.GetProvisionerSchema_Response{}
|
|
|
|
resp.Provisioner = &tfplugin5.Schema{
|
|
Block: &tfplugin5.Schema_Block{},
|
|
}
|
|
|
|
if p.schema != nil {
|
|
resp.Provisioner.Block = convert.ConfigSchemaToProto(p.schema)
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
func (p *provisioner) ValidateProvisionerConfig(_ context.Context, req *tfplugin5.ValidateProvisionerConfig_Request) (*tfplugin5.ValidateProvisionerConfig_Response, error) {
|
|
resp := &tfplugin5.ValidateProvisionerConfig_Response{}
|
|
ty := p.schema.ImpliedType()
|
|
|
|
configVal, err := decodeDynamicValue(req.Config, ty)
|
|
if err != nil {
|
|
resp.Diagnostics = convert.AppendProtoDiag(resp.Diagnostics, err)
|
|
return resp, nil
|
|
}
|
|
|
|
validateResp := p.provisioner.ValidateProvisionerConfig(provisioners.ValidateProvisionerConfigRequest{
|
|
Config: configVal,
|
|
})
|
|
|
|
resp.Diagnostics = convert.AppendProtoDiag(resp.Diagnostics, validateResp.Diagnostics)
|
|
return resp, nil
|
|
}
|
|
|
|
func (p *provisioner) ProvisionResource(req *tfplugin5.ProvisionResource_Request, srv tfplugin5.Provisioner_ProvisionResourceServer) error {
|
|
// We send back a diagnostics over the stream if there was a
|
|
// provisioner-side problem.
|
|
srvResp := &tfplugin5.ProvisionResource_Response{}
|
|
|
|
ty := p.schema.ImpliedType()
|
|
configVal, err := decodeDynamicValue(req.Config, ty)
|
|
if err != nil {
|
|
srvResp.Diagnostics = convert.AppendProtoDiag(srvResp.Diagnostics, err)
|
|
srv.Send(srvResp)
|
|
return nil
|
|
}
|
|
|
|
connVal, err := decodeDynamicValue(req.Connection, shared.ConnectionBlockSupersetSchema.ImpliedType())
|
|
if err != nil {
|
|
srvResp.Diagnostics = convert.AppendProtoDiag(srvResp.Diagnostics, err)
|
|
srv.Send(srvResp)
|
|
return nil
|
|
}
|
|
|
|
resp := p.provisioner.ProvisionResource(provisioners.ProvisionResourceRequest{
|
|
Config: configVal,
|
|
Connection: connVal,
|
|
UIOutput: uiOutput{srv},
|
|
})
|
|
|
|
srvResp.Diagnostics = convert.AppendProtoDiag(srvResp.Diagnostics, resp.Diagnostics)
|
|
srv.Send(srvResp)
|
|
return nil
|
|
}
|
|
|
|
func (p *provisioner) Stop(context.Context, *tfplugin5.Stop_Request) (*tfplugin5.Stop_Response, error) {
|
|
resp := &tfplugin5.Stop_Response{}
|
|
err := p.provisioner.Stop()
|
|
if err != nil {
|
|
resp.Error = err.Error()
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
// uiOutput implements the terraform.UIOutput interface to adapt the grpc
|
|
// stream to the legacy Provisioner.Apply method.
|
|
type uiOutput struct {
|
|
srv tfplugin5.Provisioner_ProvisionResourceServer
|
|
}
|
|
|
|
func (o uiOutput) Output(s string) {
|
|
err := o.srv.Send(&tfplugin5.ProvisionResource_Response{
|
|
Output: strings.ToValidUTF8(s, string(utf8.RuneError)),
|
|
})
|
|
if err != nil {
|
|
log.Printf("[ERROR] %s", err)
|
|
}
|
|
}
|