@@ -46,6 +46,7 @@ import { isCanary } from "./config";
4646import { App } from "./common/app" ;
4747import { redactableError } from "./pure/errors" ;
4848import { LocalDatabasesCommands } from "./common/commands" ;
49+ import { createMultiSelectionCommand } from "./common/selection-commands" ;
4950
5051enum SortOrder {
5152 NameAsc = "NameAsc" ,
@@ -241,7 +242,9 @@ export class DatabaseUI extends DisposableObject {
241242 "codeQLDatabases.sortByName" : this . handleSortByName . bind ( this ) ,
242243 "codeQLDatabases.sortByDateAdded" : this . handleSortByDateAdded . bind ( this ) ,
243244 "codeQLDatabases.removeDatabase" : this . handleRemoveDatabase . bind ( this ) ,
244- "codeQLDatabases.upgradeDatabase" : this . handleUpgradeDatabase . bind ( this ) ,
245+ "codeQLDatabases.upgradeDatabase" : createMultiSelectionCommand (
246+ this . handleUpgradeDatabase . bind ( this ) ,
247+ ) ,
245248 "codeQLDatabases.renameDatabase" : this . handleRenameDatabase . bind ( this ) ,
246249 "codeQLDatabases.openDatabaseFolder" : this . handleOpenFolder . bind ( this ) ,
247250 "codeQLDatabases.addDatabaseSource" : this . handleAddSource . bind ( this ) ,
@@ -529,17 +532,14 @@ export class DatabaseUI extends DisposableObject {
529532 }
530533
531534 private async handleUpgradeDatabase (
532- databaseItem : DatabaseItem | undefined ,
533- multiSelect : DatabaseItem [ ] | undefined ,
535+ databaseItems : DatabaseItem [ ] ,
534536 ) : Promise < void > {
535537 return withProgress (
536538 async ( progress , token ) => {
537539 return await this . handleUpgradeDatabasesInternal (
538540 progress ,
539541 token ,
540- multiSelect === undefined && databaseItem !== undefined
541- ? [ databaseItem ]
542- : multiSelect || [ ] ,
542+ databaseItems ,
543543 ) ;
544544 } ,
545545 {
0 commit comments