yomichan/ext/bg/js/search-query-parser.js

181 lines
6.2 KiB
JavaScript
Raw Normal View History

2019-10-29 21:49:36 +00:00
/*
2020-01-01 17:00:00 +00:00
* Copyright (C) 2019-2020 Alex Yatskov <alex@foosoft.net>
2019-10-29 21:49:36 +00:00
* Author: Alex Yatskov <alex@foosoft.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
2020-01-01 17:00:31 +00:00
* along with this program. If not, see <https://www.gnu.org/licenses/>.
2019-10-29 21:49:36 +00:00
*/
2020-01-16 21:22:38 +00:00
class QueryParser extends TextScanner {
2019-10-29 21:49:36 +00:00
constructor(search) {
2020-02-06 02:00:02 +00:00
super(document.querySelector('#query-parser-content'), [], [], []);
2019-10-29 21:49:36 +00:00
this.search = search;
2019-11-12 21:57:21 +00:00
this.parseResults = [];
this.selectedParser = null;
2019-11-12 21:57:21 +00:00
2020-02-06 02:00:02 +00:00
this.queryParser = document.querySelector('#query-parser-content');
this.queryParserSelect = document.querySelector('#query-parser-select-container');
this.queryParserGenerator = new QueryParserGenerator();
2019-10-29 21:49:36 +00:00
}
2020-02-10 20:09:23 +00:00
async prepare() {
await this.queryParserGenerator.prepare();
}
2019-10-29 21:49:36 +00:00
onError(error) {
logError(error, false);
}
2020-01-16 21:22:38 +00:00
onClick(e) {
super.onClick(e);
this.searchAt(e.clientX, e.clientY, 'click');
}
2020-01-16 21:22:38 +00:00
async onSearchSource(textSource, cause) {
if (textSource === null) { return null; }
2019-10-30 01:58:24 +00:00
2020-01-16 21:22:38 +00:00
this.setTextSourceScanLength(textSource, this.search.options.scanning.length);
const searchText = textSource.text();
if (searchText.length === 0) { return; }
2019-10-30 01:58:24 +00:00
2020-01-16 21:22:38 +00:00
const {definitions, length} = await apiTermsFind(searchText, {}, this.search.getOptionsContext());
if (definitions.length === 0) { return null; }
2019-10-30 01:58:24 +00:00
2020-01-16 21:22:38 +00:00
textSource.setEndOffset(length);
2019-10-30 01:58:24 +00:00
2020-01-16 21:22:38 +00:00
this.search.setContent('terms', {definitions, context: {
focus: false,
disableHistory: cause === 'mouse' ? true : false,
sentence: {text: searchText, offset: 0},
url: window.location.href
}});
2020-01-16 21:22:38 +00:00
return {definitions, type: 'terms'};
2019-10-29 21:49:36 +00:00
}
2019-11-13 11:11:41 +00:00
onParserChange(e) {
2019-11-12 21:57:21 +00:00
const selectedParser = e.target.value;
this.selectedParser = selectedParser;
2019-11-12 21:57:21 +00:00
apiOptionsSet({parsing: {selectedParser}}, this.search.getOptionsContext());
2020-02-06 02:00:02 +00:00
this.renderParseResult();
2019-11-12 21:57:21 +00:00
}
2020-01-16 21:22:38 +00:00
getMouseEventListeners() {
return [
[this.node, 'click', this.onClick.bind(this)],
[this.node, 'mousedown', this.onMouseDown.bind(this)],
[this.node, 'mousemove', this.onMouseMove.bind(this)],
[this.node, 'mouseover', this.onMouseOver.bind(this)],
[this.node, 'mouseout', this.onMouseOut.bind(this)]
];
}
getTouchEventListeners() {
return [
[this.node, 'auxclick', this.onAuxClick.bind(this)],
[this.node, 'touchstart', this.onTouchStart.bind(this)],
[this.node, 'touchend', this.onTouchEnd.bind(this)],
[this.node, 'touchcancel', this.onTouchCancel.bind(this)],
[this.node, 'touchmove', this.onTouchMove.bind(this), {passive: false}],
[this.node, 'contextmenu', this.onContextMenu.bind(this)]
];
}
setOptions(options) {
super.setOptions(options);
this.queryParser.dataset.termSpacing = `${options.parsing.termSpacing}`;
}
2019-11-13 13:29:53 +00:00
refreshSelectedParser() {
2019-11-12 21:57:21 +00:00
if (this.parseResults.length > 0) {
if (this.selectedParser === null) {
this.selectedParser = this.search.options.parsing.selectedParser;
}
if (this.selectedParser === null || !this.getParseResult()) {
2019-11-12 21:57:21 +00:00
const selectedParser = this.parseResults[0].id;
this.selectedParser = selectedParser;
2019-11-12 21:57:21 +00:00
apiOptionsSet({parsing: {selectedParser}}, this.search.getOptionsContext());
}
}
2019-11-13 13:29:53 +00:00
}
getParseResult() {
2019-11-27 03:01:54 +00:00
return this.parseResults.find((r) => r.id === this.selectedParser);
2019-11-13 13:29:53 +00:00
}
async setText(text) {
this.search.setSpinnerVisible(true);
2020-02-06 02:00:02 +00:00
this.setPreview(text);
2019-11-13 13:29:53 +00:00
this.parseResults = await this.parseText(text);
this.refreshSelectedParser();
2019-11-12 21:57:21 +00:00
this.renderParserSelect();
2020-02-06 02:00:02 +00:00
this.renderParseResult();
2019-11-12 21:57:21 +00:00
this.search.setSpinnerVisible(false);
}
async parseText(text) {
const results = [];
2019-11-05 13:56:45 +00:00
if (this.search.options.parsing.enableScanningParser) {
2019-11-12 21:57:21 +00:00
results.push({
name: 'Scanning parser',
id: 'scan',
parsedText: await apiTextParse(text, this.search.getOptionsContext())
});
2019-11-05 13:56:45 +00:00
}
if (this.search.options.parsing.enableMecabParser) {
const mecabResults = await apiTextParseMecab(text, this.search.getOptionsContext());
2019-11-05 13:56:45 +00:00
for (const mecabDictName in mecabResults) {
2019-11-12 21:57:21 +00:00
results.push({
name: `MeCab: ${mecabDictName}`,
id: `mecab-${mecabDictName}`,
parsedText: mecabResults[mecabDictName]
});
2019-11-05 13:56:45 +00:00
}
}
2019-11-12 21:57:21 +00:00
return results;
2019-10-30 01:58:24 +00:00
}
2020-02-06 02:00:02 +00:00
setPreview(text) {
const previewTerms = [];
2019-12-08 20:13:22 +00:00
for (let i = 0, ii = text.length; i < ii; i += 2) {
const tempText = text.substring(i, i + 2);
2020-02-06 02:00:02 +00:00
previewTerms.push([{text: tempText}]);
}
2020-02-06 02:00:02 +00:00
this.queryParser.textContent = '';
this.queryParser.appendChild(this.queryParserGenerator.createParseResult(previewTerms, true));
}
2019-11-12 21:57:21 +00:00
renderParserSelect() {
this.queryParserSelect.innerHTML = '';
if (this.parseResults.length > 1) {
2020-02-06 02:00:02 +00:00
const select = this.queryParserGenerator.createParserSelect(this.parseResults, this.selectedParser);
2019-11-12 21:57:21 +00:00
select.addEventListener('change', this.onParserChange.bind(this));
this.queryParserSelect.appendChild(select);
}
}
2020-02-06 02:00:02 +00:00
renderParseResult() {
2019-11-12 21:57:21 +00:00
const parseResult = this.getParseResult();
2020-02-06 02:00:02 +00:00
this.queryParser.textContent = '';
if (!parseResult) { return; }
this.queryParser.appendChild(this.queryParserGenerator.createParseResult(parseResult.parsedText));
2019-11-12 21:57:21 +00:00
}
2019-10-29 21:49:36 +00:00
}