# ==============================================================
#  import-shots.ps1
#
#  Place this script inside a folder whose SIBLINGS are the
#  render directories you want to scan, e.g.:
#
#   /renders/
#     import-shots.ps1          <- this file
#     UNG_103_010_010_BG01_TT_v001/
#     UNG_103_010_010_FG01_TT_v001/
#     UNG_103_020_010_BG01_TT_v001/
#
#  Folder naming convention expected:
#    {showId}_{episode}_{scene}_{shot}_{...anything...}
#
#  The first four underscore-delimited segments form the shot
#  code, e.g. UNG_103_010_010.
# ==============================================================

# -- CONFIG ----------------------------------------------------
$BaseUrl   = "https://review.twotalesvfx.com"
$ApiKey    = "am3O0PWUtqMJkAqsZ+bO7lho4cxQItxgukF6FHteAx4="
$ProjectId = "cmp6l5mzq0001ua0gz07bk72f"
# --------------------------------------------------------------

$ErrorActionPreference = "Stop"

$headers = @{
    "Authorization" = "Bearer $ApiKey"
    "Content-Type"  = "application/json"
}

# -- 1. Fetch existing shots for the project -------------------
Write-Host ""
Write-Host "Fetching existing shots for project $ProjectId ..." -ForegroundColor Cyan

$existingUrl = "$BaseUrl/api/ext/shots?projectId=$ProjectId"
try {
    $response      = Invoke-RestMethod -Uri $existingUrl -Method GET -Headers $headers
    $existingCodes = [System.Collections.Generic.HashSet[string]]($response.shots | ForEach-Object { $_.shotCode })
    Write-Host "  Found $($existingCodes.Count) existing shot(s)." -ForegroundColor DarkGray
} catch {
    Write-Error "Failed to fetch existing shots: $_"
    exit 1
}

# -- 2. Scan sibling directories for EXR files ----------------
$scriptDir  = Split-Path -Parent $MyInvocation.MyCommand.Path
$parentDir  = Split-Path -Parent $scriptDir
$candidates = Get-ChildItem -Path $parentDir -Directory

Write-Host ""
Write-Host "Scanning '$parentDir' for directories containing EXR files ..." -ForegroundColor Cyan

$shotMap = [ordered]@{}

foreach ($dir in $candidates) {
    if ($dir.FullName -eq $scriptDir) { continue }

    $hasExr = Get-ChildItem -Path $dir.FullName -Filter "*.exr" -Recurse -ErrorAction SilentlyContinue |
              Select-Object -First 1

    if (-not $hasExr) { continue }

    $parts = $dir.Name -split "_"
    if ($parts.Count -lt 4) {
        Write-Warning "  Skipping '$($dir.Name)' - fewer than 4 underscore-delimited segments."
        continue
    }

    $shotCode = "$($parts[0])_$($parts[1])_$($parts[2])_$($parts[3])"
    $episode  = $parts[1]
    $scene    = $parts[2]

    if (-not $shotMap.Contains($shotCode)) {
        $shotMap[$shotCode] = @{ Episode = $episode; Scene = $scene }
        Write-Host "  Found EXR dir: $($dir.Name)  ->  shot code: $shotCode" -ForegroundColor DarkGray
    }
}

if ($shotMap.Count -eq 0) {
    Write-Host ""
    Write-Host "No directories containing EXR files found. Nothing to import." -ForegroundColor Yellow
    exit 0
}

Write-Host ""
Write-Host "Unique shot codes to process: $($shotMap.Count)" -ForegroundColor Cyan

# -- 3. Create shots, skipping existing ones ------------------
$created = 0
$skipped = 0
$failed  = 0

foreach ($shotCode in $shotMap.Keys) {
    $info = $shotMap[$shotCode]

    if ($existingCodes.Contains($shotCode)) {
        Write-Host "  [SKIP]    $shotCode  (already exists)" -ForegroundColor DarkGray
        $skipped++
        continue
    }

    $body = @{
        projectId = $ProjectId
        scene     = $info.Scene
        episode   = $info.Episode
        shotCode  = $shotCode
    } | ConvertTo-Json

    try {
        $result = Invoke-RestMethod -Uri "$BaseUrl/api/ext/shots" -Method POST -Headers $headers -Body $body
        Write-Host "  [CREATED] $shotCode  (id: $($result.shot.id))" -ForegroundColor Green
        $created++
    } catch {
        $statusCode = $null
        if ($_.Exception.Response) { $statusCode = [int]$_.Exception.Response.StatusCode }
        if ($statusCode -eq 409) {
            Write-Host "  [SKIP]    $shotCode  (conflict - already exists)" -ForegroundColor DarkGray
            $skipped++
        } else {
            Write-Host "  [ERROR]   $shotCode  - $($_.Exception.Message)" -ForegroundColor Red
            $failed++
        }
    }
}

# -- 4. Summary -----------------------------------------------
Write-Host ""
Write-Host "Done." -ForegroundColor Cyan
Write-Host "  Created : $created" -ForegroundColor Green
Write-Host "  Skipped : $skipped" -ForegroundColor DarkGray
if ($failed -gt 0) {
    Write-Host "  Failed  : $failed" -ForegroundColor Red
}