$PathList = @(
    'extensions/assets-manager'
    'extensions/cocostudio'
    'extensions/GUI'
    'extensions/Inspector'
    'extensions/JSONDefault'
    'extensions/Particle3D'
    'extensions/physics-nodes'
    'extensions/scripting'
    'core'
)


function Replace-GuardInclude {
    param(
        [Parameter(Mandatory = $true)]
        [string]$Path   # 要处理的根目录
    )

    $files = Get-ChildItem -Path $Path -Recurse -Include *.h, *.hpp

    foreach ($filepath in $files) {
        # echo "Processing file $filepath"

        $lines = Get-Content $filepath -Encoding UTF8
        $lines = [System.Collections.ArrayList]$lines

        # 正则：匹配文件开头的 include guard（#ifndef/#define）
        $patternStart = '(?ms)^\s*#\s*ifndef\s+([A-Za-z_][A-Za-z0-9_]*)\s*'
        $patternDefine = '\s*#\s*define\s+'
                    
        # 正则：匹配文件结尾的 #endif（支持注释）
        $patternEnd   = '(?ms)#\s*endif(?:\s*//.*)?\s*'

        $firstIf = $null
        $firstDefine = $null
        $lastEndif = $null

        $firstIfIdx = -1
        $firstDefineIdx = -1
        $lastEndifIdx = -1

        $lineIdx = 0

        $has_pargam_once = $false
        
        foreach($lineText in $lines) {
            if ($lineText -match '^\s*#\s*pragma\s+once\b') {
                Write-Host "Skipped (has pragma once): $filepath"
                $has_pargam_once = $true
                break
            }
            elseif($lineText -match $patternStart) {
                if(!$firstIf) {
                    $firstIf = $lineText
                    $firstIfIdx = $lineIdx
                }
            }
            elseif($lineText -match $patternDefine) {
                if($firstDefineIdx -eq -1) {
                    $firstDefineIdx = $lineIdx
                    $firstDefine = $lineText
                }
            }
            elseif($lineText -match $patternEnd) {
                $lastEndif = $lineText
                $lastEndifIdx = $lineIdx
            }
            ++$lineIdx
        }

        if ($has_pargam_once) {
            continue
        }

        if($firstIfIdx -ge $firstDefineIdx) {
            throw "$firstIfIdx >= $firstDefineIdx"
        }

        if ($firstDefineIdx -ge $lastEndifIdx) {
            throw "$firstDefineIdx -ge $lastEndifIdx"
        }

        # echo "$filepath include guard info: firstIfIdx=$firstIfIdx, firstDefineIdx=$firstDefineIdx, lastEndifIdx=$lastEndifIdx, firstIf=$firstIf, lastEndif=$lastEndif"

        # step.1 remove lastEndif
        $lines.RemoveAt($lastEndifIdx)

        $lines[$firstIfIdx] = '#pragma once'
        if (-not $lines[$firstDefineIdx + 1] -match '^\s*$') {
            echo "$filepath no empty line after first define: $firstDefine"
            $lines[$firstDefineIdx] = ''
        }
        else {
            $lines.RemoveAt($firstDefineIdx)
        }

        Set-Content -Path $filepath -Value $lines -Encoding UTF8
    }

}

foreach ($Path in $PathList) {

    $Path = Join-Path $PSScriptRoot $Path

    Replace-GuardInclude $Path
}