# Prerequisites: Ensure WinRM is enabled and running.
# Set-Service WinRM -StartupType Automatic
# Start-Service WinRM

# 1. 役割機能ファイル (.psrc) の作成
$roleCapabilityPath = "C:\ProgramData\JEAConfiguration\RoleCapabilities\ServiceRestarter.psrc"
$transcriptDirectory = "C:\ProgramData\JEAConfiguration\Transcripts"

if (-not (Test-Path (Split-Path $roleCapabilityPath))) {
    $null = New-Item -ItemType Directory -Path (Split-Path $roleCapabilityPath) -Force
}
if (-not (Test-Path $transcriptDirectory)) {
    $null = New-Item -ItemType Directory -Path $transcriptDirectory -Force
}

$roleParams = @{
    Path = $roleCapabilityPath
    VisibleCmdlets = @(
        @{ Name = 'Get-Service'; Parameters = @{ Name = 'Name'; ValidateSet = 'WinRM','Spooler' } },
        @{ Name = 'Restart-Service'; Parameters = @{ Name = 'Name'; ValidateSet = 'WinRM','Spooler' } }
    )
}
New-PSRoleCapabilityFile @roleParams -Force
Write-Host "役割機能ファイルを作成しました: $roleCapabilityPath" -ForegroundColor Green

# 2. セッション構成ファイル (.pssc) の作成
$sessionConfigurationPath = "C:\ProgramData\JEAConfiguration\ServiceRestarter.pssc"
$securityGroup = 'JEA Service Restarters'

if (-not (Get-LocalGroup -Name $securityGroup -ErrorAction SilentlyContinue)) {
    New-LocalGroup -Name $securityGroup -Description "JEA Service Restarters for specific services."
    Write-Host "ローカルセキュリティグループ '$securityGroup' を作成しました。" -ForegroundColor Yellow
}

$sessionParams = @{
    Path = $sessionConfigurationPath
    SchemaVersion = '2.0.0.0'
    Author = 'YourName'
    Description = 'JEA endpoint for restarting specific services.'
    RoleDefinitions = @{
        "$Env:COMPUTERNAME\$securityGroup" = @{ RoleCapabilities = $roleCapabilityPath }
    }
    RunAsVirtualAccount = $true
    TranscriptDirectory = $transcriptDirectory
    TranscriptLogPathFromVariable = 'User'
    TranscriptSeverity = 2
    LanguageMode = 'ConstrainedLanguage'
    IdleTimeoutSec = 1200
}
New-PSSessionConfigurationFile @sessionParams -Force
Write-Host "セッション構成ファイルを作成しました: $sessionConfigurationPath" -ForegroundColor Green

# 3. JEAエンドポイントの登録
Register-PSSessionConfiguration -Path $sessionConfigurationPath -Name 'ServiceRestarter' -Force
Write-Host "JEAセッション構成 'ServiceRestarter' を登録しました。" -ForegroundColor Green

# JEAクライアントスクリプト: 複数ホストに対する並列実行、エラーハンドリング、ロギング
$scriptStartTime = Get-Date -Format "yyyy-MM-dd_HH-mm-ss"
$logDirectory = "C:\JEA_ClientLogs"
if (-not (Test-Path $logDirectory)) {
    New-Item -ItemType Directory -Path $logDirectory -Force | Out-Null
}
$logFilePath = Join-Path $logDirectory "JEA_Client_Transaction_$(hostname)_$scriptStartTime.log"

Start-Transcript -Path $logFilePath -Append -Force
Write-Host "JEAクライアントスクリプトを開始します。ログファイル: $logFilePath" -ForegroundColor Cyan

$targetHosts = @("localhost", "127.0.0.1")
$sessionConfigurationName = "ServiceRestarter"
$maxRetries = 3
$retryDelaySeconds = 5
$ErrorActionPreference = 'Stop'

$measureResult = Measure-Command {
    $targetHosts | ForEach-Object -Parallel {
        param($HostName)
        $attempt = 0
        $success = $false
        $result = $null

        while ($attempt -lt $using:maxRetries -and -not $success) {
            $attempt++
            Write-Host "[$HostName] 接続試行 $attempt / $($using:maxRetries)..."
            try {
                $result = Invoke-Command -ComputerName $HostName `
                                       -SessionConfigurationName $using:sessionConfigurationName `
                                       -ScriptBlock {
                                           $service = Get-Service -Name "WinRM" -ErrorAction Stop
                                           if ($service.Status -ne "Running") {
                                               Write-Host "[$HostName] WinRMサービスが停止しています。再起動を試みます..."
                                               Restart-Service -Name "WinRM" -ErrorAction Stop
                                               Write-Host "[$HostName] WinRMサービスを再起動しました。"
                                           } else {
                                               Write-Host "[$HostName] WinRMサービスは既に実行中です。"
                                           }
                                           Get-Service -Name "WinRM"
                                       } `
                                       -ErrorAction Stop
                $success = $true
                Write-Host "[$HostName] コマンド実行に成功しました。" -ForegroundColor Green
            } catch {
                Write-Host "[$HostName] エラー発生: $($_.Exception.Message)" -ForegroundColor Red
                if ($attempt -lt $using:maxRetries) {
                    Write-Host "[$HostName] $using:retryDelaySeconds 秒待機してリトライします..." -ForegroundColor Yellow
                    Start-Sleep -Seconds $using:retryDelaySeconds
                }
            }
        }

        if (-not $success) {
            Write-Warning "[$HostName] 全ての試行が失敗しました。このホストでの処理をスキップします。"
        }

        [PSCustomObject]@{
            HostName = $HostName
            Status = if ($success) {"Success"} else {"Failed"}
            Output = if ($success) {$result | Out-String} else {$null}
            Error = if ($success) {$null} else {$_.Exception.Message}
        }
    } -ThrottleLimit 5
}

Write-Host "`n処理全体の性能計測結果:" -ForegroundColor Cyan
$measureResult | Format-List

$measureResult.PSObject.BaseObject | ForEach-Object {
    Write-Host "--- ホスト: $($_.HostName) ---"
    Write-Host "  ステータス: $($_.Status)"
    if ($_.Status -eq "Success") {
        Write-Host "  出力:"
        $_.Output
    } else {
        Write-Host "  エラー: $($_.Error)" -ForegroundColor Red
    }
}

Stop-Transcript
Write-Host "JEAクライアントスクリプトが完了しました。ログファイル: $logFilePath" -ForegroundColor Cyan