# Define path for HTML report
$outputFile = "$env:USERPROFILE\Downloads\SharedMailboxMessageCopySettingsReport(Navitas).html"
# Define path for server address and credentials
$serverAddressFile = ".\DefaultServer.txt"
$credentialFile = ".\AdminCredential.xml"

# --- HTML Styling (CSS) ---
$cssStyle = @"
<style>
body { font-family: Segoe UI, Calibri, Arial, sans-serif; font-size: 10pt; }
h1 { color: #2E4053; font-size: 16pt; margin-bottom: 15px;} /* Added margin below H1 */
h2 { color: #17202A; font-size: 14pt; margin-top: 25px; border-bottom: 1px solid #AEB6BF; padding-bottom: 5px; }
table { border-collapse: collapse; width: 90%; margin-top: 10px; margin-bottom: 15px; border: 1px solid #AEB6BF; }
th { background-color: #4682B4; color: white; padding: 8px; text-align: left; border: 1px solid #AEB6BF; }
td { padding: 8px; text-align: left; border: 1px solid #D5DBDB; }
tr:nth-child(even) { background-color: #f2f2f2; }
tr:hover { background-color: #ddd; }
.true { color: green; font-weight: bold; }
.false { color: red; font-weight: bold; }
.description { margin-top: 10px; margin-bottom: 20px; padding: 10px; background-color: #f9f9f9; border-left: 4px solid #4682B4; } /* Added bottom margin */
.description p { margin-top: 0; margin-bottom: 5px; }
.description ul { margin-top: 5px; margin-bottom: 5px; padding-left: 20px; }
.description li { margin-bottom: 3px; }
.status-update { margin-top: 15px; font-style: italic; color: #555; }
.error { color: red; font-weight: bold; }
.warning { color: orange; font-weight: bold; }
.success { color: green; font-weight: bold; }
.timestamp { font-size: 8pt; color: #888; margin-top: 25px; }
code { font-family: Consolas, monospace; background-color: #eee; padding: 2px 4px; border-radius: 3px; }
</style>
"@

# --- Script Logic ---
$session = $null # Initialize session variable

# --- Initialize Report Body with Title and Descriptions ---
# Start with the main H1 title
$reportBody = "<h1>Shared Mailbox Message Copy Settings Report</h1>"

# Define and add the descriptions right after the title
$descriptionHtml = @"
<div class='description'>
<p><strong>Understanding Message Copy Settings:</strong></p>
<ul>
    <li><strong>SendOnBehalf Message Copy (<code>MessageCopyForSendOnBehalfEnabled</code>):</strong> When <code>True</code>, a copy of emails sent 'on behalf of' this shared mailbox is saved to the <i>shared mailbox's</i> Sent Items folder. When <code>False</code>, the copy is saved only to the sender's Sent Items folder.</li>
    <li><strong>SentAs Message Copy (<code>MessageCopyForSentAsEnabled</code>):</strong> When <code>True</code>, a copy of emails sent 'as' this shared mailbox is saved to the <i>shared mailbox's</i> Sent Items folder. When <code>False</code>, the copy is saved only to the sender's Sent Items folder.</li>
</ul>
<p><i>This script reports on these settings and allows enabling 'SentAs Message Copy' (<code>MessageCopyForSentAsEnabled = $true</code>) in bulk.</i></p>
</div>
"@
$reportBody += $descriptionHtml

try {
    # --- Input Validation ---
    if (-not (Test-Path $serverAddressFile)) {
        throw "Server address file not found: $serverAddressFile"
    }
    if (-not (Test-Path $credentialFile)) {
        throw "Credential file not found: $credentialFile"
    }

    # Import admin credentials
    $serverAddress = Get-Content -Path $serverAddressFile
    $credential = Import-Clixml -Path $credentialFile

    # Establish session with Exchange Server
    Write-Host "Connecting to Exchange Server: $serverAddress..."
    $session = New-PSSession -ConfigurationName Microsoft.Exchange -ConnectionUri "http://$serverAddress/PowerShell/" -Authentication Kerberos -Credential $credential -ErrorAction Stop
    Import-PSSession $session -DisableNameChecking -ErrorAction Stop
    Write-Host "Successfully connected to Exchange." -ForegroundColor Green

    # --- Data Retrieval ---
    Write-Host "Retrieving shared mailboxes..."
    $mailboxes = Get-Mailbox -RecipientTypeDetails SharedMailbox -ResultSize Unlimited | Select-Object DisplayName, PrimarySmtpAddress, MessageCopyForSendOnBehalfEnabled, MessageCopyForSentAsEnabled -ErrorAction Stop
    Write-Host "Found $($mailboxes.Count) shared mailboxes."

    # --- Generate "BEFORE" Report Section ---
    # This will append to the existing $reportBody (which already has Title + Description)
    $reportBody += "<h2>Shared Mailbox Message Copy Settings BEFORE Update</h2>"
    if ($mailboxes.Count -gt 0) {
        $formattedMailboxesBefore = $mailboxes | Select-Object DisplayName, PrimarySmtpAddress, @{Name='SendOnBehalfEnabled';Expression={$_.MessageCopyForSendOnBehalfEnabled}}, @{Name='SentAsEnabled';Expression={$_.MessageCopyForSentAsEnabled}}
        $htmlFragmentBefore = $formattedMailboxesBefore | ConvertTo-Html -Fragment -Property DisplayName, PrimarySmtpAddress, @{Label='SendOnBehalf Message Copy'; Expression = {"<span class='$($_.SendOnBehalfEnabled)'>$($_.SendOnBehalfEnabled)</span>"}}, @{Label='SentAs Message Copy'; Expression = {"<span class='$($_.SentAsEnabled)'>$($_.SentAsEnabled)</span>"}}
        $reportBody += [System.Net.WebUtility]::HtmlDecode($htmlFragmentBefore)
    } else {
        $reportBody += "<p>No shared mailboxes found.</p>"
    }

    # --- Filter Mailboxes for Update ---
    $mailboxesToUpdate = $mailboxes | Where-Object { $null -ne $_.MessageCopyForSentAsEnabled -and [bool]($_.MessageCopyForSentAsEnabled) -eq $false }

    # --- Update Logic ---
    if ($mailboxesToUpdate.Count -gt 0) {
        Write-Host "$($mailboxesToUpdate.Count) mailboxes identified with SentAs Message Copy disabled:" -ForegroundColor Yellow
        $mailboxesToUpdate | ForEach-Object { Write-Host " - $($_.DisplayName) ($($_.PrimarySmtpAddress))" }

        $response = Read-Host "Do you want to ENABLE SentAs Message Copy for these $($mailboxesToUpdate.Count) shared mailboxes? (Y/N)"
        if ($response -eq 'Y') {
            Write-Host "Starting update process..."
            $updateLog = foreach ($mb in $mailboxesToUpdate) {
                $updateStatus = "OK"
                $updateMessage = "Successfully enabled SentAs Message Copy for $($mb.DisplayName) ($($mb.PrimarySmtpAddress))."
                try {
                    Set-Mailbox -Identity $mb.PrimarySmtpAddress -MessageCopyForSentAsEnabled $true -ErrorAction Stop
                    Write-Host $updateMessage -ForegroundColor Green
                } catch {
                    $updateStatus = "Error"
                    $updateMessage = "Failed to enable SentAs Message Copy for $($mb.DisplayName) ($($mb.PrimarySmtpAddress)). Error: $($_.Exception.Message)"
                    Write-Host $updateMessage -ForegroundColor Red
                }
                [PSCustomObject]@{
                    Mailbox = "$($mb.DisplayName) ($($mb.PrimarySmtpAddress))"
                    Status = $updateStatus
                    Details = $updateMessage
                }
            }

            $reportBody += "<h2>Update Results</h2>"
            $reportBody += $updateLog | ConvertTo-Html -Fragment -Property Mailbox, Status, Details

            Write-Host "Waiting 5 seconds for changes to apply..."
            Start-Sleep -Seconds 5

            # --- Generate "AFTER" Report Section ---
            Write-Host "Retrieving updated shared mailbox settings..."
            $updatedMailboxes = Get-Mailbox -RecipientTypeDetails SharedMailbox -ResultSize Unlimited | Select-Object DisplayName, PrimarySmtpAddress, MessageCopyForSendOnBehalfEnabled, MessageCopyForSentAsEnabled -ErrorAction Stop

            $reportBody += "<h2>Shared Mailbox Message Copy Settings AFTER Update</h2>"
             if ($updatedMailboxes.Count -gt 0) {
                 $formattedMailboxesAfter = $updatedMailboxes | Select-Object DisplayName, PrimarySmtpAddress, @{Name='SendOnBehalfEnabled';Expression={$_.MessageCopyForSendOnBehalfEnabled}}, @{Name='SentAsEnabled';Expression={$_.MessageCopyForSentAsEnabled}}
                 $htmlFragmentAfter = $formattedMailboxesAfter | ConvertTo-Html -Fragment -Property DisplayName, PrimarySmtpAddress, @{Label='SendOnBehalf Message Copy'; Expression = {"<span class='$($_.SendOnBehalfEnabled)'>$($_.SendOnBehalfEnabled)</span>"}}, @{Label='SentAs Message Copy'; Expression = {"<span class='$($_.SentAsEnabled)'>$($_.SentAsEnabled)</span>"}}
                 $reportBody += [System.Net.WebUtility]::HtmlDecode($htmlFragmentAfter)
            } else {
                $reportBody += "<p>Could not retrieve updated mailbox information.</p>"
            }

            Write-Host "Bulk update process completed." -ForegroundColor Green
            $reportBody += "<p class='status-update success'>Bulk update process completed.</p>"

        } else {
            Write-Host "Bulk update canceled by user. No changes were made." -ForegroundColor Yellow
            $reportBody += "<p class='status-update warning'>Bulk update canceled by user. No changes were made.</p>"
        }
    } else {
        Write-Host "No shared mailboxes found requiring an update (SentAs Message Copy is already enabled or mailbox property is null)." -ForegroundColor Green
        $reportBody += "<p class='status-update'>No shared mailboxes required an update for the 'SentAs Message Copy' setting.</p>"
    }

} catch {
    Write-Host "An error occurred: $($_.Exception.Message)" -ForegroundColor Red
    # Append error details to the report body (which already has title/description)
    $reportBody += "<h2 class='error'>Script Execution Error</h2>"
    $reportBody += "<p class='error'>An error occurred while processing shared mailbox settings.</p>"
    $reportBody += "<p class='error'><strong>Error Details:</strong><br/>$([System.Net.WebUtility]::HtmlEncode($_.Exception.Message))</p>"
    if ($_.Exception.StackTrace) {
         $reportBody += "<p class='error'><strong>Stack Trace:</strong><br/>$([System.Net.WebUtility]::HtmlEncode($_.Exception.StackTrace) -replace "`n", "<br/>")</p>"
     }
     if ($_.InvocationInfo) {
         $reportBody += "<p class='error'><strong>Script Line:</strong> $($_.InvocationInfo.ScriptLineNumber)</p>"
     }
} finally {
    # --- Generate Final HTML Report ---
    $reportFooter = "<p class='timestamp'>Report generated on: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss') by $($env:USERNAME)</p>"

    # Assemble the full HTML page
    # Use -Body for the content we built. No -PreContent needed now.
    ConvertTo-Html -Head $cssStyle `
                   -Title "Shared Mailbox Message Copy Settings Report" `
                   -Body $reportBody `
                   -PostContent $reportFooter | Out-File $outputFile -Encoding UTF8 -ErrorAction SilentlyContinue

    # Check if the file was actually created before trying to open it
    if (Test-Path $outputFile) {
        Write-Host "HTML Report saved to: $outputFile"
        Invoke-Item -Path $outputFile
    } else {
         Write-Host "Failed to create the report file at: $outputFile" -ForegroundColor Red
    }

    # Clean up Exchange session
    if ($session -ne $null) {
        Write-Host "Removing Exchange PSSession..."
        Remove-PSSession $session
        Write-Host "Session removed."
    }
}