cli/pkg/common/timing.go

// Copyright (C) 2020-2022-2025, Lux Industries Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package common

import (
	"fmt"
	"math/rand"
	"os"
	"time"

	"github.com/luxfi/constants"
)

// TimedFunction executes a function and returns its result with error
func TimedFunction(f func() (any, error), actionMsg string) (any, error) {
	fmt.Printf("  %s...", actionMsg)
	start := time.Now()
	result, err := f()
	elapsed := time.Since(start)
	if err != nil {
		fmt.Printf(" failed (%v)\n", elapsed)
		return nil, err
	}
	fmt.Printf(" done (%v)\n", elapsed)
	return result, nil
}

// TimedFunctionWithRetry executes a function with retry logic
func TimedFunctionWithRetry[T any](f func() (T, error), actionMsg string, numRetries int, sleepBetweenRetries time.Duration) (T, error) {
	var result T
	var err error
	for i := 0; i <= numRetries; i++ {
		result, err = f()
		if err == nil {
			return result, nil
		}
		if i < numRetries {
			time.Sleep(sleepBetweenRetries)
		}
	}
	return result, err
}

// RandomString generates a random string of the given length
func RandomString(length int) string {
	const charset = "abcdefghijklmnopqrstuvwxyz0123456789"
	b := make([]byte, length)
	for i := range b {
		b[i] = charset[rand.Intn(len(charset))] //nolint:gosec // G404: Non-security random for temp identifiers
	}
	return string(b)
}

// IsE2E checks if running in E2E test mode
func IsE2E() bool {
	return os.Getenv("RUN_E2E") != ""
}

// E2EConvertIP converts IP for E2E testing
func E2EConvertIP(ip string) string {
	if os.Getenv("RUN_E2E") != "" {
		return constants.E2EDockerLoopbackHost
	}
	return ip
}