find.ks.js

// Info

let PLUGIN_INFO =
<KeySnailPlugin>
    <name>Find</name>
    <description>Find from KeySnail</description>
    <description lang="ja">ページ内やすべてのタブを検索</description>
    <updateURL>https://gist.github.com/raw/905297/find.ks.js</updateURL>
    <iconURL>https://sites.google.com/site/958site/Home/files/find.ks.png</iconURL>
    <author>958</author>
    <version>0.1.1</version>
    <license>MIT</license>
    <minVersion>1.8.0</minVersion>
    <include>main</include>
    <detail lang="ja"><![CDATA[
=== 使い方 ===
カレントタブやすべてのタブ内のテキストなどを検索します
上記エクステ入力後、promptに検索文字列を入力してください

>||
prompt.useMigemo = true;
||<
してあれば、 XUL/Migemo を使って検索します

=== コマンド ===
勝手に「find」という組み込みコマンドを追加します

>||
find word [-l or -lt or -lu] [*]

 word: 検索文字列

 -l: リンクテキストとリンクのURLを検索
 -lt: リンクテキストを検索
 -lu: リンク URL を検索

 *: 全てのタブを検索
||<

以下のような感じで使ってください

カレントタブのテキストを検索
>||
find hoge
||<

カレントタブのリンクテキストを検索
>||
find hoge -lt
||<

カレントタブのリンク URL を検索
>||
find hoge -lu
||<

カレントタブのリンクテキストと URL を検索
>||
find hoge -l
||<

全てのタブのテキストを検索
>||
find hoge *
||<

全てのタブのリンクテキストを検索
>||
find hoge -lt *
||<

全てのタブのリンク URL を検索
>||
find hoge -lu *
||<

全てのタブのリンクテキストと URL を検索
>||
find hoge -l *
||<
]]></detail>
</KeySnailPlugin>;

// Option

let pOptions = plugins.setupOptions("find", {
    "keymap": {
        preset: {
            "C-z"   : "prompt-toggle-edit-mode",
            "SPC"   : "prompt-next-page",
            "b"     : "prompt-previous-page",
            "j"     : "prompt-next-completion",
            "k"     : "prompt-previous-completion",
            "g"     : "prompt-beginning-of-candidates",
            "G"     : "prompt-end-of-candidates",
            "q"     : "prompt-cancel",
        },
        description: M({
            ja: "メイン画面の操作用キーマップ",
            en: "Local keymap for manipulation"
        })
    },
    'text_pre_pos_length': {
        preset: 30,
        description: M({
            ja: "テキスト検索時に、前後の文脈を省略する際の文字数 (初期値: 30)",
            en: "Number of characters when context when text about is retrieved is omitted (default: 30)"
        })
    },
    'command': {
        preset: 'find',
        description: M({
            ja: "追加するコマンド (初期値: find)",
            en: "Added command (default: find)"
        })
    },
    'result_view_widths': {
        preset: [80, 20],
        description: M({
            ja: "検索結果における文字列とタブ名の幅 (初期値: [80, 20])",
            en: "Proportion of search result text and tab name (default: [80, 20])"
        })
    }
}, PLUGIN_INFO);

// Add ext

const MAX_PRE_POST_TEXT_LEN = pOptions['text_pre_pos_length'];

let gPrompt = {
    get visible() !document.getElementById("keysnail-prompt").hidden,
    focus  : function () {
        document.getElementById("keysnail-prompt-textbox").focus();
    },
    forced : false,
    close  : function () {
        if (!gPrompt.forced)
            return;

        gPrompt.forced = false;

        if (gPrompt.visible)
            prompt.finish(true);
    }
};

function makeKeywordRegExp(word) {
	let re;
	if (window.xulMigemoCore)
		re = new RegExp(window.xulMigemoCore.getRegExp(word), "ig");
	else
		re = new RegExp(word, "ig");
	return re;
}

