#!/usr/bin/env python3
"""Check what's in the source database."""

import os

from neo4j import GraphDatabase

NEO4J_URI = 'bolt://192.168.1.25:7687'
NEO4J_USER = 'neo4j'
NEO4J_PASSWORD = '!"MiTa1205'

SOURCE_DATABASE = 'neo4j'
SOURCE_GROUP_ID = 'lvarming73'

driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USER, NEO4J_PASSWORD))

print('=' * 70)
print('Checking Source Database')
print('=' * 70)

with driver.session(database=SOURCE_DATABASE) as session:
    # Check total nodes
    result = session.run(
        """
        MATCH (n {group_id: $group_id})
        RETURN count(n) as total
    """,
        group_id=SOURCE_GROUP_ID,
    )

    total = result.single()['total']
    print(f"\n✓ Total nodes with group_id '{SOURCE_GROUP_ID}': {total}")

    # Check date range
    result = session.run(
        """
        MATCH (n:Episodic {group_id: $group_id})
        WHERE n.created_at IS NOT NULL
        RETURN
            min(n.created_at) as earliest,
            max(n.created_at) as latest,
            count(n) as total
    """,
        group_id=SOURCE_GROUP_ID,
    )

    dates = result.single()
    if dates and dates['total'] > 0:
        print(f'\n✓ Episodic date range:')
        print(f'   Earliest: {dates["earliest"]}')
        print(f'   Latest: {dates["latest"]}')
        print(f'   Total episodes: {dates["total"]}')
    else:
        print('\n⚠️  No episodic nodes with dates found')

    # Sample episodic nodes by date
    result = session.run(
        """
        MATCH (n:Episodic {group_id: $group_id})
        RETURN n.name as name, n.created_at as created_at
        ORDER BY n.created_at
        LIMIT 10
    """,
        group_id=SOURCE_GROUP_ID,
    )

    print(f'\n✓ Oldest episodic nodes:')
    for record in result:
        print(f'   - {record["name"]}: {record["created_at"]}')

    # Check for other group_ids in neo4j database
    result = session.run("""
        MATCH (n)
        WHERE n.group_id IS NOT NULL
        RETURN DISTINCT n.group_id as group_id, count(n) as count
        ORDER BY count DESC
    """)

    print(f"\n✓ All group_ids in '{SOURCE_DATABASE}' database:")
    for record in result:
        print(f'   {record["group_id"]}: {record["count"]} nodes')

driver.close()
print('\n' + '=' * 70)