#!/usr/bin/env python3
"""
Conditional USB Mode Script for WLED-MM

This script automatically sets ARDUINO_USB_MODE based on the build context:
- For development builds: ARDUINO_USB_MODE=1 (allows USB debugging)  
- For release builds (CI): ARDUINO_USB_MODE=0 (allows normal boot without USB debugger)

The script detects release builds by checking for the WLED_RELEASE environment variable
which is set to True in the GitHub Actions CI workflow.

CRITICAL: This change only applies to boards with USB-OTG (ARDUINO_USB_CDC_ON_BOOT=1).
For boards with classical UART-to-USB chips (ARDUINO_USB_CDC_ON_BOOT=0), 
ARDUINO_USB_MODE=1 is harmless and left unchanged.
"""
## This script was created with the help of an AI, reviewed and tested by @softhack007

Import('env')
import os

def has_cdc_on_boot_enabled(env):
    """
    Check if ARDUINO_USB_CDC_ON_BOOT is set to 1 in the build configuration.
    
    Returns True if CDC_ON_BOOT=1, False otherwise.
    This is used to identify boards with USB-OTG (native USB) vs UART-to-USB chips.
    """
    cpp_defines = env.get('CPPDEFINES', [])
    build_flags = env.get('BUILD_FLAGS', [])
    
    # Check in CPPDEFINES
    for define in cpp_defines:
        if isinstance(define, (list, tuple)) and len(define) == 2:
            if define[0] == 'ARDUINO_USB_CDC_ON_BOOT' and define[1] == '1':
                return True
    
    # Check in raw build flags
    for flag in build_flags:
        if isinstance(flag, str) and 'ARDUINO_USB_CDC_ON_BOOT=1' in flag:
            return True
    
    return False

def has_usb_mode_enabled(env):
    """
    Check if ARDUINO_USB_MODE is set to 1 in the build configuration.
    
    Returns True if USB_MODE=1, False otherwise.
    """
    cpp_defines = env.get('CPPDEFINES', [])
    build_flags = env.get('BUILD_FLAGS', [])
    
    # Check in CPPDEFINES
    for define in cpp_defines:
        if isinstance(define, (list, tuple)) and len(define) == 2:
            if define[0] == 'ARDUINO_USB_MODE' and define[1] == '1':
                return True
    
    # Check in raw build flags
    for flag in build_flags:
        if isinstance(flag, str) and 'ARDUINO_USB_MODE=1' in flag:
            return True
    
    return False

def conditional_usb_mode(env):
    """
    Conditionally set ARDUINO_USB_MODE based on build context.
    
    For ESP32-C3, ESP32-S2, and ESP32-S3 variants with USB-OTG (CDC_ON_BOOT=1):
    - Development builds: ARDUINO_USB_MODE=1 (default, good for debugging)
    - Release builds: ARDUINO_USB_MODE=0 (prevents hanging without USB debugger)
    
    For boards with classical UART-to-USB chip (CDC_ON_BOOT=0):
    - ARDUINO_USB_MODE=1 is harmless and left unchanged
    """
    
    # Check if this is a release build (CI sets WLED_RELEASE=True)
    is_release_build = os.environ.get('WLED_RELEASE', '').lower() in ('true', '1', 'yes')
    
    if is_release_build:
        # Check if this board uses USB-OTG (CDC_ON_BOOT=1)
        if not has_cdc_on_boot_enabled(env):
            print("WLED Release build detected - board uses UART-to-USB chip (CDC_ON_BOOT=0)")
            print("  Keeping ARDUINO_USB_MODE=1 (harmless for UART-to-USB boards)")
            return
        
        print("WLED Release build detected - board uses USB-OTG (CDC_ON_BOOT=1)")
        print("  Setting ARDUINO_USB_MODE=0 for production")
        
        # Check if ARDUINO_USB_MODE=1 is present
        if has_usb_mode_enabled(env):
            # Remove the old definition and add the new one
            # This approach properly handles flag inheritance in PlatformIO
            env.Append(BUILD_UNFLAGS=["-DARDUINO_USB_MODE=1"])
            env.Append(CPPDEFINES=[("ARDUINO_USB_MODE", "0")])
            print(f"  Changed ARDUINO_USB_MODE from 1 to 0")
            
    else:
        # Development build
        has_usb_mode = has_usb_mode_enabled(env)
        has_cdc_boot = has_cdc_on_boot_enabled(env)
        
        if has_usb_mode and has_cdc_boot:
            print("Development build detected - keeping ARDUINO_USB_MODE=1 for USB-OTG debugging")
            # Warning in orange/yellow using ANSI color codes
            print("\033[93m  WARNING: This build is NOT suitable for production devices!\033[0m")
            print("\033[93m  Production builds require WLED_RELEASE=True environment variable.\033[0m")
        elif has_cdc_boot:
            # CDC_ON_BOOT=1 present but not USB_MODE=1 
            print("Development build detected - USB-OTG enabled, but ARDUINO_USB_MODE=1 missing for debugging")
        elif has_usb_mode:
            # USB_MODE=1 present but not CDC_ON_BOOT=1 (UART-to-USB board)
            print("Development build detected - board uses UART-to-USB chip")
        # If neither flag is present, don't print anything

# Apply the conditional USB mode logic
conditional_usb_mode(env)