# --- Author: zetod1ce (github.com/ztd38f) --- #
# --- TG-Sort: Fixes file timestamps from Telegram exports (JSON & HTML) --- #

param(
    [string]$ExportFolder = "."
)

# -- ANSI Colors -- #
$ESC     = [char]27
$C_RESET = "$ESC[0m"
$C_BOLD  = "$ESC[1m"
$C_CYAN  = "$ESC[96m"
$C_GREEN = "$ESC[92m"
$C_YEL   = "$ESC[93m"
$C_RED   = "$ESC[91m"
$C_GRAY  = "$ESC[90m"
$C_MAG   = "$ESC[95m"
$C_WHITE = "$ESC[97m"

# ─────────────────────────────────────────
#  UI PRIMITIVES  (inline, no external deps)
# ─────────────────────────────────────────

function Write-Status([string]$msg, [string]$type = "info") {
    $icon = switch ($type) {
        "info"    { "${C_CYAN}[i]${C_RESET}" }
        "ok"      { "${C_GREEN}[✓]${C_RESET}" }
        "warn"    { "${C_YEL}[!]${C_RESET}" }
        "error"   { "${C_RED}[✗]${C_RESET}" }
        "loading" { "${C_MAG}[…]${C_RESET}" }
        default   { "   " }
    }
    Write-Host "   $icon $msg"
}

function PS.UI([string]$title) {
    Clear-Host
    Write-Host ""
    Write-Host "   ${C_BOLD}${C_CYAN}$title${C_RESET}"
    Write-Host "   ${C_GRAY}$('─' * ([Math]::Max($title.Length + 2, 40)))${C_RESET}"
    Write-Host ""
}

function Ask-Menu {
    param(
        [string]   $Title,
        [string[]] $Items,
        [string]   $Color = "Cyan"
    )

    $colorCode = switch ($Color) {
        "Cyan"    { $C_CYAN  }
        "Green"   { $C_GREEN }
        "Yellow"  { $C_YEL  }
        "Magenta" { $C_MAG  }
        default   { $C_CYAN  }
    }

    $idx   = 0
    $count = $Items.Count
    $first = $true
    # linesDrawn = title + blank line + N items
    $linesDrawn = 2 + $count

    [Console]::CursorVisible = $false

    try {
        while ($true) {
            # On redraw: move cursor up and erase each previous line
            if (-not $first) {
                for ($l = 0; $l -lt $linesDrawn; $l++) {
                    Write-Host -NoNewline "$ESC[1A$ESC[2K"
                }
            }
            $first = $false

            # Draw menu
            Write-Host "   ${C_BOLD}${colorCode}$Title${C_RESET}"
            Write-Host ""
            for ($i = 0; $i -lt $count; $i++) {
                if ($i -eq $idx) {
                    Write-Host "   ${colorCode}●${C_RESET} $($Items[$i])"
                } else {
                    Write-Host "   ${C_GRAY}○${C_RESET} $($Items[$i])"
                }
            }

            $key = [Console]::ReadKey($true)

            switch ($key.Key) {
                "UpArrow"   { if ($idx -gt 0)          { $idx-- } else { $idx = $count - 1 } }
                "DownArrow" { if ($idx -lt $count - 1) { $idx++ } else { $idx = 0 } }
                "Enter"     { [Console]::CursorVisible = $true; return $idx }
                "Escape"    { [Console]::CursorVisible = $true; return $null }
                "Q"         { [Console]::CursorVisible = $true; return $null }
            }
        }
    } finally {
        [Console]::CursorVisible = $true
    }
}

# -- Draw header -- #
PS.UI "TG-Sort [github.com/ztd38f]"

