update-all-repos.ps1

#Requires -Version 5.1
# Installs gitleaks pre-commit hooks in existing repositories (Windows).
# Usage:
#   .\update-all-repos.ps1                    # All local drives (C:\, D:\, E:\, etc.) - single command for all repos
#   .\update-all-repos.ps1 C:\Projects        # Only repos under C:\Projects
#   .\update-all-repos.ps1 C:\Projects C:\Sites
# Optional: $MAX_DEPTH = 3; .\update-all-repos.ps1 C:\  # Limit depth on a drive

param(
    [Parameter(ValueFromRemainingArguments = $true)]
    [string[]]$TargetPaths
)

$ErrorActionPreference = "Stop"
$TEMPLATE_DIR = Join-Path $env:USERPROFILE ".git-template"
$TEMPLATE_HOOKS = Join-Path $TEMPLATE_DIR "hooks"

function Write-Step { param($Message) Write-Host $Message -ForegroundColor Cyan }
function Write-Ok    { param($Message) Write-Host "  OK  $Message" -ForegroundColor Green }
function Write-Warn  { param($Message) Write-Host "  !!  $Message" -ForegroundColor Yellow }
function Write-Fail  { param($Message) Write-Host "  X   $Message" -ForegroundColor Red }

if (-not (Get-Command gitleaks -ErrorAction SilentlyContinue)) {
    Write-Fail "gitleaks is not installed or not in PATH."
    Write-Host "Run .\install-gitleaks-global.ps1 first, or install from https://github.com/gitleaks/gitleaks/releases"
    exit 1
}

if (-not (Test-Path (Join-Path $TEMPLATE_HOOKS "pre-commit"))) {
    Write-Fail "Git template hooks not found at $TEMPLATE_HOOKS"
    Write-Host "Run .\install-gitleaks-global.ps1 first."
    exit 1
}

# Function to sync global gitleaks config from repository
function Sync-GlobalConfig {
    $scriptDir = Split-Path -Parent $PSCommandPath
    $sourceConfig = Join-Path $scriptDir ".gitleaks.toml"
    $configDir = Join-Path $env:USERPROFILE ".config\gitleaks"
    $targetConfig = Join-Path $configDir "gitleaks.toml"
    
    # Check if source config exists
    if (-not (Test-Path $sourceConfig)) {
        Write-Warn "Source config not found: $sourceConfig"
        Write-Host "  Skipping config sync" -ForegroundColor Gray
        return $false
    }
    
    # Create config directory if it doesn't exist
    try {
        New-Item -ItemType Directory -Path $configDir -Force -ErrorAction Stop | Out-Null
    } catch {
        Write-Fail "Failed to create config directory: $configDir"
        return $false
    }
    
    # Copy the config file
    try {
        Copy-Item -Path $sourceConfig -Destination $targetConfig -Force -ErrorAction Stop
        Write-Ok "Synced global config: $targetConfig"
        return $true
    } catch {
        Write-Fail "Failed to sync config to: $targetConfig"
        return $false
    }
}

$preCommitSrc = Join-Path $TEMPLATE_HOOKS "pre-commit"
$commitMsgSrc = Join-Path $TEMPLATE_HOOKS "commit-msg"

# Sync global gitleaks config from repository
Write-Step "Syncing global gitleaks configuration..."
Sync-GlobalConfig | Out-Null
Write-Host ""

