Functions/GenXdev.Webbrowser/Close-Webbrowser.ps1

################################################################################
<#
.SYNOPSIS
Closes one or more webbrowser instances selectively.
 
.DESCRIPTION
Provides granular control over closing web browser instances. Can target specific
browsers (Edge, Chrome, Firefox) or close all browsers. Supports closing both main
windows and background processes.
 
.PARAMETER Edge
Closes all Microsoft Edge browser instances.
 
.PARAMETER Chrome
Closes all Google Chrome browser instances.
 
.PARAMETER Chromium
Closes the default Chromium-based browser (Edge or Chrome).
 
.PARAMETER Firefox
Closes all Firefox browser instances.
 
.PARAMETER All
Closes all detected modern browser instances.
 
.PARAMETER IncludeBackgroundProcesses
Also closes background processes and tasks for the selected browsers.
 
.EXAMPLE
Close-Webbrowser -Chrome -Firefox -IncludeBackgroundProcesses
# Closes all Chrome and Firefox instances including background processes
 
.EXAMPLE
wbc -a -bg
# Closes all browser instances including background processes using aliases
#>

function Close-Webbrowser {

    [CmdletBinding(DefaultParameterSetName = 'Specific')]
    [Alias("wbc")]

    param(
        ########################################################################
        [Alias("e")]
        [parameter(
            Mandatory = $false,
            Position = 0,
            ParameterSetName = 'Specific',
            HelpMessage = "Closes Microsoft Edge browser instances"
        )]
        [switch] $Edge,
        ########################################################################
        [Alias("ch")]
        [parameter(
            Mandatory = $false,
            Position = 1,
            ParameterSetName = 'Specific',
            HelpMessage = "Closes Google Chrome browser instances"
        )]
        [switch] $Chrome,
        ########################################################################
        [Alias("c")]
        [parameter(
            Mandatory = $false,
            Position = 2,
            ParameterSetName = 'Specific',
            HelpMessage = "Closes default chromium-based browser"
        )]
        [switch] $Chromium,
        ########################################################################
        [Alias("ff")]
        [parameter(
            Mandatory = $false,
            Position = 3,
            ParameterSetName = 'Specific',
            HelpMessage = "Closes Firefox browser instances"
        )]
        [switch] $Firefox,
        ########################################################################
        [Alias("a")]
        [parameter(
            Mandatory = $false,
            Position = 0,
            ParameterSetName = 'All',
            HelpMessage = "Closes all registered modern browsers"
        )]
        [switch] $All,
        ########################################################################
        [Alias("bg", "Force")]
        [parameter(
            Mandatory = $false,
            Position = 4,
            HelpMessage = "Closes all instances including background tasks"
        )]
        [switch] $IncludeBackgroundProcesses
        ########################################################################
    )

    begin {

        # query system for installed browser information
        $installedBrowsers = Get-Webbrowser

        # determine system default browser
        $defaultBrowser = Get-DefaultWebbrowser

        Write-Verbose "Found $($installedBrowsers.Count) installed browsers"
        Write-Verbose "Default browser: $($defaultBrowser.Name)"
    }

    process {

        function Close-BrowserInstance {
            param (
                [object] $Browser
            )

            Write-Verbose "Attempting to close $($Browser.Name)"

            # extract process name without extension for matching
            $processName = [System.IO.Path]::GetFileNameWithoutExtension($Browser.Path)

            # find and process all matching browser instances
            Get-Process -Name $processName -ErrorAction SilentlyContinue |
            ForEach-Object {

                $currentProcess = $_

                # handle background processes based on user preference
                if ((-not $IncludeBackgroundProcesses) -and
                    ($currentProcess.MainWindowHandle -eq 0)) {

                    Write-Verbose "Skipping background process $($currentProcess.Id)"
                    return
                }
                elseif ($currentProcess.MainWindowHandle -ne 0) {

                    # attempt graceful window close for processes with UI
                    [GenXdev.Helpers.WindowObj]::GetMainWindow($currentProcess) |
                    ForEach-Object {

                        $startTime = [DateTime]::UtcNow
                        $window = $_

                        # try graceful close
                        $null = $window.Close()

                        # wait up to 4 seconds for process to exit
                        while (!$currentProcess.HasExited -and
                            ([datetime]::UtcNow - $startTime -lt
                            [TimeSpan]::FromSeconds(4))) {

                            Start-Sleep -Milliseconds 20
                        }

                        if ($currentProcess.HasExited) {
                            Set-Variable -Scope Global `
                                -Name "_LastClose$($Browser.Name)" `
                                -Value ([DateTime]::UtcNow.AddSeconds(-1))
                            return
                        }
                    }
                }

                # force terminate if process still running
                try {
                    $currentProcess.Kill()
                    Set-Variable -Scope Global -Name "_LastClose$($Browser.Name)" `
                        -Value ([DateTime]::UtcNow)
                }
                catch {
                    Write-Warning "Failed to kill $($Browser.Name) process: $_"
                }
            }
        }

        # close all browsers if requested
        if ($All) {
            Write-Verbose "Closing all browsers"
            $installedBrowsers | ForEach-Object { Close-BrowserInstance $_ }
            return
        }

        # handle default chromium browser closure
        if ($Chromium) {
            if ($defaultBrowser.Name -like "*Chrome*" -or
                $defaultBrowser.Name -like "*Edge*") {

                Close-BrowserInstance $defaultBrowser
                return
            }

            # fallback to first available chromium browser
            $installedBrowsers |
            Where-Object { $_.Name -like "*Edge*" -or $_.Name -like "*Chrome*" } |
            Select-Object -First 1 |
            ForEach-Object {
                Close-BrowserInstance $_
            }
            return
        }

        # handle specific browser closures
        if ($Chrome) {
            $installedBrowsers |
            Where-Object { $_.Name -like "*Chrome*" } |
            ForEach-Object { Close-BrowserInstance $_ }
        }

        if ($Edge) {
            $installedBrowsers |
            Where-Object { $_.Name -like "*Edge*" } |
            ForEach-Object { Close-BrowserInstance $_ }
        }

        if ($Firefox) {
            $installedBrowsers |
            Where-Object { $_.Name -like "*Firefox*" } |
            ForEach-Object { Close-BrowserInstance $_ }
        }

        # close default browser if no specific browser selected
        if (-not ($Chromium -or $Chrome -or $Edge -or $Firefox)) {
            Close-BrowserInstance $defaultBrowser
        }
    }

    end {
    }
}
################################################################################