forked from whudqw/pinai-imagegen
96 lines
3.2 KiB
PowerShell
96 lines
3.2 KiB
PowerShell
param(
|
|
[switch]$ValidateOnly
|
|
)
|
|
|
|
Add-Type -AssemblyName System.Windows.Forms
|
|
Add-Type -AssemblyName System.Drawing
|
|
|
|
if ($ValidateOnly) {
|
|
exit 0
|
|
}
|
|
|
|
$form = New-Object System.Windows.Forms.Form
|
|
$form.Text = '设置 PinAI Key'
|
|
$form.Size = New-Object System.Drawing.Size(540, 250)
|
|
$form.StartPosition = 'CenterScreen'
|
|
$form.FormBorderStyle = 'FixedDialog'
|
|
$form.MaximizeBox = $false
|
|
$form.MinimizeBox = $false
|
|
$form.ShowInTaskbar = $true
|
|
|
|
$title = New-Object System.Windows.Forms.Label
|
|
$title.Text = '粘贴你的 PinAI API Key'
|
|
$title.Font = New-Object System.Drawing.Font('Microsoft YaHei UI', 12, [System.Drawing.FontStyle]::Bold)
|
|
$title.AutoSize = $true
|
|
$title.Location = New-Object System.Drawing.Point(28, 24)
|
|
$form.Controls.Add($title)
|
|
|
|
$hint = New-Object System.Windows.Forms.Label
|
|
$hint.Text = 'Key 会以圆点隐藏，并只保存到当前 Windows 用户环境变量。'
|
|
$hint.AutoSize = $true
|
|
$hint.Location = New-Object System.Drawing.Point(30, 58)
|
|
$form.Controls.Add($hint)
|
|
|
|
$keyBox = New-Object System.Windows.Forms.TextBox
|
|
$keyBox.Location = New-Object System.Drawing.Point(30, 90)
|
|
$keyBox.Size = New-Object System.Drawing.Size(465, 30)
|
|
$keyBox.UseSystemPasswordChar = $true
|
|
$keyBox.Font = New-Object System.Drawing.Font('Segoe UI', 11)
|
|
$form.Controls.Add($keyBox)
|
|
|
|
$saveButton = New-Object System.Windows.Forms.Button
|
|
$saveButton.Text = '保存 Key'
|
|
$saveButton.Size = New-Object System.Drawing.Size(105, 36)
|
|
$saveButton.Location = New-Object System.Drawing.Point(275, 145)
|
|
$saveButton.DialogResult = [System.Windows.Forms.DialogResult]::None
|
|
$form.Controls.Add($saveButton)
|
|
|
|
$cancelButton = New-Object System.Windows.Forms.Button
|
|
$cancelButton.Text = '取消'
|
|
$cancelButton.Size = New-Object System.Drawing.Size(105, 36)
|
|
$cancelButton.Location = New-Object System.Drawing.Point(390, 145)
|
|
$cancelButton.DialogResult = [System.Windows.Forms.DialogResult]::Cancel
|
|
$form.Controls.Add($cancelButton)
|
|
|
|
$form.AcceptButton = $saveButton
|
|
$form.CancelButton = $cancelButton
|
|
$form.Add_Shown({ $keyBox.Focus() })
|
|
|
|
$saveButton.Add_Click({
|
|
$key = $keyBox.Text.Trim()
|
|
if ([string]::IsNullOrWhiteSpace($key)) {
|
|
[System.Windows.Forms.MessageBox]::Show(
|
|
'请先粘贴 PinAI API Key。',
|
|
'未填写 Key',
|
|
[System.Windows.Forms.MessageBoxButtons]::OK,
|
|
[System.Windows.Forms.MessageBoxIcon]::Warning
|
|
) | Out-Null
|
|
return
|
|
}
|
|
|
|
try {
|
|
[Environment]::SetEnvironmentVariable('PINAI_API_KEY', $key, 'User')
|
|
[Environment]::SetEnvironmentVariable('PINAI_API_KEY', $key, 'Process')
|
|
}
|
|
catch {
|
|
[System.Windows.Forms.MessageBox]::Show(
|
|
'保存失败。请确认当前 Windows 用户有写入环境变量的权限。',
|
|
'保存失败',
|
|
[System.Windows.Forms.MessageBoxButtons]::OK,
|
|
[System.Windows.Forms.MessageBoxIcon]::Error
|
|
) | Out-Null
|
|
return
|
|
}
|
|
|
|
[System.Windows.Forms.MessageBox]::Show(
|
|
'PinAI Key 已保存。请完全退出并重新打开 Codex，然后即可生图。',
|
|
'设置完成',
|
|
[System.Windows.Forms.MessageBoxButtons]::OK,
|
|
[System.Windows.Forms.MessageBoxIcon]::Information
|
|
) | Out-Null
|
|
$form.Close()
|
|
})
|
|
|
|
[void]$form.ShowDialog()
|
|
|