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>

codex/setup-codex.sh

@@ -28,6 +28,74 @@ BASE_URL=""
 TEST_ONLY=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
 if [ -n "$API_KEY" ]; then
     API_KEY_FROM_ENV="$API_KEY"
@@ -193,6 +261,15 @@ EOF
 
     print_success "Codex configuration written to: $HOME/.codex/config.toml"
     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
 }
 
@@ -247,6 +324,8 @@ main() {
             read -p "Enter your API key: " API_KEY
             if [ -z "$API_KEY" ]; then
                 print_warning "API key is required"
+            elif ! validate_api_key "$API_KEY"; then
+                API_KEY=""
             fi
         done
     fi
@@ -258,6 +337,11 @@ main() {
         exit 1
     fi
 
+    # Validate API key format
+    if ! validate_api_key "$API_KEY"; then
+        exit 1
+    fi
+
     # Mask API key for display
     if [ ${#API_KEY} -gt 12 ]; then
         masked_key="${API_KEY:0:8}...${API_KEY: -4}"