Temp refactor

Author: rjmholt

src/PowerShellEditorServices/Services/DebugAdapter/DebugEventHandlerService.cs

@@ -92,7 +92,7 @@ private void PowerShellContext_RunspaceChanged(object sender, RunspaceChangedEve
         {
             if (_debugStateService.WaitingForAttach &&
                 e.ChangeAction == RunspaceChangeAction.Enter &&
-                e.NewRunspace.Context == RunspaceContext.DebuggedRunspace)
+                e.NewRunspace.Context == RunspaceOrigin.DebuggedRunspace)
             {
                 // Send the InitializedEvent so that the debugger will continue
                 // sending configuration requests

src/PowerShellEditorServices/Services/DebugAdapter/DebugService.cs

@@ -414,7 +414,7 @@ private async Task OnExecutionCompletedAsync(Task executeTask)
             if (_debugStateService.IsAttachSession)
             {
                // Pop the sessions
-               if (_powerShellContextService.CurrentRunspace.Context == RunspaceContext.EnteredProcess)
+               if (_powerShellContextService.CurrentRunspace.Context == RunspaceOrigin.EnteredProcess)
                {
                    try
                    {

src/PowerShellEditorServices/Services/DebugAdapter/Handlers/LaunchAndAttachHandler.cs

@@ -0,0 +1,22 @@
+using System;
+using System.Management.Automation;
+using System.Management.Automation.Runspaces;
+using SMA = System.Management.Automation;
+
+namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Context
+{
+    internal interface IPowerShellContext : IDisposable
+    {
+        SMA.PowerShell CurrentPowerShell { get; }
+
+        bool IsRunspacePushed { get; }
+
+        void SetShouldExit(int exitCode);
+
+        void ProcessDebuggerResult(DebuggerCommandResults debuggerResult);
+
+        void PushNestedPowerShell();
+
+        void PushPowerShell(Runspace runspaceToUse);
+    }
+}

src/PowerShellEditorServices/Services/PowerShell/Context/IPowerShellContext.cs

@@ -0,0 +1,47 @@
+using Microsoft.PowerShell.EditorServices.Services.PowerShell.Context;
+using System;
+using System.Collections.Generic;
+using System.Management.Automation.Runspaces;
+using System.Runtime.CompilerServices;
+using System.Text;
+using SMA = System.Management.Automation;
+
+namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Context
+{
+    internal class PowerShellContext : IPowerShellContext
+    {
+        private readonly Stack<PowerShellContextFrame> _psFrameStack;
+
+        public PowerShellContext()
+        {
+            _psFrameStack = new Stack<PowerShellContextFrame>();
+        }
+
+        public SMA.PowerShell CurrentPowerShell => _psFrameStack.Peek().PowerShell;
+
+        private void PushFrame(PowerShellContextFrame frame)
+        {
+            if (_psFrameStack.Count > 0)
+            {
+                RemoveRunspaceEventHandlers(CurrentPowerShell.Runspace);
+            }
+            AddRunspaceEventHandlers(frame.PowerShell.Runspace);
+            _psFrameStack.Push(frame);
+            RunPowerShellLoop(frame.FrameType);
+        }
+
+        private void AddRunspaceEventHandlers(Runspace runspace)
+        {
+            runspace.Debugger.DebuggerStop += OnDebuggerStopped;
+            runspace.Debugger.BreakpointUpdated += OnBreakpointUpdated;
+            runspace.StateChanged += OnRunspaceStateChanged;
+        }
+
+        private void RemoveRunspaceEventHandlers(Runspace runspace)
+        {
+            runspace.Debugger.DebuggerStop -= OnDebuggerStopped;
+            runspace.Debugger.BreakpointUpdated -= OnBreakpointUpdated;
+            runspace.StateChanged -= OnRunspaceStateChanged;
+        }
+    }
+}

src/PowerShellEditorServices/Services/PowerShell/Context/PowerShellContext.cs

@@ -0,0 +1,46 @@
+using Microsoft.PowerShell.EditorServices.Services.PowerShell.Context;
+using System;
+using System.Threading;
+using SMA = System.Management.Automation;
+
+namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Context
+{
+    internal class PowerShellContextFrame : IDisposable
+    {
+        private bool disposedValue;
+
+        public PowerShellContextFrame(SMA.PowerShell powerShell, PowerShellFrameType frameType, CancellationTokenSource cancellationTokenSource)
+        {
+            PowerShell = powerShell;
+            FrameType = frameType;
+            CancellationTokenSource = cancellationTokenSource;
+        }
+
+        public SMA.PowerShell PowerShell { get; }
+
+        public PowerShellFrameType FrameType { get; }
+
+        public CancellationTokenSource CancellationTokenSource { get; }
+
+        protected virtual void Dispose(bool disposing)
+        {
+            if (!disposedValue)
+            {
+                if (disposing)
+                {
+                    PowerShell.Dispose();
+                    CancellationTokenSource.Dispose();
+                }
+
+                disposedValue = true;
+            }
+        }
+
+        public void Dispose()
+        {
+            // Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method
+            Dispose(disposing: true);
+            GC.SuppressFinalize(this);
+        }
+    }
+}

src/PowerShellEditorServices/Services/PowerShell/Context/PowerShellContextFrame.cs

@@ -1,6 +1,6 @@
 using System;
 
-namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Execution
+namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Context
 {
     [Flags]
     internal enum PowerShellFrameType

src/PowerShellEditorServices/Services/PowerShell/Execution/PowerShellFrameType.cs renamed to src/PowerShellEditorServices/Services/PowerShell/Context/PowerShellFrameType.cs

@@ -1,6 +1,6 @@
 using System.Management.Automation;
 
-namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Execution
+namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Debugging
 {
     internal class DebuggerResumingEventArgs
     {

src/PowerShellEditorServices/Services/PowerShell/Execution/DebuggerResumingEventArgs.cs renamed to src/PowerShellEditorServices/Services/PowerShell/Debugging/DebuggerResumingEventArgs.cs

@@ -0,0 +1,32 @@
+using Microsoft.PowerShell.EditorServices.Services.PowerShellContext;
+using System;
+using System.Collections.Generic;
+using System.Management.Automation;
+using System.Text;
+using System.Threading;
+
+namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Debugging
+{
+    internal interface IPowerShellDebugContext
+    {
+        bool IsStopped { get; }
+
+        DscBreakpointCapability DscBreakpointCapability { get; }
+
+        DebuggerStopEventArgs LastStopEventArgs { get; }
+
+        CancellationToken OnResumeCancellationToken { get; }
+
+        void Continue();
+
+        void StepOver();
+
+        void StepInto();
+
+        void StepOut();
+
+        void Break();
+
+        void Abort();
+    }
+}