From 4159e0afa0250de27672a94a1d6e8d04d1c99884 Mon Sep 17 00:00:00 2001
From: Gabin L <gabin.lanore@gmail.com>
Date: Tue, 14 Jan 2025 19:04:07 +0100
Subject: [PATCH 1/2] feat: add acls mngt

Signed-off-by: Gabin L <gabin.lanore@gmail.com>
---
 labctl/commands/admin/vpn.py | 60 ++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/labctl/commands/admin/vpn.py b/labctl/commands/admin/vpn.py
index 67ec480..05e87e1 100644
--- a/labctl/commands/admin/vpn.py
+++ b/labctl/commands/admin/vpn.py
@@ -7,8 +7,10 @@
 app = typer.Typer()
 
 app_group = typer.Typer()
+app_acl = typer.Typer()
 
 app.add_typer(app_group, name="group")
+app.add_typer(app_acl, name="acl")
 
 @app_group.command(name="add-user")
 def add_user(username: str, group: str):
@@ -33,3 +35,61 @@ def del_user(username: str, group: str):
         console.print(f"[red]Error: {rsp.text}[/red]")
         return
     console.print(f"User {username} deleted from group {group}")
+
+@app_acl.command(name="list")
+def list_acls():
+    """
+    List VPN ACLs
+    """
+    api_driver = APIDriver()
+    rsp = api_driver.get("/headscale/acls/")
+    if rsp.status_code >= 400:
+        console.print(f"[red]Error: {rsp.text}[/red]")
+        return
+    acls = rsp.json()
+    table = Table(title="VPN ACLs")
+    table.add_column("ID", style="cyan bold")
+    table.add_column("Action", style="green")
+    table.add_column("Source", style="magenta")
+    table.add_column("Destination", style="yellow")
+    table.add_column("Protocol", style="blue")
+
+    for acl in acls:
+        table.add_row(
+            str(acl.get("id")),
+            acl.get("action"),
+            "\n".join(acl.get("src", [])),
+            "\n".join(acl.get("dst", [])),
+            acl.get("proto")
+        )
+    console.print(table)
+
+@app_acl.command(name="add")
+def add_acl(action: str, src: str, dst: str, proto: str = None):
+    """
+    Add VPN ACL
+    """
+    api_driver = APIDriver()
+    data = {
+        "action": action,
+        "src": src.split(","),
+        "dst": dst.split(","),
+        "proto": proto
+    }
+    rsp = api_driver.post("/headscale/acls/", json=data)
+    if rsp.status_code >= 400:
+        console.print(f"[red]Error: {rsp.text}[/red]")
+        return
+    console.print("ACL added")
+
+@app_acl.command(name="del")
+def del_acl(acl_id: int):
+    """
+    Delete VPN ACL
+    """
+    api_driver = APIDriver()
+    rsp = api_driver.delete(f"/headscale/acls/{acl_id}")
+    if rsp.status_code >= 400:
+        console.print(f"[red]Error: {rsp.text}[/red]")
+        return
+    console.print("ACL deleted")

From 863cb7211cfc1fe460af7b056e343b2bbc5d9cbb Mon Sep 17 00:00:00 2001
From: Gabin L <gabin.lanore@gmail.com>
Date: Tue, 14 Jan 2025 19:22:36 +0100
Subject: [PATCH 2/2] feat: add host ip bind headscale mngt

Signed-off-by: Gabin L <gabin.lanore@gmail.com>
---
 labctl/commands/admin/vpn.py | 54 ++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/labctl/commands/admin/vpn.py b/labctl/commands/admin/vpn.py
index 05e87e1..83683b5 100644
--- a/labctl/commands/admin/vpn.py
+++ b/labctl/commands/admin/vpn.py
@@ -8,9 +8,11 @@
 
 app_group = typer.Typer()
 app_acl = typer.Typer()
+app_host = typer.Typer()
 
 app.add_typer(app_group, name="group")
 app.add_typer(app_acl, name="acl")
+app.add_typer(app_host, name="host")
 
 @app_group.command(name="add-user")
 def add_user(username: str, group: str):
@@ -93,3 +95,55 @@ def del_acl(acl_id: int):
         console.print(f"[red]Error: {rsp.text}[/red]")
         return
     console.print("ACL deleted")
+
+@app_host.command(name="list")
+def list_hosts():
+    """
+    List VPN hosts
+    """
+    api_driver = APIDriver()
+    rsp = api_driver.get("/headscale/host/")
+    if rsp.status_code >= 400:
+        console.print(f"[red]Error: {rsp.text}[/red]")
+        return
+    hosts = rsp.json()
+    table = Table(title="VPN Hosts IP -> Host binding")
+    table.add_column("ID", style="cyan bold")
+    table.add_column("Host", style="green")
+    table.add_column("IP", style="magenta")
+
+    for host in hosts:
+        table.add_row(
+            str(host.get("id")),
+            host.get("name"),
+            host.get("ip")
+        )
+    console.print(table)
+
+@app_host.command(name="add")
+def add_host(name: str, ip: str):
+    """
+    Add VPN host
+    """
+    api_driver = APIDriver()
+    data = {
+        "name": name,
+        "ip": ip
+    }
+    rsp = api_driver.post("/headscale/host/", json=data)
+    if rsp.status_code >= 400:
+        console.print(f"[red]Error: {rsp.text}[/red]")
+        return
+    console.print("Host added")
+
+@app_host.command(name="del")
+def del_host(host_id: int):
+    """
+    Delete VPN host
+    """
+    api_driver = APIDriver()
+    rsp = api_driver.delete(f"/headscale/host/{host_id}")
+    if rsp.status_code >= 400:
+        console.print(f"[red]Error: {rsp.text}[/red]")
+        return
+    console.print("Host deleted")
\ No newline at end of file