function grepText(word, isAll) {
	let _nsIFind = Components.classes['@mozilla.org/embedcomp/rangefind;1'].getService(Components.interfaces.nsIFind);
	_nsIFind.findBackwards = false;
	_nsIFind.caseSensitive = true;

	let _wordExp = makeKeywordRegExp(word);

	let _collection = [];

	if (isAll) {
		let tabs = gBrowser.mTabContainer.childNodes;
		for (let i = 0; i < tabs.length; i++) {
			Array.prototype.push.apply(_collection, _grepTargetWindow(tabs[i].linkedBrowser.contentWindow, i));
		}
	} else {
		_collection = _grepTargetWindow(window.content, gBrowser.mTabContainer.selectedIndex);
	}
	if (_collection.length > 0)
		_showSelector();

	function _grepTargetWindow(win, tabIndex){
		let collection = _grep(win, tabIndex);
		for (let i = 0; i < win.frames.length; i++) {
			if (win.frames[i])
				Array.prototype.push.apply(collection, _grepTargetWindow(win.frames[i], tabIndex));
		}
		return collection;
	}

	function _grep(win, tabIndex) {
		let doc = win.document;
		let title = doc.title;
		let favicon = util.getFaviconPath(win.location.href);

		let collection = [];

		$X('//text()[ancestor::body and not(ancestor::script or ancestor::style or ancestor::aside or ancestor::head)]|//input[ancestor::body and not(@hidden)]|//textarea[ancestor::body]', doc).forEach(function(node)
		{
			let text = (node.nodeType == 3) ? node.textContent : node.value;

			if (!_wordExp.test(text)) return;

			let arResult = text.match(_wordExp);

			if (!arResult || arResult.length == 0) return;
			let findRange = doc.createRange();
			findRange.selectNode(node);

			let startPt = doc.createRange();
			startPt.selectNode(node);
			startPt.collapse(true);

			let endPt = doc.createRange();
			endPt.selectNode(node);
			endPt.collapse(false);

			try{
				for (let i = 0, offset = 0; i < arResult.length; i++) {
					let rsltRange = _nsIFind.Find(arResult[i], findRange, startPt, endPt);
					if (rsltRange == null)
						break;

					let desc = (function(){
						let ret;
						let orgText = text;
						if (orgText.length > MAX_PRE_POST_TEXT_LEN * 2) {
							let sttOver = (rsltRange.startOffset > MAX_PRE_POST_TEXT_LEN);
							let endOver = (orgText.length - rsltRange.endOffset > MAX_PRE_POST_TEXT_LEN);
							ret = (sttOver ? '... ' : '') +
								orgText.substring(
									sttOver ? rsltRange.startOffset - MAX_PRE_POST_TEXT_LEN : 0,
									endOver ? rsltRange.endOffset + MAX_PRE_POST_TEXT_LEN : orgText.length) +
								(endOver ? ' ...' : '');
						} else {
							ret = orgText;
						}
						return ret;
					})();
					collection.push([desc, favicon, title, tabIndex, rsltRange]);
					offset = rsltRange.endOffset;
					startPt.setStart(node, offset);
					startPt.setEnd(node, offset);
				}
			}catch(e){ }

			endPt.detach();
			startPt.detach();
			findRange.detach();
		});
		return collection;
	}

	function _showSelector() {
//		let beforeNode = null;

		prompt.selector({
			message    : "pattern:",
			collection : _collection,
			flags      : [0, ICON | IGNORE, 0, IGNORE | HIDDEN, IGNORE | HIDDEN],
			style      : [style.prompt.description],
			header     : ["Text", 'Tab'],
			width      : pOptions["result_view_widths"],
			keymap     : pOptions["keymap"],
			onChange : function (arg) {
				if (!arg.row) return;
//				if (beforeNode) beforeNode.classList.remove('keysnail-find-element-highlight')

				if (gBrowser.mTabContainer.selectedIndex != arg.row[3]) {
					gBrowser.mTabContainer.selectedIndex = arg.row[3];
					gPrompt.focus();
				}
				try {
//					let beforeNode = $X('ancestor-or-self::input', arg.row[4].startContainer);
//					if (beforeNode)
//						beforeNode.classList.add('keysnail-find-element-highlight');
					selectAndScroll(arg.row[4]);
				} catch (e) {
					display.echoStatusBar(M({ja: '非表示テキストです', en: 'Not visible'}), 3000);
					fbug(e);
				}
			},
			actions    : [
				[
					function (index) {
						let doc = _collection[index][4].startContainer.ownerDocument;
						let nodes = $X(
							'ancestor-or-self::a | ancestor-or-self::input | ancestor-or-self::textarea | ancestor-or-self::*[@onclick or @onmouseover or @onmousedown or @onmouseup or @oncommand or @role="link"]',
							_collection[index][4].startContainer);
						if (nodes.length > 0) {
							if (plugins.hok)
								plugins.hok.followLink(nodes[0], 4);
							else
								nodes[0].focus();
						}
					}, "Open or Focus", "Open-or-focus"
				]
			]
		});
	}
}

