/*
 * LICENSE
 *
 * POCKET MARKS
 *
 * Notwithstanding the permitted uses of the Software (as defined below) pursuant to the license set forth below, "Pocket," "Read It Later" and the Pocket icon and logos (collectively, the “Pocket Marks”) are registered and common law trademarks of Read It Later, Inc. This means that, while you have considerable freedom to redistribute and modify the Software, there are tight restrictions on your ability to use the Pocket Marks. This license does not grant you any rights to use the Pocket Marks except as they are embodied in the Software.
 *
 * ---
 *
 * SOFTWARE
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

/*
 * Pocket UI module
 *
 * Handles interactions with Pocket buttons, panels and menus.
 *
 */

// TODO : Get the toolbar icons from Firefox's build (Nikki needs to give us a red saved icon)
// TODO : [needs clarificaiton from Fx] Firefox's plan was to hide Pocket from context menus until the user logs in. Now that it's an extension I'm wondering if we still need to do this.
// TODO : [needs clarificaiton from Fx] Reader mode (might be a something they need to do since it's in html, need to investigate their code)
// TODO : [needs clarificaiton from Fx] Move prefs within pktApi.s to sqlite or a local file so it's not editable (and is safer)
// TODO : [nice to have] - Immediately save, buffer the actions in a local queue and send (so it works offline, works like our native extensions)

/* eslint-disable no-shadow */
/* eslint-env mozilla/browser-window */

ChromeUtils.defineModuleGetter(
  this,
  "PrivateBrowsingUtils",
  "resource://gre/modules/PrivateBrowsingUtils.jsm"
);
ChromeUtils.defineModuleGetter(
  this,
  "ReaderMode",
  "resource://gre/modules/ReaderMode.jsm"
);
ChromeUtils.defineModuleGetter(
  this,
  "pktApi",
  "chrome://pocket/content/pktApi.jsm"
);
ChromeUtils.defineModuleGetter(
  this,
  "pktTelemetry",
  "chrome://pocket/content/pktTelemetry.jsm"
);

const POCKET_ONSAVERECS_PREF = "extensions.pocket.onSaveRecs";
const POCKET_ONSAVERECS_LOCLES_PREF = "extensions.pocket.onSaveRecs.locales";

var pktUI = (function() {
  // -- Initialization (on startup and new windows) -- //

  // Init panel id at 0. The first actual panel id will have the number 1 so
  // in case at some point any panel has the id 0 we know there is something
  // wrong
  var _panelId = 0;

  let _titleToSave = "";
  let _urlToSave = "";

  // Initial sizes are only here to help visual load jank before the panel is ready.
  const initialPanelSize = {
    signup: {
      control: { height: 442, width: 300 },
      button_control: { height: 442, width: 300 },
      variant_a: { height: 408, width: 300 },
      variant_b: { height: 383, width: 300 },
      variant_c: { height: 424, width: 300 },
      button_variant: { height: 388, width: 300 },
    },
    saved: {
      control: { height: 133, width: 350 },
    },
  };

  var onSaveRecsEnabledPref;
  var onSaveRecsLocalesPref;

  function initPrefs() {
    onSaveRecsEnabledPref = Services.prefs.getBoolPref(
      POCKET_ONSAVERECS_PREF,
      false
    );
    onSaveRecsLocalesPref = Services.prefs.getStringPref(
      POCKET_ONSAVERECS_LOCLES_PREF,
      ""
    );
  }
  initPrefs();

  // -- Communication to API -- //

  /**
   * Either save or attempt to log the user in
   */
  function tryToSaveCurrentPage() {
    tryToSaveUrl(getCurrentUrl(), getCurrentTitle());
  }

  function tryToSaveUrl(url, title) {
    // Validate input parameter
    if (typeof url !== "undefined" && url.startsWith("about:reader?url=")) {
      url = ReaderMode.getOriginalUrl(url);
    }

    // If the user is logged in, go ahead and save the current page
    if (pktApi.isUserLoggedIn()) {
      _titleToSave = title;
      _urlToSave = url;
      saveAndShowConfirmation();
      return;
    }

    // If the user is not logged in, show the logged-out state to prompt them to authenticate
    showSignUp();
  }

  // -- Panel UI -- //

  /**
   * Show the sign-up panel
   */
  function showSignUp() {
    // AB test: Direct logged-out users to tab vs panel
    if (pktApi.getSignupPanelTabTestVariant() == "v2") {
      let site = Services.prefs.getCharPref("extensions.pocket.site");
      openTabWithUrl(
        "https://" +
          site +
          "/firefox_learnmore?s=ffi&t=autoredirect&tv=page_learnmore&src=ff_ext",
        Services.scriptSecurityManager.createNullPrincipal({}),
        null
      );

      // force the panel closed before it opens
      getPanel().hidePopup();

      return;
    }

    // Control: Show panel as normal
    getFirefoxAccountSignedInUser(function(userdata) {
      var controlvariant = pktApi.getSignupPanelTabTestVariant() == "control";
      var variant = "storyboard_lm";
      const loggedOutVariant =
        Services.prefs.getCharPref("extensions.pocket.loggedOutVariant") ||
        "control";
      let sizes = initialPanelSize.signup.control;
      if (loggedOutVariant && initialPanelSize.signup[loggedOutVariant]) {
        sizes = initialPanelSize.signup[loggedOutVariant];
      }

      showPanel(
        "about:pocket-signup?pockethost=" +
          Services.prefs.getCharPref("extensions.pocket.site") +
          "&loggedOutVariant=" +
          loggedOutVariant +
          "&variant=" +
          variant +
          "&controlvariant=" +
          controlvariant +
          "&locale=" +
          getUILocale(),
        sizes
      );
    });
  }

  /**
   * Get a list of recs for item and show them in the panel.
   */
  function getAndShowRecsForItem(item, options) {
    var onSaveRecsEnabled =
      onSaveRecsEnabledPref && onSaveRecsLocalesPref.includes(getUILocale());

    if (
      onSaveRecsEnabled &&
      item &&
      item.resolved_id &&
      item.resolved_id !== "0"
    ) {
      pktApi.getRecsForItem(item.resolved_id, options);
    }
  }

  /**
   * Show the logged-out state / sign-up panel
   */
  function saveAndShowConfirmation() {
    getFirefoxAccountSignedInUser(function(userdata) {
      const variant = "control";
      const sizes = initialPanelSize.saved[variant];
      showPanel(
        "about:pocket-saved?pockethost=" +
          Services.prefs.getCharPref("extensions.pocket.site") +
          "&premiumStatus=" +
          (pktApi.isPremiumUser() ? "1" : "0") +
          "&fxasignedin=" +
          (typeof userdata == "object" && userdata !== null ? "1" : "0") +
          "&locale=" +
          getUILocale(),
        sizes
      );
    });
  }

  /**
   * Open a generic panel
   */
  function showPanel(url, options) {
    // Add new panel id
    _panelId += 1;
    url += "&panelId=" + _panelId;

    resizePanel({
      width: options.width,
      height: options.height,
    });

    // We don't have to hide and show the panel again if it's already shown
    // as if the user tries to click again on the toolbar button the overlay
    // will close instead of the button will be clicked
    var iframe = getPanelFrame();

    // Load the iframe
    iframe.setAttribute("src", url);
  }

  function onShowSignup() {
    // A successful button click, for logged out users.
    pktTelemetry.sendStructuredIngestionEvent(
      pktTelemetry.createPingPayload({
        events: [
          {
            action: "click",
            source: "save_button",
          },
        ],
      })
    );
  }

  function onShowSaved() {
    var saveLinkMessageId = "PKT_saveLink";
    getPanelFrame().setAttribute("itemAdded", "false");

    // Send error message for invalid url
    if (!isValidURL()) {
      // TODO: Pass key for localized error in error object
      let error = {
        message: "Only links can be saved",
        localizedKey: "onlylinkssaved",
      };
      pktUIMessaging.sendErrorMessageToPanel(
        saveLinkMessageId,
        _panelId,
        error
      );
      return;
    }

    // Check online state
    if (!navigator.onLine) {
      // TODO: Pass key for localized error in error object
      let error = {
        message:
          "You must be connected to the Internet in order to save to Pocket. Please connect to the Internet and try again.",
      };
      pktUIMessaging.sendErrorMessageToPanel(
        saveLinkMessageId,
        _panelId,
        error
      );
      return;
    }

    // A successful button click, for logged in users.
    pktTelemetry.sendStructuredIngestionEvent(
      pktTelemetry.createPingPayload({
        events: [
          {
            action: "click",
            source: "save_button",
          },
        ],
      })
    );

    // Add url
    var options = {
      success(data, request) {
        var item = data.item;
        var ho2 = data.ho2;
        var accountState = data.account_state;
        var displayName = data.display_name;
        var successResponse = {
          status: "success",
          accountState,
          displayName,
          item,
          ho2,
        };
        pktUIMessaging.sendMessageToPanel(
          saveLinkMessageId,
          _panelId,
          successResponse
        );
        getPanelFrame().setAttribute("itemAdded", "true");

        getAndShowRecsForItem(item, {
          success(data) {
            pktUIMessaging.sendMessageToPanel(
              "PKT_renderItemRecs",
              _panelId,
              data
            );
            if (data?.recommendations?.[0]?.experiment) {
              const payload = pktTelemetry.createPingPayload({
                // This is the ML model used to recommend the story.
                // Right now this value is the same for all three items returned together,
                // so we can just use the first item's value for all.
                model: data.recommendations[0].experiment,
                // Create an impression event for each item rendered.
                events: data.recommendations.map((item, index) => ({
                  action: "impression",
                  position: index,
                  source: "on_save_recs",
                })),
              });
              // Send view impression ping.
              pktTelemetry.sendStructuredIngestionEvent(payload);
            }
          },
        });
      },
      error(error, request) {
        // If user is not authorized show singup page
        if (request.status === 401) {
          showSignUp();
          return;
        }

        // If there is no error message in the error use a
        // complete catch-all
        var errorMessage =
          error.message || "There was an error when trying to save to Pocket.";
        var panelError = { message: errorMessage };

        // Send error message to panel
        pktUIMessaging.sendErrorMessageToPanel(
          saveLinkMessageId,
          _panelId,
          panelError
        );
      },
    };

    // Add title if given
    if (typeof _titleToSave !== "undefined") {
      options.title = _titleToSave;
    }

    // Send the link
    pktApi.addLink(_urlToSave, options);
  }

  /**
   * Resize the panel
   * options = {
   *  width: ,
   *  height: ,
   * }
   */
  function resizePanel(options = {}) {
    var iframe = getPanelFrame();

    // Set an explicit size, panel will adapt.
    iframe.style.width = options.width + "px";
    iframe.style.height = options.height + "px";
  }

  // -- Browser Navigation -- //

  /**
   * Open a new tab with a given url and notify the iframe panel that it was opened
   */

  function openTabWithUrl(url, aTriggeringPrincipal, aCsp) {
    let recentWindow = Services.wm.getMostRecentWindow("navigator:browser");
    if (!recentWindow) {
      Cu.reportError("Pocket: No open browser windows to openTabWithUrl");
      return;
    }

    // If the user is in permanent private browsing than this is not an issue,
    // since the current window will always share the same cookie jar as the other
    // windows.
    if (
      !PrivateBrowsingUtils.isWindowPrivate(recentWindow) ||
      PrivateBrowsingUtils.permanentPrivateBrowsing
    ) {
      recentWindow.openWebLinkIn(url, "tab", {
        triggeringPrincipal: aTriggeringPrincipal,
        csp: aCsp,
      });
      return;
    }

    for (let win of Services.wm.getEnumerator("navigator:browser")) {
      if (!PrivateBrowsingUtils.isWindowPrivate(win)) {
        win.openWebLinkIn(url, "tab", {
          triggeringPrincipal: aTriggeringPrincipal,
          csp: aCsp,
        });
        return;
      }
    }

    // If there were no non-private windows opened already.
    recentWindow.openWebLinkIn(url, "window", {
      triggeringPrincipal: aTriggeringPrincipal,
      csp: aCsp,
    });
  }

  // Open a new tab with a given url
  function onOpenTabWithUrl(panelId, data, contentPrincipal, csp) {
    try {
      urlSecurityCheck(
        data.url,
        contentPrincipal,
        Services.scriptSecurityManager.DISALLOW_INHERIT_PRINCIPAL
      );
    } catch (ex) {
      return;
    }

    // We don't track every click, only clicks with a known source.
    if (data.source) {
      const payload = pktTelemetry.createPingPayload({
        events: [
          {
            action: "click",
            source: data.source,
          },
        ],
      });
      // Send click event ping.
      pktTelemetry.sendStructuredIngestionEvent(payload);
    }

    var url = data.url;
    openTabWithUrl(url, contentPrincipal, csp);
  }

  // Open a new tab with a Pocket story url
  function onOpenTabWithPocketUrl(panelId, data, contentPrincipal, csp) {
    try {
      urlSecurityCheck(
        data.url,
        contentPrincipal,
        Services.scriptSecurityManager.DISALLOW_INHERIT_PRINCIPAL
      );
    } catch (ex) {
      return;
    }

    const { url, position, model } = data;
    // Check to see if we need to and can fire valid telemetry.
    if (model && (position || position === 0)) {
      const payload = pktTelemetry.createPingPayload({
        model,
        events: [
          {
            action: "click",
            position,
            source: "on_save_recs",
          },
        ],
      });
      // Send click event ping.
      pktTelemetry.sendStructuredIngestionEvent(payload);
    }

    openTabWithUrl(url, contentPrincipal, csp);
  }

  // -- Helper Functions -- //

  function getCurrentUrl() {
    return gBrowser.currentURI.spec;
  }

  function getCurrentTitle() {
    return gBrowser.contentTitle;
  }

  function closePanel() {
    getPanel().hidePopup();
  }

  function getPanel() {
    var frame = getPanelFrame();
    var panel = frame;
    while (panel && panel.localName != "panel") {
      panel = panel.parentNode;
    }
    return panel;
  }

  var photonPageActionPanelFrame;

  function setPhotonPageActionPanelFrame(frame) {
    photonPageActionPanelFrame = frame;
  }

  function getPanelFrame() {
    return photonPageActionPanelFrame;
  }

  function isValidURL() {
    return (
      typeof _urlToSave !== "undefined" &&
      (_urlToSave.startsWith("http") || _urlToSave.startsWith("https"))
    );
  }

  function getFirefoxAccountSignedInUser(callback) {
    fxAccounts
      .getSignedInUser()
      .then(userData => {
        callback(userData);
      })
      .then(null, error => {
        callback();
      });
  }

  function getUILocale() {
    return Services.locale.appLocaleAsBCP47;
  }

  /**
   * Public functions
   */
  return {
    setPhotonPageActionPanelFrame,
    getPanelFrame,
    initPrefs,

    openTabWithUrl,
    onOpenTabWithUrl,
    onOpenTabWithPocketUrl,
    onShowSaved,
    onShowSignup,

    getAndShowRecsForItem,
    tryToSaveUrl,
    tryToSaveCurrentPage,
    resizePanel,
    closePanel,
  };
})();

