import logging

from typeguard import install_import_hook, TypeguardFinder, config, ForwardRefPolicy, CollectionCheckStrategy

# COPY-PASTE SOLUTION: Enhanced Debug and Verbosity Configuration

# Step 1: Configure comprehensive debug and verbosity
config.debug_instrumentation = True  # Set to False to disable debug
config.forward_ref_policy = ForwardRefPolicy.WARN  # WARN, ERROR, or IGNORE
config.collection_check_strategy = CollectionCheckStrategy.FIRST_ITEM  # ALL_ITEMS for more thorough checking

# Enable verbose logging
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger('typeguard')
logger.setLevel(logging.DEBUG)

# Step 2: Define packages to instrument
packages_to_instrument = [
    'grpc.aio',
    'grpc.aio._channel',
    'grpc.aio._server',
    'grpc.aio._utils',
    'grpc.aio._interceptor',
    'grpc.aio._base_channel',
    'grpc.aio._base_server',
    'grpc.aio._typing',
    'grpc.aio._call',
    'grpc.aio._metadata',
    # SYNC - Test only grpc._server to confirm it's causing the HandlerCallDetails TypeCheckError
    'grpc._server',
]

# Step 3: Create enhanced custom TypeguardFinder with verbose logging
class VerboseTypeGuardFinder(TypeguardFinder):
    def should_instrument(self, module_name: str) -> bool:
        result = super().should_instrument(module_name)
        if result:
            print(f"🔧 INSTRUMENTING: {module_name}")
            logger.info(f"Instrumenting module: {module_name}")
        else:
            print(f"⏭️  SKIPPING: {module_name}")
            logger.debug(f"Skipping module: {module_name}")
        return result

# Step 4: Install import hook with verbose finder
print("📋 Installing TypeguardFinder with debug enabled...")
print(f"📦 Packages to instrument: {len(packages_to_instrument)}")
for pkg in packages_to_instrument:
    print(f"   • {pkg}")

install_import_hook(packages_to_instrument, cls=VerboseTypeGuardFinder)
print("✅ TypeguardFinder installed successfully!")