boards-data-menu-updater.ts

import{
Disposable,
DisposableCollection,
}from'@theia/core/lib/common/disposable';
import{nls}from'@theia/core/lib/common/nls';
import{inject,injectable}from'@theia/core/shared/inversify';
importPQueuefrom'p-queue';
import{
BoardIdentifier,
ConfigOption,
isBoardIdentifierChangeEvent,
Programmer,
}from'../../common/protocol';
import{BoardsDataStore}from'../boards/boards-data-store';
import{BoardsServiceProvider}from'../boards/boards-service-provider';
import{ArduinoMenus,unregisterSubmenu}from'../menu/arduino-menus';
import{
CommandRegistry,
Contribution,
MenuModelRegistry,
}from'./contribution';

@injectable()
exportclassBoardsDataMenuUpdaterextendsContribution{
 @inject(CommandRegistry)
privatereadonlycommandRegistry: CommandRegistry;
 @inject(MenuModelRegistry)
privatereadonlymenuRegistry: MenuModelRegistry;
 @inject(BoardsDataStore)
privatereadonlyboardsDataStore: BoardsDataStore;
 @inject(BoardsServiceProvider)
privatereadonlyboardsServiceProvider: BoardsServiceProvider;

privatereadonlyqueue=newPQueue({autoStart: true,concurrency: 1});
privatereadonlytoDisposeOnBoardChange=newDisposableCollection();

overrideonStart(): void{
this.boardsDataStore.onDidChange(()=>
this.updateMenuActions(
this.boardsServiceProvider.boardsConfig.selectedBoard
)
);
this.boardsServiceProvider.onBoardsConfigDidChange((event)=>{
if(isBoardIdentifierChangeEvent(event)){
this.updateMenuActions(event.selectedBoard);
}
});
}

overrideonReady(): void{
this.boardsServiceProvider.ready.then(()=>
this.updateMenuActions(
this.boardsServiceProvider.boardsConfig.selectedBoard
)
);
}

privateasyncupdateMenuActions(
selectedBoard: BoardIdentifier|undefined
): Promise<void>{
returnthis.queue.add(async()=>{
this.toDisposeOnBoardChange.dispose();
this.menuManager.update();
if(selectedBoard){
const{ fqbn }=selectedBoard;
if(fqbn){
const{ configOptions, programmers, selectedProgrammer }=
awaitthis.boardsDataStore.getData(fqbn);
if(configOptions.length){
constboardsConfigMenuPath=[
 ...ArduinoMenus.TOOLS__BOARD_SETTINGS_GROUP,
'z01_boardsConfig',
];// `z_` is for ordering.
for(const{ label, option, values }ofconfigOptions.sort(
ConfigOption.LABEL_COMPARATOR
)){
constmenuPath=[...boardsConfigMenuPath,`${option}`];
constcommands=newMap<
string,
Disposable&{label: string}
>();
letselectedValue='';
for(constvalueofvalues){
constid=`${fqbn}-${option}--${value.value}`;
constcommand={ id };
consthandler={
execute: ()=>
this.boardsDataStore.selectConfigOption({
 fqbn,
optionsToUpdate: [{ option,selectedValue: value.value}],
}),
isToggled: ()=>value.selected,
};
commands.set(
id,
Object.assign(
this.commandRegistry.registerCommand(command,handler),
{label: value.label}
)
);
if(value.selected){
selectedValue=value.label;
}
}
this.menuRegistry.registerSubmenu(
menuPath,
`${label}${selectedValue ? `: "${selectedValue}"` : ''}`
);
this.toDisposeOnBoardChange.pushAll([
 ...commands.values(),
Disposable.create(()=>
unregisterSubmenu(menuPath,this.menuRegistry)
),
 ...Array.from(commands.keys()).map((commandId,i)=>{
const{ label }=commands.get(commandId)!;
this.menuRegistry.registerMenuAction(menuPath,{
 commandId,
order: String(i).padStart(4),
 label,
});
returnDisposable.create(()=>
this.menuRegistry.unregisterMenuAction(commandId)
);
}),
]);
}
}
if(programmers.length){
constprogrammersMenuPath=[
 ...ArduinoMenus.TOOLS__BOARD_SETTINGS_GROUP,
'z02_programmers',
];
constprogrammerNls=nls.localize(
'arduino/board/programmer',
'Programmer'
);
constlabel=selectedProgrammer
 ? `${programmerNls}: "${selectedProgrammer.name}"`
 : programmerNls;
this.menuRegistry.registerSubmenu(programmersMenuPath,label);
this.toDisposeOnBoardChange.push(
Disposable.create(()=>
unregisterSubmenu(programmersMenuPath,this.menuRegistry)
)
);
for(constprogrammerofprogrammers){
const{ id, name }=programmer;
constcommand={id: `${fqbn}-programmer--${id}`};
consthandler={
execute: ()=>
this.boardsDataStore.selectProgrammer({
 fqbn,
selectedProgrammer: programmer,
}),
isToggled: ()=>
Programmer.equals(programmer,selectedProgrammer),
};
this.menuRegistry.registerMenuAction(programmersMenuPath,{
commandId: command.id,
label: name,
});
this.commandRegistry.registerCommand(command,handler);
this.toDisposeOnBoardChange.pushAll([
Disposable.create(()=>
this.commandRegistry.unregisterCommand(command)
),
Disposable.create(()=>
this.menuRegistry.unregisterMenuAction(command.id)
),
]);
}
}
this.menuManager.update();
}
}
});
}
}