ui-menus.js 21 KB

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