# 查询所有虚拟机的基本信息
Get-VM | Select-Object Name, State, CPUUsage, MemoryAssigned, Uptime,
    @{N = 'MemoryMB'; E = { [math]::Round($_.MemoryAssigned / 1MB, 0) } } |
    Format-Table -AutoSize

# 仅查询正在运行的虚拟机
Get-VM | Where-Object { $_.State -eq 'Running' } |
    Select-Object Name, @{N = 'CPU(%)'; E = { $_.CPUUsage } },
    @{N = 'MemGB'; E = { [math]::Round($_.MemoryAssigned / 1GB, 2) } }

# 批量创建虚拟机的配置参数
$vmConfigs = @(
    @{
        Name   = 'Test-Web01'
        Memory = 2GB
        VHDSize = 60GB
        Switch = 'Default Switch'
    }
    @{
        Name   = 'Test-DB01'
        Memory = 4GB
        VHDSize = 120GB
        Switch = 'Default Switch'
    }
    @{
        Name   = 'Test-API01'
        Memory = 2GB
        VHDSize = 60GB
        Switch = 'Default Switch'
    }
)

# 获取 VHD 默认存储路径
$vhdPath = (Get-VMHost).VirtualHardDiskPath

foreach ($config in $vmConfigs) {
    $vhdFilePath = Join-Path $vhdPath "$($config.Name).vhdx"

    # 检查虚拟机是否已存在
    $existing = Get-VM -Name $config.Name -ErrorAction SilentlyContinue
    if ($existing) {
        Write-Host "跳过: 虚拟机 '$($config.Name)' 已存在" -ForegroundColor Yellow
        continue
    }

    Write-Host "正在创建虚拟机: $($config.Name)..." -ForegroundColor Cyan

    # 创建动态扩展的虚拟硬盘
    New-VHD -Path $vhdFilePath -SizeBytes $config.VHDSize -Dynamic |
        Out-Null

    # 创建虚拟机并挂载虚拟硬盘
    New-VM -Name $config.Name -MemoryStartupBytes $config.Memory `
        -VHDPath $vhdFilePath -SwitchName $config.Switch |
        Out-Null

    # 设置处理器数量
    Set-VM -Name $config.Name -ProcessorCount 2

    Write-Host "  完成: $($config.Name) (内存: $($config.Memory / 1GB)GB, " `
        "磁盘: $($config.VHDSize / 1GB)GB, 2 vCPU)" -ForegroundColor Green
}

Write-Host "`n批量创建完成!" -ForegroundColor Green

# 为所有运行中的虚拟机创建带时间戳的检查点
$timestamp = Get-Date -Format 'yyyy-MM-dd_HHmm'
$runningVMs = Get-VM | Where-Object { $_.State -eq 'Running' }

foreach ($vm in $runningVMs) {
    $checkpointName = "AutoBackup_$timestamp"
    Write-Host "正在为 '$($vm.Name)' 创建检查点: $checkpointName"
    Checkpoint-VM -VM $vm -SnapshotName $checkpointName
}

Write-Host "`n所有检查点创建完成" -ForegroundColor Green

# 查看指定虚拟机的检查点列表
$vmName = 'Win2022-Test'
Get-VMSnapshot -VMName $vmName |
    Select-Object Name, CreationTime, @{N = 'Type'; E = { $_.SnapshotType } } |
    Sort-Object CreationTime -Descending |
    Format-Table -AutoSize

# 回滚虚拟机到指定检查点
$snapshot = Get-VMSnapshot -VMName 'Win2022-Test' |
    Where-Object { $_.Name -eq 'BeforePatch_2025-11-20' }

if ($snapshot) {
    Write-Host "正在将 'Win2022-Test' 回滚到 '$($snapshot.Name)'..."
    Restore-VMSnapshot -VMSnapshot $snapshot -Confirm:$false
    Write-Host "回滚完成" -ForegroundColor Green
} else {
    Write-Host "未找到指定的检查点" -ForegroundColor Red
}

# 按顺序启动虚拟机组
$startOrder = @('Test-DB01', 'Test-API01', 'Test-Web01')

