ui-menus.js 21 KB

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