package wait

import (
	"context"
	"fmt"
	"time"

	"github.com/cenkalti/backoff/v5"
	"github.com/go-acme/lego/v4/log"
)

// For polls the given function 'f', once every 'interval', up to 'timeout'.
func For(msg string, timeout, interval time.Duration, f func() (bool, error)) error {
	log.Infof("Wait for %s [timeout: %s, interval: %s]", msg, timeout, interval)

	var lastErr error

	timeUp := time.After(timeout)

	for {
		select {
		case <-timeUp:
			if lastErr == nil {
				return fmt.Errorf("%s: time limit exceeded", msg)
			}

			return fmt.Errorf("%s: time limit exceeded: last error: %w", msg, lastErr)
		default:
		}

		stop, err := f()
		if stop {
			return err
		}

		if err != nil {
			lastErr = err
		}

		time.Sleep(interval)
	}
}

// Retry retries the given operation until it succeeds or the context is canceled.
// Similar to [backoff.Retry] but with a different signature.
func Retry(ctx context.Context, operation func() error, opts ...backoff.RetryOption) error {
	_, err := backoff.Retry(ctx, func() (any, error) {
		return nil, operation()
	}, opts...)

	return err
}