# ─────────────────────────────────────────
#  DATE PATTERN DETECTION
#  Extracts a DateTime from a filename
#  supporting all universal patterns (no prefix dependency)
# ─────────────────────────────────────────
function Get-DateFromFilename([string]$filename) {

    # Strip directory / extension for matching
    $name = [System.IO.Path]::GetFileNameWithoutExtension($filename)

    # --- Telegram photo: photo_N@DD-MM-YYYY_HH-MM-SS ---
    if ($name -match '@(\d{2})-(\d{2})-(\d{4})_(\d{2})-(\d{2})-(\d{2})$') {
        try {
            $dt = [DateTime]::new(
                [int]$Matches[3], [int]$Matches[2], [int]$Matches[1],
                [int]$Matches[4], [int]$Matches[5], [int]$Matches[6]
            )
            if ($dt.Year -ge 2000 -and $dt.Year -le 2035) { return $dt }
        } catch {}
    }

    # --- Epoch ms (13 digits) ---
    if ($name -match '(?<![0-9])(\d{13})(?![0-9])') {
        try {
            $ms    = [long]$Matches[1]
            $epoch = [DateTimeOffset]::FromUnixTimeMilliseconds($ms)
            return $epoch.LocalDateTime
        } catch {}
    }

    # --- Epoch s (10 digits) ---
    if ($name -match '(?<![0-9])(\d{10})(?![0-9])') {
        try {
            $s     = [long]$Matches[1]
            $epoch = [DateTimeOffset]::FromUnixTimeSeconds($s)
            return $epoch.LocalDateTime
        } catch {}
    }

    # Ordered from most-specific to least-specific to avoid false positives
    $patterns = @(
        # yyyyMMdd_HHmmss
        @{ re = '(\d{4})(\d{2})(\d{2})[_\- ](\d{2})(\d{2})(\d{2})'; g = 6 },
        # yyyy-MM-dd_HH-mm-ss  /  yyyy-MM-dd-HH-mm-ss  /  yyyy-MM-dd HH-mm-ss
        @{ re = '(\d{4})-(\d{2})-(\d{2})[_\- ](\d{2})-(\d{2})-(\d{2})'; g = 6 },
        # yyyyMMddHHmmss (14 digits)
        @{ re = '(\d{4})(\d{2})(\d{2})(\d{2})(\d{2})(\d{2})'; g = 6 },
        # yyMMddHHmmss (12 digits)
        @{ re = '(\d{2})(\d{2})(\d{2})(\d{2})(\d{2})(\d{2})'; g = 6; short = $true },
        # yyyyMMdd (8 digits, date only)
        @{ re = '(\d{4})(\d{2})(\d{2})'; g = 3 },
        # yyMMdd (6 digits, date only)
        @{ re = '(\d{2})(\d{2})(\d{2})'; g = 3; short = $true }
    )

    foreach ($p in $patterns) {
        if ($name -match $p.re) {
            try {
                $m = $Matches
                if ($p.g -eq 6) {
                    $yy = $m[1]; $mo = $m[2]; $dd = $m[3]
                    $hh = $m[4]; $mi = $m[5]; $ss = $m[6]
                    if ($p.short) { $yy = if ([int]$yy -ge 70) { "19$yy" } else { "20$yy" } }
                    $dt = [DateTime]::new([int]$yy, [int]$mo, [int]$dd, [int]$hh, [int]$mi, [int]$ss)
                } else {
                    $yy = $m[1]; $mo = $m[2]; $dd = $m[3]
                    if ($p.short) { $yy = if ([int]$yy -ge 70) { "19$yy" } else { "20$yy" } }
                    $dt = [DateTime]::new([int]$yy, [int]$mo, [int]$dd)
                }
                # Sanity: year between 2000-2035, month 1-12, day 1-31
                if ($dt.Year -ge 2000 -and $dt.Year -le 2035) { return $dt }
            } catch {}
        }
    }

    return $null
}

# ─────────────────────────────────────────
#  HTML TIMESTAMP PARSER
#  Parses Telegram's HTML export date format:
#  title="DD.MM.YYYY HH:MM:SS UTC+HH:00"
# ─────────────────────────────────────────
function Parse-TgHtmlDate([string]$title) {
    # e.g. "24.03.2024 19:59:13 UTC+02:00"
    if ($title -match '(\d{2})\.(\d{2})\.(\d{4})\s+(\d{2}):(\d{2}):(\d{2})\s+UTC([+-]\d{2}):(\d{2})') {
        try {
            $dd = [int]$Matches[1]; $mo = [int]$Matches[2]; $yy = [int]$Matches[3]
            $hh = [int]$Matches[4]; $mi = [int]$Matches[5]; $ss = [int]$Matches[6]
            $tzH = [int]$Matches[7]; $tzM = [int]$Matches[8]
            $dto = [DateTimeOffset]::new($yy, $mo, $dd, $hh, $mi, $ss,
                   [TimeSpan]::new($tzH, $tzM, 0))
            return $dto.LocalDateTime
        } catch {}
    }
    return $null
}

# ─────────────────────────────────────────
#  APPLY TIMESTAMP HELPER
# ─────────────────────────────────────────
function Set-FileTimestamp([System.IO.FileInfo]$item, [DateTime]$dt) {
    $item.LastWriteTime  = $dt
    $item.CreationTime   = $dt
    $item.LastAccessTime = $dt
}

# ─────────────────────────────────────────
#  CORE OPERATIONS
# ─────────────────────────────────────────

