feat: Add v2.KeptnInterface that adds context.Context support to api.KeptnInterface

pkg/api/utils/apiServiceUtils.go

@@ -2,6 +2,7 @@ package api
 
 import (
 	"bytes"
+	"context"
 	"crypto/tls"
 	"fmt"
 	"io/ioutil"
@@ -61,11 +62,66 @@ func getClientTransport(rt http.RoundTripper) http.RoundTripper {
 		return tr
 	}
 	return rt
+}
+
+func getAndExpectOK(ctx context.Context, uri string, api APIService) ([]byte, *models.Error) {
+	body, statusCode, status, err := get(ctx, uri, api)
+	if err != nil {
+		return nil, err
+	}
 
+	if statusCode == 200 {
+		return body, nil
+	}
+
+	if len(body) > 0 {
+		return nil, handleErrStatusCode(statusCode, body)
+	}
+
+	return nil, buildErrorResponse(fmt.Sprintf("Received unexpected response: %d %s", statusCode, status))
 }
 
-func putWithEventContext(uri string, data []byte, api APIService) (*models.EventContext, *models.Error) {
-	req, err := http.NewRequest("PUT", uri, bytes.NewBuffer(data))
+func getAndExpectSuccess(ctx context.Context, uri string, api APIService) ([]byte, *models.Error) {
+	body, statusCode, status, err := get(ctx, uri, api)
+	if err != nil {
+		return nil, err
+	}
+
+	if statusCode >= 200 && statusCode < 300 {
+		return body, nil
+	}
+
+	if len(body) > 0 {
+		return nil, handleErrStatusCode(statusCode, body)
+	}
+
+	return nil, buildErrorResponse(fmt.Sprintf("Received unexpected response: %d %s", statusCode, status))
+}
+
+func get(ctx context.Context, uri string, api APIService) ([]byte, int, string, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "GET", uri, nil)
+	if err != nil {
+		return nil, 0, "", buildErrorResponse(err.Error())
+	}
+	req.Header.Set("Content-Type", "application/json")
+	addAuthHeader(req, api)
+
+	resp, err := api.getHTTPClient().Do(req)
+	if err != nil {
+		return nil, 0, "", buildErrorResponse(err.Error())
+	}
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, 0, "", buildErrorResponse(err.Error())
+	}
+
+	return body, resp.StatusCode, resp.Status, nil
+}
+
+func putWithEventContext(ctx context.Context, uri string, data []byte, api APIService) (*models.EventContext, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "PUT", uri, bytes.NewBuffer(data))
 	if err != nil {
 		return nil, buildErrorResponse(err.Error())
 	}
