Changes necessary for exposing the RawResponse (#147)

Author: Tiberriver256

lib/Polaris.Class.ps1

@@ -29,12 +29,11 @@ class Polaris {
             $Polaris.Listener.BeginGetContext($Polaris.ContextHandler, $Polaris)
 
             [System.Net.HttpListenerRequest]$RawRequest = $Context.Request
-            [System.Net.HttpListenerResponse]$RawResponse = $Context.Response
 
             $Polaris.Log("request came in: " + $RawRequest.HttpMethod + " " + $RawRequest.RawUrl)
 
             [PolarisRequest]$Request = [PolarisRequest]::new($RawRequest)
-            [PolarisResponse]$Response = [PolarisResponse]::new()
+            [PolarisResponse]$Response = [PolarisResponse]::new($Context.Response)
 
 
             [string]$Route = $RawRequest.Url.AbsolutePath
@@ -118,15 +117,15 @@ class Polaris {
                     $Response.ByteResponse.CopyTo($Bytes, $LogBytes.Length)
                     $Response.ByteResponse = $Bytes
                 }
-                [Polaris]::Send($RawResponse, $Response)
+                [Polaris]::Send($Response)
 
             }
             catch {
                 $Polaris.Log(($_ | Out-String))
                 $Response.SetStatusCode(500)
                 $Response.Send($_)
                 try{
-                    [Polaris]::Send($RawResponse, $Response)
+                    [Polaris]::Send($Response)
                 } catch {
                     $Polaris.Log($_)
                 }
@@ -305,12 +304,11 @@ class Polaris {
     }
 
     static [void] Send (
-        [System.Net.HttpListenerResponse]$RawResponse, 
         [PolarisResponse]$Response
     ) {
         if ($Response.StreamResponse) {
             [Polaris]::Send(
-                $RawResponse,
+                $Response.RawResponse,
                 $Response.StreamResponse,
                 $Response.StatusCode,
                 $Response.ContentType,
@@ -319,7 +317,7 @@ class Polaris {
         }
         else {
             [Polaris]::Send(
-                $RawResponse,
+                $Response.RawResponse,
                 $Response.ByteResponse,
                 $Response.StatusCode,
                 $Response.ContentType,

lib/PolarisResponse.Class.ps1

@@ -4,6 +4,11 @@ class PolarisResponse {
     [System.Net.WebHeaderCollection]$Headers = [System.Net.WebHeaderCollection]::new()
     [int]$StatusCode = 200
     [System.IO.Stream]$StreamResponse
+    [System.Net.HttpListenerResponse]$RawResponse
+
+    PolarisResponse ([System.Net.HttpListenerResponse]$RawResponse) {
+        $this.RawResponse = $RawResponse
+    }
 
     Send ([string]$stringResponse) {
         $this.ByteResponse = [System.Text.Encoding]::UTF8.GetBytes($stringResponse)