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

198 lines
6.8 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-01-16 21:22:38 +00:00
super(document.querySelector('#query-parser'), [], [], []);
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
2019-10-29 21:49:36 +00:00
this.queryParser = document.querySelector('#query-parser');
2019-11-12 21:57:21 +00:00
this.queryParserSelect = document.querySelector('#query-parser-select');
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());
this.renderParseResult(this.getParseResult());
}
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)]
];
}
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);
await this.setPreview(text);
this.parseResults = await this.parseText(text);
this.refreshSelectedParser();
2019-11-12 21:57:21 +00:00
this.renderParserSelect();
await this.renderParseResult();
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
}
async 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);
previewTerms.push([{text: tempText.split('')}]);
}
this.queryParser.innerHTML = await apiTemplateRender('query-parser.html', {
terms: previewTerms,
preview: true
});
}
2019-11-12 21:57:21 +00:00
renderParserSelect() {
this.queryParserSelect.innerHTML = '';
if (this.parseResults.length > 1) {
const select = document.createElement('select');
select.classList.add('form-control');
for (const parseResult of this.parseResults) {
const option = document.createElement('option');
option.value = parseResult.id;
option.innerText = parseResult.name;
option.defaultSelected = this.selectedParser === parseResult.id;
2019-11-12 21:57:21 +00:00
select.appendChild(option);
}
select.addEventListener('change', this.onParserChange.bind(this));
this.queryParserSelect.appendChild(select);
}
}
async renderParseResult() {
const parseResult = this.getParseResult();
if (!parseResult) {
this.queryParser.innerHTML = '';
return;
}
this.queryParser.innerHTML = await apiTemplateRender(
'query-parser.html',
{terms: QueryParser.processParseResultForDisplay(parseResult.parsedText)}
);
}
2019-11-12 21:57:21 +00:00
static processParseResultForDisplay(result) {
return result.map((term) => {
2019-11-27 03:01:54 +00:00
return term.filter((part) => part.text.trim()).map((part) => {
2019-11-12 21:57:21 +00:00
return {
2019-12-08 20:20:44 +00:00
text: part.text.split(''),
2019-11-12 21:57:21 +00:00
reading: part.reading,
2019-11-25 19:41:26 +00:00
raw: !part.reading || !part.reading.trim()
2019-11-12 21:57:21 +00:00
};
});
});
}
2019-10-29 21:49:36 +00:00
}