Class ExcludedCallersUI

src/ui/excludedCallersUI.ts:73–172

@injectable()
export class ExcludedCallersUI
  implements vscode.TreeDataProvider<ExcludedCaller>, IExtensionContribution
{
  private readonly _onDidChangeTreeData = new vscode.EventEmitter<ExcludedCaller | undefined>();
  private allCallers = new Map<string, ExcludedCaller>();
  private lastHadCallers = false;

  constructor(
    @inject(DebugSessionTracker) private readonly sessionTracker: DebugSessionTracker,
  ) {}

  /** @inheritdoc */
  register(context: vscode.ExtensionContext): void {
    context.subscriptions.push(
      vscode.window.createTreeView(CustomViews.ExcludedCallers, {
        treeDataProvider: this,
      }),
      registerCommand(vscode.commands, Commands.CallersAdd, async (_uri, context) => {
        const stack = await this.sessionTracker
          .getById(context.sessionId)
          ?.customRequest('stackTrace', {
            threadId: 0, // js-debug doesn't do threads, so ID is always 0
            startFrame: 0,
            levels: 1,
          });

        if (!stack?.stackFrames.length) {
          return;
        }

        const topOfStack = stack.stackFrames[0];
        const caller = new ExcludedCaller(
          {
            name: context.frameName,
            column: context.frameLocation.range.startColumn,
            line: context.frameLocation.range.startLineNumber,
            source: context.frameLocation.source,
          },
          {
            name: topOfStack.name,
            column: topOfStack.column,
            line: topOfStack.line,
            source: topOfStack.source,
          },
        );

        this.allCallers.set(caller.id, caller);
        this.triggerUpdate();
      }),
      registerCommand(vscode.commands, Commands.CallersGoToCaller, c => revealLocation(c.caller)),
      registerCommand(vscode.commands, Commands.CallersGoToTarget, c => revealLocation(c.target)),
      registerCommand(vscode.commands, Commands.CallersRemove, async c => {
        this.allCallers.delete(c.id);
        this.triggerUpdate();
      }),
      registerCommand(vscode.commands, Commands.CallersRemoveAll, () => {
        this.allCallers.clear();
        this.triggerUpdate();