Merge branch 'workflow-child' of https://github.com/packethost/rover …

…into workflow-child

grpc-server/workflow.go

@@ -33,6 +33,7 @@ func (s *server) CreateWorkflow(ctx context.Context, in *workflow.CreateRequest)
 	labels["op"] = "createworkflow"
 	msg = "creating a new workflow"
 	id := uuid.NewV4()
+	//var data string
 	fn := func() error {
 		wf := db.Workflow{
 			ID:       id.String(),
@@ -52,7 +53,6 @@ func (s *server) CreateWorkflow(ctx context.Context, in *workflow.CreateRequest)
 		if err != nil {
 			return err
 		}
-
 		return nil
 	}
 

worker/action.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"io/ioutil"
 	"log"
 	"os"
 
@@ -19,6 +20,11 @@ var (
 )
 
 func executeAction(ctx context.Context, action *pb.WorkflowAction) error {
+	err := pullActionImage(ctx, action.GetImage())
+	if err != nil {
+		log.Fatalln("Action: ", action.Name, "Failed for error: ", err)
+		return err
+	}
 	id, err := createContainer(ctx, action)
 	if err != nil {
 		log.Fatalln("Action: ", action.Name, "Failed for error: ", err)
@@ -31,17 +37,40 @@ func executeAction(ctx context.Context, action *pb.WorkflowAction) error {
 		log.Fatalln("Action: ", action.Name, "Failed for error: ", err)
 		return err
 	}
-	// TODO: use "github.com/docker/docker/pkg/stdcopy"
+
 	io.Copy(os.Stdout, out)
 	return nil
 }
 
+func pullActionImage(ctx context.Context, image string) error {
+	f, err := os.Open("/" + registry + "/ca.crt")
+	defer f.Close()
+	if err != nil {
+		log.Fatalln("Failed to LOAD the certificate for registry:", registry)
+		return err
+	}
+	auth, err := ioutil.ReadAll(f)
+	if err != nil {
+		log.Fatalln("Failed to READ the certificate for registry:", registry)
+		return err
+	}
+
+	rw, err := cli.ImagePull(ctx, registry+"/"+image, types.ImagePullOptions{RegistryAuth: string(auth)})
+	defer rw.Close()
+	if err != nil {
+		log.Fatalln("Failed to pull Docker image", err)
+		return err
+	}
+	return nil
+}
+
 func createContainer(ctx context.Context, action *pb.WorkflowAction) (string, error) {
 	config := &container.Config{
 		Image:        registry + "/" + action.GetImage(),
 		AttachStdout: true,
 		AttachStderr: true,
 	}
+
 	resp, err := cli.ContainerCreate(ctx, config, nil, nil, action.GetName())
 	if err != nil {
 		log.Fatalln(err)