Commands/Sources/Set-OBSWindowSource.ps1
|
function Set-OBSWindowSource { <# .SYNOPSIS Adds or sets a window capture source .DESCRIPTION Adds or sets a windows capture source in OBS. This captures the contents of a window. .EXAMPLE Get-Process -id $PID | Set-OBSWindowCaptureSource -Name CurrentWindow #> [Alias('Add-OBSWindowSource','Set-OBSWindowCaptureSource','Add-OBSWindowCaptureSource')] [CmdletBinding()] param( # The monitor number. # This the number of the monitor you would like to capture. [Parameter(ValueFromPipelineByPropertyName)] [Alias('ItemValue','ItemName','WindowName','MainWindowTitle')] [string] $WindowTitle, # The number of the capture method. By default, automatic (0). [Parameter(ValueFromPipelineByPropertyName)] [ComponentModel.DefaultBindingProperty("method")] [int] $CaptureMethod, # The capture priority. [Parameter(ValueFromPipelineByPropertyName)] [ValidateSet('ExactMatch','SameType','SameExecutable')] [string] $CapturePriority, # If set, will capture the cursor. # This will be set by default. # If explicitly set to false, the cursor will not be captured. [Parameter(ValueFromPipelineByPropertyName)] [ComponentModel.DefaultBindingProperty("cursor")] [switch] $CaptureCursor, # If set, will capture the client area. # This will be set by default. [Parameter(ValueFromPipelineByPropertyName)] [ComponentModel.DefaultBindingProperty("client_area")] [switch] $ClientArea, # If set, will force SDR. [Parameter(ValueFromPipelineByPropertyName)] [ComponentModel.DefaultBindingProperty("force_sdr")] [switch] $ForceSDR, # The name of the scene. # If no scene name is provided, the current program scene will be used. [Parameter(ValueFromPipelineByPropertyName)] [Alias('SceneName')] [string] $Scene, # The name of the input. # If no name is provided, "Display $($Monitor + 1)" will be the input source name. [Parameter(ValueFromPipelineByPropertyName)] [Alias('InputName')] [string] $Name, # If set, will check if the source exists in the scene before creating it and removing any existing sources found. # If not set, you will get an error if a source with the same name exists. [Parameter(ValueFromPipelineByPropertyName)] [switch] $Force ) dynamicParam { $baseCommand = if (-not $script:AddOBSInput) { $script:AddOBSInput = $executionContext.SessionState.InvokeCommand.GetCommand('Add-OBSInput','Function') $script:AddOBSInput } else { $script:AddOBSInput } $IncludeParameter = @() $ExcludeParameter = 'inputKind','sceneName','inputName' $DynamicParameters = [Management.Automation.RuntimeDefinedParameterDictionary]::new() :nextInputParameter foreach ($paramName in ([Management.Automation.CommandMetaData]$baseCommand).Parameters.Keys) { if ($ExcludeParameter) { foreach ($exclude in $ExcludeParameter) { if ($paramName -like $exclude) { continue nextInputParameter} } } if ($IncludeParameter) { $shouldInclude = foreach ($include in $IncludeParameter) { if ($paramName -like $include) { $true;break} } if (-not $shouldInclude) { continue nextInputParameter } } $DynamicParameters.Add($paramName, [Management.Automation.RuntimeDefinedParameter]::new( $baseCommand.Parameters[$paramName].Name, $baseCommand.Parameters[$paramName].ParameterType, $baseCommand.Parameters[$paramName].Attributes )) } $DynamicParameters } process { $myParameters = [Ordered]@{} + $PSBoundParameters if (-not $myParameters["Scene"]) { $myParameters["Scene"] = Get-OBSCurrentProgramScene } if (-not $myParameters["WindowTitle"]) { while ($_ -is [Diagnostics.Process] -and -not $_.MainWindowTitle) { $_ = $_.Parent } if ($_.MainWindowTitle) { $WindowTitle = $myParameters["WindowTitle"] = "$($_.MainWindowTitle)" } } # Window capture is a bit of a tricky one. # In order to get the WindowTitle to match that OBS needs, we need to look thru the input properties list. # and for that, an input needs to exist. if (-not $myParameters["Name"]) { if ($myParameters["WindowTitle"]) { $Name = $myParameters["Name"] = "WindowCapture-" + $myParameters["WindowTitle"] } } $myParameterData = [Ordered]@{} foreach ($parameter in $MyInvocation.MyCommand.Parameters.Values) { $bindToPropertyName = $null foreach ($attribute in $parameter.Attributes) { if ($attribute -is [ComponentModel.DefaultBindingPropertyAttribute]) { $bindToPropertyName = $attribute.Name break } } if (-not $bindToPropertyName) { continue } if ($myParameters.Contains($parameter.Name)) { $myParameterData[$bindToPropertyName] = $myParameters[$parameter.Name] if ($myParameters[$parameter.Name] -is [switch]) { $myParameterData[$bindToPropertyName] = $parameter.Name -as [bool] } } } if ($null -ne $CaptureMethod) { $myParameterData["method"] = $CaptureMethod } if ($CapturePriority -eq 'ExactMatch') { $myParameterData["priority"] = 1 } elseif ($CapturePriority -eq 'SameType') { $myParameterData["priority"] = 0 } elseif ($CapturePriority -eq 'SameExecutable') { $myParameterData["priority"] = 2 } $addSplat = @{ sceneName = $myParameters["Scene"] inputName = $myParameters["Name"] inputKind = "window_capture" inputSettings = $myParameterData } # If -SceneItemEnabled was passed, if ($myParameters.Contains('SceneItemEnabled')) { # propagate it to Add-OBSInput. $addSplat.SceneItemEnabled = $myParameters['SceneItemEnabled'] -as [bool] } # Add the input. $outputAddedResult = Add-OBSInput @addSplat *>&1 $possibleWindows = Get-OBSInputPropertiesListPropertyItems -InputName $addSplat.inputName -PropertyName window foreach ($windowInfo in $possibleWindows) { if (-not $WindowTitle) { continue } if ( ($windowInfo.itemName -eq $WindowTitle) -or ($windowInfo.ItemValue -eq $WindowTitle) -or ($windowInfo.ItemName -replace '\[[^\[\]]+\]\:\s' -eq $WindowTitle) -or ($windowInfo.ItemValue -like "*$WindowTitle*") -or ($windowInfo.ItemName -like "*$WindowTitle*") ) { $myParameterData["window"] = $windowInfo.itemValue break } } # If -PassThru was passed if ($MyParameters["PassThru"]) { # pass it down to each command # Otherwise, remove SceneItemEnabled, InputKind, and SceneName $addSplat.PassThru = $true $addSplat.Remove('SceneItemEnabled') $addSplat.Remove('inputKind') $addSplat.Remove('sceneName') # and passthru Set-OBSInputSettings. Set-OBSInputSettings @addSplat return } if ($Force) { # If we do, remove the input Remove-OBSInput -InputName $addSplat.inputName # and re-add our result. $outputAddedResult = Add-OBSInput @addSplat *>&1 # If the output was still an error if ($outputAddedResult -is [Management.Automation.ErrorRecord]) { # use $psCmdlet.WriteError so that it shows the error correctly. $psCmdlet.WriteError($outputAddedResult) } # Otherwise, if we had a result elseif ($outputAddedResult) { # get the input from the scene. Get-OBSSceneItem -sceneName $myParameters["Scene"] | Where-Object SourceName -eq $name } } else { # Otherwise, get the input from the scene, $sceneItem = Get-OBSSceneItem -sceneName $myParameters["Scene"] | Where-Object SourceName -eq $myParameters["Name"] # update the input settings $sceneItem.Input.Settings = $addSplat.inputSettings $sceneItem # and return the scene item. } } } |