# Prerequisites:


# - PowerShell 7.x or later installed on the client.


# - SecretManagement module and a vault configured (e.g., using SecretStore module).


#   Install-Module -Name SecretManagement, SecretStore -Repository PSGallery -Force


#   Register-SecretVault -Name MySecretStore -ModuleName SecretStore -DefaultVault


#   Set-Secret -Name "MyRemoteCreds" -Secret (Get-Credential) -Vault MySecretStore


# - Target hosts have WinRM enabled and allow inbound connections (port 5985/5986).


# - Client has network access to target hosts.

# 2024年5月18日 (JST) 更新

# 1. パラメーターの定義と認証情報の取得

param (
    [string[]]$ComputerName = @("Server01", "Server02", "Server03", "NonExistentHost"),
    [string]$CredentialName = "MyRemoteCreds", # SecretManagementで登録された認証情報名
    [int]$ThrottleLimit = 5,                  # 同時接続数
    [int]$MaxRetryAttempts = 3,               # 最大再試行回数
    [int]$RetryDelaySeconds = 5,              # 再試行間隔（秒）
    [int]$CimSessionTimeoutSeconds = 10,      # CIMセッション確立タイムアウト（秒）
    [int]$CimOperationTimeoutSeconds = 30     # CIM操作実行タイムアウト（秒）
)

# 認証情報の安全な取得

try {

    # SecretManagement モジュールを使用し、保存された認証情報を取得

    $Credential = Get-Secret -Name $CredentialName -AsPlainText -Vault (Get-SecretVault | Select-Object -ExpandProperty Name | Select-Object -First 1)
}
catch {
    Write-Error "認証情報 '$CredentialName' の取得に失敗しました。SecretManagementのVault設定を確認してください。エラー: $($_.Exception.Message)"
    exit 1
}

$SessionOptions = New-CimSessionOption -TimeoutInSeconds $CimSessionTimeoutSeconds -OperationTimeoutSeconds $CimOperationTimeoutSeconds

# 結果を格納する配列

$results = [System.Collections.ArrayList]::Synchronized(@())

Write-Host "CIMセッションを使ったリモート操作を開始します。対象ホスト: $($ComputerName.Count)台, 並列数: $ThrottleLimit"

# 2. 並列処理の実行 (PowerShell 7.x以降)

$scriptBlock = {
    param($Computer, $Credential, $SessionOptions, $MaxRetryAttempts, $RetryDelaySeconds, $results)

    $retryCount = 0
    $success = $false
    $hostResult = [PSCustomObject]@{
        ComputerName = $Computer
        Status       = "Pending"
        Message      = ""
        OSName       = ""
        Version      = ""
        LastError    = ""
    }

    while ($retryCount -lt $MaxRetryAttempts -and -not $success) {
        try {
            Write-Verbose "[$Computer] CIMセッション確立試行 $(($retryCount + 1))/$MaxRetryAttempts..."

            # CIMセッションの確立

            $cimSession = New-CimSession -ComputerName $Computer -Credential $Credential -SessionOption $SessionOptions -ErrorAction Stop

            Write-Verbose "[$Computer] WMI情報取得試行 $(($retryCount + 1))/$MaxRetryAttempts..."

            # WMI (Win32_OperatingSystem) 情報の取得

            $osInfo = Invoke-CimMethod -Session $cimSession -ClassName Win32_OperatingSystem -MethodName Get | Select-Object Caption, Version -ErrorAction Stop

            # 結果の更新

            $hostResult.Status = "Success"
            $hostResult.Message = "OS情報取得成功"
            $hostResult.OSName = $osInfo.Caption
            $hostResult.Version = $osInfo.Version
            $success = $true
        }
        catch {
            $errorMessage = $_.Exception.Message
            $hostResult.LastError = $errorMessage
            Write-Warning "[$Computer] エラー発生 (試行 $(($retryCount + 1))/$MaxRetryAttempts): $errorMessage"

            if ($cimSession) {
                Write-Verbose "[$Computer] 失敗したCIMセッションをクリーンアップ中..."
                Remove-CimSession -CimSession $cimSession -ErrorAction SilentlyContinue
                $cimSession = $null
            }

            $retryCount++
            if ($retryCount -lt $MaxRetryAttempts) {
                Write-Verbose "[$Computer] $RetryDelaySeconds 秒後に再試行します..."
                Start-Sleep -Seconds $RetryDelaySeconds
            }
        }
        finally {

            # 処理が成功した場合のみセッションを削除

            if ($success -and $cimSession) {
                Write-Verbose "[$Computer] CIMセッションをクリーンアップ中..."
                Remove-CimSession -CimSession $cimSession -ErrorAction SilentlyContinue
            }
        }
    }

    if (-not $success) {
        $hostResult.Status = "Failed"
        $hostResult.Message = "複数回の再試行後も処理に失敗しました。"
    }

    # 結果をスレッドセーフなArrayListに追加

    $results.Add($hostResult) | Out-Null
}

# Measure-Command を使って実行時間を計測

$totalExecutionTime = Measure-Command {
    $ComputerName | ForEach-Object -Parallel $scriptBlock -ThrottleLimit $ThrottleLimit `
        -ArgumentList $Credential, $SessionOptions, $MaxRetryAttempts, $RetryDelaySeconds, $results `
        -ErrorAction Stop
}

# 3. 結果の表示とログ出力

Write-Host "`n--- 処理結果 ---"
$results | Format-Table -AutoSize

Write-Host "`n--- 実行サマリー ---"
Write-Host "合計処理時間: $($totalExecutionTime.TotalSeconds) 秒"
Write-Host "成功ホスト数: $($results | Where-Object Status -eq 'Success' | Measure-Object).Count"
Write-Host "失敗ホスト数: $($results | Where-Object Status -eq 'Failed' | Measure-Object).Count"

# 構造化ログとしてJSON出力 (例)

$logFilePath = ".\CimOperationLog-$(Get-Date -Format 'yyyyMMdd-HHmmss').json"
$results | ConvertTo-Json -Depth 3 | Set-Content -Path $logFilePath -Encoding Utf8
Write-Host "詳細ログは '$logFilePath' に出力されました。"

# Output: No explicit return value needed, as results are collected in $results ArrayList

# Prerequisites: Log file from previous run (e.g., CimOperationLog-20240518-100000.json)

# 2024年5月18日 (JST) 更新

$previousLogPath = ".\CimOperationLog-20240518-100000.json" # 前回のログファイルのパス

if (Test-Path $previousLogPath) {
    $failedHosts = (Get-Content -Path $previousLogPath | ConvertFrom-Json) |
                   Where-Object Status -eq 'Failed' |
                   Select-Object -ExpandProperty ComputerName

    if ($failedHosts.Count -gt 0) {
        Write-Host "以下のホストで再実行します: $($failedHosts -join ', ')"

        # ここで、上記のコア実装スクリプトを $ComputerName = $failedHosts として呼び出す


        # 例: .\Invoke-CimParallelOperation.ps1 -ComputerName $failedHosts -CredentialName "MyRemoteCreds"

    } else {
        Write-Host "再実行が必要な失敗ホストは見つかりませんでした。"
    }
} else {
    Write-Warning "指定されたログファイルが見つかりません: $previousLogPath"
}