feat: 添加环境变量写入功能
- Claude Code: 写入 ANTHROPIC_BASE_URL, ANTHROPIC_AUTH_TOKEN - Codex: 写入 OPENAI_API_KEY, OPENAI_BASE_URL - Bash 脚本: 自动检测 shell 配置文件 (.bashrc/.zshrc/.profile) - PowerShell 脚本: 使用 [Environment]::SetEnvironmentVariable 设置用户环境变量 - 添加 API key 格式验证 - 修复文件末尾换行问题,使用单引号包裹值更安全 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -18,6 +18,19 @@ $DefaultBaseUrl = "https://api2.xcodecli.com"
|
|||||||
$ClaudeConfigDir = "$env:USERPROFILE\.claude"
|
$ClaudeConfigDir = "$env:USERPROFILE\.claude"
|
||||||
$ClaudeSettingsFile = "$ClaudeConfigDir\settings.json"
|
$ClaudeSettingsFile = "$ClaudeConfigDir\settings.json"
|
||||||
|
|
||||||
|
# Function to write environment variable (User level)
|
||||||
|
function Set-EnvVariable {
|
||||||
|
param(
|
||||||
|
[string]$Name,
|
||||||
|
[string]$Value
|
||||||
|
)
|
||||||
|
# Set for current session
|
||||||
|
[Environment]::SetEnvironmentVariable($Name, $Value, [System.EnvironmentVariableTarget]::Process)
|
||||||
|
# Set permanently for user
|
||||||
|
[Environment]::SetEnvironmentVariable($Name, $Value, [System.EnvironmentVariableTarget]::User)
|
||||||
|
Write-Info "Environment variable set: $Name"
|
||||||
|
}
|
||||||
|
|
||||||
# Color functions for output
|
# Color functions for output
|
||||||
function Write-Info {
|
function Write-Info {
|
||||||
param([string]$Message)
|
param([string]$Message)
|
||||||
@@ -186,6 +199,13 @@ function New-Settings {
|
|||||||
$configJson = @{ primaryApiKey = "xcodecli" }
|
$configJson = @{ primaryApiKey = "xcodecli" }
|
||||||
$configJson | ConvertTo-Json | Set-Content -Path $configJsonPath -Encoding UTF8
|
$configJson | ConvertTo-Json | Set-Content -Path $configJsonPath -Encoding UTF8
|
||||||
Write-Success "VSCode Claude config written to: $configJsonPath"
|
Write-Success "VSCode Claude config written to: $configJsonPath"
|
||||||
|
|
||||||
|
# Set environment variables
|
||||||
|
Write-Info "Setting environment variables..."
|
||||||
|
Set-EnvVariable -Name "ANTHROPIC_BASE_URL" -Value $BaseUrl
|
||||||
|
Set-EnvVariable -Name "ANTHROPIC_AUTH_TOKEN" -Value $ApiKey
|
||||||
|
Write-Success "Environment variables configured"
|
||||||
|
|
||||||
return $true
|
return $true
|
||||||
}
|
}
|
||||||
catch {
|
catch {
|
||||||
|
|||||||
@@ -18,6 +18,64 @@ DEFAULT_BASE_URL="https://api2.xcodecli.com"
|
|||||||
CLAUDE_CONFIG_DIR="$HOME/.claude"
|
CLAUDE_CONFIG_DIR="$HOME/.claude"
|
||||||
CLAUDE_SETTINGS_FILE="$CLAUDE_CONFIG_DIR/settings.json"
|
CLAUDE_SETTINGS_FILE="$CLAUDE_CONFIG_DIR/settings.json"
|
||||||
|
|
||||||
|
# ========== Shell 环境变量配置 ==========
|
||||||
|
# 检测当前 shell 配置文件
|
||||||
|
get_shell_rc() {
|
||||||
|
if [ -n "${ZSH_VERSION:-}" ] || [ "${SHELL##*/}" = "zsh" ]; then
|
||||||
|
echo "$HOME/.zshrc"
|
||||||
|
elif [ -n "${BASH_VERSION:-}" ] || [ "${SHELL##*/}" = "bash" ]; then
|
||||||
|
if [ -f "$HOME/.bashrc" ]; then
|
||||||
|
echo "$HOME/.bashrc"
|
||||||
|
else
|
||||||
|
echo "$HOME/.bash_profile"
|
||||||
|
fi
|
||||||
|
elif [ "${SHELL##*/}" = "fish" ]; then
|
||||||
|
echo "$HOME/.config/fish/config.fish"
|
||||||
|
else
|
||||||
|
echo "$HOME/.profile"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 写入环境变量到 shell 配置文件
|
||||||
|
write_env_to_shell() {
|
||||||
|
local var_name="$1"
|
||||||
|
local var_value="$2"
|
||||||
|
local rc_file
|
||||||
|
rc_file=$(get_shell_rc)
|
||||||
|
|
||||||
|
# 确保配置文件存在
|
||||||
|
mkdir -p "$(dirname "$rc_file")"
|
||||||
|
touch "$rc_file"
|
||||||
|
|
||||||
|
# 确保文件末尾有换行
|
||||||
|
if [ -s "$rc_file" ] && [ "$(tail -c1 "$rc_file" | wc -l)" -eq 0 ]; then
|
||||||
|
echo "" >>"$rc_file"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 转义特殊字符(使用单引号包裹更安全)
|
||||||
|
local export_line
|
||||||
|
if [ "${SHELL##*/}" = "fish" ]; then
|
||||||
|
export_line="set -Ux $var_name '$var_value'"
|
||||||
|
else
|
||||||
|
export_line="export $var_name='$var_value'"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 删除旧的同名变量行,添加新行
|
||||||
|
local tmp_file
|
||||||
|
tmp_file=$(mktemp)
|
||||||
|
if [ -s "$rc_file" ]; then
|
||||||
|
grep -v "^export $var_name=" "$rc_file" | grep -v "^set -Ux $var_name " >"$tmp_file" 2>/dev/null || true
|
||||||
|
fi
|
||||||
|
echo "$export_line" >>"$tmp_file"
|
||||||
|
|
||||||
|
# 保留原文件权限
|
||||||
|
cat "$tmp_file" >"$rc_file"
|
||||||
|
rm -f "$tmp_file"
|
||||||
|
|
||||||
|
# 立即生效
|
||||||
|
export "$var_name=$var_value"
|
||||||
|
}
|
||||||
|
|
||||||
# Function to print colored output
|
# Function to print colored output
|
||||||
print_info() {
|
print_info() {
|
||||||
echo -e "${BLUE}[INFO]${NC} $1"
|
echo -e "${BLUE}[INFO]${NC} $1"
|
||||||
@@ -168,6 +226,14 @@ create_settings() {
|
|||||||
}
|
}
|
||||||
EOF
|
EOF
|
||||||
print_success "VSCode Claude config written to: $config_json_path"
|
print_success "VSCode Claude config written to: $config_json_path"
|
||||||
|
|
||||||
|
# Write environment variables to shell config
|
||||||
|
local rc_file
|
||||||
|
rc_file=$(get_shell_rc)
|
||||||
|
print_info "Writing environment variables to: $rc_file"
|
||||||
|
write_env_to_shell "ANTHROPIC_BASE_URL" "$base_url"
|
||||||
|
write_env_to_shell "ANTHROPIC_AUTH_TOKEN" "$api_key"
|
||||||
|
print_success "Environment variables written to shell config"
|
||||||
}
|
}
|
||||||
|
|
||||||
# Function to display current settings
|
# Function to display current settings
|
||||||
|
|||||||
@@ -18,6 +18,19 @@ $DefaultBaseUrl = "https://api2.xcodecli.com"
|
|||||||
$CodexConfigDir = "$env:USERPROFILE\.codex"
|
$CodexConfigDir = "$env:USERPROFILE\.codex"
|
||||||
$CodexConfigFile = "$CodexConfigDir\config.toml"
|
$CodexConfigFile = "$CodexConfigDir\config.toml"
|
||||||
|
|
||||||
|
# Function to write environment variable (User level)
|
||||||
|
function Set-EnvVariable {
|
||||||
|
param(
|
||||||
|
[string]$Name,
|
||||||
|
[string]$Value
|
||||||
|
)
|
||||||
|
# Set for current session
|
||||||
|
[Environment]::SetEnvironmentVariable($Name, $Value, [System.EnvironmentVariableTarget]::Process)
|
||||||
|
# Set permanently for user
|
||||||
|
[Environment]::SetEnvironmentVariable($Name, $Value, [System.EnvironmentVariableTarget]::User)
|
||||||
|
Write-Info "Environment variable set: $Name"
|
||||||
|
}
|
||||||
|
|
||||||
# Color functions for output
|
# Color functions for output
|
||||||
function Write-Info {
|
function Write-Info {
|
||||||
param([string]$Message)
|
param([string]$Message)
|
||||||
@@ -192,6 +205,13 @@ env_key = "OPENAI_API_KEY"
|
|||||||
$authJsonPath = "$CodexConfigDir\auth.json"
|
$authJsonPath = "$CodexConfigDir\auth.json"
|
||||||
$authJson | ConvertTo-Json | Set-Content -Path $authJsonPath -Encoding UTF8
|
$authJson | ConvertTo-Json | Set-Content -Path $authJsonPath -Encoding UTF8
|
||||||
Write-Success "Codex auth file written to: $authJsonPath"
|
Write-Success "Codex auth file written to: $authJsonPath"
|
||||||
|
|
||||||
|
# Set environment variables
|
||||||
|
Write-Info "Setting environment variables..."
|
||||||
|
Set-EnvVariable -Name "OPENAI_API_KEY" -Value $ApiKey
|
||||||
|
Set-EnvVariable -Name "OPENAI_BASE_URL" -Value "$BaseUrl/v1"
|
||||||
|
Write-Success "Environment variables configured"
|
||||||
|
|
||||||
return $true
|
return $true
|
||||||
}
|
}
|
||||||
catch {
|
catch {
|
||||||
|
|||||||
@@ -28,6 +28,74 @@ BASE_URL=""
|
|||||||
TEST_ONLY=false
|
TEST_ONLY=false
|
||||||
SHOW_SETTINGS=false
|
SHOW_SETTINGS=false
|
||||||
|
|
||||||
|
# ========== Shell 环境变量配置 ==========
|
||||||
|
# 检测当前 shell 配置文件
|
||||||
|
get_shell_rc() {
|
||||||
|
if [ -n "${ZSH_VERSION:-}" ] || [ "${SHELL##*/}" = "zsh" ]; then
|
||||||
|
echo "$HOME/.zshrc"
|
||||||
|
elif [ -n "${BASH_VERSION:-}" ] || [ "${SHELL##*/}" = "bash" ]; then
|
||||||
|
if [ -f "$HOME/.bashrc" ]; then
|
||||||
|
echo "$HOME/.bashrc"
|
||||||
|
else
|
||||||
|
echo "$HOME/.bash_profile"
|
||||||
|
fi
|
||||||
|
elif [ "${SHELL##*/}" = "fish" ]; then
|
||||||
|
echo "$HOME/.config/fish/config.fish"
|
||||||
|
else
|
||||||
|
echo "$HOME/.profile"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 写入环境变量到 shell 配置文件
|
||||||
|
write_env_to_shell() {
|
||||||
|
local var_name="$1"
|
||||||
|
local var_value="$2"
|
||||||
|
local rc_file
|
||||||
|
rc_file=$(get_shell_rc)
|
||||||
|
|
||||||
|
# 确保配置文件存在
|
||||||
|
mkdir -p "$(dirname "$rc_file")"
|
||||||
|
touch "$rc_file"
|
||||||
|
|
||||||
|
# 确保文件末尾有换行
|
||||||
|
if [ -s "$rc_file" ] && [ "$(tail -c1 "$rc_file" | wc -l)" -eq 0 ]; then
|
||||||
|
echo "" >> "$rc_file"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 转义特殊字符(使用单引号包裹更安全)
|
||||||
|
local export_line
|
||||||
|
if [ "${SHELL##*/}" = "fish" ]; then
|
||||||
|
export_line="set -Ux $var_name '$var_value'"
|
||||||
|
else
|
||||||
|
export_line="export $var_name='$var_value'"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 删除旧的同名变量行,添加新行
|
||||||
|
local tmp_file
|
||||||
|
tmp_file=$(mktemp)
|
||||||
|
if [ -s "$rc_file" ]; then
|
||||||
|
grep -v "^export $var_name=" "$rc_file" | grep -v "^set -Ux $var_name " > "$tmp_file" 2>/dev/null || true
|
||||||
|
fi
|
||||||
|
echo "$export_line" >> "$tmp_file"
|
||||||
|
|
||||||
|
# 保留原文件权限
|
||||||
|
cat "$tmp_file" > "$rc_file"
|
||||||
|
rm -f "$tmp_file"
|
||||||
|
|
||||||
|
# 立即生效
|
||||||
|
export "$var_name=$var_value"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Function to validate API key format
|
||||||
|
validate_api_key() {
|
||||||
|
local api_key="$1"
|
||||||
|
if [[ ! "$api_key" =~ ^[A-Za-z0-9_-]+$ ]]; then
|
||||||
|
print_error "Invalid API key format. API key should contain only alphanumeric characters, hyphens, and underscores."
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
# Check for environment variable API_KEY
|
# Check for environment variable API_KEY
|
||||||
if [ -n "$API_KEY" ]; then
|
if [ -n "$API_KEY" ]; then
|
||||||
API_KEY_FROM_ENV="$API_KEY"
|
API_KEY_FROM_ENV="$API_KEY"
|
||||||
@@ -193,6 +261,15 @@ EOF
|
|||||||
|
|
||||||
print_success "Codex configuration written to: $HOME/.codex/config.toml"
|
print_success "Codex configuration written to: $HOME/.codex/config.toml"
|
||||||
print_success "Codex auth file written to: $HOME/.codex/auth.json"
|
print_success "Codex auth file written to: $HOME/.codex/auth.json"
|
||||||
|
|
||||||
|
# Write environment variables to shell config
|
||||||
|
local rc_file
|
||||||
|
rc_file=$(get_shell_rc)
|
||||||
|
print_info "Writing environment variables to: $rc_file"
|
||||||
|
write_env_to_shell "OPENAI_API_KEY" "$api_key"
|
||||||
|
write_env_to_shell "OPENAI_BASE_URL" "${base_url}/v1"
|
||||||
|
print_success "Environment variables written to shell config"
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -247,6 +324,8 @@ main() {
|
|||||||
read -p "Enter your API key: " API_KEY
|
read -p "Enter your API key: " API_KEY
|
||||||
if [ -z "$API_KEY" ]; then
|
if [ -z "$API_KEY" ]; then
|
||||||
print_warning "API key is required"
|
print_warning "API key is required"
|
||||||
|
elif ! validate_api_key "$API_KEY"; then
|
||||||
|
API_KEY=""
|
||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
fi
|
fi
|
||||||
@@ -258,6 +337,11 @@ main() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Validate API key format
|
||||||
|
if ! validate_api_key "$API_KEY"; then
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
# Mask API key for display
|
# Mask API key for display
|
||||||
if [ ${#API_KEY} -gt 12 ]; then
|
if [ ${#API_KEY} -gt 12 ]; then
|
||||||
masked_key="${API_KEY:0:8}...${API_KEY: -4}"
|
masked_key="${API_KEY:0:8}...${API_KEY: -4}"
|
||||||
|
|||||||
Reference in New Issue
Block a user