desktop/src/zen/common/modules/ZenSessionStore.mjs

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

import { nsZenPreloadedFeature } from 'chrome://browser/content/zen-components/ZenCommonUtils.mjs';

class ZenSessionStore extends nsZenPreloadedFeature {
  init() {
    this.#waitAndCleanup();
  }

  promiseInitialized = new Promise((resolve) => {
    this._resolveInitialized = resolve;
  });

  restoreInitialTabData(tab, tabData) {
    if (tabData.zenWorkspace) {
      tab.setAttribute('zen-workspace-id', tabData.zenWorkspace);
    }
    // Keep for now, for backward compatibility for window sync to work.
    if (tabData.zenSyncId || tabData.zenPinnedId) {
      tab.setAttribute('id', tabData.zenSyncId || tabData.zenPinnedId);
    }
    if (tabData.zenStaticLabel) {
      tab.zenStaticLabel = tabData.zenStaticLabel;
    }
    if (tabData.zenHasStaticIcon) {
      tab.setAttribute('zen-has-static-icon', 'true');
    }
    if (tabData.zenEssential) {
      tab.setAttribute('zen-essential', 'true');
    }
    if (tabData.zenDefaultUserContextId) {
      tab.setAttribute('zenDefaultUserContextId', 'true');
    }
    if (tabData.zenPinnedEntry) {
      tab.setAttribute('zen-pinned-entry', tabData.zenPinnedEntry);
    }
    if (tabData._zenPinnedInitialState) {
      tab._zenPinnedInitialState = tabData._zenPinnedInitialState;
    }
  }

  async #waitAndCleanup() {
    await SessionStore.promiseInitialized;
    this.#cleanup();
  }

  #cleanup() {
    this._resolveInitialized();
  }
}

window.gZenSessionStore = new ZenSessionStore();