import MessageListener from '../../MessageListener'; import Hint from './hint'; import * as dom from '../../../shared/utils/dom'; import * as messages from '../../../shared/messages'; import * as keyUtils from '../../../shared/utils/keys'; const TARGET_SELECTOR = [ 'a', 'button', 'input', 'textarea', 'area', '[contenteditable=true]', '[contenteditable=""]', '[tabindex]', '[role="button"]', 'summary' ].join(','); interface Size { width: number; height: number; } interface Point { x: number; y: number; } const inViewport = ( win: Window, element: Element, viewSize: Size, framePosition: Point, ): boolean => { let { top, left, bottom, right } = dom.viewportRect(element); let doc = win.document; let frameWidth = doc.documentElement.clientWidth; let frameHeight = doc.documentElement.clientHeight; if (right < 0 || bottom < 0 || top > frameHeight || left > frameWidth) { // out of frame return false; } if (right + framePosition.x < 0 || bottom + framePosition.y < 0 || left + framePosition.x > viewSize.width || top + framePosition.y > viewSize.height) { // out of viewport return false; } return true; }; const isAriaHiddenOrAriaDisabled = (win: Window, element: Element): boolean => { if (!element || win.document.documentElement === element) { return false; } for (let attr of ['aria-hidden', 'aria-disabled']) { let value = element.getAttribute(attr); if (value !== null) { let hidden = value.toLowerCase(); if (hidden === '' || hidden === 'true') { return true; } } } return isAriaHiddenOrAriaDisabled(win, element.parentElement as Element); }; export default class Follow { private win: Window; private newTab: boolean; private background: boolean; private hints: {[key: string]: Hint }; private targets: HTMLElement[] = []; constructor(win: Window) { this.win = win; this.newTab = false; this.background = false; this.hints = {}; this.targets = []; new MessageListener().onWebMessage(this.onMessage.bind(this)); } key(key: keyUtils.Key): boolean { if (Object.keys(this.hints).length === 0) { return false; } this.win.parent.postMessage(JSON.stringify({ type: messages.FOLLOW_KEY_PRESS, key: key.key, ctrlKey: key.ctrlKey, }), '*'); return true; } openLink(element: HTMLAreaElement|HTMLAnchorElement) { // Browser prevent new tab by link with target='_blank' if (!this.newTab && element.getAttribute('target') !== '_blank') { element.click(); return; } let href = element.getAttribute('href'); // eslint-disable-next-line no-script-url if (!href || href === '#' || href.toLowerCase().startsWith('javascript:')) { return; } return browser.runtime.sendMessage({ type: messages.OPEN_URL, url: element.href, newTab: true, background: this.background, }); } countHints(sender: any, viewSize: Size, framePosition: Point) { this.targets = Follow.getTargetElements(this.win, viewSize, framePosition); sender.postMessage(JSON.stringify({ type: messages.FOLLOW_RESPONSE_COUNT_TARGETS, count: this.targets.length, }), '*'); } createHints(keysArray: string[], newTab: boolean, background: boolean) { if (keysArray.length !== this.targets.length) { throw new Error('illegal hint count'); } this.newTab = newTab; this.background = background; this.hints = {}; for (let i = 0; i < keysArray.length; ++i) { let keys = keysArray[i]; let hint = new Hint(this.targets[i], keys); this.hints[keys] = hint; } } showHints(keys: string) { Object.keys(this.hints).filter(key => key.startsWith(keys)) .forEach(key => this.hints[key].show()); Object.keys(this.hints).filter(key => !key.startsWith(keys)) .forEach(key => this.hints[key].hide()); } removeHints() { Object.keys(this.hints).forEach((key) => { this.hints[key].remove(); }); this.hints = {}; this.targets = []; } activateHints(keys: string) { let hint = this.hints[keys]; if (!hint) { return; } let element = hint.getTarget(); switch (element.tagName.toLowerCase()) { case 'a': return this.openLink(element as HTMLAnchorElement); case 'area': return this.openLink(element as HTMLAreaElement); case 'input': switch ((element as HTMLInputElement).type) { case 'file': case 'checkbox': case 'radio': case 'submit': case 'reset': case 'button': case 'image': case 'color': return element.click(); default: return element.focus(); } case 'textarea': return element.focus(); case 'button': case 'summary': return element.click(); default: if (dom.isContentEditable(element)) { return element.focus(); } else if (element.hasAttribute('tabindex')) { return element.click(); } } } onMessage(message: messages.Message, sender: any) { switch (message.type) { case messages.FOLLOW_REQUEST_COUNT_TARGETS: return this.countHints(sender, message.viewSize, message.framePosition); case messages.FOLLOW_CREATE_HINTS: return this.createHints( message.keysArray, message.newTab, message.background); case messages.FOLLOW_SHOW_HINTS: return this.showHints(message.keys); case messages.FOLLOW_ACTIVATE: return this.activateHints(message.keys); case messages.FOLLOW_REMOVE_HINTS: return this.removeHints(); } } static getTargetElements( win: Window, viewSize: Size, framePosition: Point, ): HTMLElement[] { let all = win.document.querySelectorAll(TARGET_SELECTOR); let filtered = Array.prototype.filter.call(all, (element: HTMLElement) => { let style = win.getComputedStyle(element); // AREA's 'display' in Browser style is 'none' return (element.tagName === 'AREA' || style.display !== 'none') && style.visibility !== 'hidden' && (element as HTMLInputElement).type !== 'hidden' && element.offsetHeight > 0 && !isAriaHiddenOrAriaDisabled(win, element) && inViewport(win, element, viewSize, framePosition); }); return filtered; } }