revanced-manager/lib/services/patcher_api.dart

import 'dart:io';
import 'package:app_installer/app_installer.dart';
import 'package:device_apps/device_apps.dart';
import 'package:flutter/services.dart';
import 'package:flutter_cache_manager/flutter_cache_manager.dart';
import 'package:injectable/injectable.dart';
import 'package:path_provider/path_provider.dart';
import 'package:revanced_manager/app/app.locator.dart';
import 'package:revanced_manager/models/patch.dart';
import 'package:revanced_manager/models/patched_application.dart';
import 'package:revanced_manager/services/github_api.dart';
import 'package:revanced_manager/services/root_api.dart';
import 'package:revanced_manager/ui/views/installer/installer_viewmodel.dart';
import 'package:revanced_manager/utils/string.dart';
import 'package:share_extend/share_extend.dart';

@lazySingleton
class PatcherAPI {
  static const platform = MethodChannel('app.revanced.manager/patcher');
  final GithubAPI githubAPI = GithubAPI();
  final RootAPI rootAPI = RootAPI();
  final List<ApplicationWithIcon> _filteredPackages = [];
  final Map<String, List<Patch>> _filteredPatches = <String, List<Patch>>{};
  Directory? _tmpDir;
  Directory? _workDir;
  Directory? _cacheDir;
  File? _patchBundleFile;
  File? _integrations;
  File? _inputFile;
  File? _patchedFile;
  File? _outFile;

  Future<dynamic> handlePlatformChannelMethods() async {
    platform.setMethodCallHandler((call) async {
      switch (call.method) {
        case 'updateInstallerLog':
          var message = call.arguments<String>('message');
          locator<InstallerViewModel>().addLog(message);
          return 'OK';
      }
    });
  }

  Future<bool?> loadPatches() async {
    if (_patchBundleFile == null) {
      String? dexFileUrl =
          await githubAPI.latestRelease('revanced', 'revanced-patches');
      if (dexFileUrl != null && dexFileUrl.isNotEmpty) {
        try {
          _patchBundleFile =
              await DefaultCacheManager().getSingleFile(dexFileUrl);
          return await platform.invokeMethod<bool>(
            'loadPatches',
            {
              'pathBundlesPaths': <String>[_patchBundleFile!.absolute.path],
            },
          );
        } on Exception {
          _patchBundleFile = null;
          return false;
        }
      }
      return false;
    }
    return true;
  }

  Future<List<ApplicationWithIcon>> getFilteredInstalledApps() async {
    if (_patchBundleFile != null && _filteredPackages.isEmpty) {
      try {
        List<String>? patchesPackages =
            await platform.invokeListMethod<String>('getCompatiblePackages');
        if (patchesPackages != null) {
          for (String package in patchesPackages) {
            try {
              ApplicationWithIcon? app = await DeviceApps.getApp(package, true)
                  as ApplicationWithIcon?;
              if (app != null) {
                _filteredPackages.add(app);
              }
            } catch (e) {
              continue;
            }
          }
        }
      } on Exception {
        _filteredPackages.clear();
        return List.empty();
      }
    }
    return _filteredPackages;
  }

  Future<List<Patch>?> getFilteredPatches(
    PatchedApplication? selectedApp,
  ) async {
    if (_patchBundleFile != null && selectedApp != null) {
      if (_filteredPatches[selectedApp.packageName] == null ||
          _filteredPatches[selectedApp.packageName]!.isEmpty) {
        _filteredPatches[selectedApp.packageName] = [];
        try {
          var patches = await platform.invokeListMethod<Map<dynamic, dynamic>>(
            'getFilteredPatches',
            {
              'targetPackage': selectedApp.packageName,
              'targetVersion': selectedApp.version,
              'ignoreVersion': true,
            },
          );
          if (patches != null) {
            for (var patch in patches) {
              if (!_filteredPatches[selectedApp.packageName]!
                  .any((element) => element.name == patch['name'])) {
                _filteredPatches[selectedApp.packageName]!.add(
                  Patch(
                    name: patch['name'],
                    simpleName: (patch['name'] as String)
                        .replaceAll('-', ' ')
                        .split('-')
                        .join(' ')
                        .toTitleCase(),
                    version: patch['version'] ?? '?.?.?',
                    description: patch['description'] ?? 'N/A',
                  ),
                );
              }
            }
          }
        } on Exception {
          _filteredPatches[selectedApp.packageName]!.clear();
          return List.empty();
        }
      }
    } else {
      return List.empty();
    }
    return _filteredPatches[selectedApp.packageName];
  }

