mirror of https://github.com/pulumi/pulumi.git
144 lines
3.7 KiB
Go
144 lines
3.7 KiB
Go
// Copyright 2016-2021, Pulumi Corporation. All rights reserved.
|
|
//go:build !all
|
|
// +build !all
|
|
|
|
package main
|
|
|
|
import (
|
|
"errors"
|
|
"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
|
|
First pulumi.StringOutput `pulumi:"first"`
|
|
Second pulumi.StringOutput `pulumi:"second"`
|
|
}
|
|
|
|
type ComponentArgs struct {
|
|
First pulumi.StringInput `pulumi:"first"`
|
|
Second pulumi.StringInput `pulumi:"second"`
|
|
}
|
|
|
|
func NewComponent(ctx *pulumi.Context, name string, args *ComponentArgs,
|
|
opts ...pulumi.ResourceOption,
|
|
) (*Component, error) {
|
|
if args == nil {
|
|
return nil, errors.New("args is required")
|
|
}
|
|
|
|
component := &Component{}
|
|
err := ctx.RegisterComponentResource("testcomponent:index:Component", name, component, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
component.First = args.First.ToStringOutput()
|
|
component.Second = args.Second.ToStringOutput()
|
|
|
|
if err := ctx.RegisterResourceOutputs(component, pulumi.Map{
|
|
"first": args.First,
|
|
"second": args.Second,
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return component, nil
|
|
}
|
|
|
|
type ComponentGetMessageArgs struct {
|
|
Name pulumi.StringInput `pulumi:"name"`
|
|
}
|
|
|
|
type ComponentGetMessageResult struct {
|
|
Message pulumi.StringOutput `pulumi:"message"`
|
|
}
|
|
|
|
func (c *Component) GetMessage(args *ComponentGetMessageArgs) (*ComponentGetMessageResult, error) {
|
|
return &ComponentGetMessageResult{
|
|
Message: pulumi.Sprintf("%s %s, %s!", c.First, c.Second, args.Name),
|
|
}, nil
|
|
}
|
|
|
|
const (
|
|
providerName = "testcomponent"
|
|
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)
|
|
}
|
|
|
|
args := &ComponentArgs{}
|
|
if err := inputs.CopyTo(args); err != nil {
|
|
return nil, fmt.Errorf("setting args: %w", err)
|
|
}
|
|
|
|
component, err := NewComponent(ctx, name, args, 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())
|
|
}
|
|
}
|