#!/usr/bin/env python3
"""Check for new messages using MCP bridge"""

import sys
import sqlite3
import argparse
from datetime import datetime
from pathlib import Path


def check_messages(db_path: str, conversation_id: str, token: str):
    """Check for unread messages"""
    try:
        if not Path(db_path).exists():
            print(f"⚠️  Database not found: {db_path}", file=sys.stderr)
            return

        conn = sqlite3.connect(db_path)
        conn.row_factory = sqlite3.Row

        # Get unread messages
        cursor = conn.execute(
            """SELECT id, sender, content, action_type, created_at
               FROM messages
               WHERE conversation_id = ? AND read_by_b = 0
               ORDER BY created_at ASC""",
            (conversation_id,)
        )

        messages = cursor.fetchall()

        if messages:
            print(f"\n📨 {len(messages)} new message(s):")
            for msg in messages:
                print(f"  From: {msg['sender']}")
                print(f"  Type: {msg['action_type']}")
                print(f"  Time: {msg['created_at']}")
                content = msg['content'][:100]
                if len(msg['content']) > 100:
                    content += "..."
                print(f"  Content: {content}")
                print()

                # Mark as read
                conn.execute(
                    "UPDATE messages SET read_by_b = 1 WHERE id = ?",
                    (msg['id'],)
                )

            conn.commit()
            print(f"✅ {len(messages)} message(s) marked as read")
        else:
            print("📭 No new messages")

        conn.close()

    except sqlite3.OperationalError as e:
        print(f"❌ Database error: {e}", file=sys.stderr)
        sys.exit(1)
    except Exception as e:
        print(f"❌ Error: {e}", file=sys.stderr)
        sys.exit(1)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Check for new MCP bridge messages")
    parser.add_argument("--conversation-id", required=True, help="Conversation ID")
    parser.add_argument("--token", required=True, help="Worker token")
    parser.add_argument("--db-path", default="/tmp/claude_bridge_coordinator.db", help="Database path")

    args = parser.parse_args()
    check_messages(args.db_path, args.conversation_id, args.token)