Compare commits
3 commits
0be4944a74
...
071e2958f0
| Author | SHA1 | Date | |
|---|---|---|---|
| 071e2958f0 | |||
| 3c72ea6545 | |||
| 964774ca6b |
2 changed files with 61 additions and 4 deletions
25
app.py
25
app.py
|
|
@ -11,7 +11,7 @@ from fastapi.responses import JSONResponse, RedirectResponse
|
||||||
from contextlib import asynccontextmanager
|
from contextlib import asynccontextmanager
|
||||||
|
|
||||||
log = logging.getLogger("butler")
|
log = logging.getLogger("butler")
|
||||||
VERSION = "2.3.3"
|
VERSION = "2.3.4"
|
||||||
|
|
||||||
API_DIR = os.environ.get("API_KEY_DIR", "/data/api")
|
API_DIR = os.environ.get("API_KEY_DIR", "/data/api")
|
||||||
VAULT_CACHE_DIR = os.environ.get("VAULT_CACHE_DIR", "/data/vault-cache")
|
VAULT_CACHE_DIR = os.environ.get("VAULT_CACHE_DIR", "/data/vault-cache")
|
||||||
|
|
@ -1567,10 +1567,29 @@ print("updated" if updated else "added")'''
|
||||||
async def ansible_run(request: Request, _=Depends(_verify)):
|
async def ansible_run(request: Request, _=Depends(_verify)):
|
||||||
body = await request.json()
|
body = await request.json()
|
||||||
hostname = body.get("limit", body.get("hostname", ""))
|
hostname = body.get("limit", body.get("hostname", ""))
|
||||||
template_id = body.get("template_id", 10)
|
|
||||||
if not hostname:
|
if not hostname:
|
||||||
return JSONResponse({"error": "limit/hostname required"}, status_code=400)
|
return JSONResponse({"error": "limit/hostname required"}, status_code=400)
|
||||||
rc, out, err = _ssh(AUTOMATION1, f"cd /app-config/ansible && bash pfannkuchen.sh setup {hostname}", timeout=600)
|
action = body.get("action", "setup")
|
||||||
|
if action not in {"setup", "tune", "pvetune"}:
|
||||||
|
return JSONResponse({"error": "action must be setup, tune or pvetune"}, status_code=400)
|
||||||
|
if not re.fullmatch(r"[a-zA-Z0-9_.:-]+", hostname):
|
||||||
|
return JSONResponse({"error": "invalid hostname/limit"}, status_code=400)
|
||||||
|
command = (
|
||||||
|
"cd /app-config/ansible && "
|
||||||
|
"git pull --ff-only origin master && "
|
||||||
|
f"bash pfannkuchen.sh {action} {hostname}"
|
||||||
|
)
|
||||||
|
rc, out, err = _ssh(AUTOMATION1, command, timeout=600)
|
||||||
|
_audit("/ansible/run", "POST", 200 if rc == 0 else 502, f"{action} {hostname}")
|
||||||
|
if action != "setup":
|
||||||
|
return {
|
||||||
|
"status": "ok" if rc == 0 else "error",
|
||||||
|
"action": action,
|
||||||
|
"hostname": hostname,
|
||||||
|
"rc": rc,
|
||||||
|
"output": out[-4000:],
|
||||||
|
"error": err[-1000:] if rc != 0 else "",
|
||||||
|
}
|
||||||
|
|
||||||
# After successful ansible run: sync Hawser token to Dockhand
|
# After successful ansible run: sync Hawser token to Dockhand
|
||||||
if rc == 0:
|
if rc == 0:
|
||||||
|
|
|
||||||
|
|
@ -42,7 +42,7 @@ def test_health_exposes_current_version():
|
||||||
with TestClient(app.app) as client:
|
with TestClient(app.app) as client:
|
||||||
response = client.get("/health")
|
response = client.get("/health")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert response.json()["version"] == app.VERSION == "2.3.3"
|
assert response.json()["version"] == app.VERSION == "2.3.4"
|
||||||
|
|
||||||
|
|
||||||
def test_sysctl_audit_reads_fixed_keys_from_inventory_host(monkeypatch):
|
def test_sysctl_audit_reads_fixed_keys_from_inventory_host(monkeypatch):
|
||||||
|
|
@ -108,6 +108,44 @@ def test_inventory_upsert_uses_base64_script(monkeypatch):
|
||||||
assert "\\nname =" not in calls[0][1]
|
assert "\\nname =" not in calls[0][1]
|
||||||
|
|
||||||
|
|
||||||
|
def test_ansible_run_supports_safe_tune_action_and_syncs_git(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": "emby-sascha", "action": "tune"},
|
||||||
|
)
|
||||||
|
assert response.status_code == 200
|
||||||
|
assert response.json()["action"] == "tune"
|
||||||
|
assert "git pull --ff-only origin master" in calls[0][1]
|
||||||
|
assert "bash pfannkuchen.sh tune emby-sascha" in calls[0][1]
|
||||||
|
assert len(calls) == 1
|
||||||
|
|
||||||
|
|
||||||
|
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")))
|
||||||
|
with TestClient(app.app) as client:
|
||||||
|
bad_action = client.post(
|
||||||
|
"/ansible/run",
|
||||||
|
headers={"Authorization": "Bearer test-token"},
|
||||||
|
json={"hostname": "emby-sascha", "action": "shell"},
|
||||||
|
)
|
||||||
|
bad_host = client.post(
|
||||||
|
"/ansible/run",
|
||||||
|
headers={"Authorization": "Bearer test-token"},
|
||||||
|
json={"hostname": "emby-sascha;id", "action": "tune"},
|
||||||
|
)
|
||||||
|
assert bad_action.status_code == 400
|
||||||
|
assert bad_host.status_code == 400
|
||||||
|
|
||||||
|
|
||||||
def test_docker_inspect_returns_sanitized_summary(monkeypatch):
|
def test_docker_inspect_returns_sanitized_summary(monkeypatch):
|
||||||
raw = [{
|
raw = [{
|
||||||
"Name": "/fileflows",
|
"Name": "/fileflows",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue