iframe-proxy | Sunbelt Computer Software

History

129 lines (120 loc) · 5.04 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

/*---------------------------------------------------------------------------------------------

* Licensed under the MIT License. See License.txt in the project root for license information.

*--------------------------------------------------------------------------------------------*/

import * as vscode from 'vscode';

import { DirectoryTreeNode } from './directoryTreeNode';

import { FileChangeNode } from './fileChangeNode';

import { TreeNode } from './treeNode';

export namespace TreeUtils {

export function processCheckboxUpdates(checkboxUpdates: vscode.TreeCheckboxChangeEvent<TreeNode>, selection: readonly TreeNode[]) {

const selectionContainsUpdates = selection.some(node => checkboxUpdates.items.some(update => update[0] === node));

const checkedNodes: FileChangeNode[] = [];

const uncheckedNodes: FileChangeNode[] = [];

for (const [node, newState] of checkboxUpdates.items) {

if (node instanceof FileChangeNode) {

if (newState === vscode.TreeItemCheckboxState.Checked) {

checkedNodes.push(node);

} else {

uncheckedNodes.push(node);

}

node.updateFromCheckboxChanged(newState);

} else if (node instanceof DirectoryTreeNode) {

collectAllDescendants(node, newState, checkedNodes, uncheckedNodes);

}

if (selectionContainsUpdates) {

for (const selected of selection) {

if (!(selected instanceof FileChangeNode)) {

continue;

}

if (!checkedNodes.includes(selected) && !uncheckedNodes.includes(selected)) {

// Only process files that have checkboxes (files without checkboxState, like those under commits, are skipped)

if (selected.checkboxState?.state === vscode.TreeItemCheckboxState.Unchecked) {

selected.updateFromCheckboxChanged(vscode.TreeItemCheckboxState.Checked);

checkedNodes.push(selected);

} else if (selected.checkboxState?.state === vscode.TreeItemCheckboxState.Checked) {

selected.updateFromCheckboxChanged(vscode.TreeItemCheckboxState.Unchecked);

uncheckedNodes.push(selected);

}

// Eagerly update ancestor directory checkbox states by walking from each

// affected file up through its parent directories. With manageCheckboxStateManually,

// we must set directory states before refresh rather than relying solely on

// getTreeItem(), since VS Code may not apply checkboxState changes during a refresh.

const allAffected = [...checkedNodes, ...uncheckedNodes];

for (const node of allAffected) {

let parent = node.getParent();

while (parent instanceof DirectoryTreeNode) {

parent.updateCheckboxFromChildren();

parent = parent.getParent();

}

// Refresh the tree so checkbox visual state updates.

// Refreshing the topmost affected directory will cascade to all descendants.

const refreshedDirs = new Set<DirectoryTreeNode>();

for (const node of allAffected) {

let topDir: DirectoryTreeNode | undefined;

let parent = node.getParent();

while (parent instanceof DirectoryTreeNode) {

topDir = parent;

parent = parent.getParent();

}

if (topDir && !refreshedDirs.has(topDir)) {

refreshedDirs.add(topDir);

topDir.refresh(topDir);

}

// If a directory was clicked directly, also refresh it

for (const [node] of checkboxUpdates.items) {

if (node instanceof DirectoryTreeNode && !refreshedDirs.has(node)) {

refreshedDirs.add(node);

node.refresh(node);

}

// For flat layout (files have no directory parent), refresh file nodes directly

for (const node of allAffected) {

const parent = node.getParent();

if (!(parent instanceof DirectoryTreeNode)) {

node.refresh(node);

}

// Send API requests without firing state change events (UI is already updated optimistically).

// This prevents race conditions where overlapping markFiles calls cause checkboxes to flicker.

if (checkedNodes.length > 0) {

const prModel = checkedNodes[0].pullRequest;

const filenames = checkedNodes.map(n => n.fileName);

prModel.markFiles(filenames, false, 'viewed').then(() => {

checkedNodes[0].refreshFileViewedContext();

});

}

if (uncheckedNodes.length > 0) {

const prModel = uncheckedNodes[0].pullRequest;

const filenames = uncheckedNodes.map(n => n.fileName);

prModel.markFiles(filenames, false, 'unviewed').then(() => {

uncheckedNodes[0].refreshFileViewedContext();

});

}

function collectAllDescendants(

dirNode: DirectoryTreeNode,

newState: vscode.TreeItemCheckboxState,

checkedNodes: FileChangeNode[],

uncheckedNodes: FileChangeNode[]

): void {

for (const child of dirNode._children) {

if (child instanceof FileChangeNode) {

if (newState === vscode.TreeItemCheckboxState.Checked) {

checkedNodes.push(child);

} else {

uncheckedNodes.push(child);

}

child.updateFromCheckboxChanged(newState);

} else if (child instanceof DirectoryTreeNode) {

collectAllDescendants(child, newState, checkedNodes, uncheckedNodes);

}

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

treeUtils.ts

Latest commit

History

treeUtils.ts

File metadata and controls