mirror of https://github.com/keeweb/keeweb.git
100 lines
3.4 KiB
JavaScript
100 lines
3.4 KiB
JavaScript
import { SearchResultCollection } from 'collections/search-result-collection';
|
|
import { Ranking } from 'util/data/ranking';
|
|
|
|
const urlPartsRegex = /^(\w+:\/\/)?(?:(?:www|wwws|secure)\.)?([^\/]+)\/?(.*)/;
|
|
|
|
class SelectEntryFilter {
|
|
constructor(windowInfo, appModel, files, filterOptions) {
|
|
this.title = windowInfo.title;
|
|
this.useTitle = !!windowInfo.title && !windowInfo.url;
|
|
this.url = windowInfo.url;
|
|
this.useUrl = !!windowInfo.url;
|
|
this.subdomains = true;
|
|
this.text = '';
|
|
this.appModel = appModel;
|
|
this.files = files;
|
|
this.filterOptions = filterOptions;
|
|
}
|
|
|
|
getEntries() {
|
|
const filter = {
|
|
text: this.text,
|
|
...this.filterOptions
|
|
};
|
|
let entries = this.appModel
|
|
.getEntriesByFilter(filter, this.files)
|
|
.map((e) => [e, this._getEntryRank(e)]);
|
|
if (this.useUrl || this.useTitle) {
|
|
entries = entries.filter((e) => e[1]);
|
|
}
|
|
entries = entries.sort((x, y) =>
|
|
x[1] === y[1] ? x[0].title.localeCompare(y[0].title) : y[1] - x[1]
|
|
);
|
|
entries = entries.map((p) => p[0]);
|
|
return new SearchResultCollection(entries, { comparator: 'none' });
|
|
}
|
|
|
|
_getEntryRank(entry) {
|
|
let titleRank = 0;
|
|
let urlRank = 0;
|
|
|
|
if (this.useTitle && this.title && entry.title) {
|
|
titleRank = Ranking.getStringRank(entry.title.toLowerCase(), this.title.toLowerCase());
|
|
if (!titleRank) {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
if (this.useUrl && this.url) {
|
|
const searchUrlLower = this.url.toLowerCase();
|
|
const searchUrlParts = urlPartsRegex.exec(searchUrlLower);
|
|
|
|
for (const url of entry.getAllUrls()) {
|
|
const entryUrlParts = urlPartsRegex.exec(url.toLowerCase());
|
|
if (entryUrlParts) {
|
|
const [, scheme, domain, path] = entryUrlParts;
|
|
const [, searchScheme, searchDomain, searchPath] = searchUrlParts;
|
|
if (
|
|
domain === searchDomain ||
|
|
(this.subdomains && searchDomain.indexOf('.' + domain) > 0)
|
|
) {
|
|
if (domain === searchDomain) {
|
|
urlRank += 20;
|
|
} else {
|
|
urlRank += 10;
|
|
}
|
|
if (path === searchPath) {
|
|
urlRank += 10;
|
|
} else if (path && searchPath) {
|
|
if (path.lastIndexOf(searchPath, 0) === 0) {
|
|
urlRank += 5;
|
|
} else if (searchPath.lastIndexOf(path, 0) === 0) {
|
|
urlRank += 3;
|
|
}
|
|
}
|
|
if (scheme === searchScheme) {
|
|
urlRank += 1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (entry.searchText.includes(searchUrlLower)) {
|
|
// the url is in some field; include it
|
|
urlRank += 5;
|
|
}
|
|
}
|
|
|
|
if (this.useTitle && !titleRank) {
|
|
return 0;
|
|
}
|
|
if (this.useUrl && !urlRank) {
|
|
return 0;
|
|
}
|
|
|
|
return titleRank + urlRank;
|
|
}
|
|
}
|
|
|
|
export { SelectEntryFilter };
|