# --- TG-Sort v3 --- #

iex (irm -useb psui.pages.dev)

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

# --- DATE ENGINE --- #

function Parse-FilenameDate($name) {

    $b = [IO.Path]::GetFileNameWithoutExtension($name)

    $date = $null
    $time = $null

    if ($b -match '(?<!\d)(\d{4})(0[1-9]|1[0-2])(0[1-9]|[12]\d|3[01])[_\- ](\d{2})(\d{2})(\d{2})(?!\d)') {
        try { $date = Get-Date "$($Matches[1])-$($Matches[2])-$($Matches[3])" } catch {}
        try { $time = New-TimeSpan -Hours $Matches[4] -Minutes $Matches[5] -Seconds $Matches[6] } catch {}
    }
    elseif ($b -match '(?<!\d)(\d{4})-(0[1-9]|1[0-2])-(0[1-9]|[12]\d|3[01])[_\- ](\d{2})-(\d{2})-(\d{2})(?!\d)') {
        try { $date = Get-Date "$($Matches[1])-$($Matches[2])-$($Matches[3])" } catch {}
        try { $time = New-TimeSpan -Hours $Matches[4] -Minutes $Matches[5] -Seconds $Matches[6] } catch {}
    }
    elseif ($b -match '(?<!\d)(0[1-9]|[12]\d|3[01])-(0[1-9]|1[0-2])-(\d{4})[_\- ](\d{2})-(\d{2})-(\d{2})(?!\d)') {
        try { $date = Get-Date "$($Matches[3])-$($Matches[2])-$($Matches[1])" } catch {}
        try { $time = New-TimeSpan -Hours $Matches[4] -Minutes $Matches[5] -Seconds $Matches[6] } catch {}
    }
    elseif ($b -match '(?<!\d)(0[1-9]|[12]\d|3[01])-(0[1-9]|1[0-2])-(\d{4})(?!\d)') {
        try { $date = Get-Date "$($Matches[3])-$($Matches[2])-$($Matches[1])" } catch {}
    }
    elseif ($b -match '(?<!\d)(\d{4})(0[1-9]|1[0-2])(0[1-9]|[12]\d|3[01])(?!\d)') {
        try { $date = Get-Date "$($Matches[1])-$($Matches[2])-$($Matches[3])" } catch {}
    }

    return @{
        date    = $date
        time    = $time
        hasDate = ($date -ne $null)
        hasTime = ($time -ne $null)
    }
}

function Combine-Time($jsonDate, $parts) {

    $j = [datetime]$jsonDate

    if ($parts.hasDate -and $parts.hasTime) { return $parts.date.Add($parts.time) }
    if (!$parts.hasDate -and $parts.hasTime) { return (Get-Date $j.Date.Add($parts.time)) }
    if ($parts.hasDate -and !$parts.hasTime) { return (Get-Date $parts.date.Add($j.TimeOfDay)) }

    return $j
}

# --- ANALYZE --- #

function Analyze($folder) {

    $files = ls $folder -Recurse -File
    $total = $files.Count

    Write-Host ""
    status_info "Scanning $total files..."
    Write-Host ""

    $okD = 0; $okT = 0
    foreach ($f in $files) {
        $p = Parse-FilenameDate $f.Name
        if ($p.hasDate) { $okD++ }
        if ($p.hasTime) { $okT++ }
    }

    UI.Box "Analyzed $total files — date: $okD  •  time: $okT" "Cyan"
}

# --- FIX ENGINE --- #

function Fix-All($folder, $json) {

    $files = ls $folder -Recurse -File
    $total = $files.Count
    $i = 0; $fixed = 0; $skipped = 0; $errors = 0

    Write-Host ""
    status_info "Processing $total files..."
    Write-Host ""

    $ESC = [char]27

    # Оптимизация: создаем хэш-таблицу для быстрого поиска O(1)
    $jsonMap = @{}
    if ($json -and $json.messages) {
        foreach ($msg in $json.messages) {
            $key = $null
            if ($msg.file_name) {
                $key = [IO.Path]::GetFileName($msg.file_name)
            } elseif ($msg.file -and $msg.file -notmatch "not included") {
                $key = [IO.Path]::GetFileName($msg.file)
            } elseif ($msg.photo -and $msg.photo -notmatch "not included") {
                $key = [IO.Path]::GetFileName($msg.photo)
            }
            if ($key) { $jsonMap[$key] = $msg }
        }
    }

    foreach ($f in $files) {
        $i++
        $p = Parse-FilenameDate $f.Name
        
        # Быстрый поиск по имени файла O(1)
        $entry = $jsonMap[$f.Name]
        $dt = $null

        if ($entry) {
            $dt = Combine-Time $entry.date $p
        } elseif ($p.hasDate) {
            if ($p.hasTime) { $dt = $p.date.Add($p.time) }
            else { $dt = $p.date }
        }

        if ($dt) {
            try {
                $it = Get-Item $f.FullName
                $it.CreationTime  = $dt
                $it.LastWriteTime = $dt
                $fixed++
            } catch {
                $errors++
            }
        } else {
            $skipped++
        }

        # Обновляем одну строку прогресса
        [Console]::Write("`r   ${ESC}[96m[*]${ESC}[0m [$i/$total]  fixed: ${ESC}[92m$fixed${ESC}[0m  skip: ${ESC}[90m$skipped${ESC}[0m  err: ${ESC}[91m$errors${ESC}[0m   ")
    }

    Write-Host ""
    Write-Host ""
    $boxColor = if ($errors -gt 0) { "Yellow" } else { "Green" }
    UI.Box "Done: $fixed fixed  •  $skipped skipped  •  $errors errors" $boxColor
}

# --- INIT --- #

cls
UI.Box "TG-Sort" "Cyan"

status_info "Folder: $ExportFolder"
Write-Host ""

if (!(UI.Confirm "Use this folder?")) {
    Write-Host ""
    $ExportFolder = UI.Input "Enter folder path:" $ExportFolder
}

Write-Host ""

$jsonPath = Join-Path $ExportFolder "result.json"
if (Test-Path $jsonPath) {
    status_ok "result.json found"
    $json = cat $jsonPath -Raw | ConvertFrom-Json
} else {
    status_warn "result.json not found — filename timestamps only"
    $json = $null
}

# --- MENU --- #

Write-Host ""

$choice = UI.Menu -Title "TG-Sort" -Color "Cyan" -Items @(
    "Analyze files — preview to temp folder",
    "Fix all timestamps — auto-detect",
    "Exit"
)

switch ($choice) {

    0 { Analyze $ExportFolder }

    1 {
        Write-Host ""
        if (UI.Confirm "Run FIX ALL? This will modify file timestamps.") {
            Fix-All $ExportFolder $json
        }
    }

    { $_ -eq 2 -or $_ -eq $null } {
        Write-Host ""
        status_info "Bye."
        Write-Host ""
        exit
    }
}

Write-Host ""

UI.Pause