ui-menus.js 22 KB

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