package Torello.Browser;

import static Torello.Java.C.BRED;
import static Torello.Java.C.BCYAN;
import static Torello.Java.C.RESET;

import Torello.Browser.BrowserAPI.Network;
import Torello.Browser.BrowserAPI.Page;
import Torello.Browser.BrowserAPI.Target;

import Torello.Java.StrIndent;

import java.util.ArrayList;
import java.util.Base64;
import java.util.List;

import java.nio.file.Files;
import java.nio.file.Paths;

/**
 * Demonstrates several CDP browser features without using JavaScript evaluation.
 * <EMBED CLASS='external-html' DATA-FILE-ID=EX03_EXAMPLE>
 */
public class Example03
{
    private Example03() { }

    /** The URL used for demonstrating Network, Page, Screenshot, PDF and Cookie commands. */
    protected static final String chineseURL =
        "https://zh.wikipedia.org/wiki/%E4%BA%BA%E5%B7%A5%E6%99%BA%E8%83%BD";

    /** The files generated by this example. */
    protected static final String screenshotFile = "Example03-Screenshot.png";
    protected static final String pdfFile        = "Example03-Page.pdf";

    private static ConnRecord connRec = new ConnRecord();

    static 
    {
        connRec.setEventHandler(Example03::eventHandler);
        connRec.setRawTextReceiver(System.out);
        connRec.setErrTextReceiver(System.out);
    }



    // ********************************************************************************************
    // ********************************************************************************************
    // Printing Helpers
    // ********************************************************************************************
    // ********************************************************************************************


    /** Network URLs collected by the page-level handler. */
    protected static final List<String> networkURLs = new ArrayList<>();

    /** Page-level handler; this is where Network events become useful. */
    static void eventHandler(BrowserEvent<?> be)
    {
        if (be instanceof Network.requestWillBeSent)
        {
            final Network.requestWillBeSent r = (Network.requestWillBeSent) be;

            networkURLs.add(r.request.url);

            System.out.println(
                '\n' + BCYAN + "Example03.java: " + RESET +
                BRED + "Network.requestWillBeSent" + RESET + '\n' +
                "    " + r.request.method + " " + r.request.url
            );
        }

        else if (be instanceof Network.responseReceived)
        {
            final Network.responseReceived r = (Network.responseReceived) be;

            System.out.println(
                '\n' + BCYAN + "Example03.java: " + RESET +
                BRED + "Network.responseReceived" + RESET + '\n' +
                "    " + r.response.status + " " + r.response.mimeType + '\n' +
                "    " + r.response.url
            );
        }

        else if (be instanceof Network.loadingFinished)
        {
            final Network.loadingFinished r = (Network.loadingFinished) be;

            System.out.println(
                '\n' + BCYAN + "Example03.java: " + RESET +
                BRED + "Network.loadingFinished" + RESET + '\n' +
                "    requestId: " + r.requestId + '\n' +
                "    bytes:     " + r.encodedDataLength
            );
        }
    };


    // ********************************************************************************************
    // ********************************************************************************************
    // MAIN METHOD
    // ********************************************************************************************
    // ********************************************************************************************


    /** This class is intended to be invoked from the Command Line. */
    public static void main(String[] argv) throws Exception
    {
        STEP_01_initializeBrowserAndPage();

        try
        {
            STEP_02_enablePageAndNetwork(connRec.pws);

            STEP_03_navigateAndCollectNetworkTraffic(connRec.pws);

            STEP_04_captureScreenshot(connRec.pws);

            STEP_05_printToPDF(connRec.pws);

            STEP_06_printCookies(connRec.pws);
        }

        finally
        {
            connRec.bws.disconnect();
            connRec.pws.disconnect();
        }
    }


    // ********************************************************************************************
    // ********************************************************************************************
    // The Steps
    // ********************************************************************************************
    // ********************************************************************************************


    /** Open Chrome, close old tabs, create a new tab, and return both WebSocket senders. */
    protected static void STEP_01_initializeBrowserAndPage() throws Exception
    {
        Printing.notice("Initialize BrowserConn, PageConn and WebSocketSender");

        final BrowserConn browserConn = BrowserConn.getBrowserConn(9222, false);

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            BRED + "Opened Browser Connection:\n" + RESET + browserConn.toString()
        );

