Magisk/app/src/full/java/com/topjohnwu/magisk/components/InstallMethodDialog.java

package com.topjohnwu.magisk.components;

import android.Manifest;
import android.app.Activity;
import android.content.Intent;
import android.widget.Toast;

import com.androidnetworking.AndroidNetworking;
import com.androidnetworking.error.ANError;
import com.androidnetworking.interfaces.DownloadListener;
import com.google.android.material.snackbar.Snackbar;
import com.topjohnwu.magisk.Const;
import com.topjohnwu.magisk.Data;
import com.topjohnwu.magisk.FlashActivity;
import com.topjohnwu.magisk.R;
import com.topjohnwu.magisk.utils.Download;
import com.topjohnwu.magisk.utils.Utils;

import java.util.List;

import androidx.appcompat.app.AlertDialog;

class InstallMethodDialog extends AlertDialog.Builder {

    InstallMethodDialog(BaseActivity activity, List<String> options) {
        super(activity);
        setTitle(R.string.select_method);
        setItems(options.toArray(new String [0]), (dialog, idx) -> {
            Intent intent;
            switch (idx) {
                case 1:
                    patchBoot(activity);
                    break;
                case 0:
                    downloadOnly(activity);
                    break;
                case 2:
                    intent = new Intent(activity, Data.classMap.get(FlashActivity.class))
                            .putExtra(Const.Key.FLASH_ACTION, Const.Value.FLASH_MAGISK);
                    activity.startActivity(intent);
                    break;
                case 3:
                    installInactiveSlot(activity);
                    break;
                default:
            }
        });
    }

    private void patchBoot(BaseActivity a) {
        Utils.toast(R.string.boot_file_patch_msg, Toast.LENGTH_LONG);
        Intent intent = new Intent(Intent.ACTION_GET_CONTENT).setType("*/*");
        a.runWithPermission(new String[]{Manifest.permission.WRITE_EXTERNAL_STORAGE}, () ->
                a.startActivityForResult(intent, Const.ID.SELECT_BOOT,
                        (requestCode, resultCode, data) -> {
                            if (requestCode == Const.ID.SELECT_BOOT &&
                                    resultCode == Activity.RESULT_OK && data != null) {
                                Intent i = new Intent(a, Data.classMap.get(FlashActivity.class))
                                        .putExtra(Const.Key.FLASH_SET_BOOT, data.getData())
                                        .putExtra(Const.Key.FLASH_ACTION, Const.Value.PATCH_BOOT);
                                a.startActivity(i);
                            }
                        })
        );
    }

    private void downloadOnly(BaseActivity a) {
        a.runWithPermission(new String[] {Manifest.permission.WRITE_EXTERNAL_STORAGE}, () -> {
            String filename = Utils.fmt("Magisk-v%s(%d).zip",
                    Data.remoteMagiskVersionString, Data.remoteMagiskVersionCode);
            NotificationProgress progress = new NotificationProgress(filename);
            AndroidNetworking
                    .download(Data.magiskLink, Download.EXTERNAL_PATH.getPath(), filename)
                    .build()
                    .setDownloadProgressListener(progress)
                    .startDownload(new DownloadListener() {
                        @Override
                        public void onDownloadComplete() {
                            progress.defaultDone();
                            SnackbarMaker.make(a,
                                a.getString(R.string.internal_storage, "/Download/" + filename),
                                Snackbar.LENGTH_LONG).show();
                        }

                        @Override
                        public void onError(ANError anError) {}
                    });
        });
    }

    private void installInactiveSlot(BaseActivity activity) {
        new CustomAlertDialog(activity)
                .setTitle(R.string.warning)
                .setMessage(R.string.install_inactive_slot_msg)
                .setCancelable(true)
                .setPositiveButton(R.string.yes, (d, i) -> {
                    Intent intent = new Intent(activity, Data.classMap.get(FlashActivity.class))
                            .putExtra(Const.Key.FLASH_ACTION, Const.Value.FLASH_INACTIVE_SLOT);
                    activity.startActivity(intent);
                })
                .setNegativeButton(R.string.no_thanks, null)
                .show();
    }
}