context-menu-manager.js

/*
This file is part of Talkie -- text-to-speech browser extension button.
<https://joelpurra.com/projects/talkie/>

Copyright (c) 2016, 2017 Joel Purra <https://joelpurra.com/>

Talkie 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.

Talkie 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
along with Talkie.  If not, see <https://www.gnu.org/licenses/>.
*/

import {
    promiseTry,
} from "../shared/promise";

import {
    logDebug,
} from "../shared/log";

export default class ContextMenuManager {
    constructor(commandHandler, metadataManager) {
        this.commandHandler = commandHandler;
        this.metadataManager = metadataManager;

        if (!isNaN(browser.contextMenus.ACTION_MENU_TOP_LEVEL_LIMIT) && browser.contextMenus.ACTION_MENU_TOP_LEVEL_LIMIT > 0) {
            this.actionMenuLimit = browser.contextMenus.ACTION_MENU_TOP_LEVEL_LIMIT;
        } else {
            this.actionMenuLimit = Number.MAX_SAFE_INTEGER;
        }

        this.contextMenuOptionsCollection = [
            {
                free: true,
                premium: true,
                chrome: true,
                webextension: true,
                item: {
                    id: "talkie-context-menu-start-stop",
                    title: browser.i18n.getMessage("contextMenuStartStopText"),
                    contexts: [
                        "selection",
                    ],
                },
            },
            {
                free: true,
                premium: true,
                chrome: true,
                webextension: true,
                item: {
                    id: "start-stop",
                    title: browser.i18n.getMessage("commandStartStopDescription"),
                    contexts: [
                        "browser_action",
                    ],
                },
            },
            {
                free: true,
                premium: true,
                chrome: true,
                // TODO: enable after Firefox 55 has landed?
                webextension: false,
                item: {
                    id: "read-clipboard",
                    title: browser.i18n.getMessage("commandReadClipboardDescription"),
                    contexts: [
                        "browser_action",
                        "page",
                    ],
                },
            },
            {
                free: true,
                premium: true,
                chrome: true,
                webextension: true,
                item: {
                    id: "buttonContextMenuSeparator01",
                    type: "separator",
                    contexts: [
                        "browser_action",
                    ],
                },
            },
            {
                free: true,
                premium: true,
                chrome: true,
                webextension: true,
                item: {
                    id: "open-website-main",
                    title: browser.i18n.getMessage("commandOpenWebsiteMainDescription"),
                    contexts: [
                        "browser_action",
                    ],
                },
            },
            {
                free: true,
                premium: false,
                chrome: true,
                webextension: true,
                item: {
                    id: "open-website-store-premium",
                    title: browser.i18n.getMessage("commandOpenWebsiteStoreDescription_Premium"),
                    contexts: [
                        "browser_action",
                    ],
                },
            },
            {
                free: true,
                premium: false,
                chrome: true,
                webextension: true,
                item: {
                    id: "open-website-donate",
                    title: browser.i18n.getMessage("commandOpenWebsiteDonateDescription"),
                    contexts: [
                        "browser_action",
                    ],
                },
            },
        ];
    }

    removeAll() {
        return promiseTry(
            () => {
                logDebug("Start", "Removing all context menus");

                return browser.contextMenus.removeAll()
                    .then((result) => {
                        logDebug("Done", "Removing all context menus");

                        return result;
                    });
            }
        );
    }

    createContextMenu(contextMenuOptions) {
        return new Promise(
            (resolve, reject) => {
                try {
                    logDebug("Start", "Creating context menu", contextMenuOptions);

                    // NOTE: apparently Chrome modifies the context menu object after it has been passed in, by adding generatedId.
                    // NOTE: Need to pass a clean object to avoid object reuse reference problems.
                    const contextMenu = Object.assign({}, contextMenuOptions);

                    // NOTE: Can't directly use a promise chain here, as the id is returned instead.
                    // https://github.com/mozilla/webextension-polyfill/pull/26
                    const contextMenuId = browser.contextMenus.create(
                        contextMenu,
                        () => {
                            if (browser.runtime.lastError) {
                                return reject(browser.runtime.lastError);
                            }

                            logDebug("Done", "Creating context menu", contextMenu);

                            return resolve(contextMenuId);
                        }
                    );
                } catch (error) {
                    return reject(error);
                }
            }
        );
    }

    contextMenuClickAction(info) {
        return promiseTry(
            () => {
                logDebug("Start", "contextMenuClickAction", info);

                if (!info) {
                    throw new Error("Unknown context menu click action info object.");
                }

                return promiseTry(
                    () => {
                        const id = info.menuItemId;

                        const selectionContextMenuStartStop = this.contextMenuOptionsCollection
                            .reduce(
                                (prev, obj) => {
                                    if (obj.item.id === "talkie-context-menu-start-stop") {
                                        return obj;
                                    }

                                    return prev;
                                },
                                null);

                        // TODO: use assertions?
                        if (!selectionContextMenuStartStop) {
                            throw new Error("Not found: selectionContextMenuStartStop");
                        }

                        if (id === selectionContextMenuStartStop.item.id) {
                            const selection = info.selectionText || null;

                            if (!selection || typeof selection !== "string" || selection.length === 0) {
                                throw new Error("Unknown context menu click action selection was empty.");
                            }

                            return this.commandHandler.handle("start-text", selection);
                        }

                        // NOTE: context menu items default to being commands.
                        return this.commandHandler.handle(id);
                    }
                )
                    .then(() => {
                        logDebug("Done", "contextMenuClickAction", info);

                        return undefined;
                    });
            }
        );
    }

    createContextMenus() {
        return promiseTry(
            () => {
                return Promise.all([
                    this.metadataManager.getVersionType(),
                    this.metadataManager.getSystemType(),
                ])
                    .then(([versionType, systemType]) => {
                        const applicableContextMenuOptions = this.contextMenuOptionsCollection
                            .filter((contextMenuOption) => contextMenuOption[versionType] === true && contextMenuOption[systemType] === true);

                        // // TODO: group by selected contexts before checking against limit.
                        // if (applicableContextMenuOptions > this.actionMenuLimit) {
                        //     throw new Error("Maximum number of menu items reached.");
                        // }

                        const contextMenuOptionsCollectionPromises = applicableContextMenuOptions
                            .map((contextMenuOption) => this.createContextMenu(contextMenuOption.item));

                        return Promise.all(contextMenuOptionsCollectionPromises);
                    });
            }
        );
    }
}