Happy path implementation (#50)

proto/unmango/baremetal/v1alpha1/command.proto

@@ -11,6 +11,7 @@ message CommandRequest {
   Command command = 2;
   repeated string args = 3;
   map<string, Flag> flags = 4;
+  string stdin = 5;
 }
 
 message CommandResponse {

provider/cmd/provisioner/main.go

@@ -28,6 +28,7 @@ var rootCmd = &cobra.Command{
 			return err
 		}
 
+		slog.SetDefault(log)
 		log.Info("serving", "network", network, "address", address, "verbose", verbose)
 		return provisioner.Serve(lis)
 	},

provider/pkg/provider/cmd/tee.go

@@ -58,6 +58,7 @@ func (state *TeeState) create(ctx context.Context, input TeeArgs) error {
 		Command: pb.Command_COMMAND_TEE,
 		Args:    input.Create.Files,
 		Flags:   map[string]*pb.Flag{},
+		Stdin:   input.Stdin,
 	})
 	if err != nil {
 		return err

provider/pkg/provisioner/cmd/service.go

@@ -1,26 +1,59 @@
 package cmd
 
 import (
+	"bytes"
 	"context"
 	"fmt"
+	"log/slog"
+	"os/exec"
+	"strings"
 
 	pb "github.com/unmango/pulumi-baremetal/gen/go/unmango/baremetal/v1alpha1"
 )
 
 type service struct {
 	pb.UnimplementedCommandServiceServer
+	log *slog.Logger
 }
 
 func NewServer() pb.CommandServiceServer {
-	return &service{}
+	return &service{log: slog.Default()}
 }
 
 // Command implements baremetalv1alpha1.CommandServiceServer.
 func (c *service) Command(ctx context.Context, req *pb.CommandRequest) (*pb.CommandResponse, error) {
-	switch req.Command {
+	log := c.log.With("op", req.Op)
+	bin, err := getBin(req.Command)
+	if err != nil {
+		return nil, err
+	}
+
+	cmd := exec.CommandContext(ctx, bin, req.Args...)
+	log = log.With("bin", bin, "args", req.Args)
+
+	stdout := &bytes.Buffer{}
+	stderr := &bytes.Buffer{}
+	cmd.Stdin = strings.NewReader(req.Stdin)
+	cmd.Stdout = stdout
+	cmd.Stderr = stderr
+
+	log.Info("executing command")
+	err = cmd.Run()
+	if err != nil {
+		log.Error("command failed", "err", err)
+	}
+
+	return &pb.CommandResponse{
+		Stdout: stdout.String(),
+		Stderr: stderr.String(),
+	}, nil
+}
+
+func getBin(cmd pb.Command) (string, error) {
+	switch cmd {
 	case pb.Command_COMMAND_TEE:
-		return tee(ctx, req)
+		return "tee", nil
 	}
 
-	return nil, fmt.Errorf("unrecognized command: %s", req.Command)
+	return "", fmt.Errorf("unrecognized command: %s", cmd)
 }

tests/logging.go

@@ -21,7 +21,7 @@ func NewLogger(w io.Writer) tc.Logging {
 
 // Accept implements testcontainers.LogConsumer.
 func (w *writerConsumer) Accept(log tc.Log) {
-	_, _ = w.Write(log.Content)
+	_, _ = w.Write(append(log.Content, '\n'))
 }
 
 // Printf implements testcontainers.Logging.

tests/provider.go

@@ -1,8 +1,12 @@
 package tests
 
 import (
+	"strings"
+
 	"github.com/blang/semver"
 	"github.com/pulumi/pulumi-go-provider/integration"
+	"github.com/pulumi/pulumi/sdk/v3/go/common/resource"
+	"github.com/pulumi/pulumi/sdk/v3/go/common/tokens"
 
 	baremetal "github.com/unmango/pulumi-baremetal/provider"
 )
@@ -14,3 +18,13 @@ func NewIntegrationProvider() integration.Server {
 		baremetal.Provider(),
 	)
 }
+
+func urn(typ string, mods ...string) resource.URN {
+	if len(mods) == 0 {
+		mods = []string{"index"}
+	}
+
+	tok := strings.Join(append(mods, typ), ":")
+	return resource.NewURN("stack", "proj", "",
+		tokens.Type("test:"+tok), "name")
+}

tests/provider_suite_test.go

@@ -2,6 +2,7 @@ package tests
 
 import (
 	"context"
+	"os"
 	"testing"
 
 	. "github.com/onsi/ginkgo/v2"
@@ -19,9 +20,10 @@ func TestProvider(t *testing.T) {
 
 var _ = BeforeSuite(func(ctx context.Context) {
 	By("creating a provisioner")
-	prov, err := NewTestProvisioner(ctx, GinkgoWriter)
+	prov, err := NewTestProvisioner(ctx, os.Stdout)
 	Expect(err).NotTo(HaveOccurred())
 
+	By("starting the provisioner")
 	err = prov.Start(ctx)
 	Expect(err).NotTo(HaveOccurred())
 	provisioner = prov

tests/provisioner.go

@@ -47,6 +47,7 @@ func NewTestProvisioner(ctx context.Context, logger io.Writer) (testProvisioner,
 			Cmd: []string{
 				"--network", defaultProtocol,
 				"--address", fmt.Sprintf("%s:%d", "0.0.0.0", port.Int()),
+				"--verbose",
 			},
 			ExposedPorts: []string{port.Port()},
 			WaitingFor:   wait.ForListeningPort(port),
@@ -74,6 +75,24 @@ func (p testProvisioner) Stop(ctx context.Context) error {
 	return p.ct.Stop(ctx, &timeout)
 }
 
+func (p testProvisioner) Exec(ctx context.Context, args ...string) error {
+	code, output, err := p.ct.Exec(ctx, args)
+	if err != nil {
+		return err
+	}
+
+	out, err := io.ReadAll(output)
+	if err != nil {
+		return err
+	}
+
+	if code != 0 {
+		return fmt.Errorf("unexpected return code: %d, output: %s", code, out)
+	}
+
+	return nil
+}
+
 func (prov testProvisioner) ConfigureProvider(ctx context.Context, server integration.Server) error {
 	ip, err := prov.ct.ContainerIP(ctx)
 	if err != nil {