operator
103 lines
3.7 KiB
Python
Executable File
103 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
opp-dm.py: Operator Direct Messages (fixed)
|
|
|
|
Works from container OR from bl directly.
|
|
Detects environment and acts accordingly.
|
|
Verifies writes.
|
|
|
|
For operator-to-operator: 646, operator-main.
|
|
"""
|
|
import argparse
|
|
import subprocess
|
|
import sys
|
|
import os
|
|
import datetime
|
|
from pathlib import Path
|
|
|
|
# Detect if we're on bl (has /home/super/Projects/NetVM)
|
|
ON_BL = Path("/home/super/Projects/NetVM").exists()
|
|
BL_MSG_DIR = Path("/home/super/Projects/NetVM/opp-dms")
|
|
|
|
def bl_run_container(cmd):
|
|
"""Run on bl via SSH chain (from container)."""
|
|
full = f"ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o BatchMode=yes super@100.123.153.75 '{cmd}'"
|
|
vm_cmd = f"ssh -i ~/.ssh/vm_to_gcp -o ProxyCommand=\"$HOME/workspace/bin/ssh-via-proxy %h %p\" -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o BatchMode=yes super@34.139.37.135 \"{full}\""
|
|
result = subprocess.run(vm_cmd, shell=True, capture_output=True, text=True, timeout=30)
|
|
return result.stdout.strip(), result.returncode
|
|
|
|
def opp_send(to, message, from_who="operator-main"):
|
|
"""Send operator DM. Works from container or bl."""
|
|
if to not in ["646", "operator-main"]:
|
|
print(f"ERROR: Unknown operator {to}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
ts = datetime.datetime.now().isoformat()
|
|
safe = message.replace("'", "'\"'\"'")[:1000]
|
|
line = f"{ts} [{from_who}]: {safe}"
|
|
|
|
if ON_BL:
|
|
# Direct write (we're on bl)
|
|
BL_MSG_DIR.mkdir(parents=True, exist_ok=True)
|
|
logfile = BL_MSG_DIR / f"{to}.log"
|
|
with open(logfile, "a") as f:
|
|
f.write(line + "\n")
|
|
# Verify
|
|
if logfile.exists():
|
|
print(f"OPP-DM sent to {to} (verified)")
|
|
return True
|
|
else:
|
|
print(f"ERROR: Write failed", file=sys.stderr)
|
|
sys.exit(1)
|
|
else:
|
|
# Via SSH (from container)
|
|
cmd = f"mkdir -p {BL_MSG_DIR} && echo '{line}' >> {BL_MSG_DIR}/{to}.log && test -f {BL_MSG_DIR}/{to}.log && echo OK"
|
|
out, rc = bl_run_container(cmd)
|
|
if rc == 0 and "OK" in out:
|
|
print(f"OPP-DM sent to {to} (verified)")
|
|
return True
|
|
else:
|
|
print(f"ERROR: Send failed: {out}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
def opp_read(who="operator-main", n=5):
|
|
"""Read operator DMs. Works from container or bl."""
|
|
if who not in ["646", "operator-main"]:
|
|
print(f"ERROR: Unknown operator {who}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
if ON_BL:
|
|
logfile = BL_MSG_DIR / f"{who}.log"
|
|
if not logfile.exists():
|
|
print("no messages")
|
|
return
|
|
with open(logfile) as f:
|
|
lines = f.readlines()
|
|
for l in lines[-n:]:
|
|
print(l.strip())
|
|
else:
|
|
cmd = f"tail -n {n} {BL_MSG_DIR}/{who}.log 2>/dev/null || echo 'no messages'"
|
|
out, rc = bl_run_container(cmd)
|
|
print(out)
|
|
|
|
def main():
|
|
p = argparse.ArgumentParser(description="OPP-DM: Operator DMs (fixed)")
|
|
sub = p.add_subparsers(dest='cmd', required=True)
|
|
|
|
ps = sub.add_parser('send', help='Send operator DM')
|
|
ps.add_argument('--to', required=True, choices=['646', 'operator-main'])
|
|
ps.add_argument('--from', dest='from_who', default='operator-main')
|
|
ps.add_argument('message', help='Message')
|
|
ps.set_defaults(func=lambda a: opp_send(a.to, a.message, a.from_who))
|
|
|
|
pr = sub.add_parser('read', help='Read operator DMs')
|
|
pr.add_argument('--who', default='operator-main', choices=['646', 'operator-main'])
|
|
pr.add_argument('--n', type=int, default=5)
|
|
pr.set_defaults(func=lambda a: opp_read(a.who, a.n))
|
|
|
|
args = p.parse_args()
|
|
args.func(args)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|