# 创建 OpenAI 交互函数
function Invoke-OpenAIAPI {
    param(
        [string]$ApiKey,
        [string]$Prompt,
        [ValidateSet('gpt-4', 'gpt-3.5-turbo')]
        [string]$Model = 'gpt-3.5-turbo',
        [float]$Temperature = 0.7,
        [int]$MaxTokens = 500
    )
    
    try {
        $headers = @{
            'Content-Type' = 'application/json'
            'Authorization' = "Bearer $ApiKey"
        }
        
        $body = @{
            model = $Model
            messages = @(
                @{
                    role = "user"
                    content = $Prompt
                }
            )
            temperature = $Temperature
            max_tokens = $MaxTokens
        } | ConvertTo-Json
        
        $response = Invoke-RestMethod -Uri "https://api.openai.com/v1/chat/completions" -Method Post -Headers $headers -Body $body
        return $response.choices[0].message.content
    }
    catch {
        Write-Host "OpenAI API 调用失败：$_"
    }
}

# 创建 AI 脚本生成函数
function New-AIScript {
    param(
        [string]$ApiKey,
        [string]$Description,
        [string]$OutputPath
    )
    
    try {
        $prompt = @"
生成一个 PowerShell 脚本，完成以下功能：$Description

要求：
1. 脚本应包含详细的注释
2. 包含适当的错误处理
3. 遵循 PowerShell 最佳实践
4. 只返回脚本代码，不要额外的解释
"@
        
        $script = Invoke-OpenAIAPI -ApiKey $ApiKey -Prompt $prompt -MaxTokens 2000
        
        if ($OutputPath) {
            $script | Out-File -FilePath $OutputPath -Encoding UTF8
            Write-Host "脚本已保存至：$OutputPath"
        }
        
        return $script
    }
    catch {
        Write-Host "AI 脚本生成失败：$_"
    }
}

# 创建 AI 日志分析函数
function Analyze-LogsWithAI {
    param(
        [string]$ApiKey,
        [string]$LogFilePath,
        [string]$OutputPath,
        [switch]$IncludeRecommendations
    )
    
    try {
        $logs = Get-Content -Path $LogFilePath -Raw
        
        # 将日志截断到合理的大小
        if ($logs.Length -gt 4000) {
            $logs = $logs.Substring(0, 4000) + "... [日志截断]"
        }
        
        $promptSuffix = ""
        if ($IncludeRecommendations) {
            $promptSuffix = "并提供解决方案建议。"
        }
        
        $prompt = @"
分析以下系统日志，识别可能的错误、警告和问题模式$promptSuffix

日志内容：
$logs
"@
        
        $analysis = Invoke-OpenAIAPI -ApiKey $ApiKey -Prompt $prompt -MaxTokens 1500
        
        if ($OutputPath) {
            $analysis | Out-File -FilePath $OutputPath -Encoding UTF8
            Write-Host "分析结果已保存至：$OutputPath"
        }
        
        return $analysis
    }
    catch {
        Write-Host "AI 日志分析失败：$_"
    }
}

# 创建 AI 故障排除函数
function Start-AITroubleshooting {
    param(
        [string]$ApiKey,
        [string]$Issue,
        [switch]$RunDiagnostics
    )
    
    try {
        $systemInfo = Get-ComputerInfo | ConvertTo-Json
        
        if ($RunDiagnostics) {
            $eventLogs = Get-WinEvent -LogName Application -MaxEvents 20 | Select-Object TimeCreated, LevelDisplayName, Message | ConvertTo-Json
            $services = Get-Service | Where-Object { $_.Status -eq 'Stopped' -and $_.StartType -eq 'Automatic' } | ConvertTo-Json
            
            $diagnosticInfo = @"
系统信息：
$systemInfo

最近事件日志：
$eventLogs

已停止的自动启动服务：
$services
"@
        } else {
            $diagnosticInfo = "系统信息：$systemInfo"
        }
        
        $prompt = @"
我有以下系统问题：$Issue

基于以下系统信息，提供排查步骤和可能的解决方案：

$diagnosticInfo
"@
        
        $troubleshooting = Invoke-OpenAIAPI -ApiKey $ApiKey -Prompt $prompt -MaxTokens 2000 -Model 'gpt-4'
        
        return $troubleshooting
    }
    catch {
        Write-Host "AI 故障排除失败：$_"
    }
}

# 创建 AI 脚本优化函数
function Optimize-ScriptWithAI {
    param(
        [string]$ApiKey,
        [string]$ScriptPath,
        [string]$OutputPath,
        [switch]$IncludeExplanation
    )
    
    try {
        $script = Get-Content -Path $ScriptPath -Raw
        
        $promptSuffix = ""
        if ($IncludeExplanation) {
            $promptSuffix = "并解释所做的改动和优化理由。"
        }
        
        $prompt = @"
优化以下 PowerShell 脚本的性能和代码质量$promptSuffix

脚本内容：
$script
"@
        
        $optimizedScript = Invoke-OpenAIAPI -ApiKey $ApiKey -Prompt $prompt -MaxTokens 2000 -Model 'gpt-4'
        
        if ($OutputPath) {
            $optimizedScript | Out-File -FilePath $OutputPath -Encoding UTF8
            Write-Host "优化后的脚本已保存至：$OutputPath"
        }
        
        return $optimizedScript
    }
    catch {
        Write-Host "AI 脚本优化失败：$_"
    }
}

