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

package com.topjohnwu.magisk;

import android.app.Fragment;
import android.content.DialogInterface;
import android.content.SharedPreferences;
import android.os.Bundle;
import android.preference.PreferenceManager;
import android.support.annotation.Nullable;
import android.support.v4.widget.SwipeRefreshLayout;
import android.support.v7.app.AlertDialog;
import android.support.v7.widget.RecyclerView;
import android.util.Log;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.TextView;

import com.topjohnwu.magisk.module.Repo;
import com.topjohnwu.magisk.module.RepoHelper;
import com.topjohnwu.magisk.utils.Async;
import com.topjohnwu.magisk.utils.Utils;

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

import butterknife.BindView;
import butterknife.ButterKnife;

public class ReposFragment extends Fragment {

    public static List<Repo> mListRepos = new ArrayList<>();
    public static List<Repo> mListReposToUpdate = new ArrayList<>();
    @BindView(R.id.recyclerView)
    RecyclerView recyclerView;
    @BindView(R.id.empty_rv)
    TextView emptyTv;
    @BindView(R.id.swipeRefreshLayout)
    SwipeRefreshLayout swipeRefreshLayout;
    private View mView;
    private boolean mCanUpdate;
    private boolean alertUpdate;
    private boolean ignoreAlertUpdate;
    private String alertPackage;
    private SharedPreferences prefs;

    @Nullable

    @Override
    public View onCreateView(LayoutInflater inflater, @Nullable ViewGroup container, @Nullable Bundle savedInstanceState) {
        View view = inflater.inflate(R.layout.repos_fragment, container, false);
        mView = view;
        ButterKnife.bind(this, view);
        prefs = PreferenceManager.getDefaultSharedPreferences(getActivity());
        if (prefs.contains("ignoreUpdateAlerts")) {
            ignoreAlertUpdate = prefs.getBoolean("ignoreUpdateAlerts", false);
        }
        swipeRefreshLayout.setOnRefreshListener(() -> {
            this.LoadRepo(true);
            ignoreAlertUpdate = false;
            prefs.edit().putBoolean("ignoreUpdateAlerts", false).apply();

        });
        LoadRepo(false);
        setHasOptionsMenu(false);
        alertUpdate = false;
        if (mListRepos.size() == 0) {
            emptyTv.setVisibility(View.VISIBLE);
            recyclerView.setVisibility(View.GONE);
            return view;
        }
        CheckForUpdates();
        Log.d("Magisk", "ReposFragment: ListRepos size is " + listRepos().size());
        recyclerView.setAdapter(new ReposAdapter(this, mListRepos));
        return view;
    }


    @Override
    public void onStart() {
        super.onStart();
        NotifyOfAlerts();

    }

    private void CheckForUpdates() {
        for (int i = 0; i < mListRepos.size(); i++) {
            if (mListRepos.get(i).canUpdate()) {
                alertUpdate = true;
                mListReposToUpdate.add(mListRepos.get(i));

            }
        }
    }

    @Override
    public void onAttachFragment(Fragment childFragment) {
        super.onAttachFragment(childFragment);
    }

    private void LoadRepo(boolean doReload) {
        RepoHelper.TaskDelegate taskDelegate = result -> {
            if (result.equals("Complete")) {
                Log.d("Magisk", "ReposFragment, got delegate");
                UpdateUI();
                if (mView != null) {
                    mView.invalidate();
                    mView.requestLayout();
                }

            }

        };
        Log.d("Magisk", "ReposFragment, LoadRepo called");
        new Async.LoadRepos(getActivity());
    }

    private void NotifyOfAlerts() {
        if (alertUpdate && !ignoreAlertUpdate) {
            Iterator<Repo> iterRepo = mListReposToUpdate.iterator();
            while (iterRepo.hasNext()) {
                Repo repo = iterRepo.next();
                DialogInterface.OnClickListener dialogClickListener = (dialog, which) -> {
                    switch (which) {
                        case DialogInterface.BUTTON_POSITIVE:
                            Utils.DownloadReceiver receiver = new Utils.DownloadReceiver() {
                                @Override
                                public void task(File file) {
                                    Log.d("Magisk", "Task firing");
                                    new Async.FlashZIP(getActivity(), repo.getId(), file.toString()).execute();
                                }
                            };
                            String filename = repo.getId().replace(" ", "") + ".zip";
                            Utils.downloadAndReceive(getActivity(), receiver, repo.getZipUrl(), filename);

                            break;

                        case DialogInterface.BUTTON_NEGATIVE:
                            ignoreAlertUpdate = true;
                            SharedPreferences.Editor editor = prefs.edit();
                            editor.putBoolean("ignoreUpdateAlerts", ignoreAlertUpdate);
                            editor.apply();
                            break;
                    }
                };

                AlertDialog.Builder builder = new AlertDialog.Builder(getActivity());
                builder.setMessage("An update is available for " + repo.getName() + ".  Would you like to install it?").setPositiveButton("Yes", dialogClickListener)
                        .setNegativeButton("No", dialogClickListener).show();
                iterRepo.remove();
            }

        }
    }


    @Override
    public void onResume() {
        super.onResume();
        LoadRepo(false);
        getActivity().setTitle("Magisk");


    }


    protected List<Repo> listRepos() {
        return mListRepos;
    }

    private void UpdateUI() {
        Log.d("Magisk", "ReposFragment: UpdateUI Called, size is " + listRepos().size());

        if (listRepos().size() == 0) {
            emptyTv.setVisibility(View.VISIBLE);
            recyclerView.setVisibility(View.GONE);

        }
        Log.d("Magisk", "ReposFragment: ListRepos size is " + listRepos().size());
        recyclerView.setAdapter(new ReposAdapter(this, listRepos()));
        if (swipeRefreshLayout.isRefreshing()) {
            swipeRefreshLayout.setRefreshing(false);
            CheckForUpdates();
            NotifyOfAlerts();
        }


    }


}