package com.github.catvod.spider;

import android.content.Context;
import android.os.SystemClock;

import com.github.catvod.crawler.Spider;
import com.github.catvod.crawler.SpiderDebug;
import com.github.catvod.net.OkHttp;
import com.github.catvod.utils.Path;
import com.github.catvod.utils.Shell;

import java.io.File;

public class Feiyang extends Spider {

    private static final String URL = "http://127.0.0.1:35345/";
    private static final String AIO = "allinone-linux-musl";
    private static final String SHELL = "run-allinone.sh";
    private static final File f_shell = new File(Path.files(), SHELL);
    private static final File f_aio = new File(Path.files(), AIO);
    private static Thread thread;
    private String extend;

    @Override
    public void init(Context context, String extend) throws Exception {
        this.extend = extend;
        createShell();
        findAIO(Path.tv().listFiles());
        findAIO(Path.download().listFiles());
    }

    @Override
    public String liveContent(String url) {
        int retry = 0;
        while ((OkHttp.string(URL)).isEmpty() && retry++ < 10) SystemClock.sleep(250);
        return OkHttp.string(!url.isEmpty() ? url : URL + extend);
    }

    private void createShell() {
        File cache = Path.cache("allinone.cache");
        String cachePath = cache.getAbsolutePath();
        String shellPath = f_shell.getAbsolutePath();
        String aioPath = f_aio.getAbsolutePath();
        String script = "if [ -e " + shellPath + ".kill ]; then\n" + "sh " + shellPath + ".kill;\n" + "killall -9 " + AIO + ";\n" + "sleep 1;\n" + "rm -f " + shellPath + ".kill;\n" + "fi\n" + "echo \"kill -9 $$\" >> " + shellPath + ".kill;\n" + "while(true)\n" + "do\n" + "TMPDIR=" + cachePath + " " + aioPath + " -ipv6=true " + " -port 35345 " + " >/dev/null 2>&1\n" + "pidlist=$(ps -ef|grep " + SHELL + "|grep -v grep|awk '{print $2}')\n" + "selfpid=$$\n" + "for pid in $pidlist\n" + "do\n" + "if [[ $pid != \"$selfpid\" ]] then\n" + "kill -9 $pid\n" + "fi\n" + "done\n" + "killall -9 " + AIO + "\n" + "rm -rf \"" + cachePath + "\"/*\n" + "sleep 5;\n" + "done";
        Path.write(f_shell, script);
    }

    private void findAIO(File[] files) {
        if (files == null || thread != null) return;
        for (File f : files) {
            if (f.getName().equals("allinone") || f.getName().startsWith("allinone-linux")) {
                Path.copy(f, f_aio);
                SpiderDebug.log("start run " + f.getName());
                thread = new Thread(() -> Shell.exec("nohup " + f_shell.getAbsolutePath() + " &"));
                thread.start();
            }
        }
    }

    @Override
    public void destroy() {
        try {
            Shell.exec("killall -9 " + AIO);
            if (thread != null) thread.interrupt();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            thread = null;
        }
    }
}