Merge pull request #6807 from IanMatthewHuff/release

Debug Mode Code lenses (#6792)

Author: IanMatthewHuff

news/1 Enhancements/6672.md

@@ -0,0 +1 @@
+Add debug command code lenses when in debug mode

package.json

@@ -351,6 +351,26 @@
                 "title": "%python.command.python.datascience.runcurrentcell.title%",
                 "category": "Python"
             },
+            {
+                "command": "python.datascience.debugcell",
+                "title": "%python.command.python.datascience.debugcell.title%",
+                "category": "Python"
+            },
+            {
+                "command": "python.datascience.debugstepover",
+                "title": "%python.command.python.datascience.debugstepover.title%",
+                "category": "Python"
+            },
+            {
+                "command": "python.datascience.debugstop",
+                "title": "%python.command.python.datascience.debugstop.title%",
+                "category": "Python"
+            },
+            {
+                "command": "python.datascience.debugcontinue",
+                "title": "%python.command.python.datascience.debugcontinue.title%",
+                "category": "Python"
+            },
             {
                 "command": "python.datascience.runcurrentcelladvance",
                 "title": "%python.command.python.datascience.runcurrentcelladvance.title%",
@@ -1430,6 +1450,12 @@
                     "description": "Set of commands to put as code lens above a cell. Defaults to 'python.datascience.runcell,  python.datascience.runallcellsabove, python.datascience.debugcell'",
                     "scope": "resource"
                 },
