arduino-ide/arduino-ide-extension/src/browser/contributions/sketchbook.ts

import { inject, injectable } from 'inversify';
import { CommandHandler } from '@theia/core/lib/common/command';
import { CommandRegistry, MenuModelRegistry } from './contribution';
import { ArduinoMenus } from '../menu/arduino-menus';
import { MainMenuManager } from '../../common/main-menu-manager';
import { NotificationCenter } from '../notification-center';
import { Examples } from './examples';
import { SketchContainer } from '../../common/protocol';
import { OpenSketch } from './open-sketch';
import { nls } from '@theia/core/lib/common';

@injectable()
export class Sketchbook extends Examples {
  @inject(CommandRegistry)
  protected readonly commandRegistry: CommandRegistry;

  @inject(MenuModelRegistry)
  protected readonly menuRegistry: MenuModelRegistry;

  @inject(MainMenuManager)
  protected readonly mainMenuManager: MainMenuManager;

  @inject(NotificationCenter)
  protected readonly notificationCenter: NotificationCenter;

  onStart(): void {
    this.sketchService.getSketches({}).then((container) => {
      this.register(container);
      this.mainMenuManager.update();
    });
    this.sketchServiceClient.onSketchbookDidChange(() => {
      this.sketchService.getSketches({}).then((container) => {
        this.register(container);
        this.mainMenuManager.update();
      });
    });
  }

  registerMenus(registry: MenuModelRegistry): void {
    registry.registerSubmenu(
      ArduinoMenus.FILE__SKETCHBOOK_SUBMENU,
      nls.localize('arduino/sketch/sketchbook', 'Sketchbook'),
      { order: '3' }
    );
  }

  protected register(container: SketchContainer): void {
    this.toDispose.dispose();
    this.registerRecursively(
      [...container.children, ...container.sketches],
      ArduinoMenus.FILE__SKETCHBOOK_SUBMENU,
      this.toDispose
    );
  }

  protected createHandler(uri: string): CommandHandler {
    return {
      execute: async () => {
        const sketch = await this.sketchService.loadSketch(uri);
        return this.commandService.executeCommand(
          OpenSketch.Commands.OPEN_SKETCH.id,
          sketch
        );
      },
    };
  }
}