foreach ($vmName in $startOrder) {
    $vm = Get-VM -Name $vmName -ErrorAction SilentlyContinue

    if (-not $vm) {
        Write-Host "跳过: 虚拟机 '$vmName' 不存在" -ForegroundColor Yellow
        continue
    }

    if ($vm.State -eq 'Running') {
        Write-Host "跳过: '$vmName' 已在运行" -ForegroundColor Gray
        continue
    }

    Write-Host "正在启动: $vmName" -ForegroundColor Cyan
    Start-VM -Name $vmName

    # 等待虚拟机进入运行状态
    $timeout = 60
    $elapsed = 0
    while ((Get-VM -Name $vmName).State -ne 'Running' -and $elapsed -lt $timeout) {
        Start-Sleep -Seconds 2
        $elapsed += 2
    }

    if ((Get-VM -Name $vmName).State -eq 'Running') {
        Write-Host "  已启动: $vmName (等待 $($elapsed)s)" -ForegroundColor Green
    } else {
        Write-Host "  超时: $vmName 未能在 ${timeout}s 内启动" -ForegroundColor Red
    }
}

# 优雅关闭所有运行中的虚拟机
$runningVMs = Get-VM | Where-Object { $_.State -eq 'Running' }

foreach ($vm in $runningVMs) {
    Write-Host "正在关闭: $($vm.Name)" -ForegroundColor Cyan
    Stop-VM -Name $vm.Name -Force

    # 等待虚拟机完全关闭
    $count = 0
    while ((Get-VM -Name $vm.Name).State -ne 'Off' -and $count -lt 30) {
        Start-Sleep -Seconds 2
        $count++
    }

    $finalState = (Get-VM -Name $vm.Name).State
    if ($finalState -eq 'Off') {
        Write-Host "  已关闭: $($vm.Name)" -ForegroundColor Green
    } else {
        Write-Host "  强制关闭: $($vm.Name)" -ForegroundColor Yellow
        Stop-VM -Name $vm.Name -TurnOff -Force
    }
}

# 虚拟机资源使用报告
$report = foreach ($vm in Get-VM) {
    $vhd = Get-VMHardDiskDrive -VMName $vm.Name |
        Select-Object -First 1

    $vhdInfo = $null
    if ($vhd -and (Test-Path $vhd.Path)) {
        $vhdItem = Get-VHD -Path $vhd.Path
        $vhdInfo = @{
            SizeGB   = [math]::Round($vhdItem.Size / 1GB, 1)
            UsedGB   = [math]::Round($vhdItem.FileSize / 1GB, 1)
            Fragment = [math]::Round($vhdItem.FragmentationPercentage, 1)
        }
    }

    [PSCustomObject]@{
        Name          = $vm.Name
        State         = $vm.State
        CPUUsage      = "$($vm.CPUUsage)%"
        MemoryGB      = [math]::Round($vm.MemoryAssigned / 1GB, 2)
        MemDemandGB   = [math]::Round($vm.MemoryDemand / 1GB, 2)
        DiskSizeGB    = if ($vhdInfo) { $vhdInfo.SizeGB } else { 'N/A' }
        DiskUsedGB    = if ($vhdInfo) { $vhdInfo.UsedGB } else { 'N/A' }
        Uptime        = $vm.Uptime.ToString('dd\天hh\时mm\分')
    }
}

$report | Format-Table -AutoSize

# 输出汇总统计
$totalRunning = ($report | Where-Object { $_.State -eq 'Running' }).Count
$totalMemGB = ($report | Where-Object { $_.State -eq 'Running' } |
    Measure-Object -Property MemoryGB -Sum).Sum

Write-Host "`n--- 汇总 ---" -ForegroundColor Cyan
Write-Host "虚拟机总数: $(($report).Count)"
Write-Host "运行中: $totalRunning"
Write-Host "总内存占用: $([math]::Round($totalMemGB, 1)) GB"

# 定义 Obsidian vault 路径
$VaultPath = "/Users/wubo/Obsidian/MyVault"

# 检查 vault 是否存在并包含 .obsidian 目录
function Test-Vault {
    param([string]$Path)
    $obsidianDir = Join-Path -Path $Path -ChildPath ".obsidian"
    if ((Test-Path -Path $Path) -and (Test-Path -Path $obsidianDir)) {
        return $true
    }
    return $false
}

# 获取 vault 中所有 Markdown 文件
function Get-VaultNotes {
    param([string]$Path)
    $mdFiles = Get-ChildItem -Path $Path -Filter "*.md" -Recurse -File
    return $mdFiles
}

# 调用示例
if (Test-Vault -Path $VaultPath) {
    $notes = Get-VaultNotes -Path $VaultPath
    Write-Output "Vault 中共有 $($notes.Count) 篇笔记"
    $notes | Select-Object -First 5 | Format-Table Name, Length, LastWriteTime
} else {
    Write-Output "指定的路径不是一个有效的 Obsidian vault"
}

# 定义日记模板
$DailyTemplate = @'
---
date: {DATE}
tags:
  - daily-note
  - journal
