// ==UserScript==
// @name      Browser search engine
// @author    Vitaliy V.
// @include   main
// @shutdown  window.contextsearch.destructor();
// @note      https://forum.mozilla-russia.org/viewtopic.php?pid=780283#p780283
// ==/UserScript==

(this.contextsearch = {
    topic: "browser-search-engine-modified",
    hide: "browser.search.hiddenOneOffs",
    defaultImg: "chrome://browser/skin/search-engine-placeholder.png",
    searchSelect: null,
    popup: null,
    init(that) {
        var searchSelect = this.searchSelect = document.querySelector("#context-searchselect");
        if (!searchSelect)
            return;
        var popup = this.popup = searchSelect.closest("menupopup");
        popup.addEventListener("popupshowing", this);
        that.unloadlisteners?.push("contextsearch");
    },
    destructor() {
        this.popup.removeEventListener("popupshowing", this);
        if (this.popupshowing == this.handler) {
            this.popup.removeEventListener("popuphidden", this);
            Services.obs.removeObserver(this, this.topic);
            Services.prefs.removeObserver(this.hide, this);
        }
    },
    handleEvent(e) {
        this[e.type](e);
    },
    popupshowing(e) {
        var popup = this.popup;
        var searchSelect = this.searchSelect;
        if (e.target != popup || searchSelect.hidden) return;

        var menu = document.createXULElement("menu");
        menu.className = "menu-iconic";
        var menupopup = document.createXULElement("menupopup");
        menu.append(menupopup);
        menu.ePopup = menupopup;
        searchSelect.style.setProperty("display", "none", "important");
        searchSelect.before(menu);
        menu.onclick = this.search.bind(this);
        this.handler = e => e.target != popup || (menu.hidden = searchSelect.hidden);
        this.handlerRebuild = e => this.handler(e) || this.rebuild(menu);
        this.popuphidden = ev => {
            if (ev.target != popup) return;
            menu.hidden = true;
        };
        this.popup.addEventListener("popuphidden", this);
        this.rebuild(menu);
    },
    getEngines() {
        var args = "hideOneOffButton" in Services.search.defaultEngine
            ? [e => !e.hideOneOffButton]
            : Object.defineProperty(
                [function(e) {return !this.includes(e.name);}], "1", {
                    get: () => Services.prefs.getStringPref(this.hide)?.split(",") || []
                }
            );
        return (this.getEngines = async () =>
            (await Services.search.getVisibleEngines()).filter(...args)
        )();
    },
    async rebuild(menu) {
        var de = Services.search.defaultEngine;
        de = de.wrappedJSObject || de;
        this.setAttrs(menu, de, `Искать в ${de.name} или в ...`);
        menu.ePopup.textContent = "";
        for(let engine of await this.getEngines()) {
            if (engine == de) continue;
            var menuitem = document.createXULElement("menuitem");
            menuitem.className = "menuitem-iconic";
            this.setAttrs(menuitem, engine);
            menu.ePopup.append(menuitem);
        }
        this.popupshowing = this.handler;
        Services.obs.addObserver(this, this.topic, false);
        Services.prefs.addObserver(this.hide, this);
    },
    setAttrs(node, engine, label = engine.name) {
        node.engine = engine;
        node.setAttribute("label", label);
        node.setAttribute("image", engine._iconURI ? engine._iconURI.spec : engine.iconURI ? engine.iconURI.spec : this.defaultImg);
    },
    observe() {
        this.popupshowing = this.handlerRebuild;
        Services.obs.removeObserver(this, this.topic);
        Services.prefs.removeObserver(this.hide, this);
    },
    search(e) {
        var {engine} = e.target;
        if (!engine) return;
        var searchSelect = this.searchSelect;
        var submission = engine.getSubmission(
            searchSelect.searchTerms, null, "contextmenu"
        );
        if (submission) {
            let tab = gBrowser.addTab(submission.uri.spec, {
                postData: submission.postData,
                index: (gBrowser.selectedTab._tPos + 1),
                triggeringPrincipal: searchSelect.principal
            });
            if (e.button == 0)
                gBrowser.selectedTab = tab;
        }
        var popup = this.popup;
        e.button != 1 && popup.state == "open" && popup.hidePopup();
    }
}).init(this);

/*
                    moduleURI: "chrome://user_chrome_files/content/custom_scripts/UcfTextToLinkActorChild.jsm",
*/
                    esModuleURI: "chrome://user_chrome_files/content/custom_scripts/UcfTextToLinkActorChild.mjs"

