Compare commits
3 commits
6092c432fe
...
6761089be9
| Author | SHA1 | Date | |
|---|---|---|---|
| 6761089be9 | |||
| ad863eac15 | |||
| fde0b2d469 |
2 changed files with 45 additions and 11 deletions
17
app.py
17
app.py
|
|
@ -1571,11 +1571,19 @@ 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"}:
|
||||||
|
if action == "nfs":
|
||||||
|
approved_files = (
|
||||||
|
"roles/nfs_stability/tasks/main.yml",
|
||||||
|
"nfs-stability.yml",
|
||||||
|
"pfannkuchen.sh",
|
||||||
|
)
|
||||||
|
prepare = "mkdir -p roles/nfs_stability/tasks && "
|
||||||
|
else:
|
||||||
approved_files = (
|
approved_files = (
|
||||||
"roles/sysctl/defaults/main.yml",
|
"roles/sysctl/defaults/main.yml",
|
||||||
"roles/sysctl/tasks/main.yml",
|
"roles/sysctl/tasks/main.yml",
|
||||||
|
|
@ -1583,13 +1591,14 @@ async def ansible_run(request: Request, _=Depends(_verify)):
|
||||||
"sysctl-proxmox.yaml",
|
"sysctl-proxmox.yaml",
|
||||||
"roles/sysctl_proxmox/tasks/main.yml",
|
"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:
|
||||||
|
|
|
||||||
|
|
@ -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:
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue