| | | 1 | | <# |
| | | 2 | | .SYNOPSIS |
| | | 3 | | Retrieves a request header value from the HTTP request. |
| | | 4 | | .DESCRIPTION |
| | | 5 | | This function accesses the current HTTP request context and retrieves the value |
| | | 6 | | of the specified request header by name. |
| | | 7 | | .PARAMETER Name |
| | | 8 | | The name of the request header to retrieve from the HTTP request. |
| | | 9 | | .PARAMETER AsInt |
| | | 10 | | If specified, converts the header value to an integer. |
| | | 11 | | .PARAMETER AsBool |
| | | 12 | | If specified, converts the header value to a boolean. |
| | | 13 | | .PARAMETER AsDouble |
| | | 14 | | If specified, converts the header value to a double. |
| | | 15 | | .PARAMETER AsString |
| | | 16 | | If specified, converts the header value to a string. |
| | | 17 | | .PARAMETER ThrowIfMissing |
| | | 18 | | If specified, throws an error if the header is not found. |
| | | 19 | | By default, it returns $null if not found. |
| | | 20 | | .EXAMPLE |
| | | 21 | | $value = Get-KrRequestHeader -Name "param1" |
| | | 22 | | Retrieves the value of the request header "param1" from the HTTP request. |
| | | 23 | | .EXAMPLE |
| | | 24 | | $id = Get-KrRequestHeader -Name "id" -AsInt -ThrowIfMissing |
| | | 25 | | Retrieves the value of the header "id" as an integer, throwing an error if it's missing. |
| | | 26 | | .EXAMPLE |
| | | 27 | | $flag = Get-KrRequestHeader -Name "flag" -AsBool |
| | | 28 | | Retrieves the value of the header "flag" as a boolean. |
| | | 29 | | .EXAMPLE |
| | | 30 | | $auth = Get-KrRequestHeader -Name "Authorization" -AsString -ThrowIfMissing |
| | | 31 | | Retrieves the value of the "Authorization" header as a string, throwing an error if it's missing. |
| | | 32 | | .OUTPUTS |
| | | 33 | | Returns the value of the specified request header, or $null if not found. |
| | | 34 | | .NOTES |
| | | 35 | | This function is designed to be used in the context of a Kestrun server response. |
| | | 36 | | #> |
| | | 37 | | function Get-KrRequestHeader { |
| | | 38 | | [KestrunRuntimeApi('Route')] |
| | | 39 | | [CmdletBinding(DefaultParameterSetName = 'Default')] |
| | | 40 | | [OutputType([string])] |
| | | 41 | | [OutputType([int])] |
| | | 42 | | [OutputType([bool])] |
| | | 43 | | [OutputType([double])] |
| | | 44 | | [OutputType([object])] |
| | | 45 | | param( |
| | | 46 | | [Parameter(Mandatory = $true)] |
| | | 47 | | [string]$Name, |
| | | 48 | | |
| | | 49 | | [Parameter(parameterSetName = 'Int')] |
| | | 50 | | [switch]$AsInt, |
| | | 51 | | |
| | | 52 | | [Parameter(parameterSetName = 'Bool')] |
| | | 53 | | [Alias('AsBoolean')] |
| | | 54 | | [switch]$AsBool, |
| | | 55 | | |
| | | 56 | | [Parameter(parameterSetName = 'Double')] |
| | | 57 | | [switch]$AsDouble, |
| | | 58 | | |
| | | 59 | | [Parameter(parameterSetName = 'String')] |
| | | 60 | | [switch]$AsString, |
| | | 61 | | |
| | | 62 | | [Parameter()] |
| | | 63 | | [switch]$ThrowIfMissing |
| | | 64 | | ) |
| | 0 | 65 | | if ($null -ne $Context -and $null -ne $Context.Request -and $null -ne $Context.Request.Headers) { |
| | 0 | 66 | | if ($ThrowIfMissing -and -not $Context.Request.Headers.ContainsKey($Name)) { |
| | 0 | 67 | | throw [System.ArgumentException]::new("Missing required header: $Name") |
| | | 68 | | } |
| | | 69 | | # Get the request header value from the request |
| | 0 | 70 | | $value = $Context.Request.Headers[$Name] |
| | 0 | 71 | | if ($AsInt) { |
| | 0 | 72 | | return [int]$value |
| | | 73 | | } |
| | 0 | 74 | | if ($AsBool) { |
| | 0 | 75 | | return [bool]$value |
| | | 76 | | } |
| | 0 | 77 | | if ($AsDouble) { |
| | 0 | 78 | | return [double]$value |
| | | 79 | | } |
| | 0 | 80 | | if ($AsString) { |
| | 0 | 81 | | return $value.ToString() |
| | | 82 | | } |
| | | 83 | | # Default: return as-is (string or array of strings) |
| | 0 | 84 | | return $value |
| | | 85 | | } else { |
| | | 86 | | # Outside of route context |
| | 0 | 87 | | Write-KrOutsideRouteWarning |
| | | 88 | | } |
| | | 89 | | } |