function Invoke-ModuleVulnerabilityScan {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory=$true)]
        [string]$ModuleName
    )

    # 获取模块版本信息
    $module = Get-InstalledModule -Name $ModuleName -ErrorAction Stop
    
    # 调用漏洞数据库API
    $response = Invoke-RestMethod -Uri "https://vulndb.example.com/api/modules/$($module.Name)/$($module.Version)"
    
    # 生成安全报告
    [PSCustomObject]@{
        ModuleName = $module.Name
        Version = $module.Version
        Vulnerabilities = $response.vulns.Count
        Critical = $response.vulns | Where-Object { $_.severity -eq 'Critical' } | Measure-Object | Select-Object -Expand Count
        LastUpdated = $module.PublishedDate
    } | Export-Csv -Path "$env:TEMP\ModuleSecurityScan_$(Get-Date -Format yyyyMMdd).csv" -Append
}

# 扫描常用模块
'PSReadLine', 'Pester', 'Az' | ForEach-Object {
    Invoke-ModuleVulnerabilityScan -ModuleName $_ -Verbose
}

# ForEach循环示例
$services = Get-Service
$services | ForEach-Object {
    if ($_.Status -eq 'Running') {
        Write-Host $_.DisplayName
    }
}

# While循环应用场景
$counter = 0
while ($counter -lt 5) {
    Start-Process notepad
    $counter++
}

# 设置循环断点
$i=0
1..10 | ForEach-Object {
    $i++
    if ($i -eq 5) { break }
    $_
}

# 跟踪循环变量
Set-PSDebug -Trace 1
foreach ($file in (Get-ChildItem)) {
    $file.Basename
}

# 全局作用域示例
$global:config = 'Server1'

function Show-Config {
    # 局部作用域访问全局变量
    Write-Host $global:config
    
    # 声明私有变量
    $private:connection = 'Active'
}

Show-Config
# $connection 在此不可访问

# 错误的作用域继承示例
function Set-Value {
    $value = 100
}

Set-Value
Write-Host $value  # 输出为空

# 正确的作用域传递方式
function Get-Value {
    $script:value = 200
}

Get-Value
Write-Host $script:value  # 输出200

try {
    Get-Content -Path '不存在的文件.txt' -ErrorAction Stop
}
catch [System.IO.FileNotFoundException] {
    Write-Warning "文件未找到: $($_.Exception.Message)"
}
catch {
    Write-Error "未知错误: $_"
}
finally {
    Write-Output "清理操作完成"
}

# 自定义错误记录
$ErrorActionPreference = 'Continue'
$ErrorView = 'NormalView'

function Invoke-SafeOperation {
    [CmdletBinding()]
    param([scriptblock]$ScriptBlock)
    
    try {
        & $ScriptBlock
    }
    catch {
        [PSCustomObject]@{
            Timestamp = Get-Date
            ErrorType = $_.Exception.GetType().Name
            Message = $_.Exception.Message
            ScriptLine = $_.InvocationInfo.ScriptLineNumber
        } | Export-Csv -Path 'errors.log' -Append
    }
}

# 错误信息增强处理
$Error[0] | Select-Object *
$Error[0].InvocationInfo | Format-List *
$Error[0].Exception | Format-List *

function [<scope:>]<name> [([type]$parameter1[,[type]$parameter2])]
{
  param([type]$parameter1 [,[type]$parameter2])
  dynamicparam {<statement list>}
  begin {<statement list>}
  process {<statement list>}
  end {<statement list>}
}

function Operation{
   $num1 = 8
   $num2 = 2
   Write-Host "Multiply : $($num1*$num2)"
   Write-Host "Addition : $($num1+$num2)"
   Write-Host "Subtraction : $($num1-$num2)"
   Write-Host "Divide : $($num1 / $num2)"
}
Operation

function show-Message
{
    [CmdletBinding()]
    Param (
    [ Parameter (Mandatory = $true)]

        [string] $Name
    )
    Process
    {
        Write-Host ("Hi $Name !")
        write-host $Name "today is $(Get-Date)"
    }
}

show-message

Get-ChildItem -Path “C:\\NonexistentFolder” -ErrorAction Continue**

Get-Item -Path “C:\\NonexistentFile” -ErrorAction SilentlyContinue**

Remove-Item -Path “C:\\ImportantFile” -ErrorAction Stop**

set-executionpolicy RemoteSigned

get-executionpolicy