# Prerequisites: Ensure WinRM is enabled and running.


# Set-Service WinRM -StartupType Automatic


# Start-Service WinRM

# 1. 役割機能ファイル (.psrc) の作成


# この例では、特定のサービス (WinRM) の取得と再起動のみを許可します。


# 役割機能ファイルはC:\ProgramData\JEAConfiguration\RoleCapabilities\ に格納することが推奨されます。

$roleCapabilityPath = "C:\ProgramData\JEAConfiguration\RoleCapabilities\ServiceRestarter.psrc"
$transcriptDirectory = "C:\ProgramData\JEAConfiguration\Transcripts" # トランスクリプト保存ディレクトリ

# ディレクトリが存在しない場合は作成

if (-not (Test-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
}

New-PSRoleCapabilityFile -Path $roleCapabilityPath -Force @{
    VisibleCmdlets = @(
        @{ Name = 'Get-Service'; Parameters = @{ Name = 'Name'; ValidateSet = 'WinRM','Spooler' } },
        @{ Name = 'Restart-Service'; Parameters = @{ Name = 'Name'; ValidateSet = 'WinRM','Spooler' } }
    )

    # スクリプトブロックを許可することも可能


    # VisibleFunctions = @('Get-CustomInfo')


    # ScriptsToProcess = @('C:\JEA\Scripts\Get-CustomInfo.ps1')

} | Out-Null # 出力は不要なためパイプで破棄

Write-Host "役割機能ファイルを作成しました: $roleCapabilityPath" -ForegroundColor Green

# 2. セッション構成ファイル (.pssc) の作成


# この例では、'JEA Service Restarters' というセキュリティグループのメンバーが接続できるJEAエンドポイントを作成します。


# 仮想アカウントを使用し、一時的な管理者権限でコマンドを実行します。

$sessionConfigurationPath = "C:\ProgramData\JEAConfiguration\ServiceRestarter.pssc"
$securityGroup = 'JEA Service Restarters' # 予めActive Directoryまたはローカルに作成しておくグループ

# セキュリティグループが存在しない場合は作成（ローカルグループの例）

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

New-PSSessionConfigurationFile -Path $sessionConfigurationPath -Force @{
    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'

    # ログの詳細度 (0=最小, 5=最大)

    TranscriptSeverity = 2

    # JEAセッションの言語モード (ConstrainedLanguageはセキュリティ強化のため推奨)

    LanguageMode = 'ConstrainedLanguage'

    # セッションのタイムアウト (分)

    IdleTimeoutSec = 1200 # 20分

    # スクリプトブロックの入力が許可されるかどうか (ConstrainedLanguageではデフォルトで制限)


    # ExposeCsrfTokens = $true # クロスサイトリクエストフォージェリ対策 (Webアクセスシナリオで有効)

} | Out-Null

Write-Host "セッション構成ファイルを作成しました: $sessionConfigurationPath" -ForegroundColor Green

# 3. JEAエンドポイントの登録


# 既存の構成を上書きする場合は -Force を使用します。

Register-PSSessionConfiguration -Path $sessionConfigurationPath -Force

Write-Host "JEAセッション構成 'ServiceRestarter' を登録しました。" -ForegroundColor Green
Write-Host "JEAエンドポイントの設定が完了しました。" -ForegroundColor Green

# JEAクライアントスクリプト: 複数ホストに対する並列実行、エラーハンドリング、ロギング

# 実行開始日時 (JST)

$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") # 複数のJEAエンドポイントを想定

# JEAセッション構成名

$sessionConfigurationName = "ServiceRestarter"

# 接続試行回数とリトライ間隔

$maxRetries = 3
$retryDelaySeconds = 5

# エラー処理設定

$ErrorActionPreference = 'Stop' # スクリプトで捕捉されないエラーは停止させる

# 処理全体の性能計測を開始

$measureResult = Measure-Command {

    # 複数ホストに対する並列処理 (PowerShell 7+ の ForEach-Object -Parallel を使用)


    # ThrottleLimit: 同時に実行するスクリプトブロックの最大数


    # InputObject: パイプラインで渡すオブジェクトのリスト

    $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 {

                # JEAエンドポイントへの接続とコマンド実行


                # SessionConfigurationNameを指定してJEAセッションに接続

                $result = Invoke-Command -ComputerName $HostName `
                                       -SessionConfigurationName $using:sessionConfigurationName `
                                       -ScriptBlock {

                                           # JEAセッション内で実行されるコマンド


                                           # ここではWinRMサービスの状態を取得して再起動を試みる

                                           $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サービスは既に実行中です。"
                                           }

                                           # CIM/WMIの利用例 (JEAで許可されている場合)


                                           # Get-CimInstance -ClassName Win32_OperatingSystem

                                           Get-Service -Name "WinRM" # JEA内で実行が許可されたGet-Service
                                       } `
                                       -ErrorAction Stop # Invoke-Command自身のエラーも捕捉
                $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 # 最大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

# 機密の安全な取り扱いに関する言及


# 実際には、パスワードなどの機密情報はGet-CredentialやSecretManagementモジュールを利用して安全に処理します。


# 例: $credential = Get-Secret -Name "MyJEAAdminCreds" | ConvertTo-SecureString | Get-Credential -UserName "JEAUser"


#     Invoke-Command ... -Credential $credential

# クライアントスクリプトの Measure-Command 結果の例


# これは上記のクライアントスクリプト実行時に自動的に表示されるものです。


# (例)


# TotalSeconds      : 5.234567


# TotalMilliseconds : 5234.567


# ...