Class PrettyPrintUI

src/ui/prettyPrint.ts:15–92

@injectable()
export class PrettyPrintUI implements IExtensionContribution {
  private readonly canPrettyPrintKey = new ManagedContextKey(ContextKey.CanPrettyPrint);

  constructor(@inject(DebugSessionTracker) private readonly tracker: DebugSessionTracker) {}

  /** @inheritdoc */
  public register(context: vscode.ExtensionContext): void {
    context.subscriptions.push(
      registerCommand(vscode.commands, Commands.PrettyPrint, () => this.prettifyActive()),
      vscode.window.onDidChangeActiveTextEditor(editor => this.updateEditorState(editor)),
      this.tracker.onSessionAdded(() => this.updateEditorState(vscode.window.activeTextEditor)),
      this.tracker.onSessionEnded(() => this.updateEditorState(vscode.window.activeTextEditor)),
    );
  }

  /**
   * Prettifies the active file in the editor.
   */
  public async prettifyActive() {
    const editor = vscode.window.activeTextEditor;
    if (!editor || !this.canPrettyPrint(editor)) {
      return;
    }

    const { sessionId, source } = sourceForUri(editor.document.uri);
    const session = sessionId && this.tracker.getById(sessionId);

    // For ephemeral files, they're attached to a single session, so go ahead
    // and send it to the owning session. For files on disk, send it to all
    // sessions--they will no-op if they don't know about the source.
    let prettied: { session: vscode.DebugSession; result: Dap.PrettyPrintSourceResult }[];
    if (session) {
      prettied = [{
        session,
        result: await sendPrintCommand(session, source, editor.selection.start),
      }];
    } else {
      prettied = await Promise.all(
        this.tracker.getConcreteSessions().map(async session => {
          const result = await sendPrintCommand(session, source, editor.selection.start);
          return { session, result };
        }),
      );
    }

    if (!prettied.some(p => p.result.didReveal)) {
      const reveal = prettied.find(p => p.result.source);
      if (reveal) {
        const doc = await vscode.workspace.openTextDocument(
          dapSourceToDebugUri(reveal.session, reveal.result.source!),
        );
        await vscode.window.showTextDocument(doc);
      }
    }
  }

  private canPrettyPrint(editor: vscode.TextEditor) {
    return (