feat: add ip module for automated free IP discovery in agent pool

infra_cli/dns.py

@@ -55,4 +55,25 @@ class DNSManager:
     def list(self):
         hosts = self.exec_lxc(f"cat {self.hosts_file}").stdout
         dns = self.exec_lxc(f"cat {self.dns_file}").stdout
-        return {"hosts": hosts, "dns": dns}
+        return {"hosts": hosts, "dns": dns}
+
+    def get_free_ips(self, range_start=1, range_end=254, subnet="10.32.70"):
+        """Finds free IPs in the specified range by checking both static and dynamic leases"""
+        # 1. Get all static IPs from dhcp-hosts.conf and dynamic-hosts.conf
+        static_configs = self.exec_lxc(f"cat /etc/dnsmasq.d/dhcp-hosts.conf {self.hosts_file} 2>/dev/null").stdout
+        import re
+        used_ips = set(re.findall(r'[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}', static_configs))
+        
+        # 2. Get all active dynamic leases
+        leases = self.exec_lxc("cat /var/lib/misc/dnsmasq.leases 2>/dev/null").stdout
+        used_ips.update(set(re.findall(r'[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}', leases)))
+
+        # 3. Find first available in the agent range
+        free_ips = []
+        for i in range(range_start, range_end + 1):
+            candidate = f"{subnet}.{i}"
+            if candidate not in used_ips:
+                free_ips.append(candidate)
+            if len(free_ips) >= 10: # Just return the top 10
+                break
+        return free_ips

infra_cli/main.py

@@ -162,6 +162,31 @@ def dns_list(config):
     click.echo(data['hosts'])
     click.echo(data['dns'])
 
+@cli.group()
+def ip():
+    """Manage and discover available IP addresses"""
+    pass
+
+@ip.command(name='list-free')
+@click.option('--count', default=5, help='Number of free IPs to show')
+@click.pass_obj
+def ip_list_free(config, count):
+    mgr = DNSManager(config)
+    free = mgr.get_free_ips()
+    for ip in free[:count]:
+        click.echo(ip)
+
+@ip.command(name='next-free')
+@click.pass_obj
+def ip_next_free(config):
+    mgr = DNSManager(config)
+    free = mgr.get_free_ips()
+    if free:
+        click.echo(free[0])
+    else:
+        click.echo("Error: No free IPs in agent pool!", err=True)
+        sys.exit(1)
+
 @cli.group()
 def ingress():
     """Manage HAProxy Ingress"""