// Copyright 2016-2021, Pulumi Corporation. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. //go:build !all // +build !all package main import ( "context" "crypto/rand" "fmt" "math/big" "github.com/pulumi/pulumi/sdk/v3/go/common/resource" "github.com/pulumi/pulumi/sdk/v3/go/common/resource/plugin" rpc "github.com/pulumi/pulumi/sdk/v3/proto/go" pbempty "github.com/golang/protobuf/ptypes/empty" ) type randomResourceProvider struct{} func (p *randomResourceProvider) Check(ctx context.Context, req *rpc.CheckRequest) (*rpc.CheckResponse, error) { return &rpc.CheckResponse{Inputs: req.News, Failures: nil}, nil } func (p *randomResourceProvider) Diff(ctx context.Context, req *rpc.DiffRequest) (*rpc.DiffResponse, error) { olds, err := plugin.UnmarshalProperties(req.GetOlds(), plugin.MarshalOptions{KeepUnknowns: true, SkipNulls: true}) if err != nil { return nil, err } news, err := plugin.UnmarshalProperties(req.GetNews(), plugin.MarshalOptions{KeepUnknowns: true, SkipNulls: true}) if err != nil { return nil, err } d := olds.Diff(news) var replaces []string changes := rpc.DiffResponse_DIFF_NONE if d.Changed("length") { changes = rpc.DiffResponse_DIFF_SOME replaces = append(replaces, "length") } if d.Changed("prefix") { changes = rpc.DiffResponse_DIFF_SOME replaces = append(replaces, "prefix") } return &rpc.DiffResponse{ Changes: changes, Replaces: replaces, }, nil } func (p *randomResourceProvider) Create(ctx context.Context, req *rpc.CreateRequest) (*rpc.CreateResponse, error) { inputs, err := plugin.UnmarshalProperties(req.GetProperties(), plugin.MarshalOptions{ KeepUnknowns: true, SkipNulls: true, }) if err != nil { return nil, err } if !inputs["length"].IsNumber() { return nil, fmt.Errorf("expected input property 'length' of type 'number' but got '%s", inputs["length"].TypeString()) } n := int(inputs["length"].NumberValue()) var prefix string if p, has := inputs["prefix"]; has { if !p.IsString() { return nil, fmt.Errorf("expected input property 'prefix' of type 'string' but got '%s", p.TypeString()) } prefix = p.StringValue() } // Actually "create" the random number result, err := makeRandom(n) if err != nil { return nil, err } outputs := resource.NewPropertyMapFromMap(map[string]interface{}{ "length": n, "result": prefix + result, }) outputs["result"] = resource.MakeSecret(outputs["result"]) outputProperties, err := plugin.MarshalProperties( outputs, plugin.MarshalOptions{KeepUnknowns: true, SkipNulls: true}, ) if err != nil { return nil, err } return &rpc.CreateResponse{ Id: result, Properties: outputProperties, }, nil } func (p *randomResourceProvider) Read(ctx context.Context, req *rpc.ReadRequest) (*rpc.ReadResponse, error) { // Just return back the input state. return &rpc.ReadResponse{ Id: req.Id, Properties: req.Properties, }, nil } func (p *randomResourceProvider) Update(ctx context.Context, req *rpc.UpdateRequest) (*rpc.UpdateResponse, error) { // Our Random resource will never be updated - if there is a diff, it will be a replacement. panic("Update not implemented") } func (p *randomResourceProvider) Delete(ctx context.Context, req *rpc.DeleteRequest) (*pbempty.Empty, error) { // Note that for our Random resource, we don't have to do anything on Delete. return &pbempty.Empty{}, nil } func makeRandom(length int) (string, error) { charset := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789") result := make([]rune, length) for i := range result { num, err := rand.Int(rand.Reader, big.NewInt(int64(len(charset)))) if err != nil { return "", err } result[i] = charset[num.Int64()] } return string(result), nil }