# ── Option 1: result.json ──────────────────
function Do-FixFromJson([string]$folder, [bool]$dryRun) {
    $resultJson = Join-Path $folder "result.json"
    if (!(Test-Path $resultJson)) {
        Write-Status "result.json not found in: $folder" "error"
        return
    }

    Write-Status "Reading result.json..." "loading"
    $data = Get-Content $resultJson -Raw -Encoding UTF8 | ConvertFrom-Json

    $updated = 0; $missing = 0; $skipped = 0

    foreach ($msg in $data.messages) {
        $paths = @()
        if ($msg.file      -and $msg.file      -notlike "*(File not*") { $paths += $msg.file }
        if ($msg.photo     -and $msg.photo     -notlike "*(File not*") { $paths += $msg.photo }
        if ($msg.video     -and $msg.video     -notlike "*(File not*") { $paths += $msg.video }
        if ($msg.audio     -and $msg.audio     -notlike "*(File not*") { $paths += $msg.audio }
        if ($msg.thumbnail -and $msg.thumbnail -notlike "*(File not*") { $paths += $msg.thumbnail }

        foreach ($rel in $paths) {
            $full = Join-Path $folder $rel
            if (!(Test-Path $full)) { $missing++; continue }
            try {
                $msgDate = [DateTime]::Parse($msg.date)
                if ($dryRun) {
                    Write-Host "   ${C_GRAY}[DRY]${C_RESET} $rel ${C_CYAN}→${C_RESET} $msgDate"
                } else {
                    Set-FileTimestamp (Get-Item $full) $msgDate
                    Write-Host "   ${C_GREEN}[✓]${C_RESET} $rel ${C_CYAN}→${C_RESET} $msgDate"
                }
                $updated++
            } catch {
                Write-Status "Error: $rel" "error"; $skipped++
            }
        }
    }

    Write-Host ""
    Write-Status "Done. Updated: ${C_GREEN}$updated${C_RESET}  Missing: ${C_YEL}$missing${C_RESET}  Errors: ${C_RED}$skipped${C_RESET}" "info"
}

# ── Option 2: messages*.html ───────────────
function Do-FixFromHtml([string]$folder, [bool]$dryRun) {
    $htmlFiles = Get-ChildItem -Path $folder -Filter "messages*.html" | Sort-Object Name
    if (!$htmlFiles) {
        Write-Status "No messages*.html files found in: $folder" "error"
        return
    }

    Write-Status "Parsing $($htmlFiles.Count) HTML file(s)..." "loading"

    # Build map: relative-path → DateTime
    # Each message block: <div class="message ... " id="message-...">
    #   contains:  <div class="pull_right date details" title="DD.MM.YYYY HH:MM:SS UTC+HH:00">
    #   and:       <a ... href="files/..." or href="photos/..." etc.
    $map = @{}  # key=relative path (normalized), value=DateTime

    $hrefPattern  = 'href="([^"]+)"'
    $titlePattern = 'class="pull_right date details" title="([^"]+)"'

    foreach ($hf in $htmlFiles) {
        $content = Get-Content $hf.FullName -Raw -Encoding UTF8

        # Split on message boundaries; keep delimiter by using lookahead trick
        $blocks = [regex]::Split($content, '(?=<div class="message (?:default|service)[^"]*" id="message-)')

        foreach ($block in $blocks) {
            # Extract timestamp from this block's date div
            $dtMatch = [regex]::Match($block, $titlePattern)
            if (!$dtMatch.Success) { continue }
            $dt = Parse-TgHtmlDate $dtMatch.Groups[1].Value
            if ($null -eq $dt) { continue }

            # Extract all hrefs in this block
            $hrefMatches = [regex]::Matches($block, $hrefPattern)
            foreach ($hm in $hrefMatches) {
                $rel = $hm.Groups[1].Value
                # Skip thumb images inline, CSS, JS, and anchor links
                if ($rel -match '^(css|js|#)' ) { continue }
                if ($rel -match '_thumb\.' )     { continue }
                # Normalize separators
                $rel = $rel -replace '/', '\'
                $key = $rel.ToLowerInvariant()
                # Keep earliest timestamp per file (first message wins)
                if (-not $map.ContainsKey($key)) {
                    $map[$key] = [PSCustomObject]@{ Rel = $rel; Dt = $dt }
                }
            }
        }
    }

    if ($map.Count -eq 0) {
        Write-Status "No media links found in HTML files." "warn"
        return
    }

    Write-Status "Found $($map.Count) unique media entries. Applying..." "info"
    Write-Host ""

    $updated = 0; $missing = 0; $skipped = 0

    foreach ($entry in $map.Values) {
        $full = Join-Path $folder $entry.Rel
        if (!(Test-Path $full)) { $missing++; continue }
        try {
            if ($dryRun) {
                Write-Host "   ${C_GRAY}[DRY]${C_RESET} $($entry.Rel) ${C_CYAN}→${C_RESET} $($entry.Dt)"
            } else {
                Set-FileTimestamp (Get-Item $full) $entry.Dt
                Write-Host "   ${C_GREEN}[✓]${C_RESET} $($entry.Rel) ${C_CYAN}→${C_RESET} $($entry.Dt)"
            }
            $updated++
        } catch {
            Write-Status "Error: $($entry.Rel)" "error"; $skipped++
        }
    }

    Write-Host ""
    Write-Status "Done. Updated: ${C_GREEN}$updated${C_RESET}  Missing: ${C_YEL}$missing${C_RESET}  Errors: ${C_RED}$skipped${C_RESET}" "info"
}