function grepLink(word, isAll, isTitle, isURL) {
	let _wordExp = makeKeywordRegExp(word);

	let _collection = [];

	if (isAll) {
		let tabs = gBrowser.mTabContainer.childNodes;
		for (let i = 0; i < tabs.length; i++) {
			Array.prototype.push.apply(_collection, _grepTargetWindow(tabs[i].linkedBrowser.contentWindow, i));
		}
	} else {
		_collection = _grepTargetWindow(window.content, gBrowser.mTabContainer.selectedIndex);
	}
	if (_collection.length > 0)
		_showSelector();

	function _grepTargetWindow(win, tabIndex){
		let collection = _grep(win, tabIndex);
		for (let i = 0; i < win.frames.length; i++) {
			if (win.frames[i] && win.frames[i].contentWindow)
				Array.prototype.push.apply(collection, _grepTargetWindow(win.frames[i].contentWindow, tabIndex));
		}
		return collection;
	}

	function _grep(win, tabIndex) {
		let doc = win.document;
		let title = doc.title;
		let favicon = util.getFaviconPath(win.location.href);

		let collection = [];

		function mapText(node) {
			let text = node.textContent;
			if (_wordExp.test(text)) {
				let range = doc.createRange();
				range.selectNodeContents(node);
				collection.push([text || node.href, node.href, favicon, title, tabIndex, range]);
			}
		}

		function mapURL(node) {
			let url = node.href;
			if (_wordExp.test(url)) {
				let range = doc.createRange();
				range.selectNodeContents(node);
				collection.push([node.textContent || url, url, favicon, title, tabIndex, range]);
			}
		}

		function mapTextAndURL(node) {
			let target = node.textContent + ' ' + node.href;
			if (_wordExp.test(target)) {
				let range = doc.createRange();
				range.selectNodeContents(node);
				collection.push([node.textContent || node.href, node.href, favicon, title, tabIndex, range]);
			}
		}

		let fn;
		if (isTitle && isURL)	fn = mapTextAndURL;
		else if (isURL)			fn = mapURL;
		else					fn = mapText;

		$X('//a[string-length(normalize-space(@href)) > 0]', doc).forEach(fn);

		return collection;
	}

	function _showSelector() {
//		let beforeNode = null;

		prompt.selector({
			message    : "pattern:",
			collection : _collection,
			flags      : [0, 0, ICON | IGNORE, 0, IGNORE | HIDDEN, IGNORE | HIDDEN],
			style      : [style.prompt.description, style.prompt.url, style.prompt.description],
			header     : ["Text", "URL", 'Tab'],
			width      : [40, 40, 20],
			keymap     : pOptions["keymap"],
			onChange : function (arg) {
				if (!arg.row) return;
//				if (beforeNode) beforeNode.classList.remove('keysnail-find-element-highlight')

				if (gBrowser.mTabContainer.selectedIndex != arg.row[4]) {
					gBrowser.mTabContainer.selectedIndex = arg.row[4];
					gPrompt.focus();
				}
				try {
//					let beforeNode = arg.row[5].startContainer;
//					if (beforeNode) beforeNode.classList.add('keysnail-find-element-highlight');
					selectAndScroll(arg.row[5]);
				} catch(e) {
					display.echoStatusBar(M({ja: '非表示リンクです', en: 'Not visible'}), 3000);
				}
			},
			actions    : [
				[
					function (index) {
						let node = _collection[index][5].startContainer;
						if (plugins.hok)
							plugins.hok.followLink(node, 4);
						else
							node.focus();
					}, "Open or Focus", "Open-or-focus"
				]
			]
		});
	}
};

plugins.withProvides(function (provide) {
	function checkArgAndGo(arg, fn, args) {
		if (typeof arg === 'string' && arg.length > 0)
			fn(arg, args);
		else
			prompt.read("search:", function (word) {
				if (word)
					fn.apply(this, [word].concat(args || []));
			}, null, null, null, 0);
	}

	provide("find-current-tab",
		function (ev, arg) {
			checkArgAndGo(arg, grepText);
		},
		M({ja: "Find - 現在のタブを検索", en: "Find - find current tab"}));

	provide("find-all-tab",
		function (ev, arg) {
			checkArgAndGo(arg, grepText, [true]);
		},
		M({ja: "Find - 全てのタブを検索", en: "find - find all tab"}));

	provide("find-current-tab-link-text",
		function (ev, arg) {
			checkArgAndGo(arg, grepLink, [false, true]);
		},
		M({ja: "Find - 現在のタブのリンクテキストを検索", en: "Find - find current tab link text"}));

	provide("find-all-tab-link-text",
		function (ev, arg) {
			checkArgAndGo(arg, grepLink, [false, true]);
		},
		M({ja: "Find - 全てのタブのリンクテキストを検索", en: "Find - find all tab link text"}));

	provide("find-current-tab-link-url",
		function (ev, arg) {
			checkArgAndGo(arg, grepLink, [false, false, true]);
		},
		M({ja: "Find - 現在のタブのリンク URL を検索", en: "Find - find current tab link url"}));

	provide("find-all-tab-link-url",
		function (ev, arg) {
			checkArgAndGo(arg, grepLink, [true, false, true]);
		},
		M({ja: "Find - 全てのタブのリンク URL を検索", en: "Find - find all tab link url"}));

	provide("find-current-tab-link-text-and-url",
		function (ev, arg) {
			checkArgAndGo(arg, grepLink, [false, true, true]);
		},
		M({ja: "Find - 現在のタブのリンクテキストと URL を検索", en: "Find - find current tab link text and url"}));

	provide("find-all-tab-link-text-and-url",
		function (ev, arg) {
			checkArgAndGo(arg, grepLink, [true, true, true]);
		},
		M({ja: "Find - 全てのタブのリンクテキストと URL を検索", en: "Find - find all tab link text and url"}));
}, PLUGIN_INFO);