/*
var EXPORTED_SYMBOLS = ["UcfTextToLinkActorChild"];
var {XPCOMUtils} = ChromeUtils.import("resource://gre/modules/XPCOMUtils.jsm");
XPCOMUtils.defineLazyGlobalGetters(this, ["NodeFilter", "Node"]);
XPCOMUtils.defineLazyGetter(this, "excludedTags", () => {
    return new Set(["a","svg","canvas","applet","input","button","area","embed","noembed","frame","frameset","head","iframe","img","select","option","datalist","map","meta","noscript","video","audio","object","param","script","style","textarea","code"]);
});

class UcfTextToLinkActorChild extends JSWindowActorChild {
*/
var excludedTags = Object.defineProperty({}, "has", {
    value: tag => (excludedTags = new Set([
        "a","svg","canvas","applet","input","button","area","embed","noembed","frame","frameset","head","iframe","img",
        "select","option","datalist","map","meta","noscript","video","audio","object","param","script","style","textarea","code"
    ])).has(tag)
});

export class UcfTextToLinkActorChild extends JSWindowActorChild {

/*
	try {
*/
	if(Components.interfaces.nsIWebTransportHash) { // random, Fx 123+
		var func = function() {
			func = false;
		}
		var thread = Services.tm.currentThread;
		var meth = newDis ? "disable" : "enable";
		addon[meth]({allowSystemAddons: true}).finally(func);
		while(func) thread.processNextEvent(true);
	}
	else try {

//
	fill(o, popup) {
		for (key in o) {
			var val = o[key];
			if (typeof val != "object") continue;

			var {lab, sep, sub, cmd, alt, inf, img, upd} = val;

			sep && popup.append(this.m("menuseparator"));
			var name = sub ? "menu" : "menuitem";
			var item = this.m(name);

			item.setAttribute("label", lab || key);
			if (img)
				item.className = name + "-iconic",
				item.setAttribute("image", img);

			item.alt = alt; //cmd2
			if (inf) item.tooltipText = inf;
			sub || cmd && item.setAttribute("oncommand", cmd.toString().replace(
				/cmd\(.*?\){/, "{var trg = event.target || event;"
			));
			popup.append(item);
			upd && (item.render = upd).call(item);
			sub && this.fill(val, item.appendChild(this.m("menupopup")));
		}
	},

// Switch Keyboard Layout
(async keybUtils => CustomizableUI.createWidget(({
	localized: false,
	id: "SwitchKeyboardLayout",
	init() {
		this.label = this.tooltiptext = "Switch Keyboard Layout";
		var args = ["keydown", e => {
			if (e.key != "F8" || e.ctrlKey || e.shiftKey || e.altKey || e.repeat) return;
			//e.preventDefault();
			//e.stopPropagation();
			this.switch(e.view.document);
		}, true];
		var unload = e => e.target.ownerGlobal.removeEventListener(...args);
		var obs = win => {
			win.addEventListener(...args);
			win.addEventListener("unload", unload, {once: true});
		}
		var topic = "browser-delayed-startup-finished";
		Services.obs.addObserver(obs, topic);
		Services.obs.addObserver(function quit(s, t) {
			Services.obs.removeObserver(quit, t);
			Services.obs.removeObserver(obs, topic);
		}, "quit-application-granted");
		return this;
	},
	onCreated(btn) {
		btn.linkedObj = this;
		btn._handleClick = this.click;
		btn.image = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAMAAAAoLQ9TAAAAk1BMVEX///8/Pz8BAQF8fHwJCQkCAgIJCQl8fHx9fX0AAABJSUkBAQENDQ0wMDBVVVUAAABra2t0dHR7e3thYWEBAQEAAAAAAAABAQEAAAAgICABAQEBAQH8/Pzw8PDp6en39/fj4+Pe3t41V9I/YeWKioqDg4N9fX0jRa8wUrVoaGhkhuxWeNx1dXVCZMhvb28kRqsqTL4KidXxAAAAHHRSTlMAAABndwAAAAA0blVZcm1naWhNa6hrcJ8AAKRU4jk/3gAAAHVJREFUeF6FyMUCwjAUBdEbaQupUMOjqMv/fx1dQN6yZ3aDcWnyw6ezoRWNYc15kqKQZCFlgTKQQwgllL/dn5f3+bPb+6P3CpWNw56srdCZOMzDmA61i8O9nKvRaHLVukEryFKIFtnkj7ENYxmNdd5v+5xj1BcJ/w9Kj6K7ZAAAAABJRU5ErkJggg==";
	},
	click() {
		this.linkedObj.switch(this.ownerDocument);
	},
	switch(doc) {
		var br = doc.activeElement;
		br && br.localName == "browser" && br.isRemoteBrowser
			? br.messageManager.loadFrameScript(this.url, false)
			: this.keybUtils.switchSelKeybLayout();
	},
	get url() {
		delete this.url;
		return this.url = `data:;charset=utf-8,(${
			encodeURIComponent(keybUtils)
		}).switchSelKeybLayout()`;
	},
	get keybUtils() {
		delete this.keybUtils;
		var def = "let{KeyEvent,HTMLInputElement,HTMLTextAreaElement}=Cu.getGlobalForObject(Services);";
		var url = `data:;charset=utf-8,${def}%0Athis.keybUtils=${encodeURIComponent(keybUtils)}`;
		Services.scriptloader.loadSubScript(url, this);
		var {id} = this;
		this.keybUtils.getFocusedElement = function(_subCall, _focusFixed) {
			var window = Services.focus.activeWindow, {document} = window;
			var button = document.getElementById(id);
			if(
				!_focusFixed
				&& "closeMenus" in window
				&& document.commandDispatcher.focusedElement == button
			) {
				window.closeMenus(button);
				window.setTimeout(function(_this) {
					_this.switchSelKeybLayout(_subCall, true);
				}, 0, this);
				return;
			}
			return document.commandDispatcher.focusedElement;
		}
		return this.keybUtils;
	}
}).init()))(`{
	//== Options
	noSelBehavior:
		"BeginLine",
		//"EndLine",
		//"All",
		//"Top",
		//"Bottom",
		//"PageUp",
		//"PageDown",
		// etc

	// falsy - do nothing
	// Or use string like following to call cmd_select{string}

	convTableForward: { // ru -> en
		"\\"": "@",
		":": "^",
		";": "$",
		"?": "&",
		",": "?",
		"/": "|",
		".": "/",
		"э": "'",
		"б": ",",
		"ю": ".",
		"Ж": ":",
		"ж": ";",
		"Б": "<",
		"Ю": ">",
		"Э": "\\"",
		"х": "[",
		"ъ": "]",
		"ё": "\`",
		"Х": "{",
		"Ъ": "}",
		"Ё": "~",
		"№": "#",
		"Ф": "A",
		"ф": "a",
		"И": "B",
		"и": "b",
		"С": "C",
		"с": "c",
		"В": "D",
		"в": "d",
		"У": "E",
		"у": "e",
		"А": "F",
		"а": "f",
		"П": "G",
		"п": "g",
		"Р": "H",
		"р": "h",
		"Ш": "I",
		"ш": "i",
		"О": "J",
		"о": "j",
		"Л": "K",
		"л": "k",
		"Д": "L",
		"д": "l",
		"Ь": "M",
		"ь": "m",
		"Т": "N",
		"т": "n",
		"Щ": "O",
		"щ": "o",
		"З": "P",
		"з": "p",
		"Й": "Q",
		"й": "q",
		"К": "R",
		"к": "r",
		"Ы": "S",
		"ы": "s",
		"Е": "T",
		"е": "t",
		"Г": "U",
		"г": "u",
		"М": "V",
		"м": "v",
		"Ц": "W",
		"ц": "w",
		"Ч": "X",
		"ч": "x",
		"Н": "Y",
		"н": "y",
		"Я": "Z",
		"я": "z",
		__proto__: null
	},
	//== End of options

	get convTableBackward() {
		var ctb = { __proto__: null };
		var ctf = this.convTableForward;
		for(var c in ctf)
			ctb[ctf[c]] = c;
		delete this.convTableBackward;
		return this.convTableBackward = ctb;
	},
	inPrimaryLayout: function(s) {
		for(var i = 0, l = s.length; i < l; ++i) {
			var c = s.charAt(i);
			var primary = c in this.convTableForward;
			if(primary ^ c in this.convTableBackward)
				return primary;
		}
		return false;
	},
	switchKeybLayout: function(s, convTable) {
		var res = "";
		for(var i = 0, l = s.length; i < l; ++i) {
			var c = s.charAt(i);
			res += c in convTable ? convTable[c] : c;
		}
		return res;
	},
	getFocusedElement: function() {
		return Cc["@mozilla.org/focus-manager;1"].getService(Ci.nsIFocusManager)
			.getFocusedElementForWindow(content, true, {});
	},
	switchSelKeybLayout: function(_subCall, _focusFixed) {
		var fe = this.getFocusedElement(_subCall, _focusFixed);
		if(!fe)
			return;
		if(HTMLInputElement.isInstance(fe) || HTMLTextAreaElement.isInstance(fe)) {
			var ta = fe;
			try {
				var val = ta.value;
				var sel = val.substring(ta.selectionStart, ta.selectionEnd);
			}
			catch(e) { // Non-text HTMLInputElement
				return;
			}
			if(!sel && val && this.noSelBehavior && !_subCall) {
				if(this.noSelBehavior == 1) {
					ta.selectionStart = 0;
					ta.selectionEnd = val.length;
					sel = val;
				}
				else {
					this.handleNoSel(ta);
					return;
				}
			}
			if(!sel)
				return;
			var res = this.switchKeybLayout(
				sel,
				this.inPrimaryLayout(sel)
					? this.convTableForward
					: this.convTableBackward
			);
			if(res != sel)
				this.insertText(ta, res);
		}
		else if(fe.contentEditable == "true") {
			var doc = fe.ownerDocument;

			var docURI = doc.documentURI;
			if(
				docURI.substr(0, 5) == "data:"
				&& docURI.indexOf("chrome://browser/skin/devtools/") != -1
			) {
				//~ todo: seems like we only can use paste from clipboard here...
				return;
			}

			var sel = doc.defaultView.getSelection();
			var rng = sel.rangeCount && sel.getRangeAt(0);
			var tmpNode;
			if(!rng || rng.collapsed) {
				if(!this.noSelBehavior || _subCall)
					return;
				if(this.noSelBehavior == 1) {
					var r = doc.createRange();
					r.selectNodeContents(fe);
					sel.removeAllRanges();
					sel.addRange(r);
					tmpNode = fe.cloneNode(true);
				}
				else {
					this.handleNoSel(fe);
					return;
				}
			}
			else {
				tmpNode = doc.createElementNS("http://www.w3.org/1999/xhtml", "div");
				tmpNode.appendChild(rng.cloneContents());
			}

			var orig = tmpNode.innerHTML;
			var convTable = this.inPrimaryLayout(tmpNode.textContent)
				? this.convTableForward
				: this.convTableBackward;

			var _this = this;
			var parseChildNodes = function(node) {
				if(Element.isInstance(node)) {
					var childNodes = node.childNodes;
					for(var i = childNodes.length - 1; i >= 0; --i)
						parseChildNodes(childNodes[i]);
				}
				else if(node.nodeType == node.TEXT_NODE) {
					var text = node.nodeValue;
					var newText = _this.switchKeybLayout(node.nodeValue, convTable);
					if(newText != text)
						node.parentNode.replaceChild(doc.createTextNode(newText), node);
				}
			}
			parseChildNodes(tmpNode);

			var res = tmpNode.innerHTML;
			if(res != orig)
				doc.execCommand("insertHTML", false, res);
		}
	},
	handleNoSel: function(node) {
		this.select(node);
		this.switchSelKeybLayout(true);
	},
	select: function(node) {
//		var e = this.noSelBehavior;
//		if(!e || typeof e != "object")
//			return;
//		var evt = node.ownerDocument.createEvent("KeyboardEvent");
//		evt.initKeyEvent(
//			"keypress", true /*bubbles*/, true /*cancelable*/, node.ownerDocument.defaultView,
//			e.ctrlKey, e.altKey, e.shiftKey, e.metaKey,
//			e.keyCode, e.charCode
//		);
//		node.dispatchEvent(evt);

	   var beh = this.noSelBehavior;
	   beh && node.ownerGlobal.docShell.doCommand("cmd_select" + beh);
	},
	beh2cmd: { // Ctrl_Shift_VK
		false_true_36: "cmd_selectLinePrevious", // Shift+Home
	},
	insertText: function(ta, text) {
		//var editor = ta.QueryInterface(Components.interfaces.nsIDOMNSEditableElement).editor
		var editor = ta.editor
			.QueryInterface(Components.interfaces.nsIPlaintextEditor || Ci.nsIEditor);
		if(editor.flags & editor.eEditorReadonlyMask)
			return;

		var sTop = ta.scrollTop;
		var sHeight = ta.scrollHeight;
		var sLeft = ta.scrollLeft;
		// var sWidth = ta.scrollWidth;

		if(text)
			editor.insertText(text);
		else
			editor.deleteSelection(0, 0);

		ta.scrollTop = sTop + (ta.scrollHeight - sHeight);
		ta.scrollLeft = sLeft; // + (ta.scrollWidth - sWidth);
	}
}`);