package ctbrec.ui.sites.amateurtv;

import java.io.IOException;
import java.util.Collections;
import java.util.Objects;
import java.util.function.Consumer;

import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ctbrec.sites.amateurtv.AmateurTv;
import ctbrec.ui.ExternalBrowser;
import okhttp3.Cookie;
import okhttp3.Cookie.Builder;
import okhttp3.CookieJar;
import okhttp3.HttpUrl;

public class AmateurTvElectronLoginDialog {

    private static final Logger LOG = LoggerFactory.getLogger(AmateurTvElectronLoginDialog.class);
    public static final String DOMAIN = "amateur.tv";
    public static final String URL = AmateurTv.baseUrl;
    private CookieJar cookieJar;
    private ExternalBrowser browser;

    public AmateurTvElectronLoginDialog(CookieJar cookieJar) throws IOException {
        this.cookieJar = cookieJar;
        browser = ExternalBrowser.getInstance();
        try {
            var config = new JSONObject();
            config.put("url", URL);
            config.put("w", 640);
            config.put("h", 480);
            var msg = new JSONObject();
            msg.put("config", config);
            browser.run(msg, msgHandler);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            throw new IOException("Couldn't wait for login dialog", e);
        } finally {
            browser.close();
        }
    }

    private Consumer<String> msgHandler = line -> {
        if (!line.startsWith("{")) {
            LOG.error("Didn't received a JSON object {}", line);
        } else {
            var json = new JSONObject(line);
            try {
                browser.executeJavaScript("let loginDialogVisible = document.querySelectorAll('div[class~=\"MuiDialog-container\"]').length > 1");
                browser.executeJavaScript("if (!loginDialogVisible) { document.querySelector('button').innerHTML.indexOf('I agree') >= 0 && document.querySelector('button').click(); }");
                browser.executeJavaScript("if (!loginDialogVisible) { document.querySelector('button[aria-label=\"open drawer\"]').click(); }"); // open the burger menu to get to the login button
                browser.executeJavaScript("if (!loginDialogVisible) { document.querySelectorAll('button').forEach(function(b) { if (b.textContent === 'Log in') b.click(); }); }"); // click the login button to open the login dialog
                browser.executeJavaScript("loginDialogVisible = document.querySelectorAll('div[class~=\"MuiDialog-container\"]').length > 1");
                browser.executeJavaScript("if (loginDialogVisible) throw new Error(\"Stop execution right here\")");
                //                String username = Config.getInstance().getSettings().amateurTvUsername;
                //                String password = Config.getInstance().getSettings().amateurTvPassword;
                //                browser.executeJavaScript("if (loginDialogVisible) { document.querySelectorAll('div[class~=\"MuiDialog-container\"] input').item(0).value = '" + username + "' }"); // enter username
                //                browser.executeJavaScript("if (loginDialogVisible) { document.querySelectorAll('div[class~=\"MuiDialog-container\"] input').item(1).value = '" + password + "' }"); // enter password
                //browser.executeJavaScript("console.log('submit')");
                //                browser.executeJavaScript("if(loginDialogVisible) { document.querySelector('div[class~=\"MuiDialog-container\"] button[type=\"submit\"]').click() }"); // click the submit button
            } catch(Exception e) {
                LOG.warn("Couldn't auto fill username and password for Amateur.TV", e);
            }

            var loginSuccessful = false;
            if (json.has("cookies")) {
                var cookiesFromBrowser = json.getJSONArray("cookies");
                for (var i = 0; i < cookiesFromBrowser.length(); i++) {
                    var cookie = cookiesFromBrowser.getJSONObject(i);
                    if (cookie.getString("domain").contains(DOMAIN)) {
                        Builder b = new Cookie.Builder()
                                .path(cookie.getString("path"))
                                .domain(DOMAIN)
                                .name(cookie.getString("name"))
                                .value(cookie.getString("value"))
                                .expiresAt((long) cookie.optDouble("expirationDate") * 1000l);
                        if (cookie.optBoolean("hostOnly")) {
                            b.hostOnlyDomain(DOMAIN);
                        }
                        if (cookie.optBoolean("httpOnly")) {
                            b.httpOnly();
                        }
                        if (cookie.optBoolean("secure")) {
                            b.secure();
                        }
                        Cookie c = b.build();
                        cookieJar.saveFromResponse(HttpUrl.parse(AmateurTv.baseUrl), Collections.singletonList(c));
                        LOG.debug("{}={}", c.name(), c.value());
                        if (Objects.equals(c.name(), "userType") && Objects.equals(c.value(), "registered")) {
                            loginSuccessful = true;
                        }
                    }
                }
            }

            if (loginSuccessful) {
                try {
                    browser.close();
                } catch (IOException e) {
                    LOG.error("Couldn't send shutdown request to external browser", e);
                }
            }
        }
    };
}