From e37d87033e4ec8c7754003a1149d2e2fc50921e7 Mon Sep 17 00:00:00 2001
From: Stuart McLaren <stuart.mclaren@hpe.com>
Date: Wed, 21 Aug 2024 16:38:20 +0100
Subject: [PATCH] Add function to poll async operations

This will be called, for example, when we create a new
resource. It will wait until the resource has been
created and then return the URI of the newly created resource.
---
 internal/poll/operation.go | 105 +++++++++++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 internal/poll/operation.go

diff --git a/internal/poll/operation.go b/internal/poll/operation.go
new file mode 100644
index 0000000..61a93a5
--- /dev/null
+++ b/internal/poll/operation.go
@@ -0,0 +1,105 @@
+package poll
+
+import (
+	"context"
+	"time"
+
+	"github.com/HewlettPackard/hpegl-pcbe-terraform-resources/internal/client"
+	"github.com/HewlettPackard/hpegl-pcbe-terraform-resources/internal/sdk/async/dataservices"
+	"github.com/hashicorp/terraform-plugin-framework/diag"
+)
+
+// AsyncOperation polls an asynchronous operation until it completes.
+// It returns the URI of the source resource of the operation.
+func AsyncOperation(
+	ctx context.Context,
+	client client.PCBeClient,
+	OperationID string,
+	diagsP *diag.Diagnostics,
+) *string {
+	maxPolls := int(client.Config.MaxPolls)
+	pollWaitTime := time.Duration(client.Config.PollInterval) * time.Second
+
+	asyncClient, err := client.NewAsyncClient(ctx)
+	if err != nil {
+		(*diagsP).AddError(
+			"error polling async operation "+OperationID,
+			err.Error(),
+		)
+
+		return nil
+	}
+
+	var opResp dataservices.V1beta1AsyncOperationsItemAsyncOperationsGetResponseable
+	arc := dataservices.
+		V1beta1AsyncOperationsAsyncOperationsItemRequestBuilderGetRequestConfiguration{}
+
+	for count := 1; ; count++ {
+
+		opResp, err = asyncClient.DataServices().
+			V1beta1().
+			AsyncOperations().
+			ById(OperationID).
+			Get(ctx, &arc)
+		if err != nil {
+			(*diagsP).AddError(
+				"error polling async operation "+OperationID,
+				err.Error(),
+			)
+
+			return nil
+		}
+
+		if opResp == nil {
+			msg := "nil op response"
+			(*diagsP).AddError(
+				"error polling async operation "+OperationID,
+				msg,
+			)
+
+			return nil
+		}
+
+		opRespState := opResp.GetState()
+		if opRespState == nil {
+			msg := "operation has nil state"
+			(*diagsP).AddError(
+				"error polling async operation "+OperationID,
+				msg,
+			)
+
+			return nil
+		}
+
+		// TODO: (API) Use enum not string for state when FF-28181 is fixed
+		if *opRespState == "FAILED" {
+			msg := "operation state FAILED"
+
+			(*diagsP).AddError(
+				"error polling async operation "+OperationID,
+				msg,
+			)
+
+			return nil
+		}
+
+		// TODO: (API) Use enum not string for state when FF-28181 is fixed
+		if *opRespState == "SUCCEEDED" {
+			break
+		}
+
+		if count == maxPolls {
+			msg := "max polls exceeded"
+			(*diagsP).AddError(
+				"error polling async operation "+OperationID,
+				msg,
+			)
+
+			return nil
+		}
+
+		time.Sleep(pollWaitTime)
+	}
+
+	return opResp.GetSourceResourceUri()
+}