桌面整理大师

$MODEL    = "deepseek-flash"
$BASE_URL = "https://api.deepseek.com"

$outDir = $PSScriptRoot
if (-not $outDir) { $outDir = $PWD.Path }
$outFile = Join-Path $outDir "suggestion.txt"

$desktop = [Environment]::GetFolderPath('Desktop')
Set-Location $desktop

$folders = Get-ChildItem -Directory | Select-Object -ExpandProperty Name
$files   = Get-ChildItem -File      | Select-Object -ExpandProperty Name

$selfNames = @("scan.exe", "scan.ps1", "suggestion.txt", "index.ico", "index.png")
$files = $files |
    Where-Object { $selfNames -notcontains $_ } |
    ForEach-Object { $_ -replace '\s*-\s*快捷方式(\.lnk)?$', '' -replace '\.lnk$', '' }

Add-Type -AssemblyName System.Windows.Forms
Add-Type -AssemblyName System.Drawing

$form = New-Object System.Windows.Forms.Form
$form.Text = "桌面整理助手"
$form.Size = New-Object System.Drawing.Size(460, 180)
$form.StartPosition = "CenterScreen"
$form.FormBorderStyle = "FixedDialog"
$form.MaximizeBox = $false
$form.MinimizeBox = $false

$label = New-Object System.Windows.Forms.Label
$label.Text = "请输入你的 API Key（仅本次使用，不会保存到任何文件）"
$label.Location = New-Object System.Drawing.Point(20, 20)
$label.Size = New-Object System.Drawing.Size(410, 25)
$label.Font = New-Object System.Drawing.Font("Microsoft YaHei", 10)
$form.Controls.Add($label)

$textbox = New-Object System.Windows.Forms.TextBox
$textbox.Location = New-Object System.Drawing.Point(20, 55)
$textbox.Size = New-Object System.Drawing.Size(410, 28)
$textbox.UseSystemPasswordChar = $true
$textbox.Font = New-Object System.Drawing.Font("Consolas", 10)
$form.Controls.Add($textbox)

$ok = New-Object System.Windows.Forms.Button
$ok.Text = "确定"
$ok.Location = New-Object System.Drawing.Point(260, 100)
$ok.Size = New-Object System.Drawing.Size(80, 30)
$ok.DialogResult = [System.Windows.Forms.DialogResult]::OK
$ok.Font = New-Object System.Drawing.Font("Microsoft YaHei", 10)
$form.Controls.Add($ok)

$cancel = New-Object System.Windows.Forms.Button
$cancel.Text = "取消"
$cancel.Location = New-Object System.Drawing.Point(350, 100)
$cancel.Size = New-Object System.Drawing.Size(80, 30)
$cancel.DialogResult = [System.Windows.Forms.DialogResult]::Cancel
$cancel.Font = New-Object System.Drawing.Font("Microsoft YaHei", 10)
$form.Controls.Add($cancel)

$form.AcceptButton = $ok
$form.CancelButton = $cancel

$result = $form.ShowDialog()
if ($result -eq [System.Windows.Forms.DialogResult]::OK) {
    $apiKey = $textbox.Text
} else {
    exit 1
}
$form.Dispose()

if ([string]::IsNullOrWhiteSpace($apiKey)) {
    [System.Windows.Forms.MessageBox]::Show("API Key 为空，退出。", "错误", "OK", "Error")
    exit 1
}

$prompt = @"
你是一个 Windows 桌面整理助手。以下是用户桌面上的内容：

【已有的文件夹】（用户已经建好的，可以往里放文件）
$($folders -join "`n")

【散落的文件】（这些是需要整理的对象）
$($files -join "`n")

任务规则：
1. 针对【散落的文件】给出归类建议。
2. 【优先】把文件归入【已有的文件夹】里。如果某个文件能合理地放进某个已有文件夹，就放进去，不要另建新文件夹。
3. 只有当某个文件确实找不到合适的已有文件夹时，才建议新建文件夹。
4. 新建的文件夹名不要和已有的文件夹重名，也不要和已有文件夹语义重复。
5. 每个建议的文件夹名和文件名，都用中文引号 ""..."" 包裹。
6. 文件名不要带扩展名。
7. 建议的文件夹总数控制在 3-6 个（含已有文件夹和新文件夹）。

输出格式示例：

1. 放入已有文件夹："工具"
   放入："Visual Studio Code"、"Docker Desktop"

2. 建议新建文件夹："浏览器与网络"
   放入："Google Chrome"、"Microsoft Edge"、"WireGuard"

3. 放入已有文件夹："大学"
   放入："选课指南"

