Tools/Session.ps1
|
function New-PodeSession { param ( [scriptblock] $ScriptBlock, [int] $Port = 0, [string] $IP = $null, [int] $Threads = 1, [int] $Interval = 0, [string] $ServerRoot, [ValidateSet('HTTP', 'HTTPS', 'SCRIPT', 'SERVICE', 'SMTP', 'TCP')] [string] $ServerType, [string] $Name = $null, [switch] $DisableLogging, [switch] $FileMonitor ) # set a random server name if one not supplied if (Test-Empty $Name) { $Name = Get-RandomName } # ensure threads are always >0 if ($Threads -le 0) { $Threads = 1 } # basic session object $session = New-Object -TypeName psobject | Add-Member -MemberType NoteProperty -Name Threads -Value $Threads -PassThru | Add-Member -MemberType NoteProperty -Name Timers -Value @{} -PassThru | Add-Member -MemberType NoteProperty -Name Schedules -Value @{} -PassThru | Add-Member -MemberType NoteProperty -Name RunspacePools -Value $null -PassThru | Add-Member -MemberType NoteProperty -Name Runspaces -Value $null -PassThru | Add-Member -MemberType NoteProperty -Name Tokens -Value @{} -PassThru | Add-Member -MemberType NoteProperty -Name DisableLogging -Value $DisableLogging -PassThru | Add-Member -MemberType NoteProperty -Name Loggers -Value @{} -PassThru | Add-Member -MemberType NoteProperty -Name RequestsToLog -Value $null -PassThru | Add-Member -MemberType NoteProperty -Name Lockable -Value $null -PassThru | Add-Member -MemberType NoteProperty -Name Server -Value @{} -PassThru # set the server type, name, logic and root $session.Server.Name = $Name $session.Server.Type = $ServerType $session.Server.Root = $ServerRoot $session.Server.Logic = $ScriptBlock $session.Server.Interval = $Interval $session.Server.FileMonitor = $FileMonitor # check if there is any global configuration $session.Server.Configuration = @{} $configPath = (Join-ServerRoot -Folder '.' -FilePath 'pode.json' -Root $ServerRoot) if (Test-PodePath -Path $configPath -NoStatus) { $session.Server.Configuration = (Get-Content $configPath -Raw | ConvertFrom-Json) } # set the IP address details $session.Server.IP = @{ 'Address' = $null; 'Port' = $Port; 'Name' = 'localhost'; 'Ssl' = ($ServerType -ieq 'https'); 'Certificate' = @{ 'Name' = $null; }; } # shared state between runspaces $session.Server.State = @{} # session engine for rendering views $session.Server.ViewEngine = @{ 'Engine' = 'html'; 'Extension' = 'html'; 'Script' = $null; } # routes for pages and api $session.Server.Routes = @{ 'delete' = @{}; 'get' = @{}; 'head' = @{}; 'merge' = @{}; 'options' = @{}; 'patch' = @{}; 'post' = @{}; 'put' = @{}; 'trace' = @{}; 'static' = @{}; '*' = @{}; } # handlers for tcp $session.Server.Handlers = @{ 'tcp' = $null; 'smtp' = $null; 'service' = $null; } # setup basic access placeholders $session.Server.Access = @{ 'Allow' = @{}; 'Deny' = @{}; } # setup basic limit rules $session.Server.Limits = @{ 'Rules' = @{}; 'Active' = @{}; } # cookies and session logic $session.Server.Cookies = @{ 'Session' = @{}; } # authnetication methods $session.Server.Authentications = @{} # create new cancellation tokens $session.Tokens = @{ 'Cancellation' = New-Object System.Threading.CancellationTokenSource; 'Restart' = New-Object System.Threading.CancellationTokenSource; } # requests that should be logged $session.RequestsToLog = New-Object System.Collections.ArrayList # middleware that needs to run $session.Server.Middleware = @() # endware that needs to run $session.Server.Endware = @() # runspace pools $session.RunspacePools = @{ 'Main' = $null; 'Schedules' = $null; } # session state $session.Lockable = [hashtable]::Synchronized(@{}) $state = [initialsessionstate]::CreateDefault() $state.ImportPSModule((Get-Module -Name Pode).Path) $_session = New-PodeStateSession $session $variables = @( (New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList 'PodeSession', $_session, $null), (New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList 'Console', $Host, $null) ) $variables | ForEach-Object { $state.Variables.Add($_) } # setup runspaces $session.Runspaces = @() # setup main runspace pool $threadsCounts = @{ 'Default' = 1; 'Timer' = 1; 'Log' = 1; 'Schedule' = 1; 'Misc' = 1; } $totalThreadCount = ($threadsCounts.Values | Measure-Object -Sum).Sum + $Threads $session.RunspacePools.Main = [runspacefactory]::CreateRunspacePool(1, $totalThreadCount, $state, $Host) $session.RunspacePools.Main.Open() # setup schedule runspace pool $session.RunspacePools.Schedules = [runspacefactory]::CreateRunspacePool(1, 2, $state, $Host) $session.RunspacePools.Schedules.Open() # return the new session return $session } function New-PodeStateSession { param ( [Parameter(Mandatory=$true)] [ValidateNotNull()] $Session ) return (New-Object -TypeName psobject | Add-Member -MemberType NoteProperty -Name Threads -Value $Session.Threads -PassThru | Add-Member -MemberType NoteProperty -Name Timers -Value $Session.Timers -PassThru | Add-Member -MemberType NoteProperty -Name Schedules -Value $Session.Schedules -PassThru | Add-Member -MemberType NoteProperty -Name RunspacePools -Value $Session.RunspacePools -PassThru | Add-Member -MemberType NoteProperty -Name Tokens -Value $Session.Tokens -PassThru | Add-Member -MemberType NoteProperty -Name DisableLogging -Value $Session.DisableLogging -PassThru | Add-Member -MemberType NoteProperty -Name Loggers -Value $Session.Loggers -PassThru | Add-Member -MemberType NoteProperty -Name RequestsToLog -Value $Session.RequestsToLog -PassThru | Add-Member -MemberType NoteProperty -Name Lockable -Value $Session.Lockable -PassThru | Add-Member -MemberType NoteProperty -Name Server -Value $Session.Server -PassThru) } function State { param ( [Parameter(Mandatory=$true)] [ValidateSet('set', 'get', 'remove')] [Alias('a')] [string] $Action, [Parameter(Mandatory=$true)] [ValidateNotNullOrEmpty()] [Alias('n')] [string] $Name, [Parameter()] [Alias('o')] [object] $Object ) try { if ($null -eq $PodeSession -or $null -eq $PodeSession.Server.State) { return $null } switch ($Action.ToLowerInvariant()) { 'set' { $PodeSession.Server.State[$Name] = $Object } 'get' { $Object = $PodeSession.Server.State[$Name] } 'remove' { $Object = $PodeSession.Server.State[$Name] $PodeSession.Server.State.Remove($Name) | Out-Null } } return $Object } catch { $Error[0] | Out-Default throw $_.Exception } } function Listen { param ( [Parameter(Mandatory=$true)] [ValidateNotNullOrEmpty()] [Alias('ipp')] [string] $IPPort, [Parameter()] [ValidateSet('HTTP', 'HTTPS', 'SMTP', 'TCP')] [Alias('t')] [string] $Type, [Parameter()] [Alias('cert')] [string] $Certificate = $null ) $hostRgx = '(?<host>(\[[a-z0-9\:]+\]|((\d+\.){3}\d+)|\:\:\d+|\*|all))' $portRgx = '(?<port>\d+)' $cmbdRgx = "$($hostRgx)\:$($portRgx)" # validate that we have a valid ip:port address if (!($IPPort -imatch "^$($cmbdRgx)$" -or $IPPort -imatch "^$($hostRgx)[\:]{0,1}" -or $IPPort -imatch "[\:]{0,1}$($portRgx)$")) { throw "Failed to parse '$($IPPort)' as a valid IP:Port address" } # grab the ip address $_host = $Matches['host'] if (Test-Empty $_host) { $_host = '*' } # ensure we have a valid ip address if (!(Test-IPAddress -IP $_host)) { throw "Invalid IP address has been supplied: $($IP)" } # grab the port $_port = $Matches['port'] if (Test-Empty $_port) { $_port = 0 } # ensure the port is valid if ($_port -lt 0) { throw "Port cannot be negative: $($_port)" } # set the ip for the session $PodeSession.Server.IP.Address = (Get-IPAddress $_host) if (!(Test-IPAddressLocal -IP $PodeSession.Server.IP.Address)) { $PodeSession.Server.IP.Name = $PodeSession.Server.IP.Address } # set the port for the session $PodeSession.Server.IP.Port = $_port # set the server type $PodeSession.Server.Type = $Type # if the server type is https, set cert details if ($Type -ieq 'https') { $PodeSession.Server.IP.Ssl = $true $PodeSession.Server.IP.Certificate.Name = $Certificate } } function Script { param ( [Parameter(Mandatory=$true)] [ValidateNotNullOrEmpty()] [string] $Path ) Import -Path $Path } function Import { param ( [Parameter(Mandatory=$true)] [ValidateNotNullOrEmpty()] [Alias('p')] [string] $Path ) # ensure the path exists, or it exists as a module $_path = Resolve-Path -Path $Path -ErrorAction Ignore if ([string]::IsNullOrWhiteSpace($_path)) { $_path = (Get-Module -Name $Path -ListAvailable | Select-Object -First 1).Path } # if it's still empty, error if ([string]::IsNullOrWhiteSpace($_path)) { throw "Failed to import module '$($Path)'" } # import the module into each runspace $PodeSession.RunspacePools.Values | ForEach-Object { $_.InitialSessionState.ImportPSModule($_path) } } |