  Future<File?> downloadIntegrations() async {
    String? apkFileUrl =
        await githubAPI.latestRelease('revanced', 'revanced-integrations');
    if (apkFileUrl != null && apkFileUrl.isNotEmpty) {
      return await DefaultCacheManager().getSingleFile(apkFileUrl);
    }
    return null;
  }

  Future<bool?> initPatcher() async {
    try {
      _integrations = await downloadIntegrations();
      if (_integrations != null) {
        _tmpDir = await getTemporaryDirectory();
        _workDir = _tmpDir!.createTempSync('tmp-');
        _inputFile = File('${_workDir!.path}/base.apk');
        _patchedFile = File('${_workDir!.path}/patched.apk');
        _outFile = File('${_workDir!.path}/out.apk');
        _cacheDir = Directory('${_workDir!.path}/cache');
        _cacheDir!.createSync();
        return true;
      }
    } on Exception {
      return false;
    }
    return false;
  }

  Future<bool?> copyInputFile(String originalFilePath) async {
    if (_inputFile != null) {
      try {
        return await platform.invokeMethod<bool>(
          'copyInputFile',
          {
            'originalFilePath': originalFilePath,
            'inputFilePath': _inputFile!.path,
          },
        );
      } on Exception {
        return false;
      }
    }
    return false;
  }

  Future<bool?> createPatcher(bool resourcePatching) async {
    if (_inputFile != null && _cacheDir != null) {
      try {
        return await platform.invokeMethod<bool>(
          'createPatcher',
          {
            'inputFilePath': _inputFile!.path,
            'cacheDirPath': _cacheDir!.path,
            'resourcePatching': resourcePatching,
          },
        );
      } on Exception {
        return false;
      }
    }
    return false;
  }

  Future<bool?> mergeIntegrations() async {
    try {
      return await platform.invokeMethod<bool>(
        'mergeIntegrations',
        {
          'integrationsPath': _integrations!.path,
        },
      );
    } on Exception {
      return false;
    }
  }

  Future<bool?> applyPatches(List<Patch> selectedPatches) async {
    try {
      return await platform.invokeMethod<bool>(
        'applyPatches',
        {
          'selectedPatches': selectedPatches.map((e) => e.name).toList(),
        },
      );
    } on Exception {
      return false;
    }
  }

  Future<bool?> repackPatchedFile() async {
    if (_inputFile != null && _patchedFile != null) {
      try {
        return await platform.invokeMethod<bool>(
          'repackPatchedFile',
          {
            'inputFilePath': _inputFile!.path,
            'patchedFilePath': _patchedFile!.path,
          },
        );
      } on Exception {
        return false;
      }
    }
    return false;
  }

  Future<bool?> signPatchedFile() async {
    if (_patchedFile != null && _outFile != null) {
      try {
        return await platform.invokeMethod<bool>(
          'signPatchedFile',
          {
            'patchedFilePath': _patchedFile!.path,
            'outFilePath': _outFile!.path,
          },
        );
      } on Exception {
        return false;
      }
    }
    return false;
  }

  Future<bool> installPatchedFile(PatchedApplication patchedApp) async {
    if (_outFile != null) {
      try {
        if (patchedApp.isRooted && !patchedApp.isFromStorage) {
          return rootAPI.installApp(
            patchedApp.packageName,
            patchedApp.apkFilePath,
            _outFile!.path,
          );
        } else {
          await AppInstaller.installApk(_outFile!.path);
          return true;
        }
      } on Exception {
        return false;
      }
    }
    return false;
  }

  void cleanPatcher() {
    if (_workDir != null) {
      _workDir!.deleteSync(recursive: true);
    }
  }

  bool sharePatchedFile(String appName, String version) {
    if (_outFile != null) {
      String path = _tmpDir!.path;
      String prefix = appName.toLowerCase().replaceAll(' ', '-');
      String sharePath = '$path/$prefix-revanced_v$version.apk';
      File share = _outFile!.copySync(sharePath);
      ShareExtend.share(share.path, 'file');
      return true;
    } else {
      return false;
    }
  }
}