每个条目之间必须换行。
只输出建议列表，不要解释。
"@

$body = @{
    model    = $MODEL
    messages = @(
        @{ role = "user"; content = $prompt }
    )
    temperature = 0.3
} | ConvertTo-Json -Depth 10

$uri = "$BASE_URL/chat/completions"

$loading = New-Object System.Windows.Forms.Form
$loading.Text = "请稍候"
$loading.Size = New-Object System.Drawing.Size(320, 120)
$loading.StartPosition = "CenterScreen"
$loading.FormBorderStyle = "FixedDialog"
$loading.MaximizeBox = $false
$loading.MinimizeBox = $true
$loading.ControlBox = $true
$loading.ShowInTaskbar = $true

$loadingLabel = New-Object System.Windows.Forms.Label
$loadingLabel.Text = "正在请求 AI，请稍候..."
$loadingLabel.Location = New-Object System.Drawing.Point(20, 30)
$loadingLabel.Size = New-Object System.Drawing.Size(280, 30)
$loadingLabel.Font = New-Object System.Drawing.Font("Microsoft YaHei", 10)
$loadingLabel.TextAlign = "MiddleCenter"
$loading.Controls.Add($loadingLabel)

$loading.Add_FormClosing({
    param($sender, $e)
    if ($e.CloseReason -eq [System.Windows.Forms.CloseReason]::UserClosing) {
        $e.Cancel = $true
    }
})

$loading.Show()
$loading.Refresh()
$loading.Activate()
[System.Windows.Forms.Application]::DoEvents()

Add-Type -AssemblyName System.Net.Http

$client = New-Object System.Net.Http.HttpClient
$client.Timeout = [TimeSpan]::FromSeconds(120)
$client.DefaultRequestHeaders.Authorization =
    New-Object System.Net.Http.Headers.AuthenticationHeaderValue("Bearer", $apiKey)

$content = New-Object System.Net.Http.StringContent(
    $body,
    [System.Text.Encoding]::UTF8,
    "application/json"
)

$task = $client.PostAsync($uri, $content)

while (-not $task.IsCompleted) {
    [System.Windows.Forms.Application]::DoEvents()
    Start-Sleep -Milliseconds 30
}

$loading.Close()
$loading.Dispose()

$response = $null
try {
    $response = $task.Result
} catch {
    [System.Windows.Forms.MessageBox]::Show("网络请求失败：`n$_", "错误", "OK", "Error")
    exit 1
}

$json = $response.Content.ReadAsStringAsync().Result

if (-not $response.IsSuccessStatusCode) {
    $errMsg = $json
    try {
        $errData = $json | ConvertFrom-Json
        if ($errData.error -and $errData.error.message) {
            $errMsg = $errData.error.message
        }
    } catch { }
    [System.Windows.Forms.MessageBox]::Show("API 返回错误 ($($response.StatusCode))：`n$errMsg", "错误", "OK", "Error")
    exit 1
}

try {
    $data = $json | ConvertFrom-Json
} catch {
    [System.Windows.Forms.MessageBox]::Show("解析响应失败：`n$json", "错误", "OK", "Error")
    exit 1
}

if ($data.error) {
    $errMsg = if ($data.error.message) { $data.error.message } else { $json }
    [System.Windows.Forms.MessageBox]::Show("API 错误：`n$errMsg", "错误", "OK", "Error")
    exit 1
}

if (-not $data.choices -or $data.choices.Count -eq 0) {
    [System.Windows.Forms.MessageBox]::Show("API 返回了空结果：`n$json", "错误", "OK", "Error")
    exit 1
}

$suggestion = $data.choices[0].message.content
$suggestion = $suggestion -replace "`r`n", "`n" -replace "`n", "`r`n"

$utf8NoBom = New-Object System.Text.UTF8Encoding($false)
[System.IO.File]::WriteAllText($outFile, $suggestion, $utf8NoBom)

$resultForm = New-Object System.Windows.Forms.Form
$resultForm.Text = "AI 建议"
$resultForm.Size = New-Object System.Drawing.Size(800, 600)
$resultForm.StartPosition = "CenterScreen"

$resultBox = New-Object System.Windows.Forms.TextBox
$resultBox.Multiline = $true
$resultBox.ScrollBars = "Both"
$resultBox.ReadOnly = $true
$resultBox.WordWrap = $true
$resultBox.Dock = "Fill"
$resultBox.Font = New-Object System.Drawing.Font("Microsoft YaHei", 10)
$resultBox.Text = $suggestion
$resultForm.Controls.Add($resultBox)

$resultForm.ShowDialog() | Out-Null
$resultForm.Dispose()