mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-03-15 19:11:38 -04:00
93 lines
3.2 KiB
TypeScript
93 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 azdata from 'azdata';
|
|
import { ModelViewBase } from './modelViewBase';
|
|
import { ApiWrapper } from '../../common/apiWrapper';
|
|
import * as constants from '../../common/constants';
|
|
import { IPageView, IDataComponent } from '../interfaces';
|
|
import { ModelSourcesComponent, ModelSourceType } from './modelSourcesComponent';
|
|
import { LocalModelsComponent } from './localModelsComponent';
|
|
import { AzureModelsComponent } from './azureModelsComponent';
|
|
|
|
/**
|
|
* View to pick model source
|
|
*/
|
|
export class ModelSourcePage extends ModelViewBase implements IPageView, IDataComponent<ModelSourceType> {
|
|
|
|
private _form: azdata.FormContainer | undefined;
|
|
private _formBuilder: azdata.FormBuilder | undefined;
|
|
public modelResources: ModelSourcesComponent | undefined;
|
|
public localModelsComponent: LocalModelsComponent | undefined;
|
|
public azureModelsComponent: AzureModelsComponent | undefined;
|
|
|
|
constructor(apiWrapper: ApiWrapper, parent: ModelViewBase) {
|
|
super(apiWrapper, parent.root, parent);
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param modelBuilder Register components
|
|
*/
|
|
public registerComponent(modelBuilder: azdata.ModelBuilder): azdata.Component {
|
|
|
|
this._formBuilder = modelBuilder.formContainer();
|
|
this.modelResources = new ModelSourcesComponent(this._apiWrapper, this);
|
|
this.modelResources.registerComponent(modelBuilder);
|
|
this.localModelsComponent = new LocalModelsComponent(this._apiWrapper, this);
|
|
this.localModelsComponent.registerComponent(modelBuilder);
|
|
this.azureModelsComponent = new AzureModelsComponent(this._apiWrapper, this);
|
|
this.azureModelsComponent.registerComponent(modelBuilder);
|
|
this.modelResources.addComponents(this._formBuilder);
|
|
this.refresh();
|
|
this._form = this._formBuilder.component();
|
|
return this._form;
|
|
}
|
|
|
|
/**
|
|
* Returns selected data
|
|
*/
|
|
public get data(): ModelSourceType {
|
|
return this.modelResources?.data || ModelSourceType.Local;
|
|
}
|
|
|
|
/**
|
|
* Returns the component
|
|
*/
|
|
public get component(): azdata.Component | undefined {
|
|
return this._form;
|
|
}
|
|
|
|
/**
|
|
* Refreshes the view
|
|
*/
|
|
public async refresh(): Promise<void> {
|
|
if (this._formBuilder) {
|
|
if (this.modelResources && this.modelResources.data === ModelSourceType.Local) {
|
|
if (this.localModelsComponent && this.azureModelsComponent) {
|
|
this.azureModelsComponent.removeComponents(this._formBuilder);
|
|
this.localModelsComponent.addComponents(this._formBuilder);
|
|
await this.localModelsComponent.refresh();
|
|
}
|
|
|
|
} else if (this.modelResources && this.modelResources.data === ModelSourceType.Azure) {
|
|
if (this.localModelsComponent && this.azureModelsComponent) {
|
|
this.localModelsComponent.removeComponents(this._formBuilder);
|
|
this.azureModelsComponent.addComponents(this._formBuilder);
|
|
await this.azureModelsComponent.refresh();
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns page title
|
|
*/
|
|
public get title(): string {
|
|
return constants.modelSourcePageTitle;
|
|
}
|
|
}
|