Magisk/app/src/full/java/com/topjohnwu/magisk/FlashActivity.java

package com.topjohnwu.magisk;

import android.Manifest;
import android.content.Intent;
import android.net.Uri;
import android.os.Bundle;
import android.text.TextUtils;
import android.view.View;
import android.widget.Button;
import android.widget.LinearLayout;
import android.widget.ScrollView;
import android.widget.TextView;
import android.widget.Toast;

import com.topjohnwu.magisk.asyncs.FlashZip;
import com.topjohnwu.magisk.asyncs.InstallMagisk;
import com.topjohnwu.magisk.components.BaseActivity;
import com.topjohnwu.magisk.utils.Download;
import com.topjohnwu.magisk.utils.RootUtils;
import com.topjohnwu.magisk.utils.Utils;
import com.topjohnwu.superuser.CallbackList;
import com.topjohnwu.superuser.Shell;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;
import java.util.Locale;

import androidx.appcompat.app.ActionBar;
import androidx.appcompat.widget.Toolbar;
import butterknife.BindView;
import butterknife.OnClick;

public class FlashActivity extends BaseActivity {

    @BindView(R.id.toolbar) Toolbar toolbar;
    @BindView(R.id.txtLog) TextView flashLogs;
    @BindView(R.id.button_panel) public LinearLayout buttonPanel;
    @BindView(R.id.reboot) public Button reboot;
    @BindView(R.id.scrollView) ScrollView sv;

    private List<String> logs;

    @OnClick(R.id.reboot)
    void reboot() {
        Shell.su("/system/bin/reboot").submit();
    }

    @OnClick(R.id.save_logs)
    void saveLogs() {
        runWithPermission(new String[] {Manifest.permission.WRITE_EXTERNAL_STORAGE}, () -> {
            Calendar now = Calendar.getInstance();
            String filename = String.format(Locale.US,
                    "magisk_install_log_%04d%02d%02d_%02d%02d%02d.log",
                    now.get(Calendar.YEAR), now.get(Calendar.MONTH) + 1,
                    now.get(Calendar.DAY_OF_MONTH), now.get(Calendar.HOUR_OF_DAY),
                    now.get(Calendar.MINUTE), now.get(Calendar.SECOND));

            File logFile = new File(Download.EXTERNAL_PATH, filename);
            try (FileWriter writer = new FileWriter(logFile)) {
                for (String s : logs) {
                    writer.write(s);
                    writer.write('\n');
                }
            } catch (IOException e) {
                e.printStackTrace();
                return;
            }
            Utils.toast(logFile.getPath(), Toast.LENGTH_LONG);
        });
    }

    @Override
    public int getDarkTheme() {
        return R.style.AppTheme_StatusBar_Dark;
    }

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_flash);
        new FlashActivity_ViewBinding(this);

        setSupportActionBar(toolbar);
        ActionBar ab = getSupportActionBar();
        if (ab != null) {
            ab.setTitle(R.string.flashing);
        }
        setFloating();
        setFinishOnTouchOutside(false);
        if (!Shell.rootAccess())
            reboot.setVisibility(View.GONE);

        logs = new ArrayList<>();
        CallbackList<String> console = new CallbackList<String>(new ArrayList<>()) {

            private void updateUI() {
                flashLogs.setText(TextUtils.join("\n", this));
                sv.postDelayed(() -> sv.fullScroll(ScrollView.FOCUS_DOWN), 10);
            }

            @Override
            public void onAddElement(String s) {
                logs.add(s);
                updateUI();
            }

            @Override
            public String set(int i, String s) {
                String ret = super.set(i, s);
                Data.mainHandler.post(this::updateUI);
                return ret;
            }
        };

        // We must receive a Uri of the target zip
        Intent intent = getIntent();
        Uri uri = intent.getData();

        switch (intent.getStringExtra(Const.Key.FLASH_ACTION)) {
            case Const.Value.FLASH_ZIP:
                new FlashZip(this, uri, console, logs).exec();
                break;
            case Const.Value.UNINSTALL:
                new UninstallMagisk(this, uri, console, logs).exec();
                break;
            case Const.Value.FLASH_MAGISK:
                new InstallMagisk(this, console, logs, InstallMagisk.DIRECT_MODE).exec();
                break;
            case Const.Value.FLASH_INACTIVE_SLOT:
                new InstallMagisk(this, console, logs, InstallMagisk.SECOND_SLOT_MODE).exec();
                break;
            case Const.Value.PATCH_BOOT:
                runWithPermission(new String[] {Manifest.permission.WRITE_EXTERNAL_STORAGE},
                        () -> new InstallMagisk(this, console, logs,
                                intent.getParcelableExtra(Const.Key.FLASH_SET_BOOT)).exec());
                break;
        }
    }

    @Override
    public void onBackPressed() {
        // Prevent user accidentally press back button
    }

    private static class UninstallMagisk extends FlashZip {

        private UninstallMagisk(BaseActivity context, Uri uri, List<String> console, List<String> logs) {
            super(context, uri, console, logs);
        }

        @Override
        protected void onPostExecute(Integer result) {
            if (result == 1) {
                Data.mainHandler.postDelayed(() ->
                        RootUtils.uninstallPkg(getActivity().getPackageName()), 3000);
            } else {
                super.onPostExecute(result);
            }
        }
    }
}