updated dev_handler to throw error if proxy_service is not the right type

Author: jyameo

dwds/lib/src/handlers/dev_handler.dart

@@ -559,12 +559,18 @@ class DevHandler {
     final appDebugServices = await loadAppServices(appConnection);
 
     // Initialize WebSocket proxy service
-    final proxyService = appDebugServices.proxyService;
-    if (proxyService is WebSocketProxyService) {
+    try {
+      final proxyService = appDebugServices.proxyService;
+      if (proxyService is! WebSocketProxyService) {
+        throw StateError(
+          'Expected WebSocketProxyService but got ${proxyService.runtimeType}. ',
+        );
+      }
       await proxyService.isInitialized;
       _logger.fine('WebSocket proxy service initialized successfully');
-    } else {
-      _logger.warning('WebSocket proxy service is null');
+    } catch (e) {
+      _logger.severe('Failed to initialize WebSocket proxy service: $e');
+      rethrow;
     }
 
     return DebugConnection(appDebugServices);
@@ -579,12 +585,13 @@ class DevHandler {
     // Initialize Chrome proxy service
     try {
       final proxyService = appDebugServices.proxyService;
-      if (proxyService is ChromeProxyService) {
-        await proxyService.isInitialized;
-        _logger.fine('Chrome proxy service initialized successfully');
-      } else {
-        _logger.warning('Chrome proxy service is null');
+      if (proxyService is! ChromeProxyService) {
+        throw StateError(
+          'Expected ChromeProxyService but got ${proxyService.runtimeType}. ',
+        );
       }
+      await proxyService.isInitialized;
+      _logger.fine('Chrome proxy service initialized successfully');
     } catch (e) {
       _logger.severe('Failed to initialize Chrome proxy service: $e');
       rethrow;

dwds/lib/src/injected/client.js

@@ -78,7 +78,7 @@ class ReloadingManager {
   ) async {
     final restarter = _restarter;
     if (restarter is DdcLibraryBundleRestarter) {
-      return restarter.handleServiceExtension(method, args);
+      return await restarter.handleServiceExtension(method, args);
     }
     // For other restarter types, return null to indicate not supported
     return null;