arduino-ide/arduino-ide-extension/src/browser/theia/core/connection-status-service.ts

import { inject, injectable, postConstruct } from 'inversify';
import { Disposable } from '@theia/core/lib/common/disposable';
import { StatusBarAlignment } from '@theia/core/lib/browser/status-bar/status-bar';
import {
  FrontendConnectionStatusService as TheiaFrontendConnectionStatusService,
  ApplicationConnectionStatusContribution as TheiaApplicationConnectionStatusContribution,
  ConnectionStatus,
} from '@theia/core/lib/browser/connection-status-service';
import { ArduinoDaemon } from '../../../common/protocol';
import { NotificationCenter } from '../../notification-center';

@injectable()
export class FrontendConnectionStatusService extends TheiaFrontendConnectionStatusService {
  @inject(ArduinoDaemon)
  protected readonly daemon: ArduinoDaemon;

  @inject(NotificationCenter)
  protected readonly notificationCenter: NotificationCenter;

  protected isRunning = false;

  @postConstruct()
  protected async init(): Promise<void> {
    this.schedulePing();
    try {
      this.isRunning = await this.daemon.isRunning();
    } catch {}
    this.notificationCenter.onDaemonStarted(() => (this.isRunning = true));
    this.notificationCenter.onDaemonStopped(() => (this.isRunning = false));
    this.wsConnectionProvider.onIncomingMessageActivity(() => {
      this.updateStatus(this.isRunning);
      this.schedulePing();
    });
  }
}

@injectable()
export class ApplicationConnectionStatusContribution extends TheiaApplicationConnectionStatusContribution {
  @inject(ArduinoDaemon)
  protected readonly daemon: ArduinoDaemon;

  @inject(NotificationCenter)
  protected readonly notificationCenter: NotificationCenter;

  protected isRunning = false;

  @postConstruct()
  protected async init(): Promise<void> {
    try {
      this.isRunning = await this.daemon.isRunning();
    } catch {}
    this.notificationCenter.onDaemonStarted(() => (this.isRunning = true));
    this.notificationCenter.onDaemonStopped(() => (this.isRunning = false));
  }

  protected onStateChange(state: ConnectionStatus): void {
    if (!this.isRunning && state === ConnectionStatus.ONLINE) {
      return;
    }
    super.onStateChange(state);
  }

  protected handleOffline(): void {
    this.statusBar.setElement('connection-status', {
      alignment: StatusBarAlignment.LEFT,
      text: this.isRunning ? 'Offline' : '$(bolt) CLI Daemon Offline',
      tooltip: this.isRunning
        ? 'Cannot connect to the backend.'
        : 'Cannot connect to the CLI daemon.',
      priority: 5000,
    });
    this.toDisposeOnOnline.push(
      Disposable.create(() => this.statusBar.removeElement('connection-status'))
    );
    document.body.classList.add('theia-mod-offline');
    this.toDisposeOnOnline.push(
      Disposable.create(() =>
        document.body.classList.remove('theia-mod-offline')
      )
    );
  }
}