mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-01-13 17:22:15 -05:00
f278e2a7a2
* Add controller persistence and info prompting * more stuff * clean up * Add arc tests to scripts
72 lines
3.2 KiB
TypeScript
72 lines
3.2 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the Source EULA. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import * as vscode from 'vscode';
|
|
import { TreeNode } from './treeNode';
|
|
import { MiaaTreeNode } from './miaaTreeNode';
|
|
import { ResourceType } from '../../constants';
|
|
import { PostgresTreeNode } from './postgresTreeNode';
|
|
import { ControllerModel, Registration } from '../../models/controllerModel';
|
|
import { ControllerDashboard } from '../dashboards/controller/controllerDashboard';
|
|
import { PostgresModel } from '../../models/postgresModel';
|
|
import { parseInstanceName } from '../../common/utils';
|
|
import { MiaaModel } from '../../models/miaaModel';
|
|
import { Deferred } from '../../common/promise';
|
|
import { RefreshTreeNode } from './refreshTreeNode';
|
|
|
|
/**
|
|
* The TreeNode for displaying an Azure Arc Controller
|
|
*/
|
|
export class ControllerTreeNode extends TreeNode {
|
|
|
|
private _children: TreeNode[] = [];
|
|
private _childrenRefreshPromise = new Deferred();
|
|
|
|
constructor(public model: ControllerModel, private _context: vscode.ExtensionContext) {
|
|
super(model.info.url, vscode.TreeItemCollapsibleState.Collapsed, ResourceType.dataControllers);
|
|
model.onRegistrationsUpdated(registrations => this.refreshChildren(registrations));
|
|
}
|
|
|
|
public async getChildren(): Promise<TreeNode[]> {
|
|
// First reset our deferred promise so we're sure we'll get the refreshed children
|
|
this._childrenRefreshPromise = new Deferred();
|
|
try {
|
|
await this.model.refresh();
|
|
await this._childrenRefreshPromise.promise;
|
|
} catch (err) {
|
|
// Couldn't get the children and TreeView doesn't have a way to collapse a node
|
|
// in a way that will refetch its children when expanded again so instead we
|
|
// display a tempory node that will prompt the user to re-enter credentials
|
|
return [new RefreshTreeNode(this)];
|
|
}
|
|
|
|
return this._children;
|
|
}
|
|
|
|
public async openDashboard(): Promise<void> {
|
|
const controllerDashboard = new ControllerDashboard(this.model);
|
|
await controllerDashboard.showDashboard();
|
|
}
|
|
|
|
private refreshChildren(registrations: Registration[]): void {
|
|
this._children = <TreeNode[]>registrations.map(registration => {
|
|
if (!registration.instanceNamespace || !registration.instanceName) {
|
|
console.warn('Registration is missing required namespace and name values, skipping');
|
|
return undefined;
|
|
}
|
|
switch (registration.instanceType) {
|
|
case ResourceType.postgresInstances:
|
|
const postgresModel = new PostgresModel(this.model.info.url, this.model.auth!, registration.instanceNamespace, parseInstanceName(registration.instanceName));
|
|
return new PostgresTreeNode(postgresModel, this.model, this._context);
|
|
case ResourceType.sqlManagedInstances:
|
|
const miaaModel = new MiaaModel(this.model.info.url, this.model.auth!, registration.instanceNamespace, parseInstanceName(registration.instanceName));
|
|
return new MiaaTreeNode(miaaModel, this.model);
|
|
}
|
|
return undefined;
|
|
}).filter(item => item); // filter out invalid nodes (controllers or ones without required properties)
|
|
this._childrenRefreshPromise.resolve();
|
|
}
|
|
}
|