rft-cli 0.5.2

Zero-config Docker Compose isolation for git worktrees
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use std::net::TcpListener;

use crate::ports::PortAllocation;

#[derive(Debug, PartialEq)]
pub struct PortConflict {
    pub port: u16,
    pub service_name: String,
    pub env_var: String,
}

/// Check whether a single port is available for binding.
pub fn check_port_available(port: u16) -> std::io::Result<bool> {
    match TcpListener::bind(("127.0.0.1", port)) {
        Ok(_listener) => Ok(true),
        Err(error) if error.kind() == std::io::ErrorKind::AddrInUse => Ok(false),
        Err(error) => Err(error),
    }
}

/// Check all allocated ports and return conflicts for those already in use.
pub fn check_ports(allocations: &[PortAllocation]) -> Vec<PortConflict> {
    allocations
        .iter()
        .filter_map(|allocation| match check_port_available(allocation.port) {
            Ok(true) => None,
            _ => Some(PortConflict {
                port: allocation.port,
                service_name: allocation.service_name.clone(),
                env_var: allocation.env_var.clone(),
            }),
        })
        .collect()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn available_port_returns_true() {
        // Port 0 lets OS pick a free port; use it to find one that's free
        let listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let free_port = listener.local_addr().unwrap().port();
        drop(listener);

        assert!(check_port_available(free_port).unwrap());
    }

    #[test]
    fn occupied_port_returns_false() {
        let listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let occupied_port = listener.local_addr().unwrap().port();
        // Keep listener alive so port stays occupied
        assert!(!check_port_available(occupied_port).unwrap());
        drop(listener);
    }

    #[test]
    fn check_ports_detects_conflict() {
        let listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let occupied_port = listener.local_addr().unwrap().port();

        let allocations = vec![PortAllocation {
            service_name: "web".to_string(),
            env_var: "WEB_PORT".to_string(),
            port: occupied_port,
            container_port: 3000,
        }];

        let conflicts = check_ports(&allocations);
        assert_eq!(conflicts.len(), 1);
        assert_eq!(conflicts[0].port, occupied_port);
        assert_eq!(conflicts[0].service_name, "web");
        assert_eq!(conflicts[0].env_var, "WEB_PORT");

        drop(listener);
    }

    #[test]
    fn check_ports_returns_empty_when_all_available() {
        let listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let free_port = listener.local_addr().unwrap().port();
        drop(listener);

        let allocations = vec![PortAllocation {
            service_name: "api".to_string(),
            env_var: "API_PORT".to_string(),
            port: free_port,
            container_port: 8080,
        }];

        let conflicts = check_ports(&allocations);
        assert!(conflicts.is_empty());
    }

    #[test]
    fn check_ports_mixed_available_and_occupied() {
        let listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let occupied_port = listener.local_addr().unwrap().port();

        let free_listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let free_port = free_listener.local_addr().unwrap().port();
        drop(free_listener);

        let allocations = vec![
            PortAllocation {
                service_name: "web".to_string(),
                env_var: "WEB_PORT".to_string(),
                port: occupied_port,
                container_port: 3000,
            },
            PortAllocation {
                service_name: "api".to_string(),
                env_var: "API_PORT".to_string(),
                port: free_port,
                container_port: 8080,
            },
        ];

        let conflicts = check_ports(&allocations);
        assert_eq!(conflicts.len(), 1);
        assert_eq!(conflicts[0].service_name, "web");

        drop(listener);
    }
}