terraform-provider-stackitprivatepreview/pkg/postgresflexalpha/wait/wait.go

package wait

import (
	"context"
	"fmt"
	"time"

	"github.com/hashicorp/terraform-plugin-log/tflog"
	postgresflex "github.com/mhenselin/terraform-provider-stackitprivatepreview/pkg/postgresflexalpha"

	"github.com/stackitcloud/stackit-sdk-go/core/oapierror"
	"github.com/stackitcloud/stackit-sdk-go/core/wait"
)

// "READY" "PENDING" "PROGRESSING" "FAILURE" "UNKNOWN" "TERMINATING"
const (
	InstanceStateEmpty       = ""
	InstanceStateProgressing = "PROGRESSING"
	InstanceStateSuccess     = "READY"
	InstanceStateFailed      = "FAILURE"
	InstanceStateTerminating = "TERMINATING"
	InstanceStateUnknown     = "UNKNOWN"
	InstanceStatePending     = "PENDING"
)

// APIClientInstanceInterface Interface needed for tests
type APIClientInstanceInterface interface {
	GetInstanceRequestExecute(ctx context.Context, projectId, region, instanceId string) (
		*postgresflex.GetInstanceResponse,
		error,
	)

	ListUsersRequestExecute(
		ctx context.Context,
		projectId string,
		region string,
		instanceId string,
	) (*postgresflex.ListUserResponse, error)
}

// APIClientUserInterface Interface needed for tests
type APIClientUserInterface interface {
	GetUserRequestExecute(ctx context.Context, projectId, region, instanceId string, userId int64) (
		*postgresflex.GetUserResponse,
		error,
	)
}

// CreateInstanceWaitHandler will wait for instance creation
func CreateInstanceWaitHandler(
	ctx context.Context, a APIClientInstanceInterface, projectId, region,
	instanceId string,
) *wait.AsyncActionHandler[postgresflex.GetInstanceResponse] {
	instanceCreated := false
	var instanceGetResponse *postgresflex.GetInstanceResponse

	handler := wait.New(
		func() (waitFinished bool, response *postgresflex.GetInstanceResponse, err error) {
			if !instanceCreated {
				s, err := a.GetInstanceRequestExecute(ctx, projectId, region, instanceId)
				if err != nil {
					return false, nil, err
				}
				if s == nil || s.Id == nil || *s.Id != instanceId || s.Status == nil {
					return false, nil, nil
				}
				switch *s.Status {
				default:
					return true, s, fmt.Errorf("instance with id %s has unexpected status %s", instanceId, *s.Status)
				case InstanceStateEmpty:
					return false, nil, nil
				case InstanceStatePending:
					return false, nil, nil
				case InstanceStateUnknown:
					return false, nil, nil
				case InstanceStateProgressing:
					return false, nil, nil
				case InstanceStateSuccess:
					if s.Network == nil || s.Network.InstanceAddress == nil {
						tflog.Info(ctx, "Waiting for instance_address")
						return false, nil, nil
					}
					if s.Network.RouterAddress == nil {
						tflog.Info(ctx, "Waiting for router_address")
						return false, nil, nil
					}
					instanceCreated = true
					instanceGetResponse = s
				case InstanceStateFailed:
					tflog.Warn(ctx, fmt.Sprintf("Wait handler got status FAILURE for instance: %s", instanceId))
					return false, nil, nil
					// API responds with FAILURE for some seconds and then the instance goes to READY
					// return true, s, fmt.Errorf("create failed for instance with id %s", instanceId)
				}
			}

			tflog.Info(ctx, "Waiting for instance (calling list users")
			// 	// User operations aren't available right after an instance is deemed successful
			//			// To check if they are, perform a users request
			_, err = a.ListUsersRequestExecute(ctx, projectId, region, instanceId)
			if err == nil {
				return true, instanceGetResponse, nil
			}
			oapiErr, ok := err.(*oapierror.GenericOpenAPIError) // nolint:errorlint //complaining that error.As should be used to catch wrapped errors, but this error should not be wrapped
			if !ok {
				return false, nil, err
			}
			if oapiErr.StatusCode < 500 {
				return true, instanceGetResponse, fmt.Errorf(
					"users request after instance creation returned %d status code",
					oapiErr.StatusCode,
				)
			}
			return false, nil, nil
		},
	)
	// Sleep before wait is set because sometimes API returns 404 right after creation request
	handler.SetTimeout(90 * time.Minute).SetSleepBeforeWait(30 * time.Second)
	return handler
}

// PartialUpdateInstanceWaitHandler will wait for instance update
func PartialUpdateInstanceWaitHandler(
	ctx context.Context, a APIClientInstanceInterface, projectId, region,
	instanceId string,
) *wait.AsyncActionHandler[postgresflex.GetInstanceResponse] {
	handler := wait.New(
		func() (waitFinished bool, response *postgresflex.GetInstanceResponse, err error) {
			s, err := a.GetInstanceRequestExecute(ctx, projectId, region, instanceId)
			if err != nil {
				return false, nil, err
			}
			if s == nil || s.Id == nil || *s.Id != instanceId || s.Status == nil {
				return false, nil, nil
			}
			switch *s.Status {
			default:
				return true, s, fmt.Errorf("instance with id %s has unexpected status %s", instanceId, *s.Status)
			case InstanceStateEmpty:
				return false, nil, nil
			case InstanceStateUnknown:
				return false, nil, nil
			case InstanceStateProgressing:
				return false, nil, nil
			case InstanceStatePending:
				return false, nil, nil
			case InstanceStateTerminating:
				return false, nil, nil
			case InstanceStateSuccess:
				return true, s, nil
			case InstanceStateFailed:
				return true, s, fmt.Errorf("update failed for instance with id %s", instanceId)
			}
		},
	)
	handler.SetTimeout(45 * time.Minute)
	return handler
}