Compare commits

...

3 commits

2 changed files with 45 additions and 11 deletions

31
app.py
View file

@ -1571,25 +1571,34 @@ async def ansible_run(request: Request, _=Depends(_verify)):
if not hostname: if not hostname:
return JSONResponse({"error": "limit/hostname required"}, status_code=400) return JSONResponse({"error": "limit/hostname required"}, status_code=400)
action = body.get("action", "setup") action = body.get("action", "setup")
if action not in {"setup", "tune", "pvetune"}: if action not in {"setup", "tune", "pvetune", "nfs"}:
return JSONResponse({"error": "action must be setup, tune or pvetune"}, status_code=400) return JSONResponse({"error": "action must be setup, tune, pvetune or nfs"}, status_code=400)
if not re.fullmatch(r"[a-zA-Z0-9_.:-]+", hostname): if not re.fullmatch(r"[a-zA-Z0-9_.:-]+", hostname):
return JSONResponse({"error": "invalid hostname/limit"}, status_code=400) return JSONResponse({"error": "invalid hostname/limit"}, status_code=400)
if action in {"tune", "pvetune"}: if action in {"tune", "pvetune", "nfs"}:
approved_files = ( if action == "nfs":
"roles/sysctl/defaults/main.yml", approved_files = (
"roles/sysctl/tasks/main.yml", "roles/nfs_stability/tasks/main.yml",
"group_vars/vps/sysctl.yml", "nfs-stability.yml",
"sysctl-proxmox.yaml", "pfannkuchen.sh",
"roles/sysctl_proxmox/tasks/main.yml", )
) prepare = "mkdir -p roles/nfs_stability/tasks && "
else:
approved_files = (
"roles/sysctl/defaults/main.yml",
"roles/sysctl/tasks/main.yml",
"group_vars/vps/sysctl.yml",
"sysctl-proxmox.yaml",
"roles/sysctl_proxmox/tasks/main.yml",
)
prepare = ""
file_sync = " && ".join( file_sync = " && ".join(
f"git show origin/master:{path} > {path}" for path in approved_files f"git show origin/master:{path} > {path}" for path in approved_files
) )
command = ( command = (
"cd /app-config/ansible && " "cd /app-config/ansible && "
"git fetch origin master && " "git fetch origin master && "
f"{file_sync} && " f"{prepare}{file_sync} && "
f"bash pfannkuchen.sh {action} {hostname}" f"bash pfannkuchen.sh {action} {hostname}"
) )
else: else:

View file

@ -235,6 +235,31 @@ def test_ansible_run_supports_safe_tune_action_and_syncs_approved_files(monkeypa
assert len(calls) == 1 assert len(calls) == 1
def test_ansible_run_supports_scoped_nfs_action(monkeypatch):
calls = []
def fake_ssh(host, command, timeout=600):
calls.append((host, command, timeout))
return 0, "changed=1 failed=0", ""
monkeypatch.setattr(app, "_ssh", fake_ssh)
with TestClient(app.app) as client:
response = client.post(
"/ansible/run",
headers={"Authorization": "Bearer test-token"},
json={"hostname": "arrapps", "action": "nfs"},
)
assert response.status_code == 200
assert response.json()["action"] == "nfs"
command = calls[0][1]
assert "mkdir -p roles/nfs_stability/tasks" in command
assert "git show origin/master:roles/nfs_stability/tasks/main.yml" in command
assert "git show origin/master:nfs-stability.yml" in command
assert "bash pfannkuchen.sh nfs arrapps" in command
assert "git pull --ff-only" not in command
assert len(calls) == 1
def test_ansible_run_rejects_unknown_action_and_shell_metacharacters(monkeypatch): def test_ansible_run_rejects_unknown_action_and_shell_metacharacters(monkeypatch):
monkeypatch.setattr(app, "_ssh", lambda *_args, **_kwargs: (_ for _ in ()).throw(AssertionError("must not SSH"))) monkeypatch.setattr(app, "_ssh", lambda *_args, **_kwargs: (_ for _ in ()).throw(AssertionError("must not SSH")))
with TestClient(app.app) as client: with TestClient(app.app) as client: