// ==UserScript==
// @name            Enter Selects
// @author          xiaoxiaoflood
// @include         main
// @startup         UC.enterSelects.exec(win);
// @shutdown        UC.enterSelects.destroy();
// @onlyonce
// ==/UserScript==

// inspired by: https://web.archive.org/web/20171129174355/https://addons.mozilla.org/en-US/firefox/addon/enter-selects/

UC.enterSelects = {
  AUTOSELECT_WITH_SPACE: false,

  init: function () {
    this.orig_receiveResults = this.controller.receiveResults;
    this.controller.receiveResults = function (queryContext) {
      let result = UC.enterSelects.orig_receiveResults.call(this, queryContext);
      let gURLBar = this.browserWindow.gURLBar;
      if (UC.enterSelects.shouldSelect(gURLBar, queryContext)) {
        if (!('lastResultCount' in UC.enterSelects.queryContext))
          UC.enterSelects.lockLastResultCount();

        gURLBar.view.input.setResultForCurrentValue(gURLBar.view.selectedResult);

        if (gURLBar.view.selectedRowIndex <= 0) {
          UC.enterSelects.flag = true;
          gURLBar.view.selectBy(1);

          gURLBar.view.selectedElement.parentElement.remove = function () {
            if (UC.enterSelects.shouldSelect(gURLBar, queryContext)) {
              UC.enterSelects.flag = true;
              gURLBar.view.selectBy(1);
            }

            return Element.prototype.remove.call(this);
          };
        }
      } else if (queryContext.results.length > 1) {
        UC.enterSelects.deleteLastResultCount();
      }

      return result;
    };

    this.orig_setValueFromResult = this.input.setValueFromResult;
    this.input.setValueFromResult = new Proxy(this.input.setValueFromResult, {
      apply: function(target, thisArg, args) {
        if (UC.enterSelects.flag) {
          UC.enterSelects.flag = false;
          return thisArg.setResultForCurrentValue(args[0].result);
        } else
          return target.apply(thisArg, args);
      }
    });

    this.orig_isDeferringEvents = Object.getOwnPropertyDescriptor(this.eventBuffer, 'isDeferringEvents').get;
    Object.defineProperty(this.eventBuffer, 'isDeferringEvents', {
      value: true,
    });

    xPref.lock('browser.urlbar.autoFill', false);
    xPref.lock('browser.urlbar.suggest.engines', false);
    xPref.lock('browser.urlbar.suggest.searches', false);
  },

  exec: function (win) {
    let observe = () => {
      Services.obs.removeObserver(observe, 'browser-window-before-show');
      this.addUrlbarListeners(win);
    }

    if (win.__SSi)
      this.addUrlbarListeners(win);
    else
      Services.obs.addObserver(observe, 'browser-window-before-show');
  },

  addUrlbarListeners: function (win) {
    const { gURLBar } = win;
    gURLBar.textbox.addEventListener('keydown', this.keyD, true);
    gURLBar.inputField.addEventListener('input', this.checkSpace, true);
  },

  controller: ChromeUtils.importESModule('resource:///modules/UrlbarController.sys.mjs').UrlbarController.prototype,
  eventBuffer: ChromeUtils.importESModule('resource:///modules/UrlbarEventBufferer.sys.mjs').UrlbarEventBufferer.prototype,
  input: ChromeUtils.importESModule('resource:///modules/UrlbarInput.sys.mjs').UrlbarInput.prototype,
  queryContext: ChromeUtils.importESModule('resource:///modules/UrlbarUtils.sys.mjs').UrlbarQueryContext.prototype,

  shouldSelect: function (gURLBar, queryContext) {
    if (gURLBar.searchMode?.engineName || queryContext.results.length < 2)
      return false;

    let search = gURLBar.value.trim();

    try {
      // Test if is URL
      return !Services.eTLD.getBaseDomainFromHost(search);
    } catch (ex) {}

    if (!this.AUTOSELECT_WITH_SPACE && /\s/.test(search))
      return false;

    // Test about:about, chrome://blabla ... 
    if (/[a-zA-Z][\w-]*:(\/\/)?[\w-]+/.test(search))
      return false;

    // Check if the first word is a keyword (search or bookmark)
    if (!!queryContext.results[0].payload.keyword)
      return false;

    return true;
  },

  checkSpace: function (e) {
    let { gURLBar } = e.view;
    if (!UC.enterSelects.AUTOSELECT_WITH_SPACE && /\s/.test(gURLBar.value)) {
      UC.enterSelects.deleteLastResultCount();
    }
  },

  lockLastResultCount: function () {
    Object.defineProperty(this.queryContext, 'lastResultCount', {
      get: () => 1,
      set: () => 1,
      configurable: true,
    });
  },

  deleteLastResultCount: function () {
    delete UC.enterSelects.queryContext.lastResultCount;
  },

  keyD: function (e) {
    let { gURLBar } = e.view;
    if (!gURLBar.view.isOpen)
      return;
    if (e.keyCode == e.DOM_VK_TAB) {
      let url = gURLBar.view.getResultAtIndex(1).payload.url;
      if (gURLBar.view.selectedRowIndex == 1 &&
          gURLBar.value != url &&
          new RegExp(/^(https?:\/\/(www\.)?)?/.source + gURLBar.value.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&')).test(url) &&
          gURLBar.value != url.match(/(\w*:\/\/)?[^/]+\//)[0]) {
        gURLBar.inputField.value = gURLBar._untrimmedValue = gURLBar._trimValue(url.match(/(\w*:\/\/)?[^/]+\//)[0] + '/').slice(0, -1);
        gURLBar.view.close();
      } else if (gURLBar.view.selectedRowIndex == -1) {
        gURLBar.view.selectedRowIndex = 1;
        gURLBar.value = url;
      } else if (gURLBar.view.selectedRowIndex == 1 && gURLBar.value != url) {
        gURLBar.value = url;
      }
      e.preventDefault();
      e.stopPropagation();
    }
  },

  destroy: function () {
    xPref.unlock('browser.urlbar.autoFill');
    xPref.unlock('browser.urlbar.suggest.engines');
    xPref.unlock('browser.urlbar.suggest.searches');

    Object.defineProperty(this.eventBuffer, 'isDeferringEvents', {
      get: this.orig_isDeferringEvents,
    });

    this.deleteLastResultCount();

    _uc.windows((doc, win) => {
      this.controller.receiveResults = this.orig_receiveResults;
      this.input.setValueFromResult = this.orig_setValueFromResult;
      win.gURLBar.textbox.removeEventListener('keydown', this.keyD, true);
      win.gURLBar.inputField.removeEventListener('input', this.checkSpace, true);
    });
    delete UC.enterSelects;
  }
}

UC.enterSelects.init();