#!/usr/bin/env python3

import json
import subprocess
import sys
from pathlib import Path

def main():
    session_file = "/home/wdjones/.openclaw/agents/main/sessions/2c022034-fccc-4c2c-b8f0-dce45ad22e68.jsonl"
    
    # Read all messages from the session file
    messages = []
    with open(session_file, 'r') as f:
        for line in f:
            data = json.loads(line.strip())
            if data.get('type') == 'message':
                messages.append(data)
    
    # Extract user-assistant pairs
    pairs = []
    current_user_msg = None
    
    for msg in messages:
        role = msg.get('message', {}).get('role')
        content = msg.get('message', {}).get('content', [])
        
        if role == 'user':
            # Extract text content from user message
            user_text = ""
            for item in content:
                if item.get('type') == 'text':
                    user_text += item.get('text', '') + "\n"
            current_user_msg = user_text.strip()
            
        elif role == 'assistant' and current_user_msg:
            # Extract text content from assistant message (skip thinking blocks)
            assistant_text = ""
            for item in content:
                if item.get('type') == 'text':
                    assistant_text += item.get('text', '') + "\n"
            
            if assistant_text.strip():
                pairs.append({
                    'user': current_user_msg,
                    'assistant': assistant_text.strip()
                })
                current_user_msg = None  # Reset for next pair
    
    # Take the last 10 pairs
    recent_pairs = pairs[-10:]
    
    print(f"Found {len(pairs)} total pairs, processing last {len(recent_pairs)}")
    
    # Send each pair to the auto-memory hook
    for pair in recent_pairs:
        payload = {
            "user": pair['user'],
            "assistant": pair['assistant'],
            "agent_id": "case",
            "session": "main"
        }
        
        try:
            result = subprocess.run([
                'python3', '/home/wdjones/.openclaw/workspace/tools/auto-memory-hook.py'
            ], input=json.dumps(payload), text=True, capture_output=True, cwd="/home/wdjones/.openclaw/workspace")
            
            if result.returncode != 0:
                print(f"Error processing pair: {result.stderr}")
            else:
                print(f"Processed pair successfully")
        except Exception as e:
            print(f"Exception processing pair: {e}")

if __name__ == "__main__":
    main()