// -- Communication to Background -- //
var pktUIMessaging = (function() {
  /**
   * Send a message to the panel's iframe
   */
  function sendMessageToPanel(messageId, panelId, payload) {
    if (!isPanelIdValid(panelId)) {
      return;
    }

    var panelFrame = pktUI.getPanelFrame();
    if (!isPocketPanelFrameValid(panelFrame)) {
      return;
    }

    const aboutPocketActor = panelFrame?.browsingContext?.currentWindowGlobal?.getActor(
      "AboutPocket"
    );

    // Send message to panel
    aboutPocketActor?.sendAsyncMessage(`${messageId}_${panelId}`, payload);
  }

  /**
   * Helper function to package an error object and send it to the panel
   * iframe as a message response
   */
  function sendErrorMessageToPanel(messageId, panelId, error) {
    var errorResponse = { status: "error", error };
    sendMessageToPanel(messageId, panelId, errorResponse);
  }

  /**
   * Validation
   */

  function isPanelIdValid(panelId) {
    // First check if panelId has a valid value > 0. We set the panelId to
    // 0 to start. But if for some reason the message is attempted to be
    // sent before the panel has a panelId, then it's going to send out
    // a message with panelId 0, which is never going to be heard. If this
    // happens, it means some race condition occurred where the panel was
    // trying to communicate before it should.
    if (panelId === 0) {
      console.warn("Tried to send message to panel with id 0.");
      return false;
    }

    return true;
  }

  function isPocketPanelFrameValid(panelFrame) {
    // Check if panel is available if not throw a warning and bailout.
    // We likely try to send to a panel that is not visible anymore
    if (typeof panelFrame === "undefined") {
      console.warn("Pocket panel frame is undefined");
      return false;
    }

    var contentWindow = panelFrame.contentWindow;
    if (typeof contentWindow == "undefined") {
      console.warn("Pocket panel frame content window is undefined");
      return false;
    }

    var doc = contentWindow.document;
    if (typeof doc === "undefined") {
      console.warn("Pocket panel frame content window document is undefined");
      return false;
    }

    var documentElement = doc.documentElement;
    if (typeof documentElement === "undefined") {
      console.warn(
        "Pocket panel frame content window document document element is undefined"
      );
      return false;
    }

    return true;
  }

  /**
   * Public
   */
  return {
    sendMessageToPanel,
    sendErrorMessageToPanel,
  };
})();