feat: tab containers

Author: iamEvanYT

src/main/browser/tabs/index.ts

@@ -2,18 +2,19 @@ import { Browser } from "@/browser/browser";
 import { TabGroupManager } from "@/browser/tabs/managers/tab-group-manager";
 import { ActiveTabGroupManager } from "@/browser/tabs/managers/active-tab-group-manager";
 import { TabManager } from "@/browser/tabs/managers/tab-manager";
+import { TabsContainerManager } from "@/browser/tabs/managers/tabs-container-manager";
 
 export class TabOrchestrator {
-  private readonly browser: Browser;
   public readonly tabManager: TabManager;
   public readonly tabGroupManager: TabGroupManager;
   public readonly activeTabGroupManager: ActiveTabGroupManager;
+  public readonly tabsContainerManager: TabsContainerManager;
 
   constructor(browser: Browser) {
-    this.browser = browser;
     this.tabManager = new TabManager(browser);
     this.tabGroupManager = new TabGroupManager(browser);
     this.activeTabGroupManager = new ActiveTabGroupManager(browser);
+    this.tabsContainerManager = new TabsContainerManager(browser);
   }
 
   public destroy(): void {

src/main/browser/tabs/objects/tabs-container.ts renamed to src/main/browser/tabs/objects/tab-containers/base.ts

@@ -1,46 +1,73 @@
 import { TabGroup } from "@/browser/tabs/objects/tab-group";
+import { TabFolder } from "@/browser/tabs/objects/tab-containers/tab-folder";
 import { TypedEventEmitter } from "@/modules/typed-event-emitter";
 
 // Container Child //
 type TabGroupChild = {
   type: "tab-group";
   item: TabGroup;
 };
-type TabContainerChild = {
-  type: "tab-container";
-  item: TabsContainer;
+type TabFolderChild = {
+  type: "tab-folder";
+  item: TabFolder;
 };
-export type ContainerChild = TabGroupChild | TabContainerChild;
+export type ContainerChild = TabGroupChild | TabFolderChild;
 
 // Exported Tab Data //
+export type ExportedBaseTabContainer = {
+  type: "tab-container";
+  children: ExportedTabData[];
+};
+
 export type ExportedTabGroup = {
   type: "tab-group";
 };
-export type ExportedTabsContainer = {
-  type: "tabs-container";
+export type ExportedTabContainer = {
+  type: "tab-container";
+  children: ExportedTabData[];
+};
+export type ExportedTabsFolder = {
+  type: "tab-folder";
   name: string;
   children: ExportedTabData[];
 };
-type ExportedTabData = ExportedTabGroup | ExportedTabsContainer;
+// ExportedTabContainer would not be a child: it should only be the root container.
+type ExportedTabData = ExportedTabGroup | ExportedTabsFolder;
 
-// Tabs Container //
-type TabsContainerEvents = {
+// Base Tab Container //
+export type BaseTabContainerEvents = {
   "child-added": [child: ContainerChild];
   "child-removed": [child: ContainerChild];
   "children-moved": [];
 };
 
-export class TabsContainer extends TypedEventEmitter<TabsContainerEvents> {
-  public name: string;
+export class BaseTabContainer<
+  TEvents extends BaseTabContainerEvents = BaseTabContainerEvents
+> extends TypedEventEmitter<TEvents> {
   public children: ContainerChild[];
 
-  constructor(name: string) {
+  constructor() {
     super();
 
-    this.name = name;
     this.children = [];
   }
 
+  public getAllTabGroups(): TabGroup[] {
+    const scanTabContainer = (container: BaseTabContainer): TabGroup[] => {
+      return container.children.flatMap((child) => {
+        if (child.type === "tab-group") {
+          return [child.item];
+        }
+        if (child.type === "tab-folder") {
+          return scanTabContainer(child.item);
+        }
+        return [];
+      });
+    };
+
+    return scanTabContainer(this);
+  }
+
   public addChild(child: ContainerChild): void {
     this.children.push(child);
     this.emit("child-added", child);
@@ -78,10 +105,9 @@ export class TabsContainer extends TypedEventEmitter<TabsContainerEvents> {
     return this.children.length;
   }
 
-  public export(): ExportedTabsContainer {
+  protected baseExport(): ExportedBaseTabContainer {
     return {
-      type: "tabs-container",
-      name: this.name,
+      type: "tab-container",
       children: this.children.map((child) => {
         return child.item.export();
       })

src/main/browser/tabs/objects/tab-containers/tab-container.ts

@@ -0,0 +1,27 @@
+import { BaseTabContainer, BaseTabContainerEvents, ExportedTabContainer } from "./base";
+
+type TabContainerEvents = BaseTabContainerEvents & {
+  "space-changed": [spaceId: string];
+};
+
+export class TabContainer extends BaseTabContainer<TabContainerEvents> {
+  public spaceId: string;
+
+  constructor(spaceId: string) {
+    super();
+    this.spaceId = spaceId;
+  }
+
+  public setSpace(spaceId: string): void {
+    this.spaceId = spaceId;
+    this.emit("space-changed", spaceId);
+  }
+
+  public export(): ExportedTabContainer {
+    const { children } = this.baseExport();
+    return {
+      type: "tab-container",
+      children
+    };
+  }
+}

src/main/browser/tabs/objects/tab-containers/tab-folder.ts

@@ -0,0 +1,23 @@
+import { BaseTabContainer, ExportedTabsFolder } from "./base";
+
+export class TabFolder extends BaseTabContainer {
+  public name: string;
+
+  constructor(name: string) {
+    super();
+    this.name = name;
+  }
+
+  public setName(name: string): void {
+    this.name = name;
+  }
+
+  public export(): ExportedTabsFolder {
+    const { children } = this.baseExport();
+    return {
+      type: "tab-folder",
+      name: this.name,
+      children
+    };
+  }
+}

src/main/browser/tabs/objects/tab-group/index.ts

@@ -9,7 +9,7 @@ import {
 } from "@/browser/tabs/objects/tab-group/controllers";
 import { Browser } from "@/browser/browser";
 import { TabbedBrowserWindow } from "@/browser/window";
-import { ExportedTabGroup } from "@/browser/tabs/objects/tabs-container";
+import { ExportedTabGroup } from "@/browser/tabs/objects/tab-containers/base";
 
 type TabGroupTypes = "normal" | "split" | "glance";