Magisk/app/src/main/java/com/topjohnwu/magisk/MagiskFragment.java

package com.topjohnwu.magisk;

import android.app.Fragment;
import android.content.Intent;
import android.content.SharedPreferences;
import android.net.Uri;
import android.os.Build;
import android.os.Bundle;
import android.preference.PreferenceManager;
import android.support.annotation.Nullable;
import android.support.v4.content.FileProvider;
import android.support.v4.widget.SwipeRefreshLayout;
import android.support.v7.app.AlertDialog;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.ImageView;
import android.widget.ProgressBar;
import android.widget.TextView;

import com.topjohnwu.magisk.receivers.DownloadReceiver;
import com.topjohnwu.magisk.utils.Async;
import com.topjohnwu.magisk.utils.Logger;
import com.topjohnwu.magisk.utils.Shell;
import com.topjohnwu.magisk.utils.Utils;
import com.topjohnwu.magisk.utils.ZipUtils;

import java.io.File;
import java.util.List;

import butterknife.BindColor;
import butterknife.BindView;
import butterknife.ButterKnife;

public class MagiskFragment extends Fragment {

    public static int remoteAppVersionCode = -1;
    public static double magiskVersion, remoteMagiskVersion = -1;
    public static String magiskVersionString, magiskLink, magiskChangelog, appLink, appChangelog, remoteAppVersion;

    @BindView(R.id.swipeRefreshLayout) SwipeRefreshLayout mSwipeRefreshLayout;

    @BindView(R.id.magiskStatusView) View magiskStatusView;
    @BindView(R.id.magisk_status_container) View magiskStatusContainer;
    @BindView(R.id.magisk_status_icon) ImageView magiskStatusIcon;
    @BindView(R.id.magisk_version) TextView magiskVersionText;

    @BindView(R.id.app_updateView) View appUpdateView;
    @BindView(R.id.app_check_updates_container) View appCheckUpdatesContainer;
    @BindView(R.id.app_check_updates_icon) ImageView appCheckUpdatesIcon;
    @BindView(R.id.app_check_updates_status) TextView appCheckUpdatesStatus;
    @BindView(R.id.app_check_updates_progress) ProgressBar appCheckUpdatesProgress;

    @BindView(R.id.magisk_updateView) View magiskUpdateView;
    @BindView(R.id.magisk_check_updates_container) View magiskCheckUpdatesContainer;
    @BindView(R.id.magisk_check_updates_icon) ImageView magiskCheckUpdatesIcon;
    @BindView(R.id.magisk_check_updates_status) TextView magiskCheckUpdatesStatus;
    @BindView(R.id.magisk_check_updates_progress) ProgressBar magiskCheckUpdatesProgress;

    @BindColor(R.color.green500) int colorOK;
    @BindColor(R.color.yellow500) int colorWarn;
    @BindColor(R.color.grey500) int colorNeutral;
    @BindColor(R.color.blue500) int colorInfo;

    @BindColor(android.R.color.transparent) int trans;

    int statusOK = R.drawable.ic_check_circle;
    int statusUnknown = R.drawable.ic_help;

    private AlertDialog.Builder builder;

    private SharedPreferences prefs;
    private SharedPreferences.OnSharedPreferenceChangeListener listener;

    private AlertDialog.OnClickListener flashMagisk = (dialogInterface, i) -> Utils.dlAndReceive(
            getActivity(),
            new DownloadReceiver() {
                @Override
                public void task(Uri uri) {
                    new Async.FlashZIP(mContext, uri, mFilename) {
                        @Override
                        protected boolean unzipAndCheck() {
                            publishProgress(mContext.getString(R.string.zip_install_unzip_zip_msg));
                            // We might not have busybox yet, unzip with Java
                            // We will have complete busybox after Magisk installation
                            ZipUtils.unzip(mCachedFile, new File(mCachedFile.getParent(), "magisk"));
                            Shell.su(
                                    "mkdir -p " + Async.TMP_FOLDER_PATH + "/magisk",
                                    "cp -af " + mCachedFile.getParent() + "/magisk/. " + Async.TMP_FOLDER_PATH + "/magisk"
                            );
                            return super.unzipAndCheck();
                        }

                        @Override
                        protected void done() {
                            Shell.su("setprop magisk.version " + String.valueOf(remoteMagiskVersion));
                            super.done();
                        }
                    }.exec();
                }
            },
            magiskLink,
            "Magisk-v" + String.valueOf(remoteMagiskVersion) + ".zip");

