// 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 'package:code_assets/code_assets.dart';
import 'package:hooks_runner/hooks_runner.dart';

import '../../../base/file_system.dart';
import '../../../build_info.dart';
import '../macos/native_assets_host.dart';
import '../native_assets.dart';

// TODO(dcharkes): Fetch minimum iOS version from somewhere. https://github.com/flutter/flutter/issues/145104
const targetIOSVersion = 13;

IOSSdk getIOSSdk(EnvironmentType environmentType) {
  return switch (environmentType) {
    EnvironmentType.physical => IOSSdk.iPhoneOS,
    EnvironmentType.simulator => IOSSdk.iPhoneSimulator,
  };
}

/// Extract the [Architecture] from a [DarwinArch].
Architecture getNativeIOSArchitecture(DarwinArch darwinArch) {
  return switch (darwinArch) {
    DarwinArch.armv7 => Architecture.arm,
    DarwinArch.arm64 => Architecture.arm64,
    DarwinArch.x86_64 => Architecture.x64,
  };
}

Map<KernelAssetPath, List<FlutterCodeAsset>> fatAssetTargetLocationsIOS(
  List<FlutterCodeAsset> nativeAssets,
) {
  final alreadyTakenNames = <String>{};
  final result = <KernelAssetPath, List<FlutterCodeAsset>>{};
  final idToPath = <String, KernelAssetPath>{};
  for (final asset in nativeAssets) {
    // Use same target path for all assets with the same id.
    final String assetId = asset.codeAsset.id;
    final KernelAssetPath path =
        idToPath[assetId] ?? _targetLocationIOS(asset, alreadyTakenNames).path;
    idToPath[assetId] = path;
    result[path] ??= <FlutterCodeAsset>[];
    result[path]!.add(asset);
  }
  return result;
}

Map<FlutterCodeAsset, KernelAsset> assetTargetLocationsIOS(List<FlutterCodeAsset> nativeAssets) {
  final alreadyTakenNames = <String>{};
  final idToPath = <String, KernelAssetPath>{};
  final result = <FlutterCodeAsset, KernelAsset>{};
  for (final asset in nativeAssets) {
    final String assetId = asset.codeAsset.id;
    final KernelAssetPath path =
        idToPath[assetId] ?? _targetLocationIOS(asset, alreadyTakenNames).path;
    idToPath[assetId] = path;
    result[asset] = KernelAsset(id: assetId, target: asset.target, path: path);
  }
  return result;
}

KernelAsset _targetLocationIOS(FlutterCodeAsset asset, Set<String> alreadyTakenNames) {
  final LinkMode linkMode = asset.codeAsset.linkMode;
  final KernelAssetPath kernelAssetPath;
  switch (linkMode) {
    case DynamicLoadingSystem _:
      kernelAssetPath = KernelAssetSystemPath(linkMode.uri);
    case LookupInExecutable _:
      kernelAssetPath = KernelAssetInExecutable();
    case LookupInProcess _:
      kernelAssetPath = KernelAssetInProcess();
    case DynamicLoadingBundled _:
      final String fileName = asset.codeAsset.file!.pathSegments.last;
      kernelAssetPath = KernelAssetAbsolutePath(frameworkUri(fileName, alreadyTakenNames));
    default:
      throw Exception('Unsupported asset link mode $linkMode in asset $asset');
  }
  return KernelAsset(id: asset.codeAsset.id, target: asset.target, path: kernelAssetPath);
}

/// Copies native assets into a framework per dynamic library.
///
/// For `flutter run -release` a multi-architecture solution is needed. So,
/// `lipo` is used to combine all target architectures into a single file.
///
/// The install name is set so that it matches with the place it will
/// be bundled in the final app. Install names that are referenced in dependent
/// libraries are updated to match the new install name, so that the referenced
/// library can be found by the dynamic linker.
///
/// Code signing is also done here, so that it doesn't have to be done in
/// in xcode_backend.dart.
Future<void> copyNativeCodeAssetsIOS(
  Uri buildUri,
  Map<KernelAssetPath, List<FlutterCodeAsset>> assetTargetLocations,
  String? codesignIdentity,
  BuildMode buildMode,
  FileSystem fileSystem,
) async {
  assert(assetTargetLocations.isNotEmpty);
  final oldToNewInstallNames = <String, String>{};
  final dylibs = <(File, String, Directory)>[];

  for (final MapEntry<KernelAssetPath, List<FlutterCodeAsset>> assetMapping
      in assetTargetLocations.entries) {
    final Uri target = (assetMapping.key as KernelAssetAbsolutePath).uri;
    final sources = <File>[
      for (final FlutterCodeAsset source in assetMapping.value)
        fileSystem.file(source.codeAsset.file),
    ];
    final Uri targetUri = buildUri.resolveUri(target);
    final File dylibFile = fileSystem.file(targetUri);
    final Directory frameworkDir = dylibFile.parent;
    if (!await frameworkDir.exists()) {
      await frameworkDir.create(recursive: true);
    }
    await lipoDylibs(dylibFile, sources);

    final String dylibFileName = dylibFile.basename;
    final newInstallName = '@rpath/$dylibFileName.framework/$dylibFileName';
    final Set<String> oldInstallNames = await getInstallNamesDylib(dylibFile);
    for (final oldInstallName in oldInstallNames) {
      oldToNewInstallNames[oldInstallName] = newInstallName;
    }
    dylibs.add((dylibFile, newInstallName, frameworkDir));

    await createInfoPlist(
      targetUri.pathSegments.last,
      frameworkDir,
      minimumIOSVersion: '$targetIOSVersion.0',
    );
  }

  for (final (File dylibFile, String newInstallName, Directory frameworkDir) in dylibs) {
    await setInstallNamesDylib(dylibFile, newInstallName, oldToNewInstallNames);
    await codesignDylib(codesignIdentity, buildMode, frameworkDir);
  }
}