// 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.

import 'dart:async';

import 'package:dtd/dtd.dart';
import 'package:file/file.dart';
import 'package:file_testing/file_testing.dart';
import 'package:flutter_tools/src/base/logger.dart';
import 'package:flutter_tools/src/commands/widget_preview.dart';
import 'package:flutter_tools/src/devtools_launcher.dart';
import 'package:flutter_tools/src/globals.dart' as globals;
import 'package:flutter_tools/src/widget_preview/dtd_services.dart';
import 'package:process/process.dart';

import '../src/common.dart';
import '../src/context.dart';
import '../src/fakes.dart';
import 'test_data/basic_project.dart';
import 'test_utils.dart';
import 'widget_preview_test_helpers.dart';

void main() {
  late Directory tempDir;
  Logger? logger;
  DtdLauncher? dtdLauncher;
  DevtoolsServerLauncher? devtoolsLauncher;
  final project = BasicProject();
  const ProcessManager processManager = LocalProcessManager();

  setUp(() async {
    logger = BufferLogger.test();
    tempDir = createResolvedTempDirectorySync('widget_preview_smoke_test.');
    await project.setUpIn(tempDir);
  });

  tearDown(() async {
    await dtdLauncher?.dispose();
    await devtoolsLauncher?.close();
    devtoolsLauncher = null;
    dtdLauncher = null;
    tryToDelete(tempDir);
  });

  group('flutter widget-preview start', () {
    testWithoutContext('smoke test', () async {
      await runWidgetPreview(tempDir: tempDir, expectedMessages: firstLaunchMessagesWeb);
    });

    testWithoutContext('--web-server starts a web server instance', () async {
      await runWidgetPreview(
        tempDir: tempDir,
        expectedMessages: firstLaunchMessagesWebServer,
        useWebServer: true,
      );
    });

    testWithoutContext(
      'runs flutter pub get in widget_preview_scaffold if '
      "widget_preview_scaffold/.dart_tool doesn't exist",
      () async {
        // Regression test for https://github.com/flutter/flutter/issues/178660
        // Generate the widget preview scaffold, but don't bother launching it.
        processManager.runSync(<String>[
          flutterBin,
          'widget-preview',
          'start',
          '--no-${WidgetPreviewStartCommand.kLaunchPreviewer}',
        ], workingDirectory: tempDir.path);

        // Ensure widget_preview_scaffold/.dart_tool/package_config.json exists.
        final Directory widgetPreviewScaffoldDartTool = tempDir
            .childDirectory('.dart_tool')
            .childDirectory('widget_preview_scaffold')
            .childDirectory('.dart_tool');
        expect(widgetPreviewScaffoldDartTool, exists);
        expect(widgetPreviewScaffoldDartTool.childFile('package_config.json'), exists);

        // Delete widget_preview_scaffold/.dart_tool/. This simulates an interrupted
        // flutter widget-preview start where 'flutter pub get' wasn't run after
        // the widget_preview_scaffold project was created.
        widgetPreviewScaffoldDartTool.deleteSync(recursive: true);

        // Ensure we don't crash due to the package_config.json lookup pointing to
        // the parent project's package_config.json due to
        // widget_preview_scaffold/.dart_tool/package_config.json not existing.
        await runWidgetPreview(tempDir: tempDir, expectedMessages: subsequentLaunchMessagesWeb);
      },
      // Project is always regenerated.
      skip: true, // See https://github.com/flutter/flutter/issues/179036.
    );

    testWithoutContext(
      'does not recreate project on subsequent runs',
      () async {
        // The first run of 'flutter widget-preview start' should generate a new preview scaffold
        await runWidgetPreview(tempDir: tempDir, expectedMessages: firstLaunchMessagesWeb);

        // We shouldn't regenerate the scaffold after the initial run.
        await runWidgetPreview(tempDir: tempDir, expectedMessages: subsequentLaunchMessagesWeb);
      },
      // Project is always regenerated.
      skip: true, // See https://github.com/flutter/flutter/issues/179036.
    );

    testUsingContext('can connect to an existing DTD instance', () async {
      dtdLauncher = DtdLauncher(
        logger: logger!,
        artifacts: globals.artifacts!,
        processManager: globals.processManager,
      );

      // Start a DTD instance.
      final Uri dtdUri = await dtdLauncher!.launch();

      // Connect to it and listen to the WidgetPreviewScaffold stream.
      //
      // The preview scaffold will send a 'Connected' event on this stream once it has initialized
      // and is ready.
      final DartToolingDaemon dtdConnection = await DartToolingDaemon.connect(dtdUri);
      final completer = Completer<void>();
      dtdConnection.onEvent(WidgetPreviewDtdServices.kWidgetPreviewScaffoldStreamRoot).listen((
        DTDEvent event,
      ) {
        expect(event.stream, WidgetPreviewDtdServices.kWidgetPreviewScaffoldStreamRoot);
        expect(event.kind, 'Connected');
        completer.complete();
      });
      await dtdConnection.streamListen(WidgetPreviewDtdServices.kWidgetPreviewScaffoldStreamRoot);

      // Start the widget preview and wait for the 'Connected' event.
      await runWidgetPreview(
        tempDir: tempDir,
        expectedMessages: firstLaunchMessagesWeb,
        dtdUri: dtdUri,
      );
      await completer.future;
    });

    testUsingContext('can connect to an existing DevTools instance', () async {
      devtoolsLauncher = DevtoolsServerLauncher(
        processManager: processManager,
        logger: logger!,
        botDetector: const FakeBotDetector(true),
        artifacts: globals.artifacts!,
      );

      // Start a DevTools instance.
      final Uri devtoolsUri = (await devtoolsLauncher!.serve())!.uri!;

      // Start the widget preview and wait for the DevTools message.
      await runWidgetPreview(
        tempDir: tempDir,
        expectedMessages: <Pattern>[
          'The Flutter DevTools debugger and profiler on Chrome is available at: $devtoolsUri',
        ],
        devToolsServerAddress: devtoolsUri,
      );
    });

    testUsingContext("doesn't crash on flutter clean", () async {
      // Regression test for https://github.com/flutter/flutter/issues/175058.
      dtdLauncher = DtdLauncher(
        logger: logger!,
        artifacts: globals.artifacts!,
        processManager: globals.processManager,
      );

      // Start a DTD instance.
      final Uri dtdUri = await dtdLauncher!.launch();

      // Connect to it and listen to the WidgetPreviewScaffold stream.
      final DartToolingDaemon dtdConnection = await DartToolingDaemon.connect(dtdUri);
      final completer = Completer<void>();
      var firstConnection = true;
      dtdConnection.onEvent(WidgetPreviewDtdServices.kWidgetPreviewScaffoldStreamRoot).listen((
        DTDEvent event,
      ) {
        expect(event.stream, WidgetPreviewDtdServices.kWidgetPreviewScaffoldStreamRoot);
        expect(event.kind, 'Connected');
        if (firstConnection) {
          firstConnection = false;
          runFlutterClean(tempDir);
          // Restore dependencies so the Analysis Server and Dwds can resolve packages during Hot Restart.
          // Add --offline flag to prevent attempts to reach the network during CI.
          processManager.runSync(<String>[
            flutterBin,
            'pub',
            'get',
            '--offline',
          ], workingDirectory: tempDir.path);
          dtdConnection.call(
            WidgetPreviewDtdServices.kWidgetPreviewServiceRoot,
            WidgetPreviewDtdServices.kHotRestartPreviewer,
          );
          return;
        }
        // The second `Connected` event should come after the previewer is hot restarted successfully.
        completer.complete();
      });
      await dtdConnection.streamListen(WidgetPreviewDtdServices.kWidgetPreviewScaffoldStreamRoot);

      // Start the widget preview and wait for the 'Connected' event.
      final Stream<String> stdoutStream = await runWidgetPreview(
        tempDir: tempDir,
        expectedMessages: firstLaunchMessagesWeb,
        dtdUri: dtdUri,
      );

      final processExitCompleter = Completer<void>();
      final StreamSubscription<String> subscription = stdoutStream.listen(
        null,
        onDone: () => processExitCompleter.complete(),
      );

      // Race the connection completer against process exit for clearer failure messages.
      await Future.any(<Future<void>>[completer.future, processExitCompleter.future]);
      expect(
        processExitCompleter.isCompleted,
        isFalse,
        reason: 'Widget preview process exited unexpectedly after flutter clean.',
      );
      await subscription.cancel();
    });
  });
}