#Requires -Version 7.0
#Requires -Modules Posh-SSH

function Wait-UntilTerminalIsReady {
    param (
        [Renci.SshNet.ShellStream]$ShellStream
    )
    $retry = 0
    $streamOut = $ShellStream.Read()

    # Wait until we can select terminal type and choose ossi
    while (-Not ($streamOut).Contains("Terminal Type") -and $retry -le 10) {
        Start-Sleep -s 1
        $streamOut = $ShellStream.Read()
        $retry++
    }

    if ($retry -gt 10) {
        throw "Timed out whie initializing waiting for terminal type prompt"
    }

    $retry = 0
    $ShellStream.WriteLine('ossi')
    
    # Read until stream is empty before starting to send commands
    while (($streamOut).length -ne 0 -and $retry -le 10) {
        Start-Sleep -s 1
        $streamOut = $ShellStream.Read()
    }

    if ($retry -gt 10) {
        throw "Timed out whie initializing OSSI terminal"
    }
}

function Write-CommandsToSshStream {
    param (
        [string[]]$Commands,
        [Renci.SshNet.ShellStream]$ShellStream
    )

    $retry = 0
    foreach ($Command in $Commands) {
        $ShellStream.WriteLine($Command)
    }
    $ShellStream.WriteLine('t')

    $streamOut = $ShellStream.Read()
    while (($streamOut).length -eq 0 -and $retry -le 10) {
        Start-Sleep -s 1
        $streamOut = $ShellStream.Read()
        $retry++
    }

    return $streamOut
}

function Get-AvayaSubEntities {
    param (
        $EntitiesId,
        [string]$EntitiesName,
        [string[]]$Commands,
        [Renci.SshNet.ShellStream]$ShellStream
    ) 
    $progressCount = 0
    foreach ($EntityId in $EntitiesId) {
        # Remove the trailing d in all the Avaya ids
        $EntityId = $EntityId.substring(1)

        foreach ($Command in $Commands) {
            Write-AyavaOutPutToFile "$Command $EntityId" $ShellStream
        }

        $progressCount++
        Write-Progress -activity "Getting $EntitiesName information..." -status "Fetched: $progressCount of $($EntitiesId.Count)" -percentComplete (($progressCount / $EntitiesId.Count) * 100)
    }
}

function Write-AyavaOutPutToFile {
    param (
        [string[]]$Commands,
        [Renci.SshNet.ShellStream]$ShellStream
    )
    $retry = 0
    $commandOutput = Write-CommandsToSshStream $Commands $ShellStream
    
    while (([string]::IsNullOrEmpty($commandOutput) -or ($commandOutput).Contains('Terminator received but no command active')) -and $retry -le 5) {
        Start-Sleep -s 1
        $commandOutput = Write-CommandsToSshStream $Commands $ShellStream
        $retry++
    }

    if ($retry -gt 5) {
        Write-Warning "Failed to get result for commands $Commands"
    }

    Add-Content -Path "output-avaya/avaya.txt" -Value $commandOutput
}

function Write-AyavaOutPutToFileAndGetEntities {
    param (
        [string[]]$Commands,
        [Renci.SshNet.ShellStream]$ShellStream
    )
    $retry = 0
    $commandOutput = Write-CommandsToSshStream $Commands $ShellStream
    
    while (([string]::IsNullOrEmpty($commandOutput) -or ($commandOutput).Contains('Terminator received but no command active')) -and $retry -le 5) {
        Start-Sleep -s 1
        $commandOutput = Write-CommandsToSshStream $Commands $ShellStream
        $retry++
    }

    if ($retry -gt 5) {
        Write-Warning "Failed to get result for commands $Commands"
    }

    Add-Content -Path "output-avaya/avaya.txt" -Value $commandOutput
    return $commandOutput.Split("`n") | Where-Object { $_.Trim("") -and $Commands -notcontains $_ -and $_ -ne 'n' -and $_ -ne 't' }
}


function Write-EntitiesProgressToHost {
    param (
        [Int]$PbxProgressCount
    )
    Write-Progress -activity "Gathering PBX information..." -status "Fetched: $PbxProgressCount of 42" -percentComplete (($PbxProgressCount / 42) * 100)
}

