package com.topjohnwu.magisk.asyncs;

import android.support.v4.app.FragmentActivity;

import com.topjohnwu.jarsigner.ByteArrayStream;
import com.topjohnwu.magisk.MagiskManager;
import com.topjohnwu.magisk.utils.Shell;
import com.topjohnwu.magisk.utils.WebService;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.lang.reflect.Proxy;
import java.net.HttpURLConnection;

import dalvik.system.DexClassLoader;

public class CheckSafetyNet extends ParallelTask<Void, Void, Exception> {

    public static final int SNET_VER = 2;

    private static final String SNET_URL = "https://github.com/topjohnwu/MagiskManager/releases/download/v5.4.0/snet.apk";
    private static final String PKG = "com.topjohnwu.snet";

    private File dexPath;
    private DexClassLoader loader;

    public CheckSafetyNet(FragmentActivity activity) {
        super(activity);
        dexPath = new File(activity.getCacheDir().getParent() + "/snet", "snet.apk");
    }

    @Override
    protected void onPreExecute() {
        MagiskManager mm = getMagiskManager();
        if (mm.snet_version != CheckSafetyNet.SNET_VER) {
            Shell.sh("rm -rf " + dexPath.getParent());
        }
        mm.snet_version = CheckSafetyNet.SNET_VER;
        mm.prefs.edit().putInt("snet_version", CheckSafetyNet.SNET_VER).apply();
    }

    @Override
    protected Exception doInBackground(Void... voids) {
        try {
            if (!dexPath.exists()) {
                HttpURLConnection conn = WebService.request(SNET_URL, null);
                ByteArrayStream bas = new ByteArrayStream();
                bas.readFrom(conn.getInputStream());
                conn.disconnect();
                dexPath.getParentFile().mkdir();
                try (OutputStream out = new BufferedOutputStream(new FileOutputStream(dexPath))) {
                    bas.writeTo(out);
                    out.flush();
                }
            }
            loader = new DexClassLoader(dexPath.toString(), dexPath.getParent(),
                    null, ClassLoader.getSystemClassLoader());
        } catch (Exception e) {
            return e;
        }
        return null;
    }

    @Override
    protected void onPostExecute(Exception err) {
        try {
            if (err != null) throw err;
            Class<?> helperClazz = loader.loadClass(PKG + ".SafetyNetHelper");
            Class<?> callbackClazz = loader.loadClass(PKG + ".SafetyNetCallback");
            Object helper = helperClazz.getConstructors()[0].newInstance(
                    getActivity(), Proxy.newProxyInstance(
                            loader, new Class[] { callbackClazz }, (proxy, method, args) -> {
                                getMagiskManager().safetyNetDone.publish(false, args[0]);
                                return null;
                            }));
            helperClazz.getMethod("attest").invoke(helper);
        } catch (Exception e) {
            e.printStackTrace();
            getMagiskManager().safetyNetDone.publish(false, -1);
        }
        super.onPostExecute(err);
    }
}