#!/bin/bash
# Memory Collaboration Reminder
# Reminds Claude agents to check memory and coordinate

set -e

MEMORY_DIR="$HOME/.blackroad/memory"
PROTOCOL_FILE="$HOME/CLAUDE_COLLABORATION_PROTOCOL.md"

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
NC='\033[0m'

# Show reminder banner
show_reminder() {
    cat << 'EOF'

╔══════════════════════════════════════════════════════════════╗
║  ⚠️  CLAUDE COLLABORATION REMINDER                          ║
╚══════════════════════════════════════════════════════════════╝

📋 BEFORE YOU START WORKING:

   1. Register yourself:
      MY_CLAUDE=$(~/memory-sync-daemon.sh register "claude-[your-focus]")

   2. Check [MEMORY] and [CODEX]:
      # Check memory first
      ~/memory-realtime-context.sh live $MY_CLAUDE compact

      # Check Codex for existing solutions
      python3 ~/blackroad-codex-search.py "[your-task-keywords]"

   3. Announce your work:
      ~/memory-system.sh log announce "$MY_CLAUDE" "
      Hey team! I'm working on: [PROJECT]

      Tasks: 1. ... 2. ... 3. ... 4. ... 5. ...

      Goal: [BIG IDEA]

      [MEMORY] ✅ Checked for conflicts
      [CODEX] ✅ Searched for existing code

      Checking both every 60s. Let me know if this overlaps!
      "

   4. Check memory & codex EVERY 60 SECONDS:
      ~/memory-realtime-context.sh live $MY_CLAUDE compact

   5. Look for:
      • [MEMORY] Other Claudes working on same thing (coordinate!)
      • [MEMORY] Deployments you depend on (use them!)
      • [MEMORY] Decisions affecting your work (align with them!)
      • [CODEX] Existing components you can reuse (don't rebuild!)
      • [CODEX] Similar patterns to follow (consistency!)
      • [MEMORY] Coordination requests for you (respond!)

   6. Update progress regularly:
      ~/memory-system.sh log progress "$MY_CLAUDE" "✅ Done: [X]. Next: [Y]"

━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━

⚠️  CRITICAL: Check [MEMORY] & [CODEX] every 60 seconds to avoid:
   ❌ Duplicate work (another Claude already did it)
   ❌ Conflicting deployments (memory shows conflicts)
   ❌ Integration failures (memory shows dependencies)
   ❌ Wasted effort (codex has existing solutions)
   ❌ Reinventing the wheel (8,789 components already built!)

✅ GOAL: All Claudes work together as one distributed swarm!
✅ CODEX: 56 repositories, 8,789 components - check before building!

━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━

📚 Full protocol: ~/CLAUDE_COLLABORATION_PROTOCOL.md

EOF
}

# Check if Claude is following protocol
check_compliance() {
    local instance_id="$1"

    if [ -z "$instance_id" ]; then
        echo -e "${RED}❌ No instance ID provided${NC}"
        echo -e "${YELLOW}Run: MY_CLAUDE=\$(~/memory-sync-daemon.sh register \"claude-[name]\")${NC}"
        return 1
    fi

    # Check if instance is registered
    local instance_file="$MEMORY_DIR/sync/instances/${instance_id}.json"
    if [ ! -f "$instance_file" ]; then
        echo -e "${RED}❌ Instance not registered${NC}"
        echo -e "${YELLOW}Run: ~/memory-sync-daemon.sh register \"${instance_id}\"${NC}"
        return 1
    fi

    # Check if announcement was made
    local announcements=$(grep -c "\"action\":\"announce\"" "$MEMORY_DIR/journals/master-journal.jsonl" 2>/dev/null || echo "0")
    if [ "$announcements" -eq 0 ]; then
        echo -e "${YELLOW}⚠️  No announcement found${NC}"
        echo -e "${YELLOW}Did you announce your work? (See protocol)${NC}"
        return 1
    fi

    # Check last seen time
    local last_seen=$(jq -r '.last_seen' "$instance_file")
    echo -e "${GREEN}✅ Instance registered and active${NC}"
    echo -e "${GREEN}   Last seen: $last_seen${NC}"

    # Check for progress updates
    local progress_count=$(grep -c "\"action\":\"progress\"" "$MEMORY_DIR/journals/master-journal.jsonl" 2>/dev/null || echo "0")
    echo -e "${GREEN}   Progress updates: $progress_count${NC}"

    # Check for coordination
    local coord_count=$(grep -c "\"action\":\"coordinate\"" "$MEMORY_DIR/journals/master-journal.jsonl" 2>/dev/null || echo "0")
    echo -e "${GREEN}   Coordination messages: $coord_count${NC}"

    return 0
}

# Watch mode - remind every 60 seconds
watch_mode() {
    local instance_id="$1"

    echo -e "${CYAN}╔════════════════════════════════════════════════════╗${NC}"
    echo -e "${CYAN}║  🔄 Collaboration Watch Mode (60s intervals)      ║${NC}"
    echo -e "${CYAN}╚════════════════════════════════════════════════════╝${NC}"
    echo ""
    echo -e "${YELLOW}Instance: $instance_id${NC}"
    echo -e "${YELLOW}Press Ctrl+C to stop${NC}"
    echo ""

    while true; do
        clear
        echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo -e "${CYAN}  🔍 [MEMORY] CHECK ($(date +%H:%M:%S))${NC}"
        echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo ""

        # Show live context
        ~/memory-realtime-context.sh live "$instance_id" compact 2>/dev/null

        echo ""
        echo -e "${PURPLE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo -e "${PURPLE}  📜 [CODEX] STATUS${NC}"
        echo -e "${PURPLE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo ""

        # Show Codex stats
        local total_components=$(sqlite3 ~/blackroad-codex/index/components.db "SELECT COUNT(*) FROM components" 2>/dev/null || echo "0")
        echo -e "  📦 Total Components: ${GREEN}$total_components${NC}"
        echo -e "  🔍 Search: ${YELLOW}python3 ~/blackroad-codex-search.py \"your-query\"${NC}"
        echo -e "  📐 Verify: ${YELLOW}~/blackroad-codex-verification-suite.sh verify <id> <file>${NC}"

        echo ""
        echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo -e "${YELLOW}⏰ Next check in 60 seconds...${NC}"
        echo -e "${YELLOW}💡 [MEMORY] Update progress | [CODEX] Search before building!${NC}"
        echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"

        sleep 60
    done
}

# Show active Claudes
show_active_claudes() {
    echo -e "${BLUE}╔════════════════════════════════════════════════════╗${NC}"
    echo -e "${BLUE}║  👥 Active Claude Agents                          ║${NC}"
    echo -e "${BLUE}╚════════════════════════════════════════════════════╝${NC}"
    echo ""

    ~/memory-sync-daemon.sh instances

    echo ""
    echo -e "${CYAN}Recent Announcements:${NC}"
    grep "\"action\":\"announce\"" "$MEMORY_DIR/journals/master-journal.jsonl" 2>/dev/null | \
        tail -5 | \
        jq -r '"  [" + .timestamp[11:19] + "] " + .entity' || echo "  No announcements yet"

    echo ""
    echo -e "${CYAN}Recent Progress Updates:${NC}"
    grep "\"action\":\"progress\"" "$MEMORY_DIR/journals/master-journal.jsonl" 2>/dev/null | \
        tail -5 | \
        jq -r '"  [" + .timestamp[11:19] + "] " + .details' || echo "  No progress updates yet"

    echo ""
    echo -e "${CYAN}Coordination Messages:${NC}"
    grep "\"action\":\"coordinate\"" "$MEMORY_DIR/journals/master-journal.jsonl" 2>/dev/null | \
        tail -5 | \
        jq -r '"  [" + .timestamp[11:19] + "] " + .details' || echo "  No coordination yet"
}

# Show help
show_help() {
    cat <<EOF
Memory Collaboration Reminder v1.0.0

USAGE:
    memory-collaboration-reminder.sh <command> [options]

COMMANDS:
    reminder                    Show collaboration reminder
    check <instance-id>         Check if Claude is following protocol
    watch <instance-id>         Watch mode (remind every 60s)
    active                      Show all active Claudes
    protocol                    Open protocol document
    help                        Show this help

EXAMPLES:
    # Show reminder
    ~/memory-collaboration-reminder.sh reminder

    # Check compliance
    ~/memory-collaboration-reminder.sh check claude-api

    # Watch mode (auto-check every 60s)
    ~/memory-collaboration-reminder.sh watch claude-api

    # See all active Claudes
    ~/memory-collaboration-reminder.sh active

    # Read full protocol
    ~/memory-collaboration-reminder.sh protocol

INTEGRATION:
    Add to Claude Code startup:
    ~/memory-collaboration-reminder.sh reminder

    Or use watch mode while working:
    ~/memory-collaboration-reminder.sh watch \$MY_CLAUDE

EOF
}

# Main command handler
case "${1:-reminder}" in
    reminder)
        show_reminder
        ;;
    check)
        if [ -z "$2" ]; then
            echo "Usage: $0 check <instance-id>"
            exit 1
        fi
        check_compliance "$2"
        ;;
    watch)
        if [ -z "$2" ]; then
            echo "Usage: $0 watch <instance-id>"
            exit 1
        fi
        watch_mode "$2"
        ;;
    active)
        show_active_claudes
        ;;
    protocol)
        if [ -f "$PROTOCOL_FILE" ]; then
            cat "$PROTOCOL_FILE"
        else
            echo "Protocol file not found: $PROTOCOL_FILE"
            exit 1
        fi
        ;;
    help|--help|-h)
        show_help
        ;;
    *)
        echo "Unknown command: $1"
        echo ""
        show_help
        exit 1
        ;;
esac