---

# {DATE} 日记

## 今日计划

-

## 今日完成

-

## 随想

-

## 明日计划

-
'@

# 批量创建指定日期范围的日记
function New-DailyNotes {
    param(
        [Parameter(Mandatory)]
        [string]$VaultPath,
        [Parameter(Mandatory)]
        [DateTime]$StartDate,
        [Parameter(Mandatory)]
        [DateTime]$EndDate
    )

    $dailyDir = Join-Path -Path $VaultPath -ChildPath "Daily Notes"
    if (-not (Test-Path -Path $dailyDir)) {
        New-Item -Path $dailyDir -ItemType Directory | Out-Null
        Write-Output "已创建 Daily Notes 目录"
    }

    $current = $StartDate
    $created = 0

    foreach ($day in @(while ($current -le $EndDate) { $current; $current = $current.AddDays(1) })) {
        $dateStr = $day.ToString("yyyy-MM-dd")
        $fileName = "{0}.md" -f $dateStr
        $filePath = Join-Path -Path $dailyDir -ChildPath $fileName

        if (Test-Path -Path $filePath) {
            Write-Output "跳过已存在: $fileName"
            continue
        }

        $content = $DailyTemplate -replace '\{DATE\}', $dateStr
        Set-Content -Path $filePath -Value $content -Encoding UTF8
        $created++
        Write-Output "已创建: $fileName"
    }

    Write-Output "`n总计创建 $created 篇日记"
}

# 创建本周（周一到周五）的工作日记
$monday = (Get-Date).Date
while ($monday.DayOfWeek -ne [System.DayOfWeek]::Monday) {
    $monday = $monday.AddDays(-1)
}
$friday = $monday.AddDays(4)

New-DailyNotes -VaultPath $VaultPath -StartDate $monday -EndDate $friday

# 分析 Obsidian vault 中的笔记标签使用情况
function Get-VaultTagStats {
    param(
        [Parameter(Mandatory)]
        [string]$VaultPath
    )

    $notes = Get-ChildItem -Path $VaultPath -Filter "*.md" -Recurse -File
    $tagMap = @{}
    $totalTags = 0

    foreach ($note in $notes) {
        $content = Get-Content -Path $note.FullName -Raw -ErrorAction SilentlyContinue
        if ($null -eq $content) { continue }

        # 提取 front matter 中的 tags
        if ($content -match '(?s)^---\r?\n(.*?)\r?\n---') {
            $frontMatter = $Matches[1]
            $tagLines = $frontMatter -split "`n" | Where-Object { $_ -match '^\s*-\s+' }

            foreach ($line in $tagLines) {
                $tag = ($line -replace '^\s*-\s+', '').Trim()
                if ($tag -and $tag -notmatch '^\s*$') {
                    $totalTags++
                    if ($tagMap.ContainsKey($tag)) {
                        $tagMap[$tag]++
                    } else {
                        $tagMap[$tag] = 1
                    }
                }
            }
        }
    }

    # 按使用次数排序输出
    $results = $tagMap.GetEnumerator() |
        Sort-Object -Property Value -Descending |
        Select-Object -First 20

    Write-Output "=== Vault 标签统计 ==="
    Write-Output "笔记总数: $($notes.Count)"
    Write-Output "标签总数: $totalTags"
    Write-Output "去重标签: $($tagMap.Count)"
    Write-Output ""
    Write-Output "Top 20 标签:"
    Write-Output ("{0,-25} {1,8}" -f "标签", "出现次数")
    Write-Output ("{0,-25} {1,8}" -f "----", "--------")

    foreach ($item in $results) {
        Write-Output ("{0,-25} {1,8}" -f $item.Key, $item.Value)
    }
}

Get-VaultTagStats -VaultPath $VaultPath

