's'
This commit is contained in:
58
scripts/ssh-setup-ntfy-acl.py
Normal file
58
scripts/ssh-setup-ntfy-acl.py
Normal file
@@ -0,0 +1,58 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Configure ntfy ACL: public read on nomadcna-* topics, backend publish token."""
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import re
|
||||
import secrets
|
||||
import sys
|
||||
|
||||
import paramiko
|
||||
|
||||
HOST = os.getenv("NOMAD_SSH_HOST", "82.157.112.245")
|
||||
USER = os.getenv("NOMAD_SSH_USER", "ubuntu")
|
||||
PASSWORD = os.getenv("NOMAD_SSH_PASSWORD", "Xiao4669805")
|
||||
|
||||
|
||||
def run(c: paramiko.SSHClient, cmd: str, timeout: int = 120) -> str:
|
||||
print(f"\n$ {cmd.replace(PASSWORD, '***')}")
|
||||
_, o, e = c.exec_command(f"echo '{PASSWORD}' | sudo -S bash -lc \"{cmd}\"", timeout=timeout, get_pty=True)
|
||||
out = o.read().decode("utf-8", errors="replace")
|
||||
err = e.read().decode("utf-8", errors="replace")
|
||||
sys.stdout.buffer.write(out.encode("utf-8", errors="replace"))
|
||||
sys.stdout.buffer.write(err.encode("utf-8", errors="replace"))
|
||||
return out + err
|
||||
|
||||
|
||||
def main() -> None:
|
||||
api_pass = secrets.token_urlsafe(16)
|
||||
c = paramiko.SSHClient()
|
||||
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||||
c.connect(HOST, username=USER, password=PASSWORD, timeout=30)
|
||||
try:
|
||||
run(c, "docker exec nomadro-ntfy ntfy user add --role=admin nomadcna-api --pass='{}' 2>/dev/null || true".format(api_pass))
|
||||
run(c, "docker exec nomadro-ntfy ntfy access nomadcna-* read everyone")
|
||||
run(c, "docker exec nomadro-ntfy ntfy access nomadcna-* write nomadcna-api")
|
||||
token_out = run(c, "docker exec nomadro-ntfy ntfy token add --user=nomadcna-api nomadcna-backend 2>&1")
|
||||
token_match = re.search(r"tk_[a-zA-Z0-9_-]+", token_out)
|
||||
token = token_match.group(0) if token_match else ""
|
||||
|
||||
if token:
|
||||
run(c, f"grep -q NTFY_AUTH_TOKEN /opt/nomadro-services/.env || echo 'NTFY_AUTH_TOKEN={token}' >> /opt/nomadro-services/.env")
|
||||
run(c, f"grep -q NTFY_AUTH_TOKEN /home/ubuntu/gitlab-instance-0a899031_cnomadcna/.env.local 2>/dev/null || echo 'NTFY_AUTH_TOKEN={token}' >> /home/ubuntu/gitlab-instance-0a899031_cnomadcna/.env.local")
|
||||
|
||||
test = run(
|
||||
c,
|
||||
f"curl -s -X POST https://ntfy.nomadro.cn/nomadcna-healthcheck "
|
||||
f"-H 'Authorization: Bearer {token}' -H 'Title: NomadCNA' -d 'push ok'",
|
||||
)
|
||||
print("\n=== Token (save to FastAPI env) ===")
|
||||
print(token or "(check token add output above)")
|
||||
print("=== Test response ===")
|
||||
print(test.strip()[:200])
|
||||
finally:
|
||||
c.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user