@@ -84,20 +140,21 @@ func putWithEventContext(uri string, data []byte, api APIService) (*models.Event
 	}
 
 	if resp.StatusCode >= 200 && resp.StatusCode <= 204 {
-		if len(body) > 0 {
-			eventContext := &models.EventContext{}
-			if err = eventContext.FromJSON(body); err != nil {
-				// failed to parse json
-				return nil, buildErrorResponse(err.Error() + "\n" + "-----DETAILS-----" + string(body))
-			}
-
-			if eventContext.KeptnContext != nil {
-				fmt.Println("ID of Keptn context: " + *eventContext.KeptnContext)
-			}
-			return eventContext, nil
+		if len(body) == 0 {
+			return nil, nil
+		}
+
+		eventContext := &models.EventContext{}
+
+		if err = eventContext.FromJSON(body); err != nil {
+			// failed to parse json
+			return nil, buildErrorResponse(err.Error() + "\n" + "-----DETAILS-----" + string(body))
 		}
 
-		return nil, nil
+		if eventContext.KeptnContext != nil {
+			fmt.Println("ID of Keptn context: " + *eventContext.KeptnContext)
+		}
+		return eventContext, nil
 	}
 
 	if len(body) > 0 {
@@ -107,8 +164,8 @@ func putWithEventContext(uri string, data []byte, api APIService) (*models.Event
 	return nil, buildErrorResponse(fmt.Sprintf("Received unexpected response: %d %s", resp.StatusCode, resp.Status))
 }
 
-func put(uri string, data []byte, api APIService) (string, *models.Error) {
-	req, err := http.NewRequest("PUT", uri, bytes.NewBuffer(data))
+func put(ctx context.Context, uri string, data []byte, api APIService) (string, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "PUT", uri, bytes.NewBuffer(data))
 	if err != nil {
 		return "", buildErrorResponse(err.Error())
 	}
@@ -127,11 +184,7 @@ func put(uri string, data []byte, api APIService) (string, *models.Error) {
 	}
 
 	if resp.StatusCode >= 200 && resp.StatusCode <= 204 {
-		if len(body) > 0 {
-			return string(body), nil
-		}
-
-		return "", nil
+		return string(body), nil
 	}
 
 	if len(body) > 0 {
@@ -141,8 +194,8 @@ func put(uri string, data []byte, api APIService) (string, *models.Error) {
 	return "", buildErrorResponse(fmt.Sprintf("Received unexpected response: %d %s", resp.StatusCode, resp.Status))
 }
 
-func postWithEventContext(uri string, data []byte, api APIService) (*models.EventContext, *models.Error) {
-	req, err := http.NewRequest("POST", uri, bytes.NewBuffer(data))
+func postWithEventContext(ctx context.Context, uri string, data []byte, api APIService) (*models.EventContext, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "POST", uri, bytes.NewBuffer(data))
 	if err != nil {
 		return nil, buildErrorResponse(err.Error())
 	}
@@ -161,20 +214,20 @@ func postWithEventContext(uri string, data []byte, api APIService) (*models.Even
 	}
 
 	if resp.StatusCode >= 200 && resp.StatusCode <= 204 {
-		if len(body) > 0 {
-			eventContext := &models.EventContext{}
-			if err = eventContext.FromJSON(body); err != nil {
-				// failed to parse json
-				return nil, buildErrorResponse(err.Error() + "\n" + "-----DETAILS-----" + string(body))
-			}
-
-			if eventContext.KeptnContext != nil {
-				fmt.Println("ID of Keptn context: " + *eventContext.KeptnContext)
-			}
-			return eventContext, nil
+		if len(body) == 0 {
+			return nil, nil
+		}
+
+		eventContext := &models.EventContext{}
+		if err = eventContext.FromJSON(body); err != nil {
+			// failed to parse json
+			return nil, buildErrorResponse(err.Error() + "\n" + "-----DETAILS-----" + string(body))
 		}
 
-		return nil, nil
+		if eventContext.KeptnContext != nil {
+			fmt.Println("ID of Keptn context: " + *eventContext.KeptnContext)
+		}
+		return eventContext, nil
 	}
 
 	if len(body) > 0 {
@@ -184,8 +237,8 @@ func postWithEventContext(uri string, data []byte, api APIService) (*models.Even
 	return nil, buildErrorResponse(fmt.Sprintf("Received unexpected response: %d %s", resp.StatusCode, resp.Status))
 }
 
-func post(uri string, data []byte, api APIService) (string, *models.Error) {
-	req, err := http.NewRequest("POST", uri, bytes.NewBuffer(data))
+func post(ctx context.Context, uri string, data []byte, api APIService) (string, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "POST", uri, bytes.NewBuffer(data))
 	if err != nil {
 		return "", buildErrorResponse(err.Error())
 	}
@@ -204,11 +257,7 @@ func post(uri string, data []byte, api APIService) (string, *models.Error) {
 	}
 
 	if resp.StatusCode >= 200 && resp.StatusCode <= 204 {
-		if len(body) > 0 {
-			return string(body), nil
-		}
-
-		return "", nil
+		return string(body), nil
 	}
 
 	if len(body) > 0 {
@@ -218,8 +267,8 @@ func post(uri string, data []byte, api APIService) (string, *models.Error) {
 	return "", buildErrorResponse(fmt.Sprintf("Received unexpected response: %d %s", resp.StatusCode, resp.Status))
 }
 
-func deleteWithEventContext(uri string, api APIService) (*models.EventContext, *models.Error) {
-	req, err := http.NewRequest("DELETE", uri, nil)
+func deleteWithEventContext(ctx context.Context, uri string, api APIService) (*models.EventContext, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "DELETE", uri, nil)
 	if err != nil {
 		return nil, buildErrorResponse(err.Error())
 	}
@@ -238,23 +287,23 @@ func deleteWithEventContext(uri string, api APIService) (*models.EventContext, *
 	}
 
 	if resp.StatusCode >= 200 && resp.StatusCode < 300 {
-		if len(body) > 0 {
-			eventContext := &models.EventContext{}
-			if err = eventContext.FromJSON(body); err != nil {
-				// failed to parse json
-				return nil, buildErrorResponse(err.Error() + "\n" + "-----DETAILS-----" + string(body))
-			}
-			return eventContext, nil
+		if len(body) == 0 {
+			return nil, nil
 		}
 
-		return nil, nil
+		eventContext := &models.EventContext{}
+		if err = eventContext.FromJSON(body); err != nil {
+			// failed to parse json
+			return nil, buildErrorResponse(err.Error() + "\n" + "-----DETAILS-----" + string(body))
+		}
+		return eventContext, nil
 	}
 
 	return nil, handleErrStatusCode(resp.StatusCode, body)
 }
 
-func delete(uri string, api APIService) (string, *models.Error) {
-	req, err := http.NewRequest("DELETE", uri, nil)
+func delete(ctx context.Context, uri string, api APIService) (string, *models.Error) {
+	req, err := http.NewRequestWithContext(ctx, "DELETE", uri, nil)
 	if err != nil {
 		return "", buildErrorResponse(err.Error())
 	}
@@ -273,11 +322,7 @@ func delete(uri string, api APIService) (string, *models.Error) {
 	}
 
 	if resp.StatusCode >= 200 && resp.StatusCode < 300 {
-		if len(body) > 0 {
-			return string(body), nil
-		}
-
-		return "", nil
+		return string(body), nil
 	}
 
 	return "", handleErrStatusCode(resp.StatusCode, body)