fe603568fd
We can use the new `provider.MainWithOptions` to reduce boilerplate in some of our testcomponent providers. Also, while cleaning up here, I took this as an opportunity to replace use of `github.com/pkg/errors` in the `tests` dir to use the built-in functionality of the Go standard library.
119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
// Copyright 2016-2021, Pulumi Corporation. All rights reserved.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/blang/semver"
|
|
|
|
"github.com/pulumi/pulumi/pkg/v3/resource/provider"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/util/cmdutil"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
|
|
pulumiprovider "github.com/pulumi/pulumi/sdk/v3/go/pulumi/provider"
|
|
)
|
|
|
|
type Component struct {
|
|
pulumi.ResourceState
|
|
}
|
|
|
|
func NewComponent(ctx *pulumi.Context, name string, opts ...pulumi.ResourceOption) (*Component, error) {
|
|
component := &Component{}
|
|
err := ctx.RegisterComponentResource("testcomponent:index:Component", name, component, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := ctx.RegisterResourceOutputs(component, pulumi.Map{}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return component, nil
|
|
}
|
|
|
|
type ComponentGetMessageArgs struct {
|
|
Echo pulumi.StringInput `pulumi:"echo"`
|
|
}
|
|
|
|
type ComponentGetMessageResult struct {
|
|
Message pulumi.StringOutput `pulumi:"message"`
|
|
}
|
|
|
|
func (c *Component) GetMessage(args *ComponentGetMessageArgs) (*ComponentGetMessageResult, error) {
|
|
message := args.Echo.ToStringOutput().ApplyT(func(val string) string {
|
|
panic("should not run (echo)")
|
|
}).(pulumi.StringOutput)
|
|
|
|
return &ComponentGetMessageResult{
|
|
Message: message,
|
|
}, nil
|
|
}
|
|
|
|
const providerName = "testcomponent"
|
|
const version = "0.0.1"
|
|
|
|
type module struct {
|
|
version semver.Version
|
|
}
|
|
|
|
func (m *module) Version() semver.Version {
|
|
return m.version
|
|
}
|
|
|
|
func (m *module) Construct(ctx *pulumi.Context, name, typ, urn string) (r pulumi.Resource, err error) {
|
|
switch typ {
|
|
case "testcomponent:index:Component":
|
|
r = &Component{}
|
|
default:
|
|
return nil, fmt.Errorf("unknown resource type: %s", typ)
|
|
}
|
|
|
|
err = ctx.RegisterResource(typ, name, nil, r, pulumi.URN_(urn))
|
|
return
|
|
}
|
|
|
|
func main() {
|
|
// Register any resources that can come back as resource references that need to be rehydrated.
|
|
pulumi.RegisterResourceModule("testcomponent", "index", &module{semver.MustParse(version)})
|
|
|
|
if err := provider.MainWithOptions(provider.Options{
|
|
Name: providerName,
|
|
Version: version,
|
|
Construct: func(ctx *pulumi.Context, typ, name string, inputs pulumiprovider.ConstructInputs,
|
|
options pulumi.ResourceOption) (*pulumiprovider.ConstructResult, error) {
|
|
|
|
if typ != "testcomponent:index:Component" {
|
|
return nil, fmt.Errorf("unknown resource type %s", typ)
|
|
}
|
|
|
|
component, err := NewComponent(ctx, name, options)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("creating component: %w", err)
|
|
}
|
|
|
|
return pulumiprovider.NewConstructResult(component)
|
|
},
|
|
Call: func(ctx *pulumi.Context, tok string, args pulumiprovider.CallArgs) (*pulumiprovider.CallResult, error) {
|
|
if tok != "testcomponent:index:Component/getMessage" {
|
|
return nil, fmt.Errorf("unknown method %s", tok)
|
|
}
|
|
|
|
methodArgs := &ComponentGetMessageArgs{}
|
|
res, err := args.CopyTo(methodArgs)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("setting args: %w", err)
|
|
}
|
|
component := res.(*Component)
|
|
|
|
result, err := component.GetMessage(methodArgs)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("calling method: %w", err)
|
|
}
|
|
|
|
return pulumiprovider.NewCallResult(result)
|
|
},
|
|
}); err != nil {
|
|
cmdutil.ExitError(err.Error())
|
|
}
|
|
}
|