osCommand.go

package command

import (
	"strings"

	"github.com/pulumi/pulumi-command/sdk/go/command/remote"
	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
)

// OSCommand defines the commands which are OS specifics
type OSCommand interface {
	GetTemporaryDirectory() string
	GetHomeDirectory() string

	CreateDirectory(
		runner *Runner,
		resourceName string,
		remotePath pulumi.StringInput,
		useSudo bool,
		opts ...pulumi.ResourceOption) (*remote.Command, error)

	CopyInlineFile(
		runner *Runner,
		fileContent pulumi.StringInput,
		remotePath string,
		useSudo bool,
		opts ...pulumi.ResourceOption) (*remote.Command, error)

	BuildCommandString(
		command pulumi.StringInput,
		env pulumi.StringMap,
		sudo bool,
		passwordFromStdin bool,
		user string) pulumi.StringInput

	IsPathAbsolute(path string) bool
}

// ------------------------------
// Helpers to implement osCommand
// ------------------------------

const backupExtension = "pulumi.backup"

func createDirectory(
	runner *Runner,
	name string,
	createCmd string,
	deleteCmd string,
	useSudo bool,
	opts ...pulumi.ResourceOption,
) (*remote.Command, error) {
	// If the folder was previously created, make sure to delete it before creating it.
	opts = append(opts, pulumi.DeleteBeforeReplace(true))
	return runner.Command(name,
		&Args{
			Create:   pulumi.String(createCmd),
			Delete:   pulumi.String(deleteCmd),
			Sudo:     useSudo,
			Triggers: pulumi.Array{pulumi.String(createCmd), pulumi.BoolPtr(useSudo)},
		}, opts...)
}

func copyInlineFile(
	name string,
	runner *Runner,
	fileContent pulumi.StringInput,
	useSudo bool,
	createCmd string,
	deleteCmd string,
	opts ...pulumi.ResourceOption,
) (*remote.Command, error) {
	return runner.Command(runner.namer.ResourceName("copy-file", name),
		&Args{
			Create:   pulumi.String(createCmd),
			Delete:   pulumi.String(deleteCmd),
			Stdin:    fileContent,
			Sudo:     useSudo,
			Triggers: pulumi.Array{pulumi.String(createCmd), fileContent, pulumi.BoolPtr(useSudo)},
		}, opts...)
}

func buildCommandString(
	command pulumi.StringInput,
	envVars pulumi.StringArray,
	fct func(envVarsStr pulumi.StringOutput) pulumi.StringInput,
) pulumi.StringInput {
	if command == nil {
		return nil
	}

	envVarsStr := envVars.ToStringArrayOutput().ApplyT(func(inputs []string) string {
		return strings.Join(inputs, " ")
	}).(pulumi.StringOutput)

	return fct(envVarsStr)
}