// Copyright (c) 2025, the Dart project authors. Please see the AUTHORS file
// for details. 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:convert';
import 'dart:io' as io;

import 'package:code_assets/code_assets.dart' show OS;
import 'package:dartdev/src/sdk_cache.dart';
import 'package:file/file.dart';
import 'package:file/memory.dart';
import 'package:hooks_runner/hooks_runner.dart' show Target;
import 'package:http/http.dart' as http;
import 'package:http/testing.dart' as http;
import 'package:test/test.dart';

const dartArchiveUri = 'https://storage.googleapis.com/dart-archive';

void main() {
  late SdkCache cache;
  late FileSystem fs;
  late StringBuffer stderr;
  late Map<String, http.Response> expectedRequests;
  late Map<String, io.ProcessResult> chmodRuns;

  setUp(() {
    // Make sure we support both path separators.
    fs = MemoryFileSystem(
        style: io.Platform.isWindows
            ? FileSystemStyle.windows
            : FileSystemStyle.posix);
    stderr = StringBuffer();
    expectedRequests = <String, http.Response>{};
    chmodRuns = <String, io.ProcessResult>{};

    cache = SdkCache(
        directory: fs.directory(Uri.file('/tmp/cache')).path,
        stderr: stderr,
        verbose: true,
        fs: fs,
        httpClient: http.MockClient((request) async {
          final key = '${request.method.toUpperCase()} ${request.url}';
          if (!expectedRequests.containsKey(key)) {
            throw Exception('Unexpected request $key');
          }
          return expectedRequests[key]!;
        }),
        chmod: (path) => chmodRuns[path]!);
  });

  group('resolveStage', () {
    test('Uses signed on macOS dev for executables', () {
      expect(
          SdkCache.resolveStage(
              channel: Channel.dev, isExecutable: true, hostOS: OS.macOS),
          Stage.signed);
    });

    test('Uses raw on macOS main for executables', () {
      expect(
          SdkCache.resolveStage(
              channel: Channel.main, isExecutable: true, hostOS: OS.macOS),
          Stage.raw);
    });

    test('Uses raw on macOS stable for non-executables', () {
      expect(
          SdkCache.resolveStage(
              channel: Channel.stable, isExecutable: false, hostOS: OS.macOS),
          Stage.raw);
    });
  });

  group('resolveVersion', () {
    test('Resolves release version', () async {
      final version = '3.4.4';
      final revision = '60465149414572c8ca189d8f65fdb39795c4b97d';
      final folder = await cache.resolveVersion(
          version: version,
          revision: revision,
          channelName: 'stable',
          host: Target.linuxArm64);
      expect(folder.version, version);
      expect(folder.revision, revision);
      expect(folder.channel, Channel.stable);
      expect(folder.fileUri('VERSION', stage: Stage.raw).toString(),
          '$dartArchiveUri/channels/stable/raw/hash/$revision/VERSION');
    });

    test('Falls back to the latest on empty main revision', () async {
      final version = '3.8.0';
      final revision = '';

      final latestVersion =
          '3.8.0-edge.7ee7416a62efb7eb23eb2a87eb34d2895559ba06';
      final latestRevision = '7ee7416a62efb7eb23eb2a87eb34d2895559ba06';

      expectedRequests['GET $dartArchiveUri/channels/main/raw/latest/VERSION'] =
          http.Response(
              json.encode(
                  {'version': latestVersion, 'revision': latestRevision}),
              io.HttpStatus.ok);
      final folder = await cache.resolveVersion(
          version: version,
          revision: revision,
          channelName: 'main',
          host: Target.macOSArm64);
      expect(folder.version, latestVersion);
      expect(folder.revision, latestRevision);
      expect(folder.channel, Channel.main);
    });

    test('Falls back to the latest on unknown main revision', () async {
      final version = '3.8.0';
      final revision = 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa';

      final latestVersion =
          '3.8.0-edge.7ee7416a62efb7eb23eb2a87eb34d2895559ba06';
      final latestRevision = '7ee7416a62efb7eb23eb2a87eb34d2895559ba06';

      expectedRequests.addAll({
        // Revision validation request.
        'HEAD $dartArchiveUri/channels/main/raw/hash/$revision/VERSION':
            http.Response('', io.HttpStatus.notFound),
        // Revision resolution request.
        'GET $dartArchiveUri/channels/main/raw/latest/VERSION': http.Response(
            json.encode({'version': latestVersion, 'revision': latestRevision}),
            io.HttpStatus.ok)
      });

      final folder = await cache.resolveVersion(
          version: version,
          revision: revision,
          channelName: 'main',
          host: Target.macOSArm64);
      expect(folder.version, latestVersion);
      expect(folder.revision, latestRevision);
      expect(folder.channel, Channel.main);
    });

    test('Reports unknown channel', () async {
      try {
        await cache.resolveVersion(
            channelName: 'wat',
            host: Target.linuxArm64,
            revision: '',
            version: '4.0');
        fail('expected to throw');
      } on ArgumentError catch (e) {
        expect(e.message, contains('Unsupported channel'));
      }
    });
  });

  group('ensureArtifact', () {
    test('Skips download when file exists', () async {
      final version = '3.4.4';
      final revision = '60465149414572c8ca189d8f65fdb39795c4b97d';

      final genSnapshotFile = fs.file(Uri.file(
          '/tmp/cache/$version/gen_snapshot_windows_arm64_linux_x64.exe'));
      genSnapshotFile.createSync(exclusive: true, recursive: true);

      final path = await cache.ensureGenSnapshot(
          archiveFolder: ArchiveFolder(
              channel: Channel.stable, version: version, revision: revision),
          host: Target.windowsArm64,
          target: Target.linuxX64);
      expect(path, genSnapshotFile.path);
    });

    test('Downloads signed gen_snapshot on macOS beta', () async {
      final version = '3.8.0-171.2.beta';
      final revision = '54cec4d7d36e7a5066770287998f425606a2f983';

      final archiveFolder = ArchiveFolder(
          channel: Channel.beta, version: version, revision: revision);
      expectedRequests['GET $dartArchiveUri/channels/beta/signed/hash/'
              '$revision/sdk/gen_snapshot_macos_arm64_linux_x64'] =
          http.Response('i am gen_snapshot', io.HttpStatus.ok);

      final path = await cache.ensureGenSnapshot(
          archiveFolder: archiveFolder,
          host: Target.macOSArm64,
          target: Target.linuxX64);

      expect(
          path,
          fs
              .file(Uri.file(
                  '/tmp/cache/$version/gen_snapshot_macos_arm64_linux_x64'))
              .path);
      expect(fs.file(path).readAsStringSync(), 'i am gen_snapshot');
    });

    test('Downloads raw dartaotruntime on macOS beta', () async {
      final version = '3.8.0-171.2.beta';
      final revision = '54cec4d7d36e7a5066770287998f425606a2f983';

      final archiveFolder = ArchiveFolder(
          channel: Channel.beta, version: version, revision: revision);
      expectedRequests['GET $dartArchiveUri/channels/beta/raw/hash/'
              '$revision/sdk/dartaotruntime_linux_x64'] =
          http.Response('i am dartaotruntime', io.HttpStatus.ok);

      final path = await cache.ensureDartAotRuntime(
          archiveFolder: archiveFolder,
          host: Target.macOSArm64,
          target: Target.linuxX64);

      expect(
          path,
          fs
              .file(Uri.file('/tmp/cache/$version/dartaotruntime_linux_x64'))
              .path);
      expect(fs.file(path).readAsStringSync(), 'i am dartaotruntime');
    });

    test('Downloads gen_snapshot.exe on Windows stable', () async {
      final version = '3.7.2';
      final channel = Channel.stable;
      final revision = '9594995093f642957b780603c6435d9e7a61b923';
      final binary = 'gen_snapshot_windows_x64_linux_x64.exe';

      final archiveFolder =
          ArchiveFolder(channel: channel, version: version, revision: revision);
      expectedRequests['GET $dartArchiveUri/channels/stable/raw/hash/'
              '$revision/sdk/$binary'] =
          http.Response('i am gen_snapshot', io.HttpStatus.ok);

      final path = await cache.ensureGenSnapshot(
          archiveFolder: archiveFolder,
          host: Target.windowsX64,
          target: Target.linuxX64);

      expect(path, fs.file(Uri.file('/tmp/cache/$version/$binary')).path);
      expect(fs.file(path).readAsStringSync(), 'i am gen_snapshot');
    });

    test('Downloads dartaotruntime on Windows stable', () async {
      final version = '3.7.2';
      final channel = Channel.stable;
      final revision = '9594995093f642957b780603c6435d9e7a61b923';
      final binary = 'dartaotruntime_linux_x64';

      final archiveFolder =
          ArchiveFolder(channel: channel, version: version, revision: revision);
      expectedRequests['GET $dartArchiveUri/channels/stable/raw/hash/'
              '$revision/sdk/$binary'] =
          http.Response('i am dartaotruntime', io.HttpStatus.ok);

      final path = await cache.ensureDartAotRuntime(
          archiveFolder: archiveFolder,
          host: Target.windowsX64,
          target: Target.linuxX64);

      expect(path, fs.file(Uri.file('/tmp/cache/$version/$binary')).path);
      expect(fs.file(path).readAsStringSync(), 'i am dartaotruntime');
    });
  });
}