        final WebSocketSender bws = browserConn.createSender(connRec);

        final Target.TargetInfo[] allTabs = Target
            .getTargets(null /* FilterEntry[] */)
            .exec(bws)
            .await();

        for (int i = 0; i < allTabs.length; i++)
            if ("page".equals(allTabs[i].type))
                Target.closeTarget(allTabs[i].targetId).exec(bws).await();

        final String targetID = Target
            .createTarget()
            .accept("url", "about:blank")
            .build()
            .exec(bws)
            .await();

        final PageConn pageConn = PageConn
            .getAllPageConn(9222, false)
            .filter((PageConn pc) -> pc.id.equals(targetID))
            .findFirst()
            .orElseThrow(() -> new RuntimeException("The Page-Connection was Not Found"));

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            BRED + "Opened Page Connection:\n" + RESET + pageConn.toString()
        );

        final WebSocketSender pws = pageConn.createSender(connRec);

        Thread.sleep(1000);

        connRec.bws = bws;
        connRec.pws = pws;
    }

    /** Enables the two CDP domains used throughout this example. */
    protected static void STEP_02_enablePageAndNetwork(final WebSocketSender pws) throws Exception
    {
        Printing.notice("Enable Page and Network Domains");

        System.out.println('\n' + BCYAN + "Example03.java: " + RESET + "Page.enable()");
        Page.enable(null /* enableFileChooserOpenedEvent */).exec(connRec.pws).await();

        System.out.println('\n' + BCYAN + "Example03.java: " + RESET + "Network.enable()");
        Network
            .enable(
                null,   // maxTotalBufferSize
                null,   // maxResourceBufferSize
                null,   // maxPostDataSize,
                null    // reportDirectSocketTraffic
            )
            .exec(connRec.pws)
            .await();
    }

    /** Navigate to the target page and allow Network events to arrive through the handler. */
    protected static void STEP_03_navigateAndCollectNetworkTraffic(final WebSocketSender pws)
        throws Exception
    {
        Printing.notice("Navigate and Collect Network Traffic");

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            "Navigating To: " + chineseURL
        );

        Page
            .navigate()
            .accept("url", chineseURL)
            .build()
            .exec(connRec.pws)
            .await();

        Thread.sleep(5000);

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            BRED + "Total Network Request URL's Collected: " + RESET +
            networkURLs.size()
        );

        for (int i = 0; i < networkURLs.size(); i++)
            System.out.println("    " + networkURLs.get(i));
    }

    /** Capture the rendered browser page as a PNG file. */
    protected static void STEP_04_captureScreenshot(final WebSocketSender pws) throws Exception
    {
        Printing.notice("Capture a PNG Screenshot");

        final String pngBase64 = Page
            .captureScreenshot()
            .accept("format", "png")
            .accept("fromSurface", true)
            .build()
            .exec(connRec.pws)
            .await();

        Files.write(Paths.get(screenshotFile), Base64.getDecoder().decode(pngBase64));

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            BRED + "Wrote Screenshot File: " + RESET + screenshotFile
        );
    }

    /** Print the rendered browser page to a PDF file. */
    protected static void STEP_05_printToPDF(final WebSocketSender pws) throws Exception
    {
        Printing.notice("Print the Page to a PDF File");

        final Page.printToPDF$$RET pdf = Page
            .printToPDF()
            .accept("printBackground", true)
            .build()
            .exec(connRec.pws)
            .await();

        Files.write(Paths.get(pdfFile), Base64.getDecoder().decode(pdf.data));

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            BRED + "Wrote PDF File: " + RESET + pdfFile
        );
    }

    /** Print the cookies visible to Chrome for the current page. */
    protected static void STEP_06_printCookies(final WebSocketSender pws) throws Exception
    {
        Printing.notice("Read Browser Cookies for the Current Page");

        final Network.Cookie[] cookies = Network
            .getCookies(null /* String[] urls */)
            .exec(connRec.pws)
            .await();

        System.out.println(
            '\n' + BCYAN + "Example03.java: " + RESET +
            BRED + "Cookies Returned: " + RESET + cookies.length
        );

        for (int i = 0; i < cookies.length; i++)
            System.out.println(StrIndent.indent(cookies[i].toString(), 4));
    }
}