shell.add([pOptions['command']], "find-text-or-link",
	function(args, extra) {
		function paramParser(arg) {
			const _param = ['*', '-l', '-lu', '-lt'];

			const _paramToCommand = {
				'find-current-tab'					: [false, false, false, false],
				'find-all-tab'						: [true,  false, false, false],
				'find-current-tab-link-text'		: [false, false, false, true ],
				'find-all-tab-link-text'			: [true,  false, false, true ],
				'find-current-tab-link-url'			: [false, false, true,  false],
				'find-all-tab-link-url'				: [true,  false, true,  false],
				'find-current-tab-link-text-and-url': [false, true,  false, false],
				'find-all-tab-link-text-and-url'	: [true,  true,  false, false]
			};

			let flags = [false, false, false, false];
			for (let i = 1; i < arg.length; i++) {
				for (let p = 0; p < _param.length; p++) {
					if (arg[i] == _param[p]) {
						flags[p] = true;
						break;
					}
				}
			}

			let ret = '';
			for (let i in _paramToCommand) {
				let result = true;
				for (let j = 0; j < flags.length; j++) {
					if (flags[j] != _paramToCommand[i][j]) {
						result = false;
						break;
					}
				}
				if (result == true) {
					ret = i;
				}
			}
			return ret;
		}
		let cmd;
		if (args.length == 1) {
			cmd = "find-current-tab";
		} else {
			cmd  = paramParser(args);
		}
		ext.exec(cmd, args[0]);
	},
	{ argCount: '*' }, true
);

function selectAndScroll(range, start, end) {
	let doc = range.startContainer.ownerDocument;
	let selection = doc.defaultView.getSelection();
	selection.removeAllRanges();

	selection.addRange(range);
	selection
		.QueryInterface(Components.interfaces.nsISelection2)
		.scrollIntoView(
			Components.interfaces.nsISelectionController.SELECTION_ANCHOR_REGION
				,true,50,50);
}

// http://d.hatena.ne.jp/amachang/20090917/1253179486
function escapeXPathExpr(text) {
	var matches = text.match(/[^"]+|"/g);
	function esc(t) {
		return t == '"' ? ('\'' + t + '\'') : ('"' + t + '"');
	}
	if (matches) {
		if (matches.length == 1) {
			return esc(matches[0]);
		} else {
			var results = [];
			for (var i = 0, len = matches.length; i < len; i ++) {
				results.push(esc(matches[i]));
			}
			return 'concat(' + results.join(', ') + ')';
		}
	} else {
		return '""';
	}
}

function $X(exp, context, resolver, result_type) {
	context || (context = document);
	var Doc = context.ownerDocument || context;
	var result = Doc.evaluate(exp, context, resolver, result_type || XPathResult.ORDERED_NODE_SNAPSHOT_TYPE, null);
	if (result_type) return result;
	for (var i = 0, len = result.snapshotLength, res = new Array(len); i < len; i++) {
		res[i] = result.snapshotItem(i);
	}
	return res;
}

//http://efcl.info/2011/0402/res2453/
function fbug(x) {
    var args = Array.slice(arguments);
    var windowManager = Components.classes['@mozilla.org/appshell/window-mediator;1']
                        .getService(Components.interfaces.nsIWindowMediator);
    var {Firebug} = windowManager.getMostRecentWindow("navigator:browser");
    if (Firebug.Console.isEnabled() && Firebug.toggleBar(true, 'console')) {
        Firebug.Console.logFormatted(args);
    }
    return args.length > 1 ? args : args[0];
}