Remembers the expanded/collapsed state of Launchpad View groups

Author: axosoft-ramint

src/constants.storage.ts

@@ -72,6 +72,7 @@ export type GlobalStorage = {
 	'launchpad:groups:collapsed': StoredLaunchpadGroup[];
 	'launchpad:indicator:hasLoaded': boolean;
 	'launchpad:indicator:hasInteracted': string;
+	'launchpadView:groups:expanded': StoredLaunchpadGroup[];
 	'graph:searchMode': StoredGraphSearchMode;
 } & { [key in `confirm:ai:tos:${AIProviders}`]: boolean } & {
 	[key in `provider:authentication:skip:${string}`]: boolean;

src/views/launchpadView.ts

@@ -19,7 +19,8 @@ import { configuration } from '../system/vscode/configuration';
 import { CacheableChildrenViewNode } from './nodes/abstract/cacheableChildrenViewNode';
 import type { ClipboardType, ViewNode } from './nodes/abstract/viewNode';
 import { ContextValues, getViewNodeId } from './nodes/abstract/viewNode';
-import { GroupingNode } from './nodes/groupingNode';
+import type { GroupingNode } from './nodes/groupingNode';
+import { LaunchpadViewGroupingNode } from './nodes/launchpadViewGroupingNode';
 import { getPullRequestChildren, getPullRequestTooltip } from './nodes/pullRequestNode';
 import { ViewBase } from './viewBase';
 import { registerViewCommand } from './viewCommands';
@@ -141,17 +142,25 @@ export class LaunchpadViewNode extends CacheableChildrenViewNode<
 				}
 
 				const uiGroups = groupAndSortLaunchpadItems(result.items);
+				const expanded = new Map(
+					(
+						(this.view.container.storage.get('launchpadView:groups:expanded') satisfies
+							| LaunchpadGroup[]
+							| undefined) ?? []
+					).map(g => [g, true]),
+				);
 				for (const [ui, groupItems] of uiGroups) {
 					if (!groupItems.length) continue;
 
 					const icon = launchpadGroupIconMap.get(ui)!;
 
 					children.push(
-						new GroupingNode(
+						new LaunchpadViewGroupingNode(
 							this.view,
 							launchpadGroupLabelMap.get(ui)!,
+							ui,
 							groupItems.map(i => new LaunchpadItemNode(this.view, this, ui, i)),
-							TreeItemCollapsibleState.Collapsed,
+							expanded.get(ui) ? TreeItemCollapsibleState.Expanded : TreeItemCollapsibleState.Collapsed,
 							undefined,
 							undefined,
 							new ThemeIcon(icon.substring(2, icon.length - 1)),

src/views/nodes/launchpadViewGroupingNode.ts

@@ -0,0 +1,45 @@
+import type { TreeItem } from 'vscode';
+import { Disposable, TreeItemCollapsibleState } from 'vscode';
+import type { LaunchpadGroup } from '../../plus/launchpad/launchpadProvider';
+import type { TreeViewNodeCollapsibleStateChangeEvent, View } from '../viewBase';
+import type { ViewNode } from './abstract/viewNode';
+import { GroupingNode } from './groupingNode';
+
+export class LaunchpadViewGroupingNode<TChild extends ViewNode = ViewNode> extends GroupingNode {
+	private disposable: Disposable;
+
+	constructor(
+		view: View,
+		label: string,
+		private readonly group: LaunchpadGroup,
+		childrenOrFn: TChild[] | Promise<TChild[]> | (() => TChild[] | Promise<TChild[]>),
+		collapsibleState: TreeItemCollapsibleState = TreeItemCollapsibleState.Expanded,
+		description?: string,
+		tooltip?: string,
+		iconPath?: TreeItem['iconPath'],
+		contextValue?: string,
+	) {
+		super(view, label, childrenOrFn, collapsibleState, description, tooltip, iconPath, contextValue);
+		this.disposable = Disposable.from(
+			this.view.onDidChangeNodeCollapsibleState(this.onNodeCollapsibleStateChanged, this),
+		);
+	}
+
+	override dispose() {
+		super.dispose();
+		this.disposable?.dispose();
+	}
+
+	private onNodeCollapsibleStateChanged(e: TreeViewNodeCollapsibleStateChangeEvent<ViewNode>) {
+		if (e.element === this) {
+			const storedExpandedGroups = this.view.container.storage.get('launchpadView:groups:expanded') ?? [];
+			if (e.state === TreeItemCollapsibleState.Expanded) {
+				storedExpandedGroups.push(this.group);
+			} else {
+				storedExpandedGroups.splice(storedExpandedGroups.indexOf(this.group), 1);
+			}
+
+			void this.view.container.storage.store('launchpadView:groups:expanded', storedExpandedGroups);
+		}
+	}
+}