# No path given = scan all local fixed drives (C:\, D:\, E:\, etc.)
if ($TargetPaths.Count -eq 0) {
    # Use .Name (e.g. "C:") to avoid null .Root on some Windows setups
    $TargetPaths = [System.IO.DriveInfo]::GetDrives() | Where-Object { $_.DriveType -eq 'Fixed' -and $_.IsReady } | ForEach-Object { $_.Name + '\' }
    if ($TargetPaths.Count -eq 0) {
        Write-Fail "No local drives found."
        exit 1
    }
    Write-Host "No path specified: scanning all local drives ( $($TargetPaths -join ', ') )" -ForegroundColor Cyan
    Write-Host "This may take a while on large drives. Press Ctrl+C to cancel." -ForegroundColor Gray
    Write-Host ""
}

$script:Updated = 0
$script:Skipped = 0
$script:Failed = 0

function Get-GitRepos {
    param([string]$Root, [int]$MaxDepth = 0)
    $script:ReposFoundList = [System.Collections.Generic.List[string]]::new()

    function Search-Dirs {
        param([string]$Path, [int]$Depth)
        # Log only parent-level folders (direct children of drive or of the given root)
        if ($Depth -eq 1) {
            Write-Host "  Scanning $Path" -ForegroundColor Gray
        }
        try {
            if (Test-Path (Join-Path $Path ".git") -ErrorAction SilentlyContinue) {
                $script:ReposFoundList.Add($Path) | Out-Null
            }
        } catch {
            # Silently skip paths we don't have permission to read
        }
        if ($MaxDepth -gt 0 -and $Depth -ge $MaxDepth) { return }
        Get-ChildItem -Path $Path -Directory -ErrorAction SilentlyContinue | ForEach-Object {
            Search-Dirs -Path $_.FullName -Depth ($Depth + 1)
        }
    }

    Search-Dirs -Path $Root -Depth 0
    Write-Host "  Scan complete: found $($script:ReposFoundList.Count) repos" -ForegroundColor Cyan
    $script:ReposFoundList | Sort-Object -Unique
}

function Install-Hooks {
    param([string]$RepoDir)
    $hooksDir = Join-Path $RepoDir ".git\hooks"
    # Create hooks directory if it doesn't exist
    if (-not (Test-Path $hooksDir)) {
        try {
            New-Item -ItemType Directory -Path $hooksDir -Force -ErrorAction Stop | Out-Null
        } catch {
            Write-Fail "Cannot create hooks dir: $hooksDir - $($_.Exception.Message)"
            return $false
        }
    }
    try {
        Copy-Item -Path $preCommitSrc -Destination (Join-Path $hooksDir "pre-commit") -Force -ErrorAction Stop
        Copy-Item -Path $commitMsgSrc -Destination (Join-Path $hooksDir "commit-msg") -Force -ErrorAction Stop
        return $true
    } catch {
        Write-Fail "Hook copy error: $($_.Exception.Message)"
        return $false
    }
}

foreach ($target in $TargetPaths) {
    $root = $target
    if (-not [System.IO.Path]::IsPathRooted($root)) {
        $root = Join-Path (Get-Location).Path $root
    }
    if (-not (Test-Path $root -PathType Container)) {
        Write-Fail "Directory does not exist: $root"
        continue
    }

    $maxDepth = if ($env:MAX_DEPTH) { [int]$env:MAX_DEPTH } else { 0 }
    Write-Step "Scanning $root for git repositories..."
    $repos = Get-GitRepos -Root $root -MaxDepth $maxDepth
    $i = 0
    foreach ($repo in $repos) {
        $i++
        Write-Host "  [$i] $repo" -ForegroundColor Gray
        if (Install-Hooks -RepoDir $repo) {
            $script:Updated++
            Write-Ok "Hooks installed"
        } else {
            $script:Failed++
            Write-Fail "Failed to install hooks"
        }
    }
}

Write-Host ""
Write-Host "Done. Updated: $($script:Updated), Failed: $($script:Failed)" -ForegroundColor Cyan
if ($script:Updated -eq 0 -and $script:Failed -eq 0) {
    Write-Host ""
    Write-Warn "No git repositories were found in the scanned path(s)."
    Write-Host "  To scan only a specific folder: .\update-all-repos.ps1 C:\Projects" -ForegroundColor Gray
}
if ($script:Failed -gt 0) { exit 1 }