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