/*
 * https://webfolder.io/license.html
 */
package io.webfolder.cdp.network;

import static io.webfolder.cdp.event.Events.NetworkRequestIntercepted;
import static io.webfolder.cdp.type.network.ResourceType.Document;
import static java.util.Arrays.asList;

import io.webfolder.cdp.Example;
import io.webfolder.cdp.Launcher;
import io.webfolder.cdp.command.Network;
import io.webfolder.cdp.event.network.RequestIntercepted;
import io.webfolder.cdp.session.Session;
import io.webfolder.cdp.session.SessionFactory;
import io.webfolder.cdp.type.network.RequestPattern;

@Example
public class FollowRedirects {

    @SuppressWarnings("deprecation")
    public static void main(String[] args) {
        Launcher launcher = new Launcher();

        try (SessionFactory factory = launcher.launch();
                            Session session = factory.create()) {

            Network network = session.getCommand().getNetwork();
            network.enable();
            // Disable newtork caching when intercepting
            // https://github.com/GoogleChrome/puppeteer/pull/1154
            network.setCacheDisabled(Boolean.TRUE);

            RequestPattern pattern = new RequestPattern();
            pattern.setUrlPattern("*");
            pattern.setResourceType(Document);
            network.setRequestInterception(asList(pattern));

            boolean followRedirect = true;

            session.addEventListener((e, d) -> {
                if (NetworkRequestIntercepted.equals(e)) {
                    RequestIntercepted ri = (RequestIntercepted) d;
                    boolean isRedirect = ri.getRedirectUrl() != null && ! ri.getRedirectUrl().isEmpty();

                    if (isRedirect) {
                        System.out.println("");
                        System.out.println("Redirect URL         : " + ri.getRedirectUrl());
                        System.out.println("Redirect Status Code : " + ri.getResponseStatusCode());
                        System.out.println("Redirect Header      : " + ri.getResponseHeaders());
                        System.out.println("");

                        if ( ! followRedirect ) {
                            return;
                        }
                    }

                    network.continueInterceptedRequest(ri.getInterceptionId());
                }
            });

            session.navigate("https://httpbin.org/redirect-to?url=https://webfolder.io?cdp4j");
            session.waitDocumentReady();

            session.wait(500);
            System.out.println(session.getLocation());
        } finally {
            launcher.kill();
        }
    }
}