/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/
 */

"use strict";

const { HttpServer } = ChromeUtils.import("resource://testing-common/httpd.js");

/**
 * Waits for an observer notification to fire.
 *
 * @param {String} topic The notification topic.
 * @returns {Promise} A promise that fulfills when the notification is fired.
 */
function promiseObserverNotification(topic, matchFunc) {
  return new Promise((resolve, reject) => {
    Services.obs.addObserver(function observe(subject, topic, data) {
      let matches = typeof matchFunc != "function" || matchFunc(subject, data);
      if (!matches) {
        return;
      }
      Services.obs.removeObserver(observe, topic);
      resolve({ subject, data });
    }, topic);
  });
}

registerCleanupFunction(() => {
  Services.prefs.clearUserPref("network.connectivity-service.DNSv4.domain");
  Services.prefs.clearUserPref("network.connectivity-service.DNSv6.domain");
  Services.prefs.clearUserPref("network.captive-portal-service.testMode");
  Services.prefs.clearUserPref("network.connectivity-service.IPv4.url");
  Services.prefs.clearUserPref("network.connectivity-service.IPv6.url");
});

let httpserver = null;
let httpserverv6 = null;
XPCOMUtils.defineLazyGetter(this, "URL", function() {
  return "http://localhost:" + httpserver.identity.primaryPort + "/content";
});

XPCOMUtils.defineLazyGetter(this, "URLv6", function() {
  return "http://[::1]:" + httpserverv6.identity.primaryPort + "/content";
});

function contentHandler(metadata, response) {
  response.setHeader("Content-Type", "text/plain");
  response.setHeader("Cache-Control", "no-cache");

  const responseBody = "anybody";
  response.bodyOutputStream.write(responseBody, responseBody.length);
}

const DEFAULT_WAIT_TIME = 200; // ms

const kDNSv6Domain =
  mozinfo.os == "linux" || mozinfo.os == "android"
    ? "ip6-localhost"
    : "localhost";

add_task(async function testDNS() {
  let ncs = Cc[
    "@mozilla.org/network/network-connectivity-service;1"
  ].getService(Ci.nsINetworkConnectivityService);

  // Set the endpoints, trigger a DNS recheck, and wait for it to complete.
  Services.prefs.setCharPref(
    "network.connectivity-service.DNSv4.domain",
    "example.org"
  );
  Services.prefs.setCharPref(
    "network.connectivity-service.DNSv6.domain",
    kDNSv6Domain
  );
  ncs.recheckDNS();
  await promiseObserverNotification(
    "network:connectivity-service:dns-checks-complete"
  );

  equal(
    ncs.DNSv4,
    Ci.nsINetworkConnectivityService.OK,
    "Check DNSv4 support (expect OK)"
  );
  equal(
    ncs.DNSv6,
    Ci.nsINetworkConnectivityService.OK,
    "Check DNSv6 support (expect OK)"
  );

  // Set the endpoints to non-exitant domains, trigger a DNS recheck, and wait for it to complete.
  Services.prefs.setCharPref(
    "network.connectivity-service.DNSv4.domain",
    "does-not-exist.example"
  );
  Services.prefs.setCharPref(
    "network.connectivity-service.DNSv6.domain",
    "does-not-exist.example"
  );
  ncs.recheckDNS();
  await promiseObserverNotification(
    "network:connectivity-service:dns-checks-complete"
  );

  equal(
    ncs.DNSv4,
    Ci.nsINetworkConnectivityService.NOT_AVAILABLE,
    "Check DNSv4 support (expect N/A)"
  );
  equal(
    ncs.DNSv6,
    Ci.nsINetworkConnectivityService.NOT_AVAILABLE,
    "Check DNSv6 support (expect N/A)"
  );

  // Set the endpoints back to the proper domains, and simulate a captive portal
  // event.
  Services.prefs.setCharPref(
    "network.connectivity-service.DNSv4.domain",
    "example.org"
  );
  Services.prefs.setCharPref(
    "network.connectivity-service.DNSv6.domain",
    kDNSv6Domain
  );
  Services.obs.notifyObservers(null, "network:captive-portal-connectivity");
  // This will cause the state to go to UNKNOWN for a bit, until the check is completed.
  equal(
    ncs.DNSv4,
    Ci.nsINetworkConnectivityService.UNKNOWN,
    "Check DNSv4 support (expect UNKNOWN)"
  );
  equal(
    ncs.DNSv6,
    Ci.nsINetworkConnectivityService.UNKNOWN,
    "Check DNSv6 support (expect UNKNOWN)"
  );

  await promiseObserverNotification(
    "network:connectivity-service:dns-checks-complete"
  );

  equal(
    ncs.DNSv4,
    Ci.nsINetworkConnectivityService.OK,
    "Check DNSv4 support (expect OK)"
  );
  equal(
    ncs.DNSv6,
    Ci.nsINetworkConnectivityService.OK,
    "Check DNSv6 support (expect OK)"
  );

  httpserver = new HttpServer();
  httpserver.registerPathHandler("/content", contentHandler);
  httpserver.start(-1);

  httpserverv6 = new HttpServer();
  httpserverv6.registerPathHandler("/contentt", contentHandler);
  httpserverv6._start(-1, "[::1]");

  // Before setting the pref, this status is unknown in automation
  equal(
    ncs.IPv4,
    Ci.nsINetworkConnectivityService.UNKNOWN,
    "Check IPv4 support (expect UNKNOWN)"
  );
  equal(
    ncs.IPv6,
    Ci.nsINetworkConnectivityService.UNKNOWN,
    "Check IPv6 support (expect UNKNOWN)"
  );

  Services.prefs.setBoolPref("network.captive-portal-service.testMode", true);
  Services.prefs.setCharPref("network.connectivity-service.IPv4.url", URL);
  Services.prefs.setCharPref("network.connectivity-service.IPv6.url", URLv6);
  ncs.recheckIPConnectivity();
  await promiseObserverNotification(
    "network:connectivity-service:ip-checks-complete"
  );

  equal(
    ncs.IPv4,
    Ci.nsINetworkConnectivityService.OK,
    "Check IPv4 support (expect OK)"
  );
  equal(
    ncs.IPv6,
    Ci.nsINetworkConnectivityService.OK,
    "Check IPv6 support (expect OK)"
  );

  // check that the CPS status is NOT_AVAILABLE when the endpoint is down.
  await new Promise(resolve => httpserver.stop(resolve));
  await new Promise(resolve => httpserverv6.stop(resolve));
  Services.obs.notifyObservers(null, "network:captive-portal-connectivity");
  await promiseObserverNotification(
    "network:connectivity-service:ip-checks-complete"
  );

  equal(
    ncs.IPv4,
    Ci.nsINetworkConnectivityService.NOT_AVAILABLE,
    "Check IPv4 support (expect NOT_AVAILABLE)"
  );
  equal(
    ncs.IPv6,
    Ci.nsINetworkConnectivityService.NOT_AVAILABLE,
    "Check IPv6 support (expect NOT_AVAILABLE)"
  );
});