param(
    [Parameter(Mandatory=$false)]
    [string]$ReferenceFilePath,

    [Parameter(Mandatory=$false)]
    [string]$TargetFolderPath
)

function Remove-SurroundingQuotes {
    param([string]$Text)
    if ([string]::IsNullOrWhiteSpace($Text)) { return $Text }

    $t = $Text.Trim() `
        -replace '^[\u201C\u201D]', '"' `
        -replace '[\u201C\u201D]$', '"' `
        -replace '^[\u2018\u2019]', "'" `
        -replace '[\u2018\u2019]$', "'"

    if (($t.StartsWith('"') -and $t.EndsWith('"')) -or
        ($t.StartsWith("'") -and $t.EndsWith("'"))) {
        $t = $t.Substring(1, $t.Length - 2)
    }
    return $t.Trim()
}

function Normalize-Input { param([string]$Text) Remove-SurroundingQuotes $Text }

if (-not $ReferenceFilePath) {
    $ReferenceFilePath = Read-Host 'Enter reference file path (quotes OK)'
}
if (-not $TargetFolderPath) {
    $TargetFolderPath = Read-Host 'Enter target folder path (quotes OK)'
}

$ReferenceFilePath = Normalize-Input $ReferenceFilePath
$TargetFolderPath  = Normalize-Input $TargetFolderPath

if (-not (Test-Path $ReferenceFilePath)) { throw "Reference file not found: $ReferenceFilePath" }
if (-not (Test-Path $TargetFolderPath))  { throw "Target folder not found: $TargetFolderPath" }

# ------------------ Parse mapping file ------------------

$rawLines = Get-Content -LiteralPath $ReferenceFilePath
$mapping  = New-Object System.Collections.Generic.List[object]

foreach ($line in $rawLines) {
    $line = $line.Trim()
    if (-not $line) { continue }

    $parts = $null
    if ($line -like '*---*') {
        $parts = $line -split '\s*---\s*', 2
    } else {
        $m = [regex]::Match($line, '^(.*\S)\s+(\S.*)$')
        if ($m.Success) { $parts = @($m.Groups[1].Value, $m.Groups[2].Value) }
    }

    if ($parts.Count -eq 2) {
        $replacement = (Remove-SurroundingQuotes $parts[0]).Trim()
        $search      = (Remove-SurroundingQuotes $parts[1]).Trim()

        if ($replacement -and $search) {
            $mapping.Add([pscustomobject]@{
                Search  = $search
                Replace = $replacement
            })
        }
    }
}

if ($mapping.Count -eq 0) {
    Write-Warning "No valid mappings found. Nothing to do."
    exit
}

# ------------------ Process only defined files ------------------

$targetNames = @(
    'wpsecure_new.htm','wpsecure_new.rtf','wpsecure_new.txt',
    'wpsecure_reply.htm','wpsecure_reply.rtf','wpsecure_reply.txt',
    'wpsecure_web.htm','wpsecure_web.txt'
)

$summary = New-Object System.Collections.Generic.List[object]

foreach ($name in $targetNames) {
    $path = Join-Path $TargetFolderPath $name

    if (-not (Test-Path $path)) {
        $summary.Add([pscustomobject]@{ File=$name; Status='Not found'; Mappings=0 })
        continue
    }

    try {
        $text = Get-Content -LiteralPath $path -Raw -ErrorAction Stop
    } catch {
        Write-Warning ("Failed to read {0}: {1}" -f $name, $_.Exception.Message)
        $summary.Add([pscustomobject]@{ File=$name; Status='Read error'; Mappings=0 })
        continue
    }

    if ($null -eq $text) { $text = '' }

    $isRtf = $name.ToLower().EndsWith('.rtf')
    $count = 0

    foreach ($pair in $mapping) {
        $search = $pair.Search
        $replacement = $pair.Replace

        if ($isRtf) {
            $replacement = $replacement.Replace('{','\{').Replace('}','\}')
        }

        if ($text.Contains($search)) {
            $text = $text.Replace($search, $replacement)
            $count++
        }
    }

    if ($count -gt 0) {
        try {
            Set-Content -LiteralPath $path -Value $text -NoNewline -Encoding UTF8
            $summary.Add([pscustomobject]@{ File=$name; Status='Updated'; Mappings=$count })
        } catch {
            Write-Warning ("Failed to write {0}: {1}" -f $name, $_.Exception.Message)
            $summary.Add([pscustomobject]@{ File=$name; Status='Write error'; Mappings=$count })
        }
    } else {
        $summary.Add([pscustomobject]@{ File=$name; Status='No changes'; Mappings=0 })
    }
}

$summary | Sort-Object File | Format-Table -AutoSize