Draft command

src/handler.ts

@@ -107,10 +107,12 @@ export class DebuggerHandler {
    *
    * @param widget The widget to update.
    * @param connection The session connection.
+   * @param manuallyRun - Flag for run button function manually
    */
   async update(
     widget: DebuggerHandler.SessionWidget[DebuggerHandler.SessionType],
-    connection: Session.ISessionConnection
+    connection: Session.ISessionConnection,
+    manuallyRun?: boolean
   ): Promise<void> {
     if (!connection) {
       delete this._kernelChangedHandlers[widget.id];
@@ -143,8 +145,7 @@ export class DebuggerHandler {
     }
     connection.statusChanged.connect(statusChanged);
     this._statusChangedHandlers[widget.id] = statusChanged;
-
-    return this._update(widget, connection);
+    return this._update(widget, connection, manuallyRun);
   }
 
   /**
@@ -153,10 +154,12 @@ export class DebuggerHandler {
    *
    * @param widget The widget to update.
    * @param sessionContext The session context.
+   * @param manuallyRun - Flag for run button function manually
    */
   async updateContext(
     widget: DebuggerHandler.SessionWidget[DebuggerHandler.SessionType],
-    sessionContext: ISessionContext
+    sessionContext: ISessionContext,
+    manuallyRun?: boolean
   ): Promise<void> {
     const connectionChanged = (): void => {
       const { session: connection } = sessionContext;
@@ -173,18 +176,20 @@ export class DebuggerHandler {
     this._contextKernelChangedHandlers[widget.id] = connectionChanged;
     sessionContext.kernelChanged.connect(connectionChanged);
 
-    return this.update(widget, sessionContext.session);
+    return this.update(widget, sessionContext.session, manuallyRun);
   }
 
   /**
    * Update a debug handler for the given widget.
    *
    * @param widget The widget to update.
    * @param connection The session connection.
+   * @param manuallyRun - Flag for run button function manually
    */
-  private async _update(
+  async _update(
     widget: DebuggerHandler.SessionWidget[DebuggerHandler.SessionType],
-    connection: Session.ISessionConnection
+    connection: Session.ISessionConnection,
+    manuallyRun?: boolean
   ): Promise<void> {
     if (!this._service.model) {
       return;
@@ -311,6 +316,9 @@ export class DebuggerHandler {
     await this._service.restoreState(false);
     addToolbarButton();
 
+    if (manuallyRun) {
+      await toggleDebugging();
+    }
     // check the state of the debug session
     if (!this._service.isStarted) {
       removeHandlers();

src/index.ts

@@ -69,6 +69,8 @@ export namespace CommandIDs {
   export const stepOut = 'debugger:stepOut';
 
   export const inspectVariable = 'debugger:inspect-variable';
+
+  export const startDebug = 'debugger:start-debug';
 }
 
 /**
@@ -207,6 +209,26 @@ const notebooks: JupyterFrontEndPlugin<void> = {
     notebookTracker: INotebookTracker,
     labShell: ILabShell
   ) => {
+    app.commands.addCommand(CommandIDs.startDebug, {
+      label: 'StartDebug',
+      caption: 'Start Debug',
+      icon: stepOutIcon,
+      isEnabled: () => {
+        // @todo should be something similar like threadHasStopped
+        return true;
+      },
+      execute: async () => {
+        await updateCurrentWidget();
+      }
+    });
+
+    // @todo selector probably should be different, shortcut key is draft as well
+    app.commands.addKeyBinding({
+      command: CommandIDs.startDebug,
+      keys: ['Alt A'],
+      selector: 'div'
+    });
+
     const handler = new DebuggerHandler({
       type: 'notebook',
       shell: app.shell,
@@ -216,14 +238,22 @@ const notebooks: JupyterFrontEndPlugin<void> = {
       handler.disposeAll(service);
     });
     const updateHandlerAndCommands = async (
-      widget: NotebookPanel
+      widget: NotebookPanel,
+      manuallyRun?: boolean
     ): Promise<void> => {
       const { sessionContext } = widget;
       await sessionContext.ready;
-      await handler.updateContext(widget, sessionContext);
+      await handler.updateContext(widget, sessionContext, manuallyRun);
       app.commands.notifyCommandChanged();
     };
 
+    const updateCurrentWidget = async (): Promise<void> => {
+      if (labShell) {
+        const widget = notebookTracker.currentWidget;
+        await updateHandlerAndCommands(widget, true);
+      }
+    };
+
     if (labShell) {
       labShell.currentChanged.connect(async (_, update) => {
         const widget = update.newValue;