arduino-ide/arduino-ide-extension/src/browser/theia/core/application-shell.ts

import { injectable, inject } from 'inversify';
import { EditorWidget } from '@theia/editor/lib/browser';
import { CommandService } from '@theia/core/lib/common/command';
import { MessageService } from '@theia/core/lib/common/message-service';
import { OutputWidget } from '@theia/output/lib/browser/output-widget';
import {
  ConnectionStatusService,
  ConnectionStatus,
} from '@theia/core/lib/browser/connection-status-service';
import {
  ApplicationShell as TheiaApplicationShell,
  DockPanel,
  Panel,
  Widget,
} from '@theia/core/lib/browser';
import { Sketch } from '../../../common/protocol';
import { SaveAsSketch } from '../../contributions/save-as-sketch';
import { SketchesServiceClientImpl } from '../../../common/protocol/sketches-service-client-impl';
import { nls } from '@theia/core/lib/common';
import URI from '@theia/core/lib/common/uri';

@injectable()
export class ApplicationShell extends TheiaApplicationShell {
  @inject(CommandService)
  protected readonly commandService: CommandService;

  @inject(MessageService)
  protected readonly messageService: MessageService;

  @inject(SketchesServiceClientImpl)
  protected readonly sketchesServiceClient: SketchesServiceClientImpl;

  @inject(ConnectionStatusService)
  protected readonly connectionStatusService: ConnectionStatusService;

  protected track(widget: Widget): void {
    super.track(widget);
    if (widget instanceof OutputWidget) {
      widget.title.closable = false; // TODO: https://arduino.slack.com/archives/C01698YT7S4/p1598011990133700
    }
    if (widget instanceof EditorWidget) {
      // Make the editor un-closeable asynchronously.
      this.sketchesServiceClient.currentSketch().then((sketch) => {
        if (sketch) {
          if (!this.isSketchFile(widget.editor.uri, sketch.uri)) {
              return;
          }
          if (Sketch.isInSketch(widget.editor.uri, sketch)) {
            widget.title.closable = false;
          }
        }
      });
    }
  }

  private isSketchFile(uri: URI, sketchUriString: string): boolean {
      const sketchUri = new URI(sketchUriString);
      if (uri.parent.isEqual(sketchUri)) {
          return true;
      }
      return false;
  }

  async addWidget(
    widget: Widget,
    options: Readonly<TheiaApplicationShell.WidgetOptions> = {}
  ): Promise<void> {
    // By default, Theia open a widget **next** to the currently active in the target area.
    // Instead of this logic, we want to open the new widget after the last of the target area.
    if (!widget.id) {
      console.error(
        'Widgets added to the application shell must have a unique id property.'
      );
      return;
    }
    let ref: Widget | undefined = options.ref;
    const area: TheiaApplicationShell.Area = options.area || 'main';
    if (!ref && (area === 'main' || area === 'bottom')) {
      const tabBar = this.getTabBarFor(area);
      if (tabBar) {
        const last = tabBar.titles[tabBar.titles.length - 1];
        if (last) {
          ref = last.owner;
        }
      }
    }
    return super.addWidget(widget, { ...options, ref });
  }

  handleEvent(): boolean {
    // NOOP, dragging has been disabled
    return false
  }

  // Avoid hiding top panel as we use it for arduino toolbar
  protected createTopPanel(): Panel {
    const topPanel = super.createTopPanel();
    topPanel.show();
    return topPanel;
  }

  async saveAll(): Promise<void> {
    if (
      this.connectionStatusService.currentStatus === ConnectionStatus.OFFLINE
    ) {
      this.messageService.error(
        nls.localize(
          'theia/core/couldNotSave',
          'Could not save the sketch. Please copy your unsaved work into your favorite text editor, and restart the IDE.'
        )
      );
      return; // Theia does not reject on failed save: https://github.com/eclipse-theia/theia/pull/8803
    }
    await super.saveAll();
    const options = { execOnlyIfTemp: true, openAfterMove: true };
    await this.commandService.executeCommand(
      SaveAsSketch.Commands.SAVE_AS_SKETCH.id,
      options
    );
  }
}

const originalHandleEvent = DockPanel.prototype.handleEvent;

DockPanel.prototype.handleEvent = function (event) {
  switch (event.type) {
      case 'p-dragenter':
      case 'p-dragleave':
      case 'p-dragover':
      case 'p-drop':
        return;
  }
  originalHandleEvent(event);
};