嘿，我来帮你搞定这个问题！你已经能打开SSMS和查询文件了，接下来要实现自动执行查询和关闭窗口，得靠UI自动化来模拟用户操作——毕竟SSMS本身没有直接的命令行参数支持执行查询后自动关闭。下面是具体的实现思路和代码：

1. 加载必要的.NET程序集 #

要控制SSMS的GUI操作，我们需要用到.NET的UI自动化类库和Windows Forms的键盘模拟功能：

Add-Type -AssemblyName UIAutomationClient
Add-Type -AssemblyName UIAutomationTypes
Add-Type -AssemblyName System.Windows.Forms

2. 打开SSMS并加载查询文件（你的现有代码） #

先把你已经写好的打开SSMS和查询文件的代码放这里，按需替换路径和实例名：

# 替换成你的查询文件路径和SQL实例名
$queryFilePath = "C:\Your\Path\To\Query.sql"
$sqlInstance = "YourSQLServerInstance"

# 启动SSMS并加载查询文件（Windows身份验证用-E，SQL身份验证换成-U和-P参数）
Start-Process "Ssms.exe" -ArgumentList "-S $sqlInstance -E -f $queryFilePath"

# 给SSMS一点加载时间，避免后面找不到窗口
Start-Sleep -Seconds 3

3. 定位到目标SSMS窗口 #

我们需要找到刚才打开的、加载了查询文件的SSMS窗口：

$queryFileName = [System.IO.Path]::GetFileName($queryFilePath)
$ssmsProcess = Get-Process | Where-Object { 
    $_.Name -eq "Ssms" -and $_.MainWindowTitle -match $queryFileName 
}

if (-not $ssmsProcess) {
    Write-Error "没找到加载了目标查询文件的SSMS窗口！"
    exit 1
}

4. 模拟执行查询 #

SSMS里执行查询的快捷键是F5，我们先激活窗口再发送这个快捷键：

# 把SSMS窗口调到前台
$ssmsProcess.MainWindowHandle | Set-ForegroundWindow

# 发送F5执行查询
[System.Windows.Forms.SendKeys]::SendWait("{F5}")

优化：等待查询执行完成 #

固定的Start-Sleep不够灵活，你可以通过检测SSMS状态栏的文本判断查询是否完成，比如找“命令已完成”的提示：

$automation = [System.Windows.Automation.AutomationElement]::FromHandle($ssmsProcess.MainWindowHandle)
$statusBarCondition = [System.Windows.Automation.PropertyCondition]::new(
    [System.Windows.Automation.AutomationElement]::ControlTypeProperty,
    [System.Windows.Automation.ControlType]::StatusBar
)
$statusBar = $automation.FindFirst(
    [System.Windows.Automation.TreeScope]::Descendants,
    $statusBarCondition
)

# 循环检查状态栏文本，直到出现"命令已完成"
do {
    Start-Sleep -Seconds 1
    $statusText = $statusBar.GetCurrentPropertyValue(
        [System.Windows.Automation.AutomationElement]::NameProperty
    )
} while ($statusText -notmatch "命令已完成")

5. 执行截图脚本 #

这一步直接调用你已经写好的截图脚本即可：

# 替换成你的截图脚本路径
& "C:\Your\Path\To\ScreenshotScript.ps1"

6. 关闭查询窗口或SSMS #

执行完截图后，就可以关闭窗口了：

# 关闭单个查询窗口，快捷键Ctrl+F4
[System.Windows.Forms.SendKeys]::SendWait("^{F4}")

# 要是想直接关掉整个SSMS，就用这个：
# $ssmsProcess.Kill()

一些注意事项 #

• 权限问题：如果你的系统开了UAC，建议以管理员身份运行PowerShell脚本，避免无法控制SSMS窗口。
• SQL身份验证：如果不用Windows身份验证，启动SSMS时加上-U YourUsername -P YourPassword（注意密码明文的安全问题，建议用Get-Credential来安全获取密码）。
• 等待时间调整：如果你的查询执行时间很长，记得调整初始的等待加载时间，或者用上面的状态栏检测方法更可靠。

内容的提问来源于stack exchange，提问作者Vinesh Senthilvel