Devolutions.CIEM

0.3.180

modules/Devolutions.CIEM.Graph/Public/Remove-CIEMGraphEdge.ps1

function Remove-CIEMGraphEdge {
    [CmdletBinding(SupportsShouldProcess, DefaultParameterSetName = 'ById')]
    param(
        [Parameter(Mandatory, ParameterSetName = 'ById')]
        [int]$Id,

        [Parameter(Mandatory, ParameterSetName = 'ByKind')]
        [string]$Kind,

        [Parameter(Mandatory, ParameterSetName = 'ByComputed')]
        [int]$Computed,

        [Parameter(Mandatory, ParameterSetName = 'All')]
        [switch]$All,

        [Parameter(Mandatory, ParameterSetName = 'InputObject', ValueFromPipeline)]
        [PSObject[]]$InputObject,

        [Parameter()]
        [Microsoft.Data.Sqlite.SqliteConnection]$Connection
    )

    process {
        $ErrorActionPreference = 'Stop'
        switch ($PSCmdlet.ParameterSetName) {
            'ById' {
                if ($PSCmdlet.ShouldProcess("Id $Id", 'Remove graph edge')) {
                    Write-CIEMLog -Message "DELETE graph_edges WHERE id=$Id (caller: $((Get-PSCallStack)[1].Command))" -Severity WARNING -Component 'Remove-GraphEdge'
                    $q = "DELETE FROM graph_edges WHERE id = @id"
                    $p = @{ id = $Id }
                    if ($Connection) { Invoke-PSUSQLiteQuery -Connection $Connection -Query $q -Parameters $p -AsNonQuery | Out-Null }
                    else { Invoke-CIEMQuery -Query $q -Parameters $p -AsNonQuery | Out-Null }
                }
            }
            'ByKind' {
                if ($PSCmdlet.ShouldProcess("Kind '$Kind'", 'Remove graph edges')) {
                    Write-CIEMLog -Message "DELETE graph_edges WHERE kind='$Kind' (caller: $((Get-PSCallStack)[1].Command))" -Severity WARNING -Component 'Remove-GraphEdge'
                    $q = "DELETE FROM graph_edges WHERE kind = @kind"
                    $p = @{ kind = $Kind }
                    if ($Connection) { Invoke-PSUSQLiteQuery -Connection $Connection -Query $q -Parameters $p -AsNonQuery | Out-Null }
                    else { Invoke-CIEMQuery -Query $q -Parameters $p -AsNonQuery | Out-Null }
                }
            }
            'ByComputed' {
                if ($PSCmdlet.ShouldProcess("Computed=$Computed", 'Remove graph edges')) {
                    Write-CIEMLog -Message "DELETE graph_edges WHERE computed=$Computed (caller: $((Get-PSCallStack)[1].Command))" -Severity WARNING -Component 'Remove-GraphEdge'
                    $q = "DELETE FROM graph_edges WHERE computed = @computed"
                    $p = @{ computed = $Computed }
                    if ($Connection) { Invoke-PSUSQLiteQuery -Connection $Connection -Query $q -Parameters $p -AsNonQuery | Out-Null }
                    else { Invoke-CIEMQuery -Query $q -Parameters $p -AsNonQuery | Out-Null }
                }
            }
            'All' {
                if ($PSCmdlet.ShouldProcess('all rows', 'Remove graph edges')) {
                    Write-CIEMLog -Message "DELETE graph_edges ALL ROWS (caller: $((Get-PSCallStack)[1].Command))" -Severity WARNING -Component 'Remove-GraphEdge'
                    $q = "DELETE FROM graph_edges"
                    if ($Connection) { Invoke-PSUSQLiteQuery -Connection $Connection -Query $q -AsNonQuery | Out-Null }
                    else { Invoke-CIEMQuery -Query $q -AsNonQuery | Out-Null }
                }
            }
            'InputObject' {
                foreach ($obj in $InputObject) {
                    if ($PSCmdlet.ShouldProcess("Id $($obj.Id)", 'Remove graph edge')) {
                        Write-CIEMLog -Message "DELETE graph_edges WHERE id=$($obj.Id) (caller: $((Get-PSCallStack)[1].Command))" -Severity WARNING -Component 'Remove-GraphEdge'
                        $q = "DELETE FROM graph_edges WHERE id = @id"
                        $p = @{ id = $obj.Id }
                        if ($Connection) { Invoke-PSUSQLiteQuery -Connection $Connection -Query $q -Parameters $p -AsNonQuery | Out-Null }
                        else { Invoke-CIEMQuery -Query $q -Parameters $p -AsNonQuery | Out-Null }
                    }
                }
            }
        }
    }
}