-
Notifications
You must be signed in to change notification settings - Fork 1.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Showing
21 changed files
with
1,742 additions
and
612 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| /* -------------------------------------------------------------------------------------------- | ||
| * Copyright (c) Microsoft Corporation. All Rights Reserved. | ||
| * See 'LICENSE' in the project root for license information. | ||
| * ------------------------------------------------------------------------------------------ */ | ||
|
|
||
| import * as vscode from 'vscode'; | ||
|
|
||
| /** | ||
| * Base class of nodes in all tree nodes | ||
| */ | ||
| export interface BaseNode { | ||
| /** | ||
| * Get the child nodes of this node | ||
| */ | ||
| getChildren(): Promise<BaseNode[]>; | ||
|
|
||
| /** | ||
| * Get the vscode.TreeItem associated with this node | ||
| */ | ||
| getTreeItem(): Promise<vscode.TreeItem>; | ||
| } | ||
|
|
||
| export class LabelLeafNode implements BaseNode { | ||
| constructor(private readonly label: string) { /* blank */ } | ||
|
|
||
| async getChildren(): Promise<BaseNode[]> { | ||
| return []; | ||
| } | ||
|
|
||
| getTreeItem(): Promise<vscode.TreeItem> { | ||
| return Promise.resolve(new vscode.TreeItem(this.getLabel(), vscode.TreeItemCollapsibleState.None)); | ||
| } | ||
|
|
||
| getLabel(): string { | ||
| return this.label; | ||
| } | ||
| } | ||
|
|
||
| export const refreshCppSshTargetsView: string = 'C_Cpp.refreshCppSshTargetsView'; | ||
| export const addSshTarget: string = 'C_Cpp.addSshTarget'; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,119 @@ | ||
| /* -------------------------------------------------------------------------------------------- | ||
| * Copyright (c) Microsoft Corporation. All Rights Reserved. | ||
| * See 'LICENSE' in the project root for license information. | ||
| * ------------------------------------------------------------------------------------------ */ | ||
|
|
||
| import { getSshConfigHostInfos } from '../sshHosts'; | ||
| import * as vscode from 'vscode'; | ||
| import { addSshTarget, BaseNode, LabelLeafNode, refreshCppSshTargetsView } from './common'; | ||
| import { TargetLeafNode, filesWritable, setActiveSshTarget, _activeTarget, workspaceState_activeSshTarget } from './targetNodes'; | ||
| import { extensionContext, ISshConfigHostInfo } from '../../common'; | ||
| import * as nls from 'vscode-nls'; | ||
|
|
||
| nls.config({ messageFormat: nls.MessageFormat.bundle, bundleFormat: nls.BundleFormat.standalone })(); | ||
| const localize: nls.LocalizeFunc = nls.loadMessageBundle(); | ||
|
|
||
| let _targets: Map<string, ISshConfigHostInfo> = new Map<string, ISshConfigHostInfo>(); | ||
|
|
||
| export class SshTargetsProvider implements vscode.TreeDataProvider<BaseNode>, vscode.Disposable { | ||
| private readonly _onDidChangeTreeData: vscode.EventEmitter<BaseNode | undefined> = new vscode.EventEmitter<BaseNode | undefined>(); | ||
|
|
||
| public get onDidChangeTreeData(): vscode.Event<BaseNode | undefined> { | ||
| return this._onDidChangeTreeData.event; | ||
| } | ||
|
|
||
| async getChildren(node?: BaseNode): Promise<BaseNode[]> { | ||
| if (node) { | ||
| return node.getChildren(); | ||
| } | ||
|
|
||
| const children: BaseNode[] = await this.getTargets(); | ||
| if (children.length === 0) { | ||
| return [new LabelLeafNode(localize('no.ssh.targets', 'No SSH targets'))]; | ||
| } | ||
|
|
||
| return children; | ||
| } | ||
|
|
||
| getTreeItem(node: BaseNode): Promise<vscode.TreeItem> { | ||
| return node.getTreeItem(); | ||
| } | ||
|
|
||
| refresh(node?: BaseNode): void { | ||
| this._onDidChangeTreeData.fire(node); | ||
| } | ||
|
|
||
| dispose(): void { | ||
| this._onDidChangeTreeData.dispose(); | ||
| } | ||
|
|
||
| private async getTargets(): Promise<BaseNode[]> { | ||
| filesWritable.clear(); | ||
| _targets = await getSshConfigHostInfos(); | ||
| const targetNodes: BaseNode[] = []; | ||
| // Currently, the best place to check if a connection is removed is during refresh, since the active target could be removed | ||
| // by editing the SSH config file directly. If we see any performance issue in the future, we can move this to removeSshTargetImpl, | ||
| // and the file watchers. | ||
| let activeTargetRemoved: boolean = true; | ||
| const cachedActiveTarget: string | undefined = await getActiveSshTarget(false); | ||
| for (const host of Array.from(_targets.keys())) { | ||
| const sshConfigHostInfo: ISshConfigHostInfo | undefined = _targets.get(host); | ||
| if (sshConfigHostInfo) { | ||
| targetNodes.push(new TargetLeafNode(host, sshConfigHostInfo)); | ||
| if (host === cachedActiveTarget) { | ||
| activeTargetRemoved = false; | ||
| } | ||
| } | ||
| } | ||
| if (activeTargetRemoved) { | ||
| setActiveSshTarget(undefined); | ||
| } | ||
| return targetNodes; | ||
| } | ||
| } | ||
|
|
||
| export async function initializeSshTargets(): Promise<void> { | ||
| _targets = await getSshConfigHostInfos(); | ||
| let activeTargetRemoved: boolean = true; | ||
| const cachedActiveTarget: string | undefined = await getActiveSshTarget(false); | ||
| for (const host of Array.from(_targets.keys())) { | ||
| if (host === cachedActiveTarget) { | ||
| activeTargetRemoved = false; | ||
| } | ||
| } | ||
| if (activeTargetRemoved) { | ||
| setActiveSshTarget(undefined); | ||
| } | ||
| await setActiveSshTarget(extensionContext?.workspaceState.get(workspaceState_activeSshTarget)); | ||
| } | ||
|
|
||
| export async function getActiveSshTarget(selectWhenNotSet: boolean = true): Promise<string | undefined> { | ||
| if (_targets.size === 0 && !selectWhenNotSet) { | ||
| return undefined; | ||
| } | ||
| if (!_activeTarget && selectWhenNotSet) { | ||
| const name: string | undefined = await selectSshTarget(); | ||
| if (!name) { | ||
| throw Error(localize('active.ssh.target.selection.cancelled', 'Active SSH target selection cancelled.')); | ||
| } | ||
| await setActiveSshTarget(name); | ||
| await vscode.commands.executeCommand(refreshCppSshTargetsView); | ||
| } | ||
| return _activeTarget; | ||
| } | ||
|
|
||
| const addNewSshTarget: string = localize('add.new.ssh.target', '{0} Add New SSH Target...', '$(plus)'); | ||
|
|
||
| export async function selectSshTarget(): Promise<string | undefined> { | ||
| const items: string[] = Array.from(_targets.keys()); | ||
| // Special item for adding SSH target | ||
| items.push(addNewSshTarget); | ||
| const selection: string | undefined = await vscode.window.showQuickPick(items, { title: localize('select.ssh.target', 'Select an SSH target') }); | ||
| if (!selection) { | ||
| return undefined; | ||
| } | ||
| if (selection === addNewSshTarget) { | ||
| return vscode.commands.executeCommand(addSshTarget); | ||
| } | ||
| return selection; | ||
| } |
Oops, something went wrong.