// Copyright 2014 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

@Tags(<String>['flutter-test-driver'])
library;

import 'dart:async';

import 'package:file/file.dart';
import 'package:flutter_tools/src/isolated/resident_web_runner.dart' show kNoClientConnectedMessage;
import '../src/common.dart';
import 'test_data/hot_reload_project.dart';
import 'test_data/websocket_dwds_test_common.dart';
import 'test_driver.dart';
import 'test_utils.dart';
import 'transition_test_utils.dart';

void main() {
  testAll();
}

void testAll({List<String> additionalCommandArgs = const <String>[]}) {
  group('WebSocket DWDS connection'
      '${additionalCommandArgs.isEmpty ? '' : ' with args: $additionalCommandArgs'}', () {
    // Test configuration constants
    const hotReloadTimeout = Duration(seconds: 10);

    late Directory tempDir;
    final project = HotReloadProject();
    late FlutterRunTestDriver flutter;

    setUp(() async {
      tempDir = createResolvedTempDirectorySync('hot_reload_websocket_test.');
      await project.setUpIn(tempDir);
      flutter = FlutterRunTestDriver(tempDir);
    });

    tearDown(() async {
      await flutter.stop();
      tryToDelete(tempDir);
    });

    testWithoutContext(
      'hot reload with headless Chrome WebSocket connection',
      () async {
        debugPrint('Starting WebSocket DWDS test with headless Chrome for hot reload...');

        // Set up WebSocket connection
        final WebSocketDwdsTestSetup setup = await WebSocketDwdsTestUtils.setupWebSocketConnection(
          flutter,
          additionalCommandArgs: additionalCommandArgs,
        );

        try {
          // Test hot reload functionality
          debugPrint('Testing hot reload with WebSocket connection...');
          await flutter.hotReload().timeout(
            hotReloadTimeout,
            onTimeout: () {
              throw Exception('Hot reload timed out');
            },
          );

          // Give some time for logs to capture
          await Future<void>.delayed(const Duration(seconds: 2));

          final output = setup.stdout.toString();
          expect(output, contains('Reloaded'), reason: 'Hot reload should complete successfully');
          debugPrint('✓ Hot reload completed successfully with WebSocket connection');

          // Verify the correct infrastructure was used
          WebSocketDwdsTestUtils.verifyWebSocketInfrastructure(output);

          debugPrint('✓ WebSocket DWDS test completed successfully');
          debugPrint('✓ Verified: web-server device + DWDS + WebSocket connection + hot reload');
        } finally {
          await cleanupWebSocketTestResources(setup.chromeProcess, setup.subscription);
        }
      },
      skip: !platform.isMacOS, // Skip on non-macOS platforms where Chrome paths may differ
    );

    testWithoutContext(
      'hot reload gracefully handles closed browser (no clients available)',
      () async {
        debugPrint('Starting test for no clients available scenario...');

        // Set up WebSocket connection
        final WebSocketDwdsTestSetup setup = await WebSocketDwdsTestUtils.setupWebSocketConnection(
          flutter,
          additionalCommandArgs: additionalCommandArgs,
        );

        try {
          // First, verify hot reload works with browser connected
          debugPrint('Verifying initial hot reload with browser connected...');
          await flutter.hotReload().timeout(
            hotReloadTimeout,
            onTimeout: () {
              throw Exception('Initial hot reload timed out');
            },
          );

          await Future<void>.delayed(const Duration(seconds: 1));
          final initialOutput = setup.stdout.toString();
          expect(initialOutput, contains('Reloaded'), reason: 'Initial hot reload should succeed');
          debugPrint('✓ Initial hot reload succeeded');

          // Close the browser to simulate no clients available
          debugPrint('Closing browser to simulate no clients available...');
          setup.chromeProcess.kill();
          await setup.chromeProcess.exitCode;
          debugPrint('✓ Browser closed');

          // Give DWDS time to detect the disconnection
          await Future<void>.delayed(const Duration(seconds: 2));

          // Attempt hot reload with no browser connected
          debugPrint('Attempting hot reload with no browser connected...');
          await flutter.hotReload().timeout(
            hotReloadTimeout,
            onTimeout: () {
              throw Exception('Hot reload with no clients timed out');
            },
          );

          // Give some time for logs to capture
          await Future<void>.delayed(const Duration(seconds: 2));

          final output = setup.stdout.toString();

          // Verify the graceful handling message
          expect(
            output,
            contains(kNoClientConnectedMessage),
            reason: 'Should show no client connected message',
          );

          debugPrint('✓ Hot reload handled no clients gracefully');
          debugPrint('✓ Test completed: Verified graceful handling when browser is closed');
        } finally {
          // Note: Chrome process is already killed in the test, so just cancel subscription
          await setup.subscription.cancel();
        }
      },
      skip: !platform.isMacOS, // Skip on non-macOS platforms where Chrome paths may differ
    );
  });
}