implement wait api for workflow status

Author: Oleg Sucharevich

VERSION

@@ -1 +1 @@
-0.5.1
+0.6.0

cmd/wait.go

@@ -0,0 +1,37 @@
+// Copyright © 2019 Codefresh.Inc
+//
+// 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.
+
+package cmd
+
+import (
+	"time"
+
+	"github.com/codefresh-io/go-sdk/pkg/utils"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+// waitCmd represents the logs command
+var waitCmd = &cobra.Command{
+	Use: "wait",
+	Run: func(cmd *cobra.Command, args []string) {
+		client := viper.Get("codefresh")
+		codefreshClient := utils.CastToCodefreshOrDie(client)
+		codefreshClient.Workflows().WaitForStatus(args[0], "success", 2*time.Second, 5*time.Minute)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(waitCmd)
+}

pkg/codefresh/codefresh.go

@@ -18,6 +18,7 @@ type (
 		Pipelines() IPipelineAPI
 		Tokens() ITokenAPI
 		RuntimeEnvironments() IRuntimeEnvironmentAPI
+		Workflows() IWorkflowAPI
 	}
 )
 
@@ -42,6 +43,10 @@ func (c *codefresh) RuntimeEnvironments() IRuntimeEnvironmentAPI {
 	return newRuntimeEnvironmentAPI(c)
 }
 
+func (c *codefresh) Workflows() IWorkflowAPI {
+	return newWorkflowAPI(c)
+}
+
 func (c *codefresh) requestAPI(opt *requestOptions) (*http.Response, error) {
 	var body []byte
 	finalURL := fmt.Sprintf("%s%s", c.host, opt.path)

pkg/codefresh/workflow.go

@@ -0,0 +1,76 @@
+package codefresh
+
+import (
+	"errors"
+	"fmt"
+	"time"
+)
+
+type (
+	IWorkflowAPI interface {
+		WaitForStatus(string, string, time.Duration, time.Duration) error
+	}
+
+	workflow struct {
+		codefresh Codefresh
+	}
+
+	Workflow struct {
+		ID     string `json:"id"`
+		Status string `json:"status"`
+	}
+)
+
+func newWorkflowAPI(codefresh Codefresh) IWorkflowAPI {
+	return &workflow{codefresh}
+}
+
+func (w *workflow) WaitForStatus(id string, status string, interval time.Duration, timeout time.Duration) error {
+	err := waitFor(interval, timeout, func() (bool, error) {
+
+		wf := &Workflow{}
+		resp, err := w.codefresh.requestAPI(&requestOptions{
+			path:   fmt.Sprintf("/api/builds/%s", id),
+			method: "GET",
+		})
+		// failed in api call
+		if err != nil {
+			return false, err
+		}
+		err = w.codefresh.decodeResponseInto(resp, wf)
+		// failed to decode
+		if err != nil {
+			return false, err
+		}
+		// status match
+		if wf.Status == status {
+			return true, nil
+		}
+		// status dosent match
+		return false, nil
+	})
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func waitFor(interval time.Duration, timeout time.Duration, execution func() (bool, error)) error {
+	t := time.After(timeout)
+	tick := time.Tick(interval)
+	// Keep trying until we're timed out or got a result or got an error
+	for {
+		select {
+		// Got a timeout! fail with a timeout error
+		case <-t:
+			return errors.New("timed out")
+		case <-tick:
+			ok, err := execution()
+			if err != nil {
+				return err
+			} else if ok {
+				return nil
+			}
+		}
+	}
+}