ui-menus.js 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604
  1. /*
  2. * Copyright 2010-2020 Gildas Lormeau
  3. * contact : gildas.lormeau <at> gmail.com
  4. *
  5. * This file is part of SingleFile.
  6. *
  7. * The code in this file is free software: you can redistribute it and/or
  8. * modify it under the terms of the GNU Affero General Public License
  9. * (GNU AGPL) as published by the Free Software Foundation, either version 3
  10. * of the License, or (at your option) any later version.
  11. *
  12. * The code in this file is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero
  15. * General Public License for more details.
  16. *
  17. * As additional permission under GNU AGPL version 3 section 7, you may
  18. * distribute UNMODIFIED VERSIONS OF THIS file without the copy of the GNU
  19. * AGPL normally required by section 4, provided you include this license
  20. * notice and a URL through which recipients can access the Corresponding
  21. * Source.
  22. */
  23. /* global browser, URL */
  24. import * as config from "./../../core/bg/config.js";
  25. import { queryTabs } from "./../../core/bg/tabs-util.js";
  26. import * as tabsData from "./../../core/bg/tabs-data.js";
  27. import { refreshAutoSaveTabs } from "./../../core/bg/autosave-util.js";
  28. import * as button from "./ui-button.js";
  29. const menus = browser.menus || browser.contextMenus;
  30. const BROWSER_MENUS_API_SUPPORTED = menus && menus.onClicked && menus.create && menus.update && menus.removeAll;
  31. const MENU_ID_SAVE_PAGE = "save-page";
  32. const MENU_ID_EDIT_AND_SAVE_PAGE = "edit-and-save-page";
  33. const MENU_ID_SAVE_WITH_PROFILE = "save-with-profile";
  34. const MENU_ID_SAVE_SELECTED_LINKS = "save-selected-links";
  35. const MENU_ID_VIEW_PENDINGS = "view-pendings";
  36. const MENU_ID_SELECT_PROFILE = "select-profile";
  37. const MENU_ID_SAVE_WITH_PROFILE_PREFIX = "wasve-with-profile-";
  38. const MENU_ID_SELECT_PROFILE_PREFIX = "select-profile-";
  39. const MENU_ID_ASSOCIATE_WITH_PROFILE = "associate-with-profile";
  40. const MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX = "associate-with-profile-";
  41. const MENU_ID_SAVE_SELECTED = "save-selected";
  42. const MENU_ID_SAVE_FRAME = "save-frame";
  43. const MENU_ID_SAVE_TABS = "save-tabs";
  44. const MENU_ID_SAVE_SELECTED_TABS = "save-selected-tabs";
  45. const MENU_ID_SAVE_UNPINNED_TABS = "save-unpinned-tabs";
  46. const MENU_ID_SAVE_ALL_TABS = "save-all-tabs";
  47. const MENU_ID_BUTTON_SAVE_SELECTED_TABS = "button-" + MENU_ID_SAVE_SELECTED_TABS;
  48. const MENU_ID_BUTTON_SAVE_UNPINNED_TABS = "button-" + MENU_ID_SAVE_UNPINNED_TABS;
  49. const MENU_ID_BUTTON_SAVE_ALL_TABS = "button-" + MENU_ID_SAVE_ALL_TABS;
  50. const MENU_ID_AUTO_SAVE = "auto-save";
  51. const MENU_ID_AUTO_SAVE_DISABLED = "auto-save-disabled";
  52. const MENU_ID_AUTO_SAVE_TAB = "auto-save-tab";
  53. const MENU_ID_AUTO_SAVE_UNPINNED = "auto-save-unpinned";
  54. const MENU_ID_AUTO_SAVE_ALL = "auto-save-all";
  55. const MENU_CREATE_DOMAIN_RULE_MESSAGE = browser.i18n.getMessage("menuCreateDomainRule");
  56. const MENU_UPDATE_RULE_MESSAGE = browser.i18n.getMessage("menuUpdateRule");
  57. const MENU_SAVE_PAGE_MESSAGE = browser.i18n.getMessage("menuSavePage");
  58. const MENU_SAVE_WITH_PROFILE = browser.i18n.getMessage("menuSaveWithProfile");
  59. const MENU_SAVE_SELECTED_LINKS = browser.i18n.getMessage("menuSaveSelectedLinks");
  60. const MENU_EDIT_PAGE_MESSAGE = browser.i18n.getMessage("menuEditPage");
  61. const MENU_EDIT_AND_SAVE_PAGE_MESSAGE = browser.i18n.getMessage("menuEditAndSavePage");
  62. const MENU_VIEW_PENDINGS_MESSAGE = browser.i18n.getMessage("menuViewPendingSaves");
  63. const MENU_SAVE_SELECTION_MESSAGE = browser.i18n.getMessage("menuSaveSelection");
  64. const MENU_SAVE_FRAME_MESSAGE = browser.i18n.getMessage("menuSaveFrame");
  65. const MENU_SAVE_TABS_MESSAGE = browser.i18n.getMessage("menuSaveTabs");
  66. const MENU_SAVE_SELECTED_TABS_MESSAGE = browser.i18n.getMessage("menuSaveSelectedTabs");
  67. const MENU_SAVE_UNPINNED_TABS_MESSAGE = browser.i18n.getMessage("menuSaveUnpinnedTabs");
  68. const MENU_SAVE_ALL_TABS_MESSAGE = browser.i18n.getMessage("menuSaveAllTabs");
  69. const MENU_SELECT_PROFILE_MESSAGE = browser.i18n.getMessage("menuSelectProfile");
  70. const PROFILE_DEFAULT_SETTINGS_MESSAGE = browser.i18n.getMessage("profileDefaultSettings");
  71. const MENU_AUTOSAVE_MESSAGE = browser.i18n.getMessage("menuAutoSave");
  72. const MENU_AUTOSAVE_DISABLED_MESSAGE = browser.i18n.getMessage("menuAutoSaveDisabled");
  73. const MENU_AUTOSAVE_TAB_MESSAGE = browser.i18n.getMessage("menuAutoSaveTab");
  74. const MENU_AUTOSAVE_UNPINNED_TABS_MESSAGE = browser.i18n.getMessage("menuAutoSaveUnpinnedTabs");
  75. const MENU_AUTOSAVE_ALL_TABS_MESSAGE = browser.i18n.getMessage("menuAutoSaveAllTabs");
  76. const MENU_TOP_VISIBLE_ENTRIES = [
  77. MENU_ID_EDIT_AND_SAVE_PAGE,
  78. MENU_ID_SAVE_SELECTED_LINKS,
  79. MENU_ID_SAVE_SELECTED,
  80. MENU_ID_SAVE_FRAME,
  81. MENU_ID_AUTO_SAVE,
  82. MENU_ID_ASSOCIATE_WITH_PROFILE
  83. ];
  84. const menusCheckedState = new Map();
  85. const menusTitleState = new Map();
  86. let contextMenuVisibleState = true;
  87. let allMenuVisibleState = true;
  88. let profileIndexes = new Map();
  89. let menusCreated, pendingRefresh, business;
  90. Promise.resolve().then(initialize);
  91. export {
  92. onMessage,
  93. refreshTab as onTabCreated,
  94. refreshTab as onTabActivated,
  95. refreshTab as onInit,
  96. createMenus as refreshTab,
  97. setBusiness
  98. };
  99. function setBusiness(businessApi) {
  100. business = businessApi;
  101. }
  102. function onMessage(message) {
  103. if (message.method.endsWith("refreshMenu")) {
  104. createMenus();
  105. return Promise.resolve({});
  106. }
  107. }
  108. async function createMenus(tab) {
  109. const [profiles, allTabsData] = await Promise.all([config.getProfiles(), tabsData.get()]);
  110. const options = await config.getOptions(tab && tab.url);
  111. if (BROWSER_MENUS_API_SUPPORTED && options) {
  112. const pageContextsEnabled = ["page", "frame", "image", "link", "video", "audio", "selection"];
  113. const defaultContextsDisabled = [];
  114. if (options.browserActionMenuEnabled) {
  115. defaultContextsDisabled.push("browser_action");
  116. }
  117. if (options.tabMenuEnabled) {
  118. try {
  119. menus.create({
  120. id: "temporary-id",
  121. contexts: ["tab"],
  122. title: "title"
  123. });
  124. defaultContextsDisabled.push("tab");
  125. } catch (error) {
  126. options.tabMenuEnabled = false;
  127. }
  128. }
  129. await menus.removeAll();
  130. const defaultContextsEnabled = defaultContextsDisabled.concat(...pageContextsEnabled);
  131. const defaultContexts = options.contextMenuEnabled ? defaultContextsEnabled : defaultContextsDisabled;
  132. menus.create({
  133. id: MENU_ID_SAVE_PAGE,
  134. contexts: defaultContexts,
  135. title: MENU_SAVE_PAGE_MESSAGE
  136. });
  137. menus.create({
  138. id: MENU_ID_EDIT_AND_SAVE_PAGE,
  139. contexts: defaultContexts,
  140. title: MENU_EDIT_AND_SAVE_PAGE_MESSAGE
  141. });
  142. menus.create({
  143. id: MENU_ID_SAVE_SELECTED_LINKS,
  144. contexts: options.contextMenuEnabled ? defaultContextsDisabled.concat(["selection"]) : defaultContextsDisabled,
  145. title: MENU_SAVE_SELECTED_LINKS
  146. });
  147. if (Object.keys(profiles).length > 1) {
  148. menus.create({
  149. id: MENU_ID_SAVE_WITH_PROFILE,
  150. contexts: defaultContexts,
  151. title: MENU_SAVE_WITH_PROFILE
  152. });
  153. }
  154. if (options.contextMenuEnabled) {
  155. menus.create({
  156. id: "separator-1",
  157. contexts: pageContextsEnabled,
  158. type: "separator"
  159. });
  160. }
  161. menus.create({
  162. id: MENU_ID_SAVE_SELECTED,
  163. contexts: defaultContexts,
  164. title: MENU_SAVE_SELECTION_MESSAGE
  165. });
  166. if (options.contextMenuEnabled) {
  167. menus.create({
  168. id: MENU_ID_SAVE_FRAME,
  169. contexts: ["frame"],
  170. title: MENU_SAVE_FRAME_MESSAGE
  171. });
  172. }
  173. menus.create({
  174. id: MENU_ID_SAVE_TABS,
  175. contexts: defaultContextsDisabled,
  176. title: MENU_SAVE_TABS_MESSAGE
  177. });
  178. menus.create({
  179. id: MENU_ID_BUTTON_SAVE_SELECTED_TABS,
  180. contexts: defaultContextsDisabled,
  181. title: MENU_SAVE_SELECTED_TABS_MESSAGE,
  182. parentId: MENU_ID_SAVE_TABS
  183. });
  184. menus.create({
  185. id: MENU_ID_BUTTON_SAVE_UNPINNED_TABS,
  186. contexts: defaultContextsDisabled,
  187. title: MENU_SAVE_UNPINNED_TABS_MESSAGE,
  188. parentId: MENU_ID_SAVE_TABS
  189. });
  190. menus.create({
  191. id: MENU_ID_BUTTON_SAVE_ALL_TABS,
  192. contexts: defaultContextsDisabled,
  193. title: MENU_SAVE_ALL_TABS_MESSAGE,
  194. parentId: MENU_ID_SAVE_TABS
  195. });
  196. if (options.contextMenuEnabled) {
  197. menus.create({
  198. id: MENU_ID_SAVE_SELECTED_TABS,
  199. contexts: pageContextsEnabled,
  200. title: MENU_SAVE_SELECTED_TABS_MESSAGE
  201. });
  202. menus.create({
  203. id: MENU_ID_SAVE_UNPINNED_TABS,
  204. contexts: pageContextsEnabled,
  205. title: MENU_SAVE_UNPINNED_TABS_MESSAGE
  206. });
  207. menus.create({
  208. id: MENU_ID_SAVE_ALL_TABS,
  209. contexts: pageContextsEnabled,
  210. title: MENU_SAVE_ALL_TABS_MESSAGE
  211. });
  212. menus.create({
  213. id: "separator-2",
  214. contexts: pageContextsEnabled,
  215. type: "separator"
  216. });
  217. }
  218. if (Object.keys(profiles).length > 1) {
  219. menus.create({
  220. id: MENU_ID_SELECT_PROFILE,
  221. title: MENU_SELECT_PROFILE_MESSAGE,
  222. contexts: defaultContexts,
  223. });
  224. menus.create({
  225. id: MENU_ID_SAVE_WITH_PROFILE_PREFIX + "default",
  226. contexts: defaultContexts,
  227. title: PROFILE_DEFAULT_SETTINGS_MESSAGE,
  228. parentId: MENU_ID_SAVE_WITH_PROFILE
  229. });
  230. const defaultProfileId = MENU_ID_SELECT_PROFILE_PREFIX + "default";
  231. const defaultProfileChecked = !allTabsData.profileName || allTabsData.profileName == config.DEFAULT_PROFILE_NAME;
  232. menus.create({
  233. id: defaultProfileId,
  234. type: "radio",
  235. contexts: defaultContexts,
  236. title: PROFILE_DEFAULT_SETTINGS_MESSAGE,
  237. checked: defaultProfileChecked,
  238. parentId: MENU_ID_SELECT_PROFILE
  239. });
  240. menusCheckedState.set(defaultProfileId, defaultProfileChecked);
  241. menus.create({
  242. id: MENU_ID_ASSOCIATE_WITH_PROFILE,
  243. title: MENU_CREATE_DOMAIN_RULE_MESSAGE,
  244. contexts: defaultContexts,
  245. });
  246. menusTitleState.set(MENU_ID_ASSOCIATE_WITH_PROFILE, MENU_CREATE_DOMAIN_RULE_MESSAGE);
  247. let rule;
  248. if (tab && tab.url) {
  249. rule = await config.getRule(tab.url, true);
  250. }
  251. const currentProfileId = MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + "current";
  252. const currentProfileChecked = !rule || (rule.profile == config.CURRENT_PROFILE_NAME);
  253. menus.create({
  254. id: currentProfileId,
  255. type: "radio",
  256. contexts: defaultContexts,
  257. title: config.CURRENT_PROFILE_NAME,
  258. checked: currentProfileChecked,
  259. parentId: MENU_ID_ASSOCIATE_WITH_PROFILE
  260. });
  261. menusCheckedState.set(currentProfileId, currentProfileChecked);
  262. const associatedDefaultProfileId = MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + "default";
  263. const associatedDefaultProfileChecked = Boolean(rule) && (rule.profile == config.DEFAULT_PROFILE_NAME);
  264. menus.create({
  265. id: associatedDefaultProfileId,
  266. type: "radio",
  267. contexts: defaultContexts,
  268. title: PROFILE_DEFAULT_SETTINGS_MESSAGE,
  269. checked: associatedDefaultProfileChecked,
  270. parentId: MENU_ID_ASSOCIATE_WITH_PROFILE
  271. });
  272. menusCheckedState.set(associatedDefaultProfileId, associatedDefaultProfileChecked);
  273. profileIndexes = new Map();
  274. Object.keys(profiles).forEach((profileName, profileIndex) => {
  275. if (profileName != config.DEFAULT_PROFILE_NAME) {
  276. let profileId = MENU_ID_SAVE_WITH_PROFILE_PREFIX + profileIndex;
  277. menus.create({
  278. id: profileId,
  279. contexts: defaultContexts,
  280. title: profileName,
  281. parentId: MENU_ID_SAVE_WITH_PROFILE
  282. });
  283. profileId = MENU_ID_SELECT_PROFILE_PREFIX + profileIndex;
  284. let profileChecked = allTabsData.profileName == profileName;
  285. menus.create({
  286. id: profileId,
  287. type: "radio",
  288. contexts: defaultContexts,
  289. title: profileName,
  290. checked: profileChecked,
  291. parentId: MENU_ID_SELECT_PROFILE
  292. });
  293. menusCheckedState.set(profileId, profileChecked);
  294. profileId = MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + profileIndex;
  295. profileChecked = Boolean(rule) && rule.profile == profileName;
  296. menus.create({
  297. id: profileId,
  298. type: "radio",
  299. contexts: defaultContexts,
  300. title: profileName,
  301. checked: profileChecked,
  302. parentId: MENU_ID_ASSOCIATE_WITH_PROFILE
  303. });
  304. menusCheckedState.set(profileId, profileChecked);
  305. profileIndexes.set(profileName, profileIndex);
  306. }
  307. });
  308. if (options.contextMenuEnabled) {
  309. menus.create({
  310. id: "separator-3",
  311. contexts: pageContextsEnabled,
  312. type: "separator"
  313. });
  314. }
  315. }
  316. menus.create({
  317. id: MENU_ID_AUTO_SAVE,
  318. contexts: defaultContexts,
  319. title: MENU_AUTOSAVE_MESSAGE
  320. });
  321. menus.create({
  322. id: MENU_ID_AUTO_SAVE_DISABLED,
  323. type: "radio",
  324. title: MENU_AUTOSAVE_DISABLED_MESSAGE,
  325. contexts: defaultContexts,
  326. checked: true,
  327. parentId: MENU_ID_AUTO_SAVE
  328. });
  329. menusCheckedState.set(MENU_ID_AUTO_SAVE_DISABLED, true);
  330. menus.create({
  331. id: MENU_ID_AUTO_SAVE_TAB,
  332. type: "radio",
  333. title: MENU_AUTOSAVE_TAB_MESSAGE,
  334. contexts: defaultContexts,
  335. checked: false,
  336. parentId: MENU_ID_AUTO_SAVE
  337. });
  338. menusCheckedState.set(MENU_ID_AUTO_SAVE_TAB, false);
  339. menus.create({
  340. id: MENU_ID_AUTO_SAVE_UNPINNED,
  341. type: "radio",
  342. title: MENU_AUTOSAVE_UNPINNED_TABS_MESSAGE,
  343. contexts: defaultContexts,
  344. checked: false,
  345. parentId: MENU_ID_AUTO_SAVE
  346. });
  347. menusCheckedState.set(MENU_ID_AUTO_SAVE_UNPINNED, false);
  348. menus.create({
  349. id: MENU_ID_AUTO_SAVE_ALL,
  350. type: "radio",
  351. title: MENU_AUTOSAVE_ALL_TABS_MESSAGE,
  352. contexts: defaultContexts,
  353. checked: false,
  354. parentId: MENU_ID_AUTO_SAVE
  355. });
  356. menusCheckedState.set(MENU_ID_AUTO_SAVE_ALL, false);
  357. menus.create({
  358. id: "separator-4",
  359. contexts: defaultContexts,
  360. type: "separator"
  361. });
  362. menus.create({
  363. id: MENU_ID_VIEW_PENDINGS,
  364. contexts: defaultContexts,
  365. title: MENU_VIEW_PENDINGS_MESSAGE
  366. });
  367. }
  368. menusCreated = true;
  369. if (pendingRefresh) {
  370. pendingRefresh = false;
  371. (await browser.tabs.query({})).forEach(async tab => await refreshTab(tab));
  372. }
  373. }
  374. async function initialize() {
  375. if (BROWSER_MENUS_API_SUPPORTED) {
  376. createMenus();
  377. menus.onClicked.addListener(async (event, tab) => {
  378. if (event.menuItemId == MENU_ID_SAVE_PAGE) {
  379. if (event.linkUrl) {
  380. business.saveUrls([event.linkUrl]);
  381. } else {
  382. business.saveTabs([tab]);
  383. }
  384. }
  385. if (event.menuItemId == MENU_ID_EDIT_AND_SAVE_PAGE) {
  386. const allTabsData = await tabsData.get(tab.id);
  387. if (allTabsData[tab.id].savedPageDetected) {
  388. business.openEditor(tab);
  389. } else {
  390. if (event.linkUrl) {
  391. business.saveUrls([event.linkUrl], { openEditor: true });
  392. } else {
  393. business.saveTabs([tab], { openEditor: true });
  394. }
  395. }
  396. }
  397. if (event.menuItemId == MENU_ID_SAVE_SELECTED_LINKS) {
  398. business.saveSelectedLinks(tab);
  399. }
  400. if (event.menuItemId == MENU_ID_VIEW_PENDINGS) {
  401. await browser.tabs.create({ active: true, url: "/extension/ui/pages/pendings.html" });
  402. }
  403. if (event.menuItemId == MENU_ID_SAVE_SELECTED) {
  404. business.saveTabs([tab], { selected: true });
  405. }
  406. if (event.menuItemId == MENU_ID_SAVE_FRAME) {
  407. business.saveTabs([tab], { frameId: event.frameId });
  408. }
  409. if (event.menuItemId == MENU_ID_SAVE_SELECTED_TABS || event.menuItemId == MENU_ID_BUTTON_SAVE_SELECTED_TABS) {
  410. const tabs = await queryTabs({ currentWindow: true, highlighted: true });
  411. business.saveTabs(tabs);
  412. }
  413. if (event.menuItemId == MENU_ID_SAVE_UNPINNED_TABS || event.menuItemId == MENU_ID_BUTTON_SAVE_UNPINNED_TABS) {
  414. const tabs = await queryTabs({ currentWindow: true, pinned: false });
  415. business.saveTabs(tabs);
  416. }
  417. if (event.menuItemId == MENU_ID_SAVE_ALL_TABS || event.menuItemId == MENU_ID_BUTTON_SAVE_ALL_TABS) {
  418. const tabs = await queryTabs({ currentWindow: true });
  419. business.saveTabs(tabs);
  420. }
  421. if (event.menuItemId == MENU_ID_AUTO_SAVE_TAB) {
  422. const allTabsData = await tabsData.get(tab.id);
  423. allTabsData[tab.id].autoSave = true;
  424. await tabsData.set(allTabsData);
  425. refreshExternalComponents(tab);
  426. }
  427. if (event.menuItemId == MENU_ID_AUTO_SAVE_DISABLED) {
  428. const allTabsData = await tabsData.get();
  429. Object.keys(allTabsData).forEach(tabId => {
  430. if (typeof allTabsData[tabId] == "object" && allTabsData[tabId].autoSave) {
  431. allTabsData[tabId].autoSave = false;
  432. }
  433. });
  434. allTabsData.autoSaveUnpinned = allTabsData.autoSaveAll = false;
  435. await tabsData.set(allTabsData);
  436. refreshExternalComponents(tab);
  437. }
  438. if (event.menuItemId == MENU_ID_AUTO_SAVE_ALL) {
  439. const allTabsData = await tabsData.get();
  440. allTabsData.autoSaveAll = event.checked;
  441. await tabsData.set(allTabsData);
  442. refreshExternalComponents(tab);
  443. }
  444. if (event.menuItemId == MENU_ID_AUTO_SAVE_UNPINNED) {
  445. const allTabsData = await tabsData.get();
  446. allTabsData.autoSaveUnpinned = event.checked;
  447. await tabsData.set(allTabsData);
  448. refreshExternalComponents(tab);
  449. }
  450. if (event.menuItemId.startsWith(MENU_ID_SAVE_WITH_PROFILE_PREFIX)) {
  451. const profiles = await config.getProfiles();
  452. const profileId = event.menuItemId.split(MENU_ID_SAVE_WITH_PROFILE_PREFIX)[1];
  453. let profileName;
  454. if (profileId == "default") {
  455. profileName = config.DEFAULT_PROFILE_NAME;
  456. } else {
  457. const profileIndex = Number(profileId);
  458. profileName = Object.keys(profiles)[profileIndex];
  459. }
  460. profiles[profileName].profileName = profileName;
  461. business.saveTabs([tab], profiles[profileName]);
  462. }
  463. if (event.menuItemId.startsWith(MENU_ID_SELECT_PROFILE_PREFIX)) {
  464. const [profiles, allTabsData] = await Promise.all([config.getProfiles(), tabsData.get()]);
  465. const profileId = event.menuItemId.split(MENU_ID_SELECT_PROFILE_PREFIX)[1];
  466. if (profileId == "default") {
  467. allTabsData.profileName = config.DEFAULT_PROFILE_NAME;
  468. } else {
  469. const profileIndex = Number(profileId);
  470. allTabsData.profileName = Object.keys(profiles)[profileIndex];
  471. }
  472. await tabsData.set(allTabsData);
  473. refreshExternalComponents(tab);
  474. }
  475. if (event.menuItemId.startsWith(MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX)) {
  476. const [profiles, rule] = await Promise.all([config.getProfiles(), config.getRule(tab.url, true)]);
  477. const profileId = event.menuItemId.split(MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX)[1];
  478. let profileName;
  479. if (profileId == "default") {
  480. profileName = config.DEFAULT_PROFILE_NAME;
  481. } else if (profileId == "current") {
  482. profileName = config.CURRENT_PROFILE_NAME;
  483. } else {
  484. const profileIndex = Number(profileId);
  485. profileName = Object.keys(profiles)[profileIndex];
  486. }
  487. if (rule) {
  488. await config.updateRule(rule.url, rule.url, profileName, profileName);
  489. } else {
  490. await updateTitleValue(MENU_ID_ASSOCIATE_WITH_PROFILE, MENU_UPDATE_RULE_MESSAGE);
  491. await config.addRule(new URL(tab.url).hostname, profileName, profileName);
  492. }
  493. }
  494. });
  495. if (menusCreated) {
  496. pendingRefresh = true;
  497. } else {
  498. (await browser.tabs.query({})).forEach(async tab => await refreshTab(tab));
  499. }
  500. }
  501. }
  502. async function refreshExternalComponents(tab) {
  503. const allTabsData = await tabsData.get(tab.id);
  504. await refreshAutoSaveTabs();
  505. await button.refreshTab(tab);
  506. try {
  507. await browser.runtime.sendMessage({ method: "options.refresh", profileName: allTabsData.profileName });
  508. } catch (error) {
  509. // ignored
  510. }
  511. }
  512. async function refreshTab(tab) {
  513. if (BROWSER_MENUS_API_SUPPORTED && menusCreated) {
  514. const promises = [];
  515. const allTabsData = await tabsData.get(tab.id);
  516. if (allTabsData[tab.id].editorDetected) {
  517. updateAllVisibleValues(false);
  518. } else {
  519. updateAllVisibleValues(true);
  520. promises.push(updateCheckedValue(MENU_ID_AUTO_SAVE_DISABLED, !allTabsData[tab.id].autoSave));
  521. promises.push(updateCheckedValue(MENU_ID_AUTO_SAVE_TAB, allTabsData[tab.id].autoSave));
  522. promises.push(updateCheckedValue(MENU_ID_AUTO_SAVE_UNPINNED, Boolean(allTabsData.autoSaveUnpinned)));
  523. promises.push(updateCheckedValue(MENU_ID_AUTO_SAVE_ALL, Boolean(allTabsData.autoSaveAll)));
  524. if (tab && tab.url) {
  525. const options = await config.getOptions(tab.url);
  526. promises.push(updateVisibleValue(tab, options.contextMenuEnabled));
  527. promises.push(updateTitleValue(MENU_ID_EDIT_AND_SAVE_PAGE, allTabsData[tab.id].savedPageDetected ? MENU_EDIT_PAGE_MESSAGE : MENU_EDIT_AND_SAVE_PAGE_MESSAGE));
  528. promises.push(menus.update(MENU_ID_SAVE_SELECTED, { visible: !options.saveRawPage }));
  529. promises.push(menus.update(MENU_ID_EDIT_AND_SAVE_PAGE, { visible: !options.openEditor || allTabsData[tab.id].savedPageDetected }));
  530. let selectedEntryId = MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + "default";
  531. let title = MENU_CREATE_DOMAIN_RULE_MESSAGE;
  532. const [profiles, rule] = await Promise.all([config.getProfiles(), config.getRule(tab.url)]);
  533. if (rule) {
  534. const profileIndex = profileIndexes.get(rule.profile);
  535. if (profileIndex) {
  536. selectedEntryId = MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + profileIndex;
  537. title = MENU_UPDATE_RULE_MESSAGE;
  538. }
  539. }
  540. if (Object.keys(profiles).length > 1) {
  541. Object.keys(profiles).forEach((profileName, profileIndex) => {
  542. if (profileName == config.DEFAULT_PROFILE_NAME) {
  543. promises.push(updateCheckedValue(MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + "default", selectedEntryId == MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + "default"));
  544. } else {
  545. promises.push(updateCheckedValue(MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + profileIndex, selectedEntryId == MENU_ID_ASSOCIATE_WITH_PROFILE_PREFIX + profileIndex));
  546. }
  547. });
  548. promises.push(updateTitleValue(MENU_ID_ASSOCIATE_WITH_PROFILE, title));
  549. }
  550. }
  551. }
  552. await Promise.all(promises);
  553. }
  554. }
  555. async function updateAllVisibleValues(visible) {
  556. const lastVisibleState = allMenuVisibleState;
  557. allMenuVisibleState = visible;
  558. if (lastVisibleState === undefined || lastVisibleState != visible) {
  559. const promises = [];
  560. try {
  561. MENU_TOP_VISIBLE_ENTRIES.forEach(id => promises.push(menus.update(id, { visible })));
  562. await Promise.all(promises);
  563. } catch (error) {
  564. // ignored
  565. }
  566. }
  567. }
  568. async function updateVisibleValue(tab, visible) {
  569. const lastVisibleState = contextMenuVisibleState;
  570. contextMenuVisibleState = visible;
  571. if (lastVisibleState === undefined || lastVisibleState != visible) {
  572. await createMenus(tab);
  573. }
  574. }
  575. function updateTitleValue(id, title) {
  576. const lastTitleValue = menusTitleState.get(id);
  577. menusTitleState.set(id, title);
  578. if (lastTitleValue === undefined) {
  579. return menus.update(id, { title });
  580. } else if (lastTitleValue != title) {
  581. return menus.update(id, { title });
  582. }
  583. }
  584. async function updateCheckedValue(id, checked) {
  585. checked = Boolean(checked);
  586. menusCheckedState.set(id, checked);
  587. await menus.update(id, { checked });
  588. }