/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

import { CustomizableUI } from "resource:///modules/CustomizableUI.sys.mjs";
import { PanelMultiView } from "resource:///modules/PanelMultiView.sys.mjs";
import { XPCOMUtils } from "resource://gre/modules/XPCOMUtils.sys.mjs";

const lazy = {};

XPCOMUtils.defineLazyServiceGetter(
  lazy,
  "styleSheetService",
  "@mozilla.org/content/style-sheet-service;1",
  "nsIStyleSheetService"
);

export const BrowserUtils = {
  // internal functions/props
  get mostRecentWindow() {
    return Services.wm.getMostRecentWindow("navigator:browser");
  },

  get document() {
    return Services.wm.getMostRecentWindow("navigator:browser").document;
  },
  createElement(aDoc, aTag, aAttrs) {
    // Create element
    const el = aDoc.createXULElement(aTag);
    for (const att in aAttrs) {
      // don't set null attrs
      if (aAttrs[att]) {
        el.setAttribute(att, aAttrs[att]);
      }
    }
    return el;
  },

  // api endpoints
  createAndPositionElement(aWindow, aTag, aAttrs, aAdjacentTo, aPosition) {
    const doc = aWindow.document;
    // Create element
    const el = this.createElement(doc, aTag, aAttrs);
    // Place it in certain location
    let pos = doc.getElementById(aAdjacentTo);
    if (aPosition) {
      // App Menu items are not retrieved successfully with doc.getElementById
      if (!pos) {
        pos = PanelMultiView.getViewNode(this.document, aAdjacentTo);
      }
      // If neither method to retrieve a positional element succeeded
      // don't attempt to insert as it will fail
      if (pos) {
        pos.insertAdjacentElement(aPosition, el);
      }
    } else if (aAdjacentTo === "gNavToolbox") {
      aWindow.gNavToolbox.appendChild(el);
    } else {
      pos.appendChild(el);
    }
  },

  /**
   * Helper function to execute a given function with some args in every open browser window.
   * Window must be the functions first arg, subsequent args are passed in the same manner
   * as to executeInAllWindows().
   *
   * @param func - The function to be called in each open browser window.
   * @param args - The arguments to supply to the function.
   * Example:
   * BrowserUtils.executeInAllWindows(Urlbar.addDynamicStylesheet, "chrome://browser/skin/waterfox.css")
   */
  executeInAllWindows(func, ...args) {
    const windows = Services.wm.getEnumerator("navigator:browser");
    while (windows.hasMoreElements()) {
      const window = windows.getNext();
      func(window, ...args);
    }
  },

  /**
   * Helper method to register or unregister a given stylesheet depending on the bool arg passed.
   *
   * @param uri - The URI of the stylesheet to register or unregister.
   * @param enabled - A boolean indicating whether to register or unregister the sheet.
   */
  registerOrUnregisterSheet(uri, enabled = false) {
    if (enabled) {
      BrowserUtils.registerStylesheet(uri);
    } else {
      BrowserUtils.unregisterStylesheet(uri);
    }
  },

  registerStylesheet(uri) {
    if (!this.sheetRegistered(uri)) {
      const url = Services.io.newURI(uri);
      const type = lazy.styleSheetService.USER_SHEET;
      lazy.styleSheetService.loadAndRegisterSheet(url, type);
    }
  },

  unregisterStylesheet(uri) {
    if (this.sheetRegistered(uri)) {
      const url = Services.io.newURI(uri);
      const type = lazy.styleSheetService.USER_SHEET;
      lazy.styleSheetService.unregisterSheet(url, type);
    }
  },

  sheetRegistered(uri) {
    const url = Services.io.newURI(uri);
    const type = lazy.styleSheetService.USER_SHEET;
    return lazy.styleSheetService.sheetRegistered(url, type);
  },

  setStyle(aStyleSheet) {
    const styleSheetService = Cc[
      "@mozilla.org/content/style-sheet-service;1"
    ].getService(Ci.nsIStyleSheetService);

    const url = Services.io.newURI(
      `data:text/css;charset=UTF-8,${encodeURIComponent(aStyleSheet)}`
    );
    const type = styleSheetService.USER_SHEET;

    styleSheetService.loadAndRegisterSheet(url, type);
  },

  registerArea(aArea) {
    CustomizableUI.registerArea(aArea, {});
  },
};