# ── Option 3: filenames ────────────────────
function Do-FixFromFilenames([string]$folder, [bool]$dryRun) {
    $files = Get-ChildItem -Path $folder -Recurse -File |
        Where-Object { $_.Name -ne "result.json" -and $_.Extension -ne ".html" }

    if (!$files) {
        Write-Status "No files found in: $folder" "warn"
        return
    }

    $updated = 0; $skipped = 0; $nodate = 0

    foreach ($f in $files) {
        $dt = Get-DateFromFilename $f.Name
        if ($null -eq $dt) {
            $nodate++
            Write-Host "   ${C_GRAY}[–]${C_RESET} $($f.Name)  ${C_GRAY}(no date pattern)${C_RESET}"
            continue
        }
        try {
            if ($dryRun) {
                Write-Host "   ${C_GRAY}[DRY]${C_RESET} $($f.Name) ${C_CYAN}→${C_RESET} $dt"
            } else {
                Set-FileTimestamp $f $dt
                Write-Host "   ${C_GREEN}[✓]${C_RESET} $($f.Name) ${C_CYAN}→${C_RESET} $dt"
            }
            $updated++
        } catch {
            Write-Status "Error: $($f.Name)" "error"; $skipped++
        }
    }

    Write-Host ""
    Write-Status "Done. Updated: ${C_GREEN}$updated${C_RESET}  No date: ${C_YEL}$nodate${C_RESET}  Errors: ${C_RED}$skipped${C_RESET}" "info"
}

# ── Option 5: preview patterns ─────────────
function Do-PreviewPatterns([string]$folder) {
    $files = Get-ChildItem -Path $folder -Recurse -File |
        Where-Object { $_.Name -ne "result.json" -and $_.Extension -ne ".html" }

    Write-Host ""
    Write-Host "   ${C_BOLD}${C_CYAN}Filename pattern detection preview:${C_RESET}"
    Write-Host "   ${C_GRAY}$('─' * 52)${C_RESET}"

    $hit = 0; $miss = 0
    foreach ($f in $files) {
        $dt = Get-DateFromFilename $f.Name
        if ($dt) {
            Write-Host "   ${C_GREEN}●${C_RESET} $($f.Name.PadRight(36)) ${C_CYAN}$dt${C_RESET}"
            $hit++
        } else {
            Write-Host "   ${C_GRAY}○${C_RESET} $($f.Name)"
            $miss++
        }
    }

    Write-Host "   ${C_GRAY}$('─' * 52)${C_RESET}"
    Write-Status "Detected: ${C_GREEN}$hit${C_RESET}  Undetected: ${C_YEL}$miss${C_RESET}" "info"
}


# ─────────────────────────────────────────
#  MAIN MENU  (1:1 style of CF-PagesRouter)
# ─────────────────────────────────────────

$menuItems = @(
    "Fix timestamps from result.json    (JSON export, uses message.date)",
    "Fix timestamps from messages*.html (HTML export, uses title attribute)",
    "Fix timestamps from filenames      (universal date patterns)",
    "Dry-run: result.json               (preview, no changes)",
    "Dry-run: messages*.html            (preview, no changes)",
    "Dry-run: filenames                 (preview, no changes)",
    "Preview pattern detection          (show what gets matched)"
)

Write-Host ""
$choice = Ask-Menu -Title "Select action:" -Items $menuItems -Color "Cyan"

if ($null -eq $choice) {
    Write-Status "Cancelled." "warn"
    exit
}

Write-Host ""

switch ($choice) {
    0 { Do-FixFromJson      $ExportFolder $false }
    1 { Do-FixFromHtml      $ExportFolder $false }
    2 { Do-FixFromFilenames $ExportFolder $false }
    3 { Do-FixFromJson      $ExportFolder $true  }
    4 { Do-FixFromHtml      $ExportFolder $true  }
    5 { Do-FixFromFilenames $ExportFolder $true  }
    6 { Do-PreviewPatterns  $ExportFolder        }
}

Write-Host ""