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_archive/flutter_archive.dart';
import 'package:injectable/injectable.dart';
import 'package:path_provider/path_provider.dart';
import 'package:revanced_manager/models/patch.dart';
import 'package:revanced_manager/models/patched_application.dart';
import 'package:revanced_manager/services/manager_api.dart';
import 'package:revanced_manager/services/root_api.dart';
import 'package:revanced_manager/utils/string.dart';
import 'package:share_extend/share_extend.dart';

@lazySingleton
class PatcherAPI {
  static const patcherChannel = MethodChannel(
    'app.revanced.manager/patcher',
  );
  final ManagerAPI _managerAPI = ManagerAPI();
  final RootAPI _rootAPI = RootAPI();
  Directory? _tmpDir;
  Directory? _workDir;
  Directory? _cacheDir;
  File? _zipPatchBundleFile;
  File? _integrations;
  File? _inputFile;
  File? _patchedFile;
  File? _outFile;

  Future<void> initPatcher() async {
    _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();
  }

  Future<void> loadPatches() async {
    if (_cacheDir == null) {
      await initPatcher();
    }
    if (_zipPatchBundleFile == null) {
      File? patchBundleDexFile = await _managerAPI.downloadPatches('.dex');
      File? patchBundleJarFile = await _managerAPI.downloadPatches('.jar');
      if (patchBundleDexFile != null && patchBundleJarFile != null) {
        await joinPatchBundleFiles(patchBundleDexFile, patchBundleJarFile);
        if (_zipPatchBundleFile != null) {
          await patcherChannel.invokeMethod<bool>(
            'loadPatches',
            {
              'zipPatchBundlePath': _zipPatchBundleFile!.path,
              'cacheDirPath': _cacheDir!.path,
            },
          );
        }
      }
    }
  }

  Future<List<ApplicationWithIcon>> getFilteredInstalledApps() async {
    List<ApplicationWithIcon> filteredPackages = [];
    if (_zipPatchBundleFile != null) {
      try {
        List<String>? patchesPackages = await patcherChannel
            .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 {
        return List.empty();
      }
    }
    return filteredPackages;
  }

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

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

  Future<void> mergeIntegrations(bool mergeIntegrations) async {
    if (mergeIntegrations) {
      _integrations = await _managerAPI.downloadIntegrations('.apk');
    } else {
      _integrations = null;
    }
  }

  Future<void> runPatcher(
    String originalFilePath,
    List<Patch> selectedPatches,
    bool mergeIntegrations,
    bool resourcePatching,
  ) async {
    await patcherChannel.invokeMethod(
      'runPatcher',
      {
        'originalFilePath': originalFilePath,
        'inputFilePath': _inputFile!.path,
        'patchedFilePath': _patchedFile!.path,
        'outFilePath': _outFile!.path,
        'integrationsPath': _integrations != null ? _integrations!.path : '',
        'selectedPatches': selectedPatches.map((p) => p.name).toList(),
        'cacheDirPath': _cacheDir!.path,
        'mergeIntegrations': mergeIntegrations,
        'resourcePatching': resourcePatching,
      },
    );
  }

  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 await DeviceApps.isAppInstalled(patchedApp.packageName);
        }
      } 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;
    }
  }

  Future<bool> checkOldPatch(PatchedApplication patchedApp) async {
    if (patchedApp.isRooted) {
      return await _rootAPI.checkApp(patchedApp.packageName);
    }
    return false;
  }

  Future<void> deleteOldPatch(PatchedApplication patchedApp) async {
    if (patchedApp.isRooted) {
      await _rootAPI.deleteApp(patchedApp.packageName, patchedApp.apkFilePath);
    }
  }

  Future<void> joinPatchBundleFiles(
    File patchBundleDexFile,
    File patchBundleJarFile,
  ) async {
    _zipPatchBundleFile = File('${_workDir!.path}/join.zip');
    Directory joinDir = Directory('${_cacheDir!.path}/join');
    try {
      await ZipFile.extractToDirectory(
        zipFile: patchBundleJarFile,
        destinationDir: joinDir,
      );
      patchBundleDexFile.copySync('${joinDir.path}/classes.dex');
      await ZipFile.createFromDirectory(
        sourceDir: joinDir,
        zipFile: _zipPatchBundleFile!,
        recurseSubDirs: true,
      );
    } on Exception {
      _zipPatchBundleFile = null;
    }
  }
}