+                "python.dataScience.debugCodeLenses": {
+                    "type": "string",
+                    "default": "python.datascience.debugcontinue, python.datascience.debugstop, python.datascience.debugstepover",
+                    "description": "Set of debug commands to put as code lens above a cell while debugging.",
+                    "scope": "resource"
+                },
                 "python.dataScience.ptvsdDistPath": {
                     "type": "string",
                     "default": "",

package.nls.json

@@ -39,6 +39,9 @@
     "python.command.python.datascience.runcurrentcellandallbelow.palette.title": "Run Current Cell and Below",
     "python.command.python.datascience.debugcurrentcell.palette.title": "Debug Current Cell",
     "python.command.python.datascience.debugcell.title": "Debug Cell",
+    "python.command.python.datascience.debugstepover.title": "Step Over",
+    "python.command.python.datascience.debugcontinue.title": "Continue",
+    "python.command.python.datascience.debugstop.title": "Stop",
     "python.command.python.datascience.runtoline.title": "Run To Line in Python Interactive Window",
     "python.command.python.datascience.runfromline.title": "Run From Line in Python Interactive Window",
     "python.command.python.datascience.runcurrentcell.title": "Run Current Cell",
@@ -305,6 +308,9 @@
     "DataScience.jupyterDataRateExceeded": "Cannot view variable because data rate exceeded. Please restart your server with a higher data rate limit. For example, --NotebookApp.iopub_data_rate_limit=10000000000.0",
     "DataScience.addCellBelowCommandTitle": "Add cell",
     "DataScience.debugCellCommandTitle": "Debug cell",
+    "DataScience.debugStepOverCommandTitle": "Step over",
+    "DataScience.debugContinueCommandTitle": "Continue",
+    "DataScience.debugStopCommandTitle": "Stop",
     "DataScience.runCurrentCellAndAddBelow": "Run current and add cell below",
     "DataScience.variableExplorerDisabledDuringDebugging": "Please see the Debug Side Bar's VARIABLES section.",
     "DataScience.jupyterDebuggerNotInstalledError": "Pip module ptvsd is required for debugging cells. You will need to install it to debug cells.",

src/client/common/application/commands.ts

@@ -25,6 +25,8 @@ interface ICommandNameWithoutArgumentTypeMapping {
     [Commands.Set_ShebangInterpreter]: [];
     [Commands.Run_Linter]: [];
     [Commands.Enable_Linter]: [];
+    ['workbench.action.debug.continue']: [];
+    ['workbench.action.debug.stepOver']: [];
     ['workbench.action.debug.stop']: [];
     ['workbench.action.reloadWindow']: [];
     ['editor.action.formatDocument']: [];
@@ -115,6 +117,9 @@ export interface ICommandNameArgumentTypeMapping extends ICommandNameWithoutArgu
     [DSCommands.RunFileInInteractiveWindows]: [string];
     [DSCommands.DebugFileInInteractiveWindows]: [string];
     [DSCommands.DebugCell]: [string, number, number, number, number];
+    [DSCommands.DebugStepOver]: [];
+    [DSCommands.DebugStop]: [];
+    [DSCommands.DebugContinue]: [];
     [DSCommands.RunCurrentCellAndAddBelow]: [string];
     [DSCommands.ScrollToCell]: [string, string];
 }

src/client/common/application/debugService.ts

@@ -38,6 +38,10 @@ export class DebugService implements IDebugService {
     public registerDebugConfigurationProvider(debugType: string, provider: any): Disposable {
         return debug.registerDebugConfigurationProvider(debugType, provider);
     }
+    // tslint:disable-next-line:no-any
+    public registerDebugAdapterTrackerFactory(debugType: string, provider: any): Disposable {
+        return debug.registerDebugAdapterTrackerFactory(debugType, provider);
+    }
     public startDebugging(folder: WorkspaceFolder | undefined, nameOrConfiguration: string | DebugConfiguration, parentSession?: DebugSession): Thenable<boolean> {
         return debug.startDebugging(folder, nameOrConfiguration, parentSession);
     }

src/client/common/application/types.ts

@@ -7,6 +7,7 @@ import {
     CancellationToken,
     CompletionItemProvider,
     ConfigurationChangeEvent,
+    DebugAdapterTrackerFactory,
     DebugConfiguration,
     DebugConfigurationProvider,
     DebugConsole,
@@ -764,6 +765,15 @@ export interface IDebugService {
      */
     registerDebugConfigurationProvider(debugType: string, provider: DebugConfigurationProvider): Disposable;
 
+    /**
+     * Register a debug adapter tracker factory for the given debug type.
+     *
+     * @param debugType The debug type for which the factory is registered or '*' for matching all debug types.
+     * @param factory The [debug adapter tracker factory](#DebugAdapterTrackerFactory) to register.
+     * @return A [disposable](#Disposable) that unregisters this factory when being disposed.
+     */
+    registerDebugAdapterTrackerFactory(debugType: string, factory: DebugAdapterTrackerFactory): Disposable;
+
     /**
      * Start debugging by using either a named launch or named compound configuration,
      * or by directly passing a [DebugConfiguration](#DebugConfiguration).

src/client/common/types.ts

@@ -324,6 +324,7 @@ export interface IDataScienceSettings {
     askForKernelRestart?: boolean;
     enablePlotViewer?: boolean;
     codeLenses?: string;
+    debugCodeLenses?: string;
     ptvsdDistPath?: string;
     stopOnFirstLineWhileDebugging?: boolean;
     textOutputLimit?: number;

src/client/common/utils/localize.ts

@@ -229,6 +229,9 @@ export namespace DataScience {
     export const jupyterDataRateExceeded = localize('DataScience.jupyterDataRateExceeded', 'Cannot view variable because data rate exceeded. Please restart your server with a higher data rate limit. For example, --NotebookApp.iopub_data_rate_limit=10000000000.0');
     export const addCellBelowCommandTitle = localize('DataScience.addCellBelowCommandTitle', 'Add cell');
     export const debugCellCommandTitle = localize('DataScience.debugCellCommandTitle', 'Debug cell');
+    export const debugStepOverCommandTitle = localize('DataScience.debugStepOverCommandTitle', 'Step over');
+    export const debugContinueCommandTitle = localize('DataScience.debugContinueCommandTitle', 'Continue');
+    export const debugStopCommandTitle = localize('DataScience.debugStopCommandTitle', 'Stop');
     export const runCurrentCellAndAddBelow = localize('DataScience.runCurrentCellAndAddBelow', 'Run current and add cell below');
     export const variableExplorerDisabledDuringDebugging = localize('DataScience.variableExplorerDisabledDuringDebugging', 'Please see the Debug Side Bar\'s VARIABLES section.');
     export const jupyterDebuggerNotInstalledError = localize('DataScience.jupyterDebuggerNotInstalledError', 'Pip module ptvsd is required for debugging cells. You will need to install it to debug cells.');

src/client/datascience/constants.ts

@@ -36,10 +36,22 @@ export namespace Commands {
     export const AddCellBelow = 'python.datascience.addcellbelow';
     export const DebugCurrentCellPalette = 'python.datascience.debugcurrentcell.palette';
     export const DebugCell = 'python.datascience.debugcell';
+    export const DebugStepOver = 'python.datascience.debugstepover';
+    export const DebugContinue = 'python.datascience.debugcontinue';
+    export const DebugStop = 'python.datascience.debugstop';
     export const RunCurrentCellAndAddBelow = 'python.datascience.runcurrentcellandaddbelow';
     export const ScrollToCell = 'python.datascience.scrolltocell';
 }
 
+export namespace CodeLensCommands {
+    // If not specified in the options this is the default set of commands in our design time code lenses
+    export const DefaultDesignLenses = [Commands.RunCurrentCell, Commands.RunAllCellsAbove, Commands.DebugCell];
+    // If not specified in the options this is the default set of commands in our debug time code lenses
+    export const DefaultDebuggingLenses = [Commands.DebugContinue, Commands.DebugStop, Commands.DebugStepOver];
+    // These are the commands that are allowed at debug time
+    export const DebuggerCommands = [Commands.DebugContinue, Commands.DebugStop, Commands.DebugStepOver];
+}
+
 export namespace EditorContexts {
     export const HasCodeCells = 'python.datascience.hascodecells';
     export const DataScienceEnabled = 'python.datascience.featureenabled';
@@ -152,7 +164,10 @@ export enum Telemetry {
     PtvsdPromptToInstall = 'DATASCIENCE.PTVSD_PROMPT_TO_INSTALL',
     PtvsdSuccessfullyInstalled = 'DATASCIENCE.PTVSD_SUCCESSFULLY_INSTALLED',
     PtvsdInstallFailed = 'DATASCIENCE.PTVSD_INSTALL_FAILED',
-    ScrolledToCell = 'DATASCIENCE.SCROLLED_TO_CELL'
+    ScrolledToCell = 'DATASCIENCE.SCROLLED_TO_CELL',
+    DebugStepOver = 'DATASCIENCE.DEBUG_STEP_OVER',
+    DebugContinue = 'DATASCIENCE.DEBUG_CONTINUE',
+    DebugStop = 'DATASCIENCE.DEBUG_STOP'
 }
 
 export namespace HelpLinks {

src/client/datascience/datascience.ts

@@ -8,7 +8,7 @@ import { inject, injectable } from 'inversify';
 import { URL } from 'url';
 import * as vscode from 'vscode';
 
-import { IApplicationShell, ICommandManager, IDocumentManager, IWorkspaceService } from '../common/application/types';
+import { IApplicationShell, ICommandManager, IDebugService, IDocumentManager, IWorkspaceService } from '../common/application/types';
 import { PYTHON_ALLFILES, PYTHON_LANGUAGE } from '../common/constants';
 import { ContextKey } from '../common/contextKey';
 import { traceError } from '../common/logger';
@@ -43,6 +43,7 @@ export class DataScience implements IDataScience {
         @inject(IConfigurationService) private configuration: IConfigurationService,
         @inject(IDocumentManager) private documentManager: IDocumentManager,
         @inject(IApplicationShell) private appShell: IApplicationShell,
+        @inject(IDebugService) private debugService: IDebugService,
         @inject(IWorkspaceService) private workspace: IWorkspaceService
     ) {
         this.commandListeners = this.serviceContainer.getAll<IDataScienceCommandListener>(IDataScienceCommandListener);
@@ -243,6 +244,36 @@ export class DataScience implements IDataScience {
         }
     }
 
+    @captureTelemetry(Telemetry.DebugStepOver)
+    public async debugStepOver(): Promise<void> {
+        this.dataScienceSurveyBanner.showBanner().ignoreErrors();
+
+        // Make sure that we are in debug mode
+        if (this.debugService.activeDebugSession) {
+            this.commandManager.executeCommand('workbench.action.debug.stepOver');
+        }
+    }
+
+    @captureTelemetry(Telemetry.DebugStop)
+    public async debugStop(): Promise<void> {
+        this.dataScienceSurveyBanner.showBanner().ignoreErrors();
+
+        // Make sure that we are in debug mode
+        if (this.debugService.activeDebugSession) {
+            this.commandManager.executeCommand('workbench.action.debug.stop');
+        }
+    }
+
+    @captureTelemetry(Telemetry.DebugContinue)
+    public async debugContinue(): Promise<void> {
+        this.dataScienceSurveyBanner.showBanner().ignoreErrors();
+
+        // Make sure that we are in debug mode
+        if (this.debugService.activeDebugSession) {
+            this.commandManager.executeCommand('workbench.action.debug.continue');
+        }
+    }
+
     @captureTelemetry(Telemetry.SetJupyterURIToLocal)
     private async setJupyterURIToLocal(): Promise<void> {
         await this.configuration.updateSetting('dataScience.jupyterServerURI', Settings.JupyterServerLocalLaunch, undefined, vscode.ConfigurationTarget.Workspace);
@@ -417,6 +448,12 @@ export class DataScience implements IDataScience {
         this.disposableRegistry.push(disposable);
         disposable = this.commandManager.registerCommand(Commands.DebugCell, this.debugCell, this);
         this.disposableRegistry.push(disposable);
+        disposable = this.commandManager.registerCommand(Commands.DebugStepOver, this.debugStepOver, this);
+        this.disposableRegistry.push(disposable);
+        disposable = this.commandManager.registerCommand(Commands.DebugContinue, this.debugContinue, this);
+        this.disposableRegistry.push(disposable);
+        disposable = this.commandManager.registerCommand(Commands.DebugStop, this.debugStop, this);
+        this.disposableRegistry.push(disposable);
         disposable = this.commandManager.registerCommand(Commands.DebugCurrentCellPalette, this.debugCurrentCellFromCursor, this);
         this.disposableRegistry.push(disposable);
         this.commandListeners.forEach((listener: IDataScienceCommandListener) => {