From 33fb7fead1741741b3441b80be066023655ef493 Mon Sep 17 00:00:00 2001
From: Bob <bob@umbernetworks.com>
Date: Mon, 10 Nov 2025 03:43:58 +0000
Subject: [PATCH] Add chip type detection to device detection script

- Add --probe flag to detect exact chip type using esptool
- Show USB-based chip guess by default (faster)
- Support ESP32, ESP32-S2, ESP32-S3, ESP32-C3, ESP32-C6, ESP32-H2
- Useful for identifying mixed chip types in large deployments
---
 detect_esp32.py | 75 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)

diff --git a/detect_esp32.py b/detect_esp32.py
index 7b484e0..620c380 100755
--- a/detect_esp32.py
+++ b/detect_esp32.py
@@ -5,6 +5,57 @@ Detects and lists ESP32 devices connected via USB
 """
 
 import serial.tools.list_ports
+import subprocess
+import sys
+
+
+def detect_chip_type(port):
+    """
+    Try to detect the ESP32 chip type using esptool.py
+    Returns chip type string or 'Unknown'
+    """
+    try:
+        result = subprocess.run(
+            ['esptool.py', '--port', port, 'chip_id'],
+            capture_output=True,
+            text=True,
+            timeout=10
+        )
+        
+        output = result.stdout + result.stderr
+        
+        if 'ESP32-S3' in output:
+            return 'ESP32-S3'
+        elif 'ESP32-S2' in output:
+            return 'ESP32-S2'
+        elif 'ESP32-C3' in output:
+            return 'ESP32-C3'
+        elif 'ESP32-C6' in output:
+            return 'ESP32-C6'
+        elif 'ESP32-H2' in output:
+            return 'ESP32-H2'
+        elif 'ESP32' in output:
+            return 'ESP32'
+        
+    except Exception:
+        pass
+    
+    return 'Unknown'
+
+
+def guess_chip_type_from_usb(vid, pid):
+    """
+    Make an educated guess about chip type based on USB VID:PID
+    Returns chip type string or None
+    """
+    if vid == 0x303A:  # Espressif USB
+        if pid == 0x1001:
+            return 'ESP32-S3/C3'  # Built-in USB, could be either
+        elif pid == 0x1002:
+            return 'ESP32-S3/C3'
+    
+    # For CP210x, CH340, etc., we can't tell from USB alone
+    return None
 
 
 def detect_esp32_devices():
@@ -59,6 +110,13 @@ def detect_esp32_devices():
 
 
 def main():
+    import argparse
+    
+    parser = argparse.ArgumentParser(description='ESP32 USB Device Detection')
+    parser.add_argument('--probe', action='store_true',
+                       help='Probe each device to detect exact chip type (slower)')
+    args = parser.parse_args()
+    
     print("=" * 60)
     print("ESP32 USB Device Detection")
     print("=" * 60)
@@ -79,6 +137,21 @@ def main():
             if device.vid is not None and device.pid is not None:
                 print(f"  VID:PID:     {device.vid:04X}:{device.pid:04X}")
             
+            # Chip type detection
+            chip_type = None
+            if args.probe:
+                print(f"  Probing chip type...", end='', flush=True)
+                chip_type = detect_chip_type(device.device)
+                print(f"\r  Chip Type:   {chip_type}")
+            else:
+                # Try to guess from USB VID:PID
+                if device.vid is not None and device.pid is not None:
+                    chip_type = guess_chip_type_from_usb(device.vid, device.pid)
+                    if chip_type:
+                        print(f"  Chip Guess:  {chip_type} (use --probe for exact type)")
+                    else:
+                        print(f"  Chip Type:   Unknown (use --probe to detect)")
+            
             if device.manufacturer:
                 print(f"  Manufacturer: {device.manufacturer}")
             
@@ -89,6 +162,8 @@ def main():
         
         print("=" * 60)
         print(f"Total ESP32 devices detected: {len(esp32_devices)}")
+        if not args.probe:
+            print("Tip: Use --probe flag to detect exact chip types")
         print("=" * 60)
     else:
         print("No ESP32 devices detected.")