# 从 vault 中提取所有未完成的待办事项
function Get-VaultTodoItems {
    param(
        [Parameter(Mandatory)]
        [string]$VaultPath,
        [switch]$IncludeCompleted
    )

    $notes = Get-ChildItem -Path $VaultPath -Filter "*.md" -Recurse -File
    $todoItems = [System.Collections.Generic.List[PSObject]]::new()

    foreach ($note in $notes) {
        $lines = Get-Content -Path $note.FullName -ErrorAction SilentlyContinue
        if ($null -eq $lines) { continue }

        $lineNum = 0
        foreach ($line in $lines) {
            $lineNum++

            if ($IncludeCompleted) {
                # 匹配已完成和未完成的待办
                if ($line -match '^\s*-\s+\[([ xX])\]\s+(.+)') {
                    $done = $Matches[1] -match '[xX]'
                    $todoItems += [PSCustomObject]@{
                        File     = $note.Name
                        Line     = $lineNum
                        Done     = $done
                        Task     = $Matches[2].Trim()
                        Relative = $note.FullName.Replace($VaultPath, "").TrimStart("/", "\")
                    }
                }
            } else {
                # 只匹配未完成的待办
                if ($line -match '^\s*-\s+\[\s\]\s+(.+)') {
                    $todoItems += [PSCustomObject]@{
                        File     = $note.Name
                        Line     = $lineNum
                        Task     = $Matches[1].Trim()
                        Relative = $note.FullName.Replace($VaultPath, "").TrimStart("/", "\")
                    }
                }
            }
        }
    }

    return $todoItems
}

# 查找所有未完成的待办事项
$todos = Get-VaultTodoItems -VaultPath $VaultPath

Write-Output "=== 未完成待办事项 ($($todos.Count) 项) ==="
Write-Output ""

$grouped = $todos | Group-Object -Property File |
    Sort-Object -Property Count -Descending |
    Select-Object -First 10

foreach ($group in $grouped) {
    Write-Output "--- $($group.Name) ($($group.Count) 项) ---"
    foreach ($item in $group.Group) {
        Write-Output "  [ ] $($item.Task)"
    }
    Write-Output ""
}

function Write-AppLog {
    param(
        [Parameter(Mandatory)]
        [string]$Message,

        [Parameter()]
        [ArgumentCompleter({
            param($commandName, $parameterName, $wordToComplete)
            @('Debug', 'Info', 'Warning', 'Error') |
                Where-Object { $_ -like "$wordToComplete*" }
        })]
        [string]$LogLevel = 'Info'
    )

    $Timestamp = Get-Date -Format 'yyyy-MM-dd HH:mm:ss'
    Write-Host "[$Timestamp] [$LogLevel] $Message"
}

Write-AppLog -Message '系统启动完成' -LogLevel W<TAB>

# 为 Stop-Service 的 -Name 参数注册补全器
Register-ArgumentCompleter -CommandName 'Stop-Service' -ParameterName 'Name' -ScriptBlock {
    param($commandName, $parameterName, $wordToComplete)

    $Services = Get-Service | Where-Object { $_.Status -eq 'Running' }
    foreach ($Svc in $Services) {
        if ($Svc.Name -like "$wordToComplete*") {
            [System.Management.Automation.CompletionResult]::new(
                $Svc.Name,
                $Svc.Name,
                'ParameterValue',
                "运行中 - $($Svc.DisplayName)"
            )
        }
    }
}

# 使用时按 Tab 即可看到所有正在运行的服务
Stop-Service -Name <TAB>

# 定义辅助函数：从配置文件读取环境列表
function Get-AvailableEnvironments {
    $ConfigPath = Join-Path $PSScriptRoot 'environments.json'
    if (Test-Path $ConfigPath) {
        $Config = Get-Content -Path $ConfigPath -Raw | ConvertFrom-Json
        return $Config.Environments
    }
    return @()
}

# 定义部署函数，带动态参数补全
function Publish-Project {
    param(
        [Parameter(Mandatory)]
        [ArgumentCompleter({
            param($commandName, $parameterName, $wordToComplete)
            $EnvList = Get-AvailableEnvironments
            foreach ($Env in $EnvList) {
                $Name = $Env.Name
                if ($Name -like "$wordToComplete*") {
                    $Detail = "$($Env.Region) - $($Env.Cluster)"
                    [System.Management.Automation.CompletionResult]::new(
                        $Name,
                        $Name,
                        'ParameterValue',
                        $Detail
                    )
                }
            }
        })]
        [string]$Environment,

        [Parameter(Mandatory)]
        [string]$ProjectName,

        [Parameter()]
        [ValidateSet('patch', 'minor', 'major')]
        [string]$VersionBump = 'patch'
    )

    Write-Host "正在部署项目: $ProjectName"
    Write-Host "目标环境: $Environment"
    Write-Host "版本升级类型: $VersionBump"
    Write-Host "部署完成！"
}

# 定义一个哈希表，映射参数名到对应的补全脚本块
$CompleterMap = @{
    # 服务器名称补全
    ServerName = {
        param($commandName, $parameterName, $wordToComplete)
        $Servers = @('web-prod-01', 'web-prod-02', 'web-qa-01', 'db-prod-01')
        foreach ($S in $Servers) {
            if ($S -like "$wordToComplete*") {
                [System.Management.Automation.CompletionResult]::new(
                    $S, $S, 'ParameterValue', $S
                )
            }
        }
    }

    # 日志级别补全
    LogLevel = {
        param($commandName, $parameterName, $wordToComplete)
        $Levels = @('TRACE', 'DEBUG', 'INFO', 'WARN', 'ERROR', 'FATAL')
        foreach ($L in $Levels) {
            if ($L -like "$wordToComplete*") {
                [System.Management.Automation.CompletionResult]::new(
                    $L, $L, 'ParameterValue', "日志级别: $L"
                )
            }
        }
    }
}

# 获取模块中所有函数，为匹配的参数名注册补全器
$Functions = @('Connect-AppServer', 'Get-AppLog', 'Restart-AppService')
foreach ($Func in $Functions) {
    foreach ($ParamName in $CompleterMap.Keys) {
        Register-ArgumentCompleter -CommandName $Func -ParameterName $ParamName `
            -ScriptBlock $CompleterMap[$ParamName]
    }
}

Write-Host '已为以下函数注册参数补全器:'
foreach ($Func in $Functions) {
    Write-Host "  - $Func"
}

function Remove-OldTempFiles {
    <#
    .SYNOPSIS
    清理指定目录中超过指定天数的临时文件
    #>
    [CmdletBinding(SupportsShouldProcess)]
    param(
        [Parameter(Mandatory)]
        [string]$Path,

        [int]$OlderThanDays = 30,

        [string[]]$Extension = @('.tmp', '.log', '.bak')
    )

    # 验证目标目录是否存在
    if (-not (Test-Path -Path $Path)) {
        Write-Error "目录不存在: $Path"
        return
    }

    $cutoffDate = (Get-Date).AddDays(-$OlderThanDays)
    $files = Get-ChildItem -Path $Path -File -Recurse -ErrorAction SilentlyContinue |
        Where-Object { $_.Extension -in $Extension -and $_.LastWriteTime -lt $cutoffDate }

    if ($files.Count -eq 0) {
        Write-Host "未找到符合条件的文件" -ForegroundColor Yellow
        return
    }

    Write-Host "找到 $($files.Count) 个符合条件的文件 (超过 $OlderThanDays 天)" -ForegroundColor Cyan

    $deletedCount = 0
    $totalSize = 0

    foreach ($file in $files) {
        # ShouldProcess 是核心：-WhatIf 时只输出描述，-Confirm 时弹出确认提示
        if ($PSCmdlet.ShouldProcess($file.FullName, "删除文件")) {
            try {
                $totalSize += $file.Length
                Remove-Item -Path $file.FullName -Force -ErrorAction Stop
                $deletedCount++
                Write-Host "  [已删除] $($file.FullName)" -ForegroundColor Green
            }
            catch {
                Write-Warning "删除失败: $($file.FullName) - $($_.Exception.Message)"
            }
        }
    }

    Write-Host "`n清理完成: 删除 $deletedCount 个文件, 释放 $([math]::Round($totalSize / 1MB, 2)) MB" -ForegroundColor Cyan
}

# 使用 -WhatIf 预览将要删除的文件，不会实际删除
Remove-OldTempFiles -Path 'C:\Temp' -OlderThanDays 7 -WhatIf

function Restart-CriticalService {
    <#
    .SYNOPSIS
    安全重启指定服务，关键服务需二次确认
    #>
    [CmdletBinding(SupportsShouldProcess)]
    param(
        [Parameter(Mandatory)]
        [string]$Name,

        [string]$ComputerName = $env:COMPUTERNAME
    )

    # 查询服务信息
    $service = Get-Service -Name $Name -ComputerName $ComputerName -ErrorAction SilentlyContinue
    if (-not $service) {
        Write-Error "找不到服务: $Name"
        return
    }

    # 定义关键服务列表，这些服务需要额外确认
    $criticalServices = @('Winmgmt', 'EventLog', 'Dnscache', 'LanmanServer', 'Netlogon')

    $target = "\\$ComputerName\Service:$Name"

    # 第一层：ShouldProcess 检查（受 -WhatIf 和 -Confirm 控制）
    if (-not $PSCmdlet.ShouldProcess($target, "重启服务")) {
        return
    }

    # 第二层：如果是关键服务，用 ShouldContinue 强制二次确认
    if ($Name -in $criticalServices) {
        $message = "服务 '$Name' 是关键系统服务，重启可能导致系统不稳定。"
        $caption = "确认重启关键服务"

        # ShouldContinue 支持自定义 Yes/No 提示文本
        $confirmed = $PSCmdlet.ShouldContinue($message, $caption)
        if (-not $confirmed) {
            Write-Host "操作已取消" -ForegroundColor Yellow
            return
        }
    }

    try {
        # 记录当前状态
        $originalStatus = $service.Status
        Write-Host "服务当前状态: $originalStatus" -ForegroundColor Cyan

        # 如果服务正在运行，先停止
        if ($originalStatus -eq 'Running') {
            Write-Host "正在停止服务 $Name..." -ForegroundColor Yellow
            Stop-Service -Name $Name -Force -ComputerName $ComputerName -ErrorAction Stop
            Start-Sleep -Seconds 2
        }

        # 启动服务
        Write-Host "正在启动服务 $Name..." -ForegroundColor Yellow
        Start-Service -Name $Name -ComputerName $ComputerName -ErrorAction Stop

        # 验证最终状态
        $finalStatus = (Get-Service -Name $Name -ComputerName $ComputerName).Status
        Write-Host "服务重启完成，最终状态: $finalStatus" -ForegroundColor Green
    }
    catch {
        Write-Error "服务重启失败: $($_.Exception.Message)"
    }
}

# 普通服务：仅需 ShouldProcess 确认
Restart-CriticalService -Name 'Spooler' -Confirm

# 关键服务：ShouldProcess 通过后还会弹出 ShouldContinue 二次确认
Restart-CriticalService -Name 'Winmgmt' -Confirm

function Set-ServiceState {
    <#
    .SYNOPSIS
    批量管理服务状态，支持影响级别控制
    #>
    [CmdletBinding(SupportsShouldProcess,
        ConfirmImpact = 'Medium')]
    param(
        [Parameter(Mandatory)]
        [string[]]$ServiceName,

        [ValidateSet('Start', 'Stop', 'Restart')]
        [string]$Action = 'Restart',

        [int]$DelayBetweenActions = 1
    )

    $results = @()

    foreach ($svc in $ServiceName) {
        $service = Get-Service -Name $svc -ErrorAction SilentlyContinue
        if (-not $service) {
            Write-Warning "服务不存在: $svc"
            $results += [PSCustomObject]@{
                Service = $svc
                Action  = $Action
                Status  = 'NotFound'
            }
            continue
        }

        # 根据操作类型构建不同的确认描述
        $operationDesc = switch ($Action) {
            'Start'   { '启动服务' }
            'Stop'    { '停止服务' }
            'Restart' { '重启服务' }
        }

        # ShouldProcess 第二个参数是操作描述，会显示在确认提示中
        if ($PSCmdlet.ShouldProcess($svc, $operationDesc)) {
            try {
                switch ($Action) {
                    'Start'   { Start-Service -Name $svc -ErrorAction Stop }
                    'Stop'    { Stop-Service -Name $svc -Force -ErrorAction Stop }
                    'Restart' {
                        Stop-Service -Name $svc -Force -ErrorAction Stop
                        Start-Sleep -Seconds $DelayBetweenActions
                        Start-Service -Name $svc -ErrorAction Stop
                    }
                }

                $finalState = (Get-Service -Name $svc).Status
                Write-Host "  [$Action] $svc -> $finalState" -ForegroundColor Green

                $results += [PSCustomObject]@{
                    Service = $svc
                    Action  = $Action
                    Status  = $finalState
                }
            }
            catch {
                Write-Warning "操作失败: $svc - $($_.Exception.Message)"
                $results += [PSCustomObject]@{
                    Service = $svc
                    Action  = $Action
                    Status  = "Failed: $($_.Exception.Message)"
                }
            }
        }
        else {
            $results += [PSCustomObject]@{
                Service = $svc
                Action  = $Action
                Status  = 'Skipped'
            }
        }
    }

    Write-Host "`n===== 操作结果汇总 =====" -ForegroundColor Cyan
    $results | Format-Table -AutoSize
}

# 批量重启多个服务
# ConfirmImpact 为 Medium：$ConfirmPreference 默认为 High，不会逐个确认
Set-ServiceState -ServiceName 'Spooler', 'W32Time', 'Audiosrv' -Action Restart

# 强制逐个确认（覆盖 ConfirmImpact 设置）
Set-ServiceState -ServiceName 'Spooler', 'W32Time' -Action Stop -Confirm

# 安装 Az 模块（如已安装可跳过）
# Install-Module -Name Az -Scope CurrentUser -Force

# 连接 Azure 账户
Connect-AzAccount

# 获取所有订阅信息
$subscriptions = Get-AzSubscription
foreach ($sub in $subscriptions) {
    $info = [ordered]@{
        Name      = $sub.Name
        Id        = $sub.Id
        State     = $sub.State
        TenantId  = $sub.TenantId
    }
    New-Object -TypeName PSCustomObject -Property $info
}

function Get-AzCostDetail {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory = $true)]
        [string]$SubscriptionId,

        [Parameter(Mandatory = $false)]
        [DateTime]$StartDate = (Get-Date).AddDays(-30),

        [Parameter(Mandatory = $false)]
        [DateTime]$EndDate = (Get-Date),

        [Parameter(Mandatory = $false)]
        [int]$Top = 10
    )

    # 选择目标订阅上下文
    $null = Set-AzContext -SubscriptionId $SubscriptionId

    # 构建 Cost Management REST API 请求
    $apiVersion = "2023-11-01"
    $uri = "/subscriptions/$SubscriptionId/providers/Microsoft.CostManagement/query?api-version=$apiVersion"

    # 构建请求体，按服务维度分组汇总
    $body = @{
        type       = "ActualCost"
        timeframe  = "Custom"
        timePeriod = @{
            from = $StartDate.ToString("yyyy-MM-dd")
            to   = $EndDate.ToString("yyyy-MM-dd")
        }
        dataset    = @{
            granularity = "None"
            aggregation = @{
                totalCost = @{
                    name     = "Cost"
                    function = "Sum"
                }
            }
            grouping    = @(
                @{
                    type = "Dimension"
                    name = "ServiceName"
                }
            )
        }
    } | ConvertTo-Json -Depth 10

    # 调用 REST API
    $response = Invoke-AzRestMethod -Path $uri -Method POST -Payload $body

    if ($response.StatusCode -ne 200) {
        Write-Error "查询失败，状态码：$($response.StatusCode)"
        Write-Error $response.Content
        return
    }

    # 解析返回结果
    $result = $response.Content | ConvertFrom-Json
    $columns = $result.properties.columns
    $rows = $result.properties.rows

    # 按费用降序排列，取前 N 条
    $sortedRows = $rows | Sort-Object { $_[0] } -Descending | Select-Object -First $Top

    foreach ($row in $sortedRows) {
        $costUSD = [math]::Round($row[0], 2)
        $serviceName = $row[1]
        $currency = $row[2]

        $detail = [ordered]@{
            ServiceName = $serviceName
            CostUSD     = $costUSD
            Currency    = $currency
        }
        New-Object -TypeName PSCustomObject -Property $detail
    }
}