function Get-CdrInformation {
    param (
        [string]$ServerUrl,
        [string]$Date
    )
    $sftpSession = $null;

    try {
        $answer = Read-Host "Do you want to also export CDRs information? [y/N]? "
        if ($answer -eq 'y') { 
            New-Item -Name "avaya_cdr" -ItemType Directory -Force | Out-Null
            
            $credential = Get-Credential -Message 'Enter CDR username and password'
            $sftpSession = New-SFTPSession -ComputerName $ServerUrl -Credential $credential -AcceptKey
            Write-Output "Downloading CDRs..."
            Get-SFTPItem -SFTPSession $sftpSession -Path "/var/home/ftp/CDR/" -Destination "avaya_cdr/" -Force -SkipSymLink

            $ZipFileName = "avaya_" + $date + "_cdr.zip"

            Compress-Archive -Path avaya_cdr/* -DestinationPath $ZipFileName -Force 
            Remove-Item -Path avaya_cdr -Recurse -Force
        }
        else {
            Write-Output "Skipping CDRs export..."
        }
    }
    finally {
        if ($null -ne $sftpSession) {
            Remove-SftpSession -SFTPSession $sftpSession | Out-Null
        }
    }
}
$sshsession = $null;
$stream = $null;
try {
    $serverUrl = Read-Host 'Avaya FQDN or IP Address (avayacm.mycompany.com)'
    $credential = Get-Credential -Message 'Enter administrator username and password'

    $date = (Get-Date -Format "dd-MM-yyyy_HH-mm-ss").ToString()

    New-Item -Name "output-avaya" -ItemType Directory -Force | Out-Null
    New-Item -ItemType File -Name "output-avaya/avaya.txt" -Force | Out-Null

    Get-CdrInformation $serverUrl $date
    
    $sshsession = New-SSHSession -ComputerName $serverurl -Credential $credential -Port 5022 -AcceptKey
    $stream = New-SSHShellStream -SSHSession $sshsession

    Wait-UntilTerminalIsReady $stream

    $pbxProgressCount = 0

    Write-AyavaOutPutToFile 'cdisplay system-parameters cdr' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist hunt-group' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist pickup-group' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay alias station' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay tenant 1' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 1' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 2' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 3' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 4' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 5' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 6' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 7' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 8' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 9' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay coverage remote 10' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay feature-access-codes' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount
    
    Write-AyavaOutPutToFile 'clist coverage path' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay ip-network-map' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'cdisplay capacity' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist user-profiles' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist route-pattern' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist integrated-annc-boards' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount
    
    Write-AyavaOutPutToFile 'clist ars analysis' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist media-gateway' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist public-unknown-numbering' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist off-pbx-telephone station-mapping' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist intercom-group' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist abbreviated-dialing personal' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist station' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist trunk-group' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist vector' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist vdn' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist ip-network-region monitor' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Write-AyavaOutPutToFile 'clist announcement' $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $extensionIds = Write-AyavaOutPutToFileAndGetEntities @('clist station', 'f8005ff00') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $corIds = Write-AyavaOutPutToFileAndGetEntities @('clist station', 'f8001ff00') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $cosIds = Write-AyavaOutPutToFileAndGetEntities @('clist station', 'f8002ff00') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $trunkGroupIds = Write-AyavaOutPutToFileAndGetEntities @('clist trunk-group', 'f800bff00') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $vectorIds = Write-AyavaOutPutToFileAndGetEntities @('clist vector', 'f0001ff01') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $vdnIds = Write-AyavaOutPutToFileAndGetEntities @('clist vdn', 'f8005ff01') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $ipNetworkRegionMonitorIds = Write-AyavaOutPutToFileAndGetEntities @('clist ip-network-region monitor', 'f6c00ff00') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    $announcementIds = Write-AyavaOutPutToFileAndGetEntities @('clist announcement', 'f8005ff00') $stream
    $pbxProgressCount++
    Write-EntitiesProgressToHost $pbxProgressCount

    Get-AvayaSubEntities $extensionIds 'Extensions' @('cdisplay station', 'clist bridged-extensions', 'cdisplay button-labels') $stream
    Get-AvayaSubEntities $corIds 'CORs' 'cdisplay cor' $stream
    Get-AvayaSubEntities $cosIds 'COSs' 'cdisplay cos' $stream
    Get-AvayaSubEntities $trunkGroupIds 'Trunk Groups' @('cdisplay trunk-group', 'cdisplay inc-call-handling-trmt trunk-group') $stream
    Get-AvayaSubEntities $vectorIds 'Vectors' 'cdisplay vector' $stream
    Get-AvayaSubEntities $vdnIds 'VDNs' 'cdisplay vdn' $stream
    Get-AvayaSubEntities $ipNetworkRegionMonitorIds 'IP Network Regions' 'cstatus ip-network-region' $stream
    Get-AvayaSubEntities $announcementIds 'Announcements' 'cdisplay announcement' $stream

    Write-AyavaOutPutToFile "clogoff" $stream

    $ZipFileName = "avaya_" + $date + ".zip"

    Compress-Archive -Path output-avaya/* -DestinationPath $ZipFileName -Force 
    Remove-Item -Path output-avaya -Recurse -Force
    
    Write-Host "The script ran successfully" -ForegroundColor Green
    exit 0
}
catch { 
    Write-host -f red "Encountered Error:"$_.Exception.Message
    Write-Error "Avaya information export failed"
    exit 1
}
finally {
    if ($null -ne $sshsession) {
        Remove-SSHSession -SSHSession $sshsession | Out-Null
    }
}