wip preview deployment support

Author: HomelessDinosaur

cloud/common/deploy/provider/pulumi.go

@@ -343,6 +343,86 @@ func (s *PulumiProviderServer) Down(req *deploymentspb.DeploymentDownRequest, st
 	return nil
 }
 
+// Preview - automatically called by the Nitric CLI via the `preview` command
+func (s *PulumiProviderServer) Preview(req *deploymentspb.DeploymentUpRequest, stream deploymentspb.Deployment_PreviewServer) error {
+	// Verify if dependencies are available
+	if err := checkDependencies(checkPulumiAvailable, checkDockerAvailable); err != nil {
+		return status.Error(codes.FailedPrecondition, err.Error())
+	}
+
+	projectName, stackName, err := stackAndProjectFromAttributes(req.Attributes.AsMap())
+	if err != nil {
+		return err
+	}
+
+	attributesMap := req.Attributes.AsMap()
+
+	err = s.provider.Init(attributesMap)
+	if err != nil {
+		return err
+	}
+
+	pulumiProgram := createPulumiProgramForNitricProvider(req, s.provider, s.runtime)
+
+	autoStack, err := auto.UpsertStackInlineSource(context.TODO(), fmt.Sprintf("%s-%s", projectName, stackName), projectName, pulumiProgram)
+	if err != nil {
+		return err
+	}
+
+	pulumiEventsChan := make(chan events.EngineEvent)
+
+	go func() {
+		// output the stream
+		_ = pulumix.StreamPulumiPreviewEngineEvents(stream, pulumiEventsChan)
+	}()
+
+	config, err := s.provider.Config()
+	if err != nil {
+		return err
+	}
+
+	err = autoStack.SetAllConfig(context.TODO(), config)
+	if err != nil {
+		return err
+	}
+
+	refresh, ok := attributesMap["refresh"].(bool)
+
+	options := []optup.Option{optup.EventStreams(pulumiEventsChan)}
+
+	if ok && refresh {
+		options = append(options, optup.Refresh())
+	}
+
+	result, err := autoStack.Up(context.TODO(), options...)
+	if err != nil {
+		err = handleCommonErrors(err)
+
+		for _, handler := range s.errorHandlers {
+			err = handler(err)
+		}
+
+		return err
+	}
+
+	resultStr, ok := result.Outputs[resultCtxKey].Value.(string)
+	if !ok {
+		resultStr = ""
+	}
+
+	err = stream.Send(&deploymentspb.DeploymentPreviewEvent{
+		Content: &deploymentspb.DeploymentPreviewEvent_Result{
+			Result: &deploymentspb.UpResult{
+				Content: &deploymentspb.UpResult_Text{
+					Text: resultStr,
+				},
+			},
+		},
+	})
+
+	return err
+}
+
 // Start - starts the Nitric Provider gRPC server, making it callable by the Nitric CLI during deployments.
 func (s *PulumiProviderServer) Start() {
 	port := env.PORT.String()

cloud/common/deploy/provider/terraform.go

@@ -108,6 +108,14 @@ func (s *TerraformProviderServer) Down(req *deploymentspb.DeploymentDownRequest,
 	return status.Error(codes.Unimplemented, "Down not implemented for Terraform providers, please run terraform destroy against your stack state")
 }
 
+func (s *TerraformProviderServer) Preview(req *deploymentspb.DeploymentUpRequest, stream deploymentspb.Deployment_PreviewServer) error {
+	if beta, err := env.BETA_PROVIDERS.Bool(); err != nil || !beta {
+		return status.Error(codes.FailedPrecondition, "Nitric terraform providers are currently in beta, please add beta-providers to the preview field of your nitric.yaml to enable")
+	}
+
+	return status.Error(codes.Unimplemented, "Preview not implemented for Terraform providers, please run terraform plan against your stack state")
+}
+
 func NewTerraformProviderServer(provider NitricTerraformProvider, runtime RuntimeProvider) *TerraformProviderServer {
 	return &TerraformProviderServer{
 		provider: provider,

cloud/common/deploy/pulumix/clistream.go

@@ -263,6 +263,38 @@ func StreamPulumiUpEngineEvents(stream deploymentspb.Deployment_UpServer, pulumi
 	return nil
 }
 
+func StreamPulumiPreviewEngineEvents(stream deploymentspb.Deployment_PreviewServer, pulumiEventsChan <-chan events.EngineEvent) error {
+	evtHandler := pulumiEventHandler{
+		tree: DataTree{
+			Root: &DataNode{
+				Id:       "stack",
+				Data:     nil,
+				Children: make([]*DataNode, 0),
+			},
+		},
+	}
+
+	for evt := range pulumiEventsChan {
+		// translate the engine event to a server message and send back to the CLIent
+		updateDetails, err := evtHandler.engineEventToResourceUpdate(evt)
+		if err != nil {
+			// unknown event, possibly still useful for debugging.
+			logger.Debugf("%+v", evt)
+			continue
+		}
+
+		err = stream.Send(&deploymentspb.DeploymentPreviewEvent{
+			Content: &deploymentspb.DeploymentPreviewEvent_Update{
+				Update: updateDetails,
+			},
+		})
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 func StreamPulumiDownEngineEvents(stream deploymentspb.Deployment_DownServer, pulumiEventsChan <-chan events.EngineEvent) (err error) {
 	evtHandler := pulumiEventHandler{
 		tree: DataTree{