#!/usr/bin/env python3
"""
opp-dm.py: Operator Direct Messages

For operator-to-operator communication (646, operator-main).
Operators have SSH access and infrastructure tools.
This is bidirectional and reliable.

Distinct from dev-dm.py (for task agents: muse, pip).

Usage:
  opp-dm.py send --to 646 "message"        # operator-main -> 646
  opp-dm.py send --to operator-main "msg"  # 646 -> operator-main (via SSH)
  opp-dm.py read [--from 646]              # Read operator messages
"""
import argparse
import subprocess
import sys
import time

# Operators have SSH access to bl
# Messages stored in a shared location on bl
BL_MSG_DIR = "/home/super/Projects/NetVM/opp-dms"

def bl_run(cmd):
    """Run command on bl via SSH."""
    full = f"ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o BatchMode=yes super@100.123.153.75 '{cmd}'"
    # Via VM
    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()

def opp_send(to, message, from_who="operator-main"):
    """Send an operator DM (stored on bl, both can access via SSH)."""
    import datetime
    ts = datetime.datetime.now().isoformat()
    # Escape for shell
    safe = message.replace("'", "'\"'\"'")[:1000]
    cmd = f"mkdir -p {BL_MSG_DIR} && echo '{ts} [{from_who}]: {safe}' >> {BL_MSG_DIR}/{to}.log && echo 'sent'"
    result = bl_run(cmd)
    print(f"OPP-DM sent to {to}")
    return result

def opp_read(who="operator-main", n=5):
    """Read operator DMs."""
    cmd = f"tail -n {n*2} {BL_MSG_DIR}/{who}.log 2>/dev/null || echo 'no messages'"
    result = bl_run(cmd)
    print(result)
    return result

def main():
    p = argparse.ArgumentParser(description="OPP-DM: Operator Direct Messages")
    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()