# 查询最近 30 天费用 Top 10 服务
Get-AzCostDetail -SubscriptionId "a1b2c3d4-e5f6-7890-abcd-ef1234567890" -Top 10

function Set-AzCostBudget {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory = $true)]
        [string]$SubscriptionId,

        [Parameter(Mandatory = $true)]
        [string]$BudgetName,

        [Parameter(Mandatory = $true)]
        [decimal]$Amount,

        [Parameter(Mandatory = $false)]
        [string[]]$NotificationEmails = @("admin@contoso.com"),

        [Parameter(Mandatory = $false)]
        [string]$TimeZone = "UTC"
    )

    $null = Set-AzContext -SubscriptionId $SubscriptionId

    $apiVersion = "2023-11-01"
    $uri = "/subscriptions/$SubscriptionId/providers/Microsoft.Consumption/budgets/$BudgetName?api-version=$apiVersion"

    $now = Get-Date
    $year = $now.Year
    $month = $now.Month

    # 构建预算请求体
    $body = @{
        properties = @{
            timePeriod = @{
                startDate = "$year-$($month.ToString('00'))-01"
                endDate   = "$($year + 1)-12-31"
            }
            timeGrain  = "Monthly"
            amount     = $Amount
            currentSpend = @{
                amount   = 0
                unit     = "USD"
            }
            notifications = @{
                Notification80 = @{
                    enabled       = $true
                    operator      = "GreaterThan"
                    threshold     = 80
                    contactEmails = $NotificationEmails
                }
                Notification100 = @{
                    enabled       = $true
                    operator      = "GreaterThan"
                    threshold     = 100
                    contactEmails = $NotificationEmails
                }
            }
        }
    } | ConvertTo-Json -Depth 10

    $response = Invoke-AzRestMethod -Path $uri -Method PUT -Payload $body

    if ($response.StatusCode -in @(200, 201)) {
        $result = $response.Content | ConvertFrom-Json
        Write-Host "预算 '$BudgetName' 创建成功！" -ForegroundColor Green
        Write-Host "  月度预算金额: `$$Amount USD"
        Write-Host "  告警阈值: 80%, 100%"
        Write-Host "  通知邮箱: $($NotificationEmails -join ', ')"
        Write-Host "  预算 ID: $($result.name)"
    }
    else {
        Write-Error "预算创建失败，状态码：$($response.StatusCode)"
        Write-Error $response.Content
    }
}

