function Invoke-AIScriptGeneration {
    param(
        [Parameter(Mandatory=$true)]
        [string]$Prompt,
        [ValidateSet('AWS','Azure','Windows','Linux')]
        [string]$Environment = 'Windows'
    )

    $apiKey = $env:OPENAI_API_KEY
    $headers = @{
        'Authorization' = "Bearer $apiKey"
        'Content-Type' = 'application/json'
    }

    $body = @{
        model = 'gpt-4-turbo'
        messages = @(
            @{
                role = 'system'
                content = "你是一名资深PowerShell专家，根据用户需求生成可直接执行的运维脚本。当前环境：$Environment"
            },
            @{
                role = 'user'
                content = $Prompt
            }
        )
        temperature = 0.3
        max_tokens = 1024
    } | ConvertTo-Json -Depth 5

    try {
        $response = Invoke-RestMethod -Uri 'https://api.openai.com/v1/chat/completions' \
            -Method Post \
            -Headers $headers \
            -Body $body

        $scriptBlock = [scriptblock]::Create($response.choices[0].message.content)
        Write-Host "生成脚本验证通过：" -ForegroundColor Green
        $scriptBlock.Invoke()
    }
    catch {
        Write-Error "AI脚本生成失败: $($_.Exception.Message)"
    }
}

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

    # 生成Terraform变量文件
    $tfVars = @{
        environment = $Environment
        location    = 'eastus'
        vm_count    = 3
    } | ConvertTo-Json
    $tfVars | Out-File -FilePath "./terraform.tfvars.json"

    # 初始化并应用配置
    terraform init -input=false
    terraform apply -auto-approve -var-file="./terraform.tfvars.json"

    # 获取输出变量
    $output = terraform output -json | ConvertFrom-Json
    [PSCustomObject]@{
        PublicIP = $output.public_ip.value
        StorageEndpoint = $output.storage_endpoint.value
    }
}

# 执行多环境部署
'dev','staging','prod' | ForEach-Object {
    Invoke-TerraformDeployment -Environment $_ -Verbose
}

function Invoke-K8sDeployment {
    param(
        [ValidateSet('AzureAKS','AWS-EKS')]
        [string]$ClusterType,
        [string]$Namespace,
        [string]$DeploymentFile
    )

    try {
        # 认证集群
        $kubeconfig = switch ($ClusterType) {
            'AzureAKS' { Get-AzAksCredential -Admin }
            'AWS-EKS' { Get-EksClusterCredential }
        }

        # 执行部署
        kubectl apply -f $DeploymentFile --namespace $Namespace --kubeconfig $kubeconfig

        # 实时监控
        $watchJob = Start-Job -ScriptBlock {
            kubectl get pods --namespace $using:Namespace --watch
        }
        Receive-Job $watchJob -Wait
    }
    catch {
        Write-Error "部署失败：$_"
    }
    finally {
        Remove-Job $watchJob -Force
    }
}

function Invoke-AIScriptGeneration {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory=$true)]
        [string]$NaturalLanguageQuery,
        
        [ValidateRange(1,4096)]
        [int]$MaxTokens = 1024
    )

    $apiKey = Read-Host -Prompt '输入OpenAI API密钥' -AsSecureString
    $cred = New-Object System.Management.Automation.PSCredential ('api', $apiKey)

    $body = @{
        model = 'gpt-4-turbo-preview'
        messages = @(
            @{
                role = 'system'
                content = '你是一个PowerShell专家，将自然语言查询转换为可执行的PowerShell脚本。确保代码符合最佳安全实践，包含错误处理，并添加中文注释。'
            },
            @{
                role = 'user'
                content = $NaturalLanguageQuery
            }
        )
        temperature = 0.2
        max_tokens = $MaxTokens
    } | ConvertTo-Json -Depth 5

    try {
        $response = Invoke-RestMethod -Uri 'https://api.openai.com/v1/chat/completions' \
            -Method POST \
            -Headers @{ Authorization = 'Bearer ' + $cred.GetNetworkCredential().Password } \
            -ContentType 'application/json' \
            -Body $body

        $scriptBlock = [scriptblock]::Create($response.choices[0].message.content)
        $transcript = [PSCustomObject]@{
            Timestamp = Get-Date -Format 'yyyy-MM-dd HH:mm:ss'
            OriginalQuery = $NaturalLanguageQuery
            GeneratedScript = $response.choices[0].message.content
            TokenUsage = $response.usage
        }

        $transcript | Export-Clixml -Path "$env:TEMP/AIScriptTranscript_$(Get-Date -Format yyyyMMdd).xml"
        return $scriptBlock
    }
    catch {
        Write-Error "AI脚本生成失败: $_"
    }
}