    private AlertDialog.OnClickListener installMagiskApk = (dialogInterface, i) -> Utils.dlAndReceive(
            getActivity(),
            new DownloadReceiver() {
                @Override
                public void task(Uri uri) {
                    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
                        Intent install = new Intent(Intent.ACTION_INSTALL_PACKAGE);
                        install.setFlags(Intent.FLAG_GRANT_READ_URI_PERMISSION);
                        Uri content = FileProvider.getUriForFile(getActivity(), "com.topjohnwu.magisk.provider", new File(uri.getPath()));
                        install.setData(content);
                        mContext.startActivity(install);
                    } else {
                        Intent install = new Intent(Intent.ACTION_VIEW);
                        install.setDataAndType(uri, "application/vnd.android.package-archive");
                        install.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
                        mContext.startActivity(install);
                    }
                }
            },
            appLink,
            "MagiskManager-v" + remoteAppVersion + ".apk");

    static {
        updateMagiskVersion();
    }

    @Nullable
    @Override
    public View onCreateView(LayoutInflater inflater, @Nullable ViewGroup container, @Nullable Bundle savedInstanceState) {
        View v = inflater.inflate(R.layout.magisk_fragment, container, false);
        ButterKnife.bind(this, v);

        prefs = PreferenceManager.getDefaultSharedPreferences(getActivity());

        mSwipeRefreshLayout.setOnRefreshListener(() -> {
            prefs.edit().putBoolean("update_check_done", false).apply();

            appCheckUpdatesContainer.setBackgroundColor(trans);
            magiskCheckUpdatesContainer.setBackgroundColor(trans);

            appCheckUpdatesIcon.setImageResource(0);
            magiskCheckUpdatesIcon.setImageResource(0);

            appCheckUpdatesStatus.setText(null);
            magiskCheckUpdatesStatus.setText(null);

            appCheckUpdatesProgress.setVisibility(View.VISIBLE);
            magiskCheckUpdatesProgress.setVisibility(View.VISIBLE);

            new Async.CheckUpdates(prefs).exec();
        });

        if (prefs.getBoolean("update_check_done", false)) {
            updateUI();
        }

        listener = (pref, s) -> {
            if (s.equals("update_check_done")) {
                if (pref.getBoolean(s, false)) {
                    Logger.dev("MagiskFragment: UI refresh triggered");
                    updateMagiskVersion();
                    updateUI();
                }
            }
        };

        return v;
    }

    @Override
    public void onResume() {
        super.onResume();
        prefs.registerOnSharedPreferenceChangeListener(listener);
    }

    @Override
    public void onDestroy() {
        super.onDestroy();
        prefs.unregisterOnSharedPreferenceChangeListener(listener);
    }

    private static void updateMagiskVersion() {
        List<String> ret = Shell.sh("getprop magisk.version");
        if (ret.get(0).length() == 0) {
            magiskVersion = -1;
        } else try {
            magiskVersionString = ret.get(0);
            magiskVersion = Double.parseDouble(ret.get(0));
        } catch (NumberFormatException e) {
            // Custom version don't need to receive updates
            magiskVersion = Double.POSITIVE_INFINITY;
        }
    }


    private void updateUI() {
        String theme = prefs.getString("theme", "");
        if (theme.equals("Dark")) {
            builder = new AlertDialog.Builder(getActivity(), R.style.AlertDialog_dh);
        } else {
            builder = new AlertDialog.Builder(getActivity());
        }

        if (magiskVersion == -1) {
            magiskStatusContainer.setBackgroundColor(colorNeutral);
            magiskStatusIcon.setImageResource(statusUnknown);

            magiskVersionText.setTextColor(colorNeutral);
            magiskVersionText.setText(R.string.magisk_version_error);
        } else {
            magiskStatusContainer.setBackgroundColor(colorOK);
            magiskStatusIcon.setImageResource(statusOK);

            magiskVersionText.setText(getString(R.string.magisk_version, magiskVersionString));
            magiskVersionText.setTextColor(colorOK);
        }

        if (remoteMagiskVersion == -1) {
            appCheckUpdatesContainer.setBackgroundColor(colorWarn);
            magiskCheckUpdatesContainer.setBackgroundColor(colorWarn);

            appCheckUpdatesIcon.setImageResource(R.drawable.ic_warning);
            magiskCheckUpdatesIcon.setImageResource(R.drawable.ic_warning);

            appCheckUpdatesStatus.setText(R.string.cannot_check_updates);
            appCheckUpdatesStatus.setTextColor(colorWarn);
            magiskCheckUpdatesStatus.setText(R.string.cannot_check_updates);
            magiskCheckUpdatesStatus.setTextColor(colorWarn);
        } else {
            if (remoteMagiskVersion > magiskVersion) {
                magiskCheckUpdatesContainer.setBackgroundColor(colorInfo);
                magiskCheckUpdatesIcon.setImageResource(R.drawable.ic_file_download);
                magiskCheckUpdatesStatus.setText(getString(R.string.magisk_update_available, String.valueOf(remoteMagiskVersion)));
                magiskCheckUpdatesStatus.setTextColor(colorInfo);
                magiskUpdateView.setOnClickListener(view -> builder
                        .setTitle(getString(R.string.update_title, getString(R.string.magisk)))
                        .setMessage(getString(R.string.update_msg, getString(R.string.magisk), String.valueOf(remoteMagiskVersion), magiskChangelog))
                        .setCancelable(true)
                        .setPositiveButton(R.string.download_install, flashMagisk)
                        .setNegativeButton(R.string.no_thanks, null)
                        .show());

            } else {
                magiskCheckUpdatesContainer.setBackgroundColor(colorOK);
                magiskCheckUpdatesIcon.setImageResource(R.drawable.ic_check_circle);
                magiskCheckUpdatesStatus.setText(getString(R.string.up_to_date, getString(R.string.magisk)));
                magiskCheckUpdatesStatus.setTextColor(colorOK);
                magiskUpdateView.setOnClickListener(view -> builder
                        .setTitle(getString(R.string.repo_install_title, getString(R.string.magisk)))
                        .setMessage(getString(R.string.repo_install_msg, "Magisk-v" + String.valueOf(remoteMagiskVersion)))
                        .setCancelable(true)
                        .setPositiveButton(R.string.download_install, flashMagisk)
                        .setNegativeButton(R.string.no_thanks, null)
                        .show());
            }

            if (remoteAppVersionCode > BuildConfig.VERSION_CODE) {
                appCheckUpdatesContainer.setBackgroundColor(colorInfo);
                appCheckUpdatesIcon.setImageResource(R.drawable.ic_file_download);
                appCheckUpdatesStatus.setText(getString(R.string.app_update_available, remoteAppVersion));
                appCheckUpdatesStatus.setTextColor(colorInfo);
                appUpdateView.setOnClickListener(view -> builder
                        .setTitle(getString(R.string.update_title, getString(R.string.app_name)))
                        .setMessage(getString(R.string.update_msg, getString(R.string.app_name), remoteAppVersion, appChangelog))
                        .setCancelable(true)
                        .setPositiveButton(R.string.download_install, installMagiskApk)
                        .setNegativeButton(R.string.no_thanks, null)
                        .show()
                );
            } else {
                appCheckUpdatesContainer.setBackgroundColor(colorOK);
                appCheckUpdatesIcon.setImageResource(R.drawable.ic_check_circle);
                appCheckUpdatesStatus.setText(getString(R.string.up_to_date, getString(R.string.app_name)));
                appCheckUpdatesStatus.setTextColor(colorOK);

            }
        }

        appCheckUpdatesProgress.setVisibility(View.GONE);
        magiskCheckUpdatesProgress.setVisibility(View.GONE);
        mSwipeRefreshLayout.setRefreshing(false);

        if (magiskVersion == -1) {
            builder
                    .setTitle(R.string.no_magisk_title)
                    .setMessage(R.string.no_magisk_msg)
                    .setCancelable(true)
                    .setPositiveButton(R.string.download_install, flashMagisk)
                    .setNegativeButton(R.string.no_thanks, null)
                    .show();
        }
    }
}