# 创建月度 500 美元的预算
Set-AzCostBudget `
    -SubscriptionId "a1b2c3d4-e5f6-7890-abcd-ef1234567890" `
    -BudgetName "Monthly-Production-Budget" `
    -Amount 500 `
    -NotificationEmails @("admin@contoso.com", "finance@contoso.com")

function Find-AzIdleResources {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory = $true)]
        [string]$SubscriptionId
    )

    $null = Set-AzContext -SubscriptionId $SubscriptionId
    $findings = @()

    # 1. 检查停止状态的虚拟机
    $vms = Get-AzVM -Status
    foreach ($vm in $vms) {
        $powerState = ($vm.Statuses | Where-Object { $_.Code -like "PowerState/*" }).Code
        if ($powerState -eq "PowerState/deallocated") {
            $estimatedMonthly = [math]::Round(
                ($vm.HardwareProfile.VmSize -match "Standard_" -and $true) * 0
            )
            $rgName = $vm.ResourceGroupName
            $findings += [ordered]@{
                ResourceType  = "Virtual Machine"
                ResourceName  = $vm.Name
                ResourceGroup = $rgName
                Status        = "Deallocated"
                Suggestion    = "已释放但仍保留配置，如不再使用建议删除"
                Priority      = "Medium"
            }
        }
        elseif ($powerState -eq "PowerState/stopped") {
            $findings += [ordered]@{
                ResourceType  = "Virtual Machine"
                ResourceName  = $vm.Name
                ResourceGroup = $vm.ResourceGroupName
                Status        = "Stopped (not deallocated)"
                Suggestion    = "停止但未释放，仍在计费！建议 deallocate 或删除"
                Priority      = "High"
            }
        }
    }

    # 2. 检查未挂载的磁盘
    $disks = Get-AzDisk
    foreach ($disk in $disks) {
        if ($null -eq $disk.ManagedBy -and $disk.DiskState -eq "Unattached") {
            $sizeGB = $disk.DiskSizeGB
            $sku = $disk.Sku.Name
            $findings += [ordered]@{
                ResourceType  = "Managed Disk"
                ResourceName  = $disk.Name
                ResourceGroup = $disk.ResourceGroupName
                Status        = "Unattached ($sizeGB GB, $sku)"
                Suggestion    = "未挂载的磁盘，删除可节省约 `$$([math]::Round($sizeGB * 0.05, 2))/月"
                Priority      = "High"
            }
        }
    }

    # 3. 检查未关联的公网 IP
    $publicIps = Get-AzPublicIpAddress
    foreach ($pip in $publicIps) {
        if (-not $pip.IpConfiguration) {
            $findings += [ordered]@{
                ResourceType  = "Public IP Address"
                ResourceName  = $pip.Name
                ResourceGroup = $pip.ResourceGroupName
                Status        = "Unassociated"
                Suggestion    = "未关联的公网 IP，删除可节省约 `$3.65/月"
                Priority      = "Medium"
            }
        }
    }

    # 4. 检查暂停的 SQL 数据库
    $sqlServers = Get-AzSqlServer
    foreach ($server in $sqlServers) {
        $databases = Get-AzSqlDatabase -ServerName $server.ServerName -ResourceGroupName $server.ResourceGroupName
        foreach ($db in $databases) {
            if ($db.Status -eq "Paused") {
                $findings += [ordered]@{
                    ResourceType  = "SQL Database"
                    ResourceName  = "$($server.ServerName)/$($db.DatabaseName)"
                    ResourceGroup = $db.ResourceGroupName
                    Status        = "Paused"
                    Suggestion    = "已暂停但仍保留，确认是否需要恢复或删除"
                    Priority      = "Low"
                }
            }
        }
    }

    # 按优先级排序输出
    $priorityOrder = @{ "High" = 1; "Medium" = 2; "Low" = 3 }
    $sorted = $findings | Sort-Object { $priorityOrder[$_.Priority] }

    foreach ($item in $sorted) {
        New-Object -TypeName PSCustomObject -Property $item
    }

    Write-Host "`n共发现 $($findings.Count) 个可优化项，其中 High 优先级 $(($findings | Where-Object { $_.Priority -eq 'High' }).Count) 个。" -ForegroundColor Yellow
}

# 扫描生产订阅的闲置资源
Find-AzIdleResources -SubscriptionId "a1b2c3d4-e5f6-7890-abcd-ef1234567890"