InterceptAndBlockUrlsExample.java

package com.github.kklisura.cdt.examples;

import com.github.kklisura.cdt.launch.ChromeLauncher;
import com.github.kklisura.cdt.protocol.commands.Network;
import com.github.kklisura.cdt.protocol.commands.Page;
import com.github.kklisura.cdt.protocol.types.network.ErrorReason;
import com.github.kklisura.cdt.protocol.types.network.RequestPattern;
import com.github.kklisura.cdt.services.ChromeDevToolsService;
import com.github.kklisura.cdt.services.ChromeService;
import com.github.kklisura.cdt.services.types.ChromeTab;
import java.util.Collections;

/**
 * Intercept and block per URL. Since requestIntercepted event is still Experimental it might not
 * work on your browser.
 *
 * @author Kenan Klisura
 */
public class InterceptAndBlockUrlsExample {
  public static void main(String[] args) {
    // Create chrome launcher.
    final ChromeLauncher launcher = new ChromeLauncher();

    // Launch chrome either as headless (true) or regular (false).
    final ChromeService chromeService = launcher.launch(false);

    // Create empty tab ie about:blank.
    final ChromeTab tab = chromeService.createTab();

    // Get DevTools service to this tab
    final ChromeDevToolsService devToolsService = chromeService.createDevToolsService(tab);

    // Get individual commands
    final Page page = devToolsService.getPage();
    final Network network = devToolsService.getNetwork();

    network.onRequestIntercepted(
        event -> {
          String interceptionId = event.getInterceptionId();
          boolean blocked = isBlocked(event.getRequest().getUrl());

          System.out.printf(
              "%s - %s%s",
              (blocked ? "BLOCKED" : "ALLOWED"),
              event.getRequest().getUrl(),
              System.lineSeparator());

          ErrorReason errorReason = blocked ? ErrorReason.ABORTED : null;

          network.continueInterceptedRequest(
              interceptionId, errorReason, null, null, null, null, null, null);
        });

    page.onLoadEventFired(event -> devToolsService.close());

    RequestPattern interceptionRequestPattern = new RequestPattern();
    network.setRequestInterception(Collections.singletonList(interceptionRequestPattern));
    network.enable();

    // Enable page events.
    page.enable();

    // Navigate to github.com.
    page.navigate("http://github.com");

    devToolsService.waitUntilClosed();
  }

  public static boolean isBlocked(String url) {
    return url.endsWith(".png") || url.endsWith(".css");
  }
}