#!/bin/bash
# OpenClaw Monitor Agent
set -e
SERVER="" TOKEN="" NAME="" ROLE="worker" INTERVAL=30

while getopts "s:t:n:r:i:" opt; do
  case $opt in s)SERVER="$OPTARG";;t)TOKEN="$OPTARG";;n)NAME="$OPTARG";;r)ROLE="$OPTARG";;i)INTERVAL="$OPTARG";;esac
done
[ -z "$SERVER" ] || [ -z "$TOKEN" ] && echo "Usage: $0 -s SERVER_URL -t TOKEN [-n name] [-r role]" && exit 1

NODE_ID=$(hostname | md5sum 2>/dev/null | cut -c1-12 || md5 -qs "$(hostname)" | cut -c1-12)
NAME="${NAME:-$(hostname)}"
OC_CONFIG=""
for p in "$HOME/.openclaw/openclaw.json" "/root/.openclaw/openclaw.json"; do
  [ -f "$p" ] && OC_CONFIG="$p" && break
done
OC_VERSION=$(openclaw --version 2>/dev/null | head -1 || echo "unknown")

echo "OC Monitor Agent: node=$NODE_ID name=$NAME server=$SERVER"

while true; do
  python3 - "$OC_CONFIG" "$NODE_ID" "$NAME" "$OC_VERSION" "$ROLE" << 'PYEOF' > /tmp/.oc-agent-payload.json
import json,subprocess,os,platform,time,sys,glob

def run(cmd):
    try: return subprocess.check_output(cmd,shell=True,stderr=subprocess.DEVNULL).decode().strip()
    except: return ''

cfg,nid,name,ver,role = sys.argv[1],sys.argv[2],sys.argv[3],sys.argv[4],sys.argv[5]
mac = sys.platform=='darwin'

# Host IP - try multiple interfaces on macOS
if mac:
    host = run("ifconfig | grep 'inet ' | grep -v 127 | head -1 | awk '{print $2}'")
else:
    host = run("hostname -I | awk '{print $1}'")
if not host: host = 'unknown'

# Providers
providers = []
if cfg and os.path.exists(cfg):
    try:
        c = json.load(open(cfg))
        for n,p in c.get('models',{}).get('providers',{}).items():
            if not isinstance(p,dict): continue
            for m in p.get('models',[]):
                providers.append({'name':n,'model':m.get('id',''),'api':p.get('api','')})
    except: pass

# CPU
if mac:
    raw = run("ps -A -o %cpu | tail -n +2")
    try: cpu = round(sum(float(x) for x in raw.split() if x) / os.cpu_count(), 1)
    except: cpu = 0
else:
    cpu = float(run("top -bn1 | grep 'Cpu' | awk '{print 100-$8}'") or 0)

# Memory
if mac:
    try:
        vm = run('vm_stat')
        d = {}
        for l in vm.split('\n')[1:]:
            if ':' not in l: continue
            k,v = l.split(':',1)
            d[k.strip()] = int(v.strip().rstrip('.'))
        used = d.get('Pages active',0) + d.get('Pages wired down',0)
        total = used + d.get('Pages free',0) + d.get('Pages inactive',0) + d.get('Pages speculative',0)
        mem = round(used/total*100,1) if total else 0
    except: mem = 0
else:
    mem = float(run("free | awk '/Mem/{printf \"%.1f\",$3/$2*100}'") or 0)

# Disk
disk = float(run("df / | awk 'NR==2{gsub(/%/,\"\",$5);print $5}'") or 0)

# Swap
if mac:
    try:
        sw = run('sysctl vm.swapusage')
        used_sw = float(sw.split('used = ')[1].split('M')[0])
        total_sw = float(sw.split('total = ')[1].split('M')[0])
        swap = round(used_sw/total_sw*100,1) if total_sw > 0 else 0
    except: swap = 0
else:
    swap = float(run("free | awk '/Swap/{if($2>0)printf \"%.1f\",$3/$2*100;else print 0}'") or 0)

# Uptime
if mac:
    try:
        b = run('sysctl -n kern.boottime')
        uptime = int(time.time()) - int(b.split('sec = ')[1].split(',')[0])
    except: uptime = 0
else:
    try: uptime = int(float(open('/proc/uptime').read().split()[0]))
    except: uptime = 0

# Gateway/daemon - check by process name
gw = bool(run('pgrep -f "openclaw"'))
daemon = gw  # if openclaw is running, both are likely up

# Sessions - search for session files
sessions = 0
oc_dir = os.path.expanduser('~/.openclaw')
for sf in [os.path.join(oc_dir,'sessions.json'), os.path.join(oc_dir,'data','sessions.json')]:
    if os.path.exists(sf):
        try: sessions = len(json.load(open(sf))); break
        except: pass
# Fallback: count session dirs
if sessions == 0:
    sess_dir = os.path.join(oc_dir,'sessions')
    if os.path.isdir(sess_dir):
        sessions = len([d for d in os.listdir(sess_dir) if os.path.isdir(os.path.join(sess_dir,d))])

print(json.dumps({
    'id':nid,'name':name,'host':host,
    'os':platform.system()+' '+platform.release()+' '+platform.machine(),
    'oc_version':ver,'role':role,'providers':providers,
    'cpu':cpu,'mem':mem,'disk':disk,'swap':swap,
    'sessions':sessions,'gw_ok':gw,'daemon_ok':daemon,'uptime':uptime
}))
PYEOF

  if [ -s /tmp/.oc-agent-payload.json ]; then
    curl -sS -X POST "$SERVER/api/heartbeat" \
      -H "Authorization: Bearer $TOKEN" \
      -H "Content-Type: application/json" \
      -d @/tmp/.oc-agent-payload.json -m 10 >/dev/null 2>&1 \
      && echo "[$(date +%H:%M:%S)] heartbeat ok" \
      || echo "[$(date +%H:%M:%S)] heartbeat failed"
  fi
  sleep "$INTERVAL"
done