Private/Start-HttpListener.ps1
|
<#
.SYNOPSIS Starts an HTTP listener on the given prefix. .DESCRIPTION The Start-HttpListener function starts an HTTP listener on the given prefix. It will wait for a single request and then stop the listener. .PARAMETER Prefix The prefix to start the HTTP listener on. Defaults to "http://localhost:8080/". .EXAMPLE Start-HttpListener -Prefix "http://localhost:8081/" #> function Start-HttpListener { [CmdletBinding()] param ( [string]$Prefix = "http://localhost:8080/" ) $catcher = Start-Job -Name "HttpListenerCatchCode" -ScriptBlock { param ($prefix) $httpListener = New-Object System.Net.HttpListener $httpListener.Prefixes.Add($prefix) $httpListener.Start() while ($httpListener.IsListening) { try { $context = $httpListener.GetContext() $response = $context.Response $request = $context.Request $authCode = $request.QueryString["code"] if ($authCode) { $response.StatusCode = 200 $response.ContentType = 'text/html' [byte[]]$buffer = [Text.Encoding]::UTF8.GetBytes("<html><body>Authorization code received: $authCode</body></html>") $response.ContentLength64 = $buffer.Length $response.OutputStream.Write($buffer, 0, $buffer.Length) $response.Close() $httpListener.Stop() return $authCode } else { $response.StatusCode = 400 $response.ContentType = 'text/html' $response.OutputStream.Write([Text.Encoding]::UTF8.GetBytes("<html><body>Error: Authorization code not found</body></html>"), 0, 50) $response.Close() $httpListener.Stop() } } catch { # Handle any exceptions that occur during GetContext } } } -ArgumentList $Prefix $authCode = Receive-Job -Job $catcher -Wait -AutoRemoveJob return $authCode } |