import { injectable } from 'tsyringe'; import CompletionGroup from '../domains/CompletionGroup'; import CommandDocs from '../domains/CommandDocs'; import CompletionsRepository from '../repositories/CompletionsRepository'; import * as filters from './filters'; import SettingRepository from '../repositories/SettingRepository'; import TabPresenter from '../presenters/TabPresenter'; import Properties from '../../shared/settings/Properties'; const COMPLETION_ITEM_LIMIT = 10; type Tab = browser.tabs.Tab; type HistoryItem = browser.history.HistoryItem; @injectable() export default class CompletionsUseCase { constructor( private tabPresenter: TabPresenter, private completionsRepository: CompletionsRepository, private settingRepository: SettingRepository, ) { } queryConsoleCommand(prefix: string): Promise { let keys = Object.keys(CommandDocs); let items = keys .filter(name => name.startsWith(prefix)) .map(name => ({ caption: name, content: name, url: CommandDocs[name], })); if (items.length === 0) { return Promise.resolve([]); } return Promise.resolve([{ name: 'Console Command', items }]); } async queryOpen(name: string, keywords: string): Promise { // TODO This logic contains view entities. They should be defined on // content script let settings = await this.settingRepository.get(); let groups: CompletionGroup[] = []; let complete = settings.properties.complete; for (let c of complete) { if (c === 's') { // eslint-disable-next-line no-await-in-loop let engines = await this.querySearchEngineItems(name, keywords); if (engines.length > 0) { groups.push({ name: 'Search Engines', items: engines }); } // browser.history not supported on Android } else if (c === 'h' && typeof browser.history === 'object') { // eslint-disable-next-line no-await-in-loop let histories = await this.queryHistoryItems(name, keywords); if (histories.length > 0) { groups.push({ name: 'History', items: histories }); } // browser.bookmarks not supported on Android } else if (c === 'b' && typeof browser.bookmarks === 'object') { // eslint-disable-next-line no-await-in-loop let bookmarks = await this.queryBookmarkItems(name, keywords); if (bookmarks.length > 0) { groups.push({ name: 'Bookmarks', items: bookmarks }); } } } return groups; } // eslint-disable-next-line max-statements async queryBuffer( name: string, keywords: string, ): Promise { let lastId = await this.tabPresenter.getLastSelectedId(); let trimmed = keywords.trim(); let tabs: Tab[] = []; if (trimmed.length > 0 && !isNaN(Number(trimmed))) { let all = await this.tabPresenter.getAll(); let index = parseInt(trimmed, 10) - 1; if (index >= 0 && index < all.length) { tabs = [all[index]]; } } else if (trimmed === '%') { let all = await this.tabPresenter.getAll(); let tab = all.find(t => t.active) as Tab; tabs = [tab]; } else if (trimmed === '#') { if (typeof lastId !== 'undefined' && lastId !== null) { let all = await this.tabPresenter.getAll(); let tab = all.find(t => t.id === lastId) as Tab; tabs = [tab]; } } else { tabs = await this.completionsRepository.queryTabs(keywords, false); } const flag = (tab: Tab) => { if (tab.active) { return '%'; } else if (tab.id === lastId) { return '#'; } return ' '; }; let items = tabs.map(tab => ({ caption: tab.index + 1 + ': ' + flag(tab) + ' ' + tab.title, content: name + ' ' + tab.title, url: tab.url, icon: tab.favIconUrl, })); if (items.length === 0) { return Promise.resolve([]); } return [{ name: 'Buffers', items }]; } queryBdelete(name: string, keywords: string): Promise { return this.queryTabs(name, true, keywords); } queryBdeleteForce( name: string, keywords: string, ): Promise { return this.queryTabs(name, false, keywords); } querySet(name: string, keywords: string): Promise { let items = Properties.defs().map((def) => { if (def.type === 'boolean') { return [ { caption: def.name, content: name + ' ' + def.name, url: 'Enable ' + def.description, }, { caption: 'no' + def.name, content: name + ' no' + def.name, url: 'Disable ' + def.description } ]; } return [ { caption: def.name, content: name + ' ' + def.name, url: 'Set ' + def.description, } ]; }); let flatten = items.reduce((acc, val) => acc.concat(val), []); flatten = flatten.filter((item) => { return item.caption.startsWith(keywords); }); if (flatten.length === 0) { return Promise.resolve([]); } return Promise.resolve( [{ name: 'Properties', items: flatten }], ); } async queryTabs( name: string, excludePinned: boolean, args: string, ): Promise { let tabs = await this.completionsRepository.queryTabs(args, excludePinned); let items = tabs.map(tab => ({ caption: tab.title, content: name + ' ' + tab.title, url: tab.url, icon: tab.favIconUrl })); if (items.length === 0) { return Promise.resolve([]); } return [{ name: 'Buffers', items }]; } async querySearchEngineItems(name: string, keywords: string) { let settings = await this.settingRepository.get(); let engines = Object.keys(settings.search.engines) .filter(key => key.startsWith(keywords)); return engines.map(key => ({ caption: key, content: name + ' ' + key, })); } async queryHistoryItems(name: string, keywords: string) { let histories = await this.completionsRepository.queryHistories(keywords); histories = [histories] .map(filters.filterBlankTitle) .map(filters.filterHttp) .map(filters.filterByTailingSlash) .map(pages => filters.filterByPathname(pages, COMPLETION_ITEM_LIMIT)) .map(pages => filters.filterByOrigin(pages, COMPLETION_ITEM_LIMIT))[0] .sort((x: HistoryItem, y: HistoryItem): number => { return Number(y.visitCount) - Number(x.visitCount); }) .slice(0, COMPLETION_ITEM_LIMIT); return histories.map(page => ({ caption: page.title, content: name + ' ' + page.url, url: page.url })); } async queryBookmarkItems(name: string, keywords: string) { let bookmarks = await this.completionsRepository.queryBookmarks(keywords); return bookmarks.slice(0, COMPLETION_ITEM_LIMIT) .map(page => ({ caption: page.title, content: name + ' ' + page.url, url: page.url })); } async queryProxyItems(name: string, keywords: string) { let proxies = [{ caption: "burp", url: "127.0.0.1:8080" },{ caption: "none", url: "none" } ].filter(proxy=>{ if( keywords == '' ){ return true; } if( proxy.caption.startsWith(keywords) ){ return true; } if( proxy.url.startsWith(keywords) ){ return true; } return false; }); return proxies; return proxies.map(proxy => ({ caption: proxy.caption, content: name + ' ' + proxy.url, url: proxy.url })); } async queryProxy(name: string, keywords: string): Promise { let groups: CompletionGroup[] = []; let items = await this.queryProxyItems( name, keywords ); groups.push({ name: 'Saved Proxies', items: items }); return groups; } }