stakpak 0.3.68

Stakpak: Your DevOps AI Agent. Generate infrastructure code, debug Kubernetes, configure CI/CD, automate deployments, without giving an LLM the keys to production.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use crate::utils::local_context::detect_container_environment;
use rand::Rng;
use tokio::net::TcpListener;

/// Reserved port range for ephemeral/dynamic ports (IANA recommendation)
const MIN_EPHEMERAL_PORT: u16 = 49152;
const MAX_EPHEMERAL_PORT: u16 = 65535;
const MAX_ATTEMPTS: u32 = 100;

/// Finds an available port using random selection to minimize collisions.
/// Returns a bound TcpListener that the caller must use immediately to avoid races.
async fn find_available_port_with_listener(host: &str) -> Result<(TcpListener, u16), String> {
    for _attempt in 0..MAX_ATTEMPTS {
        // Use random port selection to minimize collision probability
        let port = {
            let mut rng = rand::rng();
            rng.random_range(MIN_EPHEMERAL_PORT..=MAX_EPHEMERAL_PORT)
        }; // rng is dropped here before the await
        let addr = format!("{}:{}", host, port);

        match TcpListener::bind(&addr).await {
            Ok(listener) => {
                return Ok((listener, port));
            }
            Err(_) => {
                // Port is in use, try another random port
                continue;
            }
        }
    }

    Err(format!(
        "Failed to find available port after {} attempts in range {}-{}",
        MAX_ATTEMPTS, MIN_EPHEMERAL_PORT, MAX_EPHEMERAL_PORT
    ))
}

/// Returns a bind address string and a bound TcpListener to prevent race conditions.
/// The caller must use the listener immediately to start their server.
pub async fn find_available_bind_address_with_listener() -> Result<(String, TcpListener), String> {
    let host = match detect_container_environment() {
        true => "0.0.0.0",
        false => "127.0.0.1", // Force IPv4 to avoid IPv6 TLS issues
    };

    let (listener, port) = find_available_port_with_listener(host).await?;
    let bind_address = format!("{}:{}", host, port);

    Ok((bind_address, listener))
}