【Azure】VMをrunbookで休止状態（割当解除）にする

# Azure Automation Runbookで使用するPowerShellスクリプト
# VMの起動、再起動、休止状態（休止）を操作

param (
    [Parameter(Mandatory=$true)]
    [string]$ResourceGroupName,  # リソースグループ名

    [Parameter(Mandatory=$true)]
    [string]$HostPoolName,       # ホストプール名

    [Parameter(Mandatory=$true)]
    [string]$SessionHostName,    # 対象のVM名

    [Parameter(Mandatory=$true)]
    [ValidateSet("Start", "Restart", "Hibernate")]
    [string]$Action              # 実行するアクション: Start, Restart, Hibernate
)

# Azureに接続
Write-Output "Authenticating to Azure..."
Connect-AzAccount -Identity

# 対象のセッションホストの完全なリソースIDを取得
$sessionHostResourceId = "/subscriptions/$((Get-AzContext).Subscription.Id)/resourceGroups/$ResourceGroupName/providers/Microsoft.DesktopVirtualization/hostPools/$HostPoolName/sessionHosts/$SessionHostName"

Write-Output "Target Session Host Resource ID: $sessionHostResourceId"

# アクションの実行
switch ($Action) {
    "Start" {
        Write-Output "Starting session host: $SessionHostName..."
        Start-AzVM -ResourceGroupName $ResourceGroupName -Name $SessionHostName
    }
    "Restart" {
        Write-Output "Restarting session host: $SessionHostName..."
        Restart-AzVM -ResourceGroupName $ResourceGroupName -Name $SessionHostName
    }
    "Hibernate" {
        Write-Output "Putting session host into hibernation: $SessionHostName..."
        Stop-AzVM -ResourceGroupName $ResourceGroupName -Name $SessionHostName -Hibernate -Force
    }
    default {
        Write-Error "Unknown action: $Action"
    }
}

Write-Output "Action '$Action' completed for session host: $SessionHostName"