package client

import (
	"context"

	"go.unistack.org/micro/v4/errors"
)

// RetryFunc that returning either false or a non-nil error will result in the call not being retried
type RetryFunc func(ctx context.Context, req Request, retryCount int, err error) (bool, error)

// RetryAlways always retry on error
func RetryAlways(ctx context.Context, req Request, retryCount int, err error) (bool, error) {
	return true, nil
}

// RetryNever never retry on error
func RetryNever(ctx context.Context, req Request, retryCount int, err error) (bool, error) {
	return false, nil
}

// RetryOnError retries a request on a 500 or 408 (timeout) error
func RetryOnError(_ context.Context, _ Request, _ int, err error) (bool, error) {
	if err == nil {
		return false, nil
	}
	me := errors.FromError(err)
	switch me.Code {
	// retry on timeout or internal server error
	case 408, 500:
		return true, nil
	}
	return false, nil
}

// RetryOnErrors retries a request on specified error codes
func RetryOnErrors(codes ...int32) RetryFunc {
	return func(_ context.Context, _ Request, _ int, err error) (bool, error) {
		if err == nil {
			return false, nil
		}
		me := errors.FromError(err)
		for _, code := range codes {
			if me.Code == code {
				return true, nil
			}
		}
		return false, nil
	}
}