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,64 @@ DEFAULT_BASE_URL="https://api2.xcodecli.com"
|
||||
CLAUDE_CONFIG_DIR="$HOME/.claude"
|
||||
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
|
||||
print_info() {
|
||||
echo -e "${BLUE}[INFO]${NC} $1"
|
||||
@@ -37,7 +95,7 @@ print_error() {
|
||||
|
||||
# Function to check if jq is installed
|
||||
check_jq() {
|
||||
if ! command -v jq &> /dev/null; then
|
||||
if ! command -v jq &>/dev/null; then
|
||||
print_error "jq is required but not installed."
|
||||
print_info "Please install jq:"
|
||||
print_info " macOS: brew install jq"
|
||||
@@ -134,7 +192,7 @@ test_api_connection() {
|
||||
create_settings() {
|
||||
local base_url="$1"
|
||||
local api_key="$2"
|
||||
|
||||
|
||||
# Create JSON using jq to ensure proper escaping
|
||||
local settings_json
|
||||
settings_json=$(jq -n \
|
||||
@@ -149,25 +207,33 @@ create_settings() {
|
||||
"CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": 1
|
||||
}
|
||||
}')
|
||||
|
||||
|
||||
# Validate JSON
|
||||
if ! echo "$settings_json" | jq . > /dev/null 2>&1; then
|
||||
if ! echo "$settings_json" | jq . >/dev/null 2>&1; then
|
||||
print_error "Generated settings JSON is invalid"
|
||||
return 1
|
||||
fi
|
||||
|
||||
|
||||
# Write settings file
|
||||
echo "$settings_json" > "$CLAUDE_SETTINGS_FILE"
|
||||
echo "$settings_json" >"$CLAUDE_SETTINGS_FILE"
|
||||
print_success "Claude Code settings written to: $CLAUDE_SETTINGS_FILE"
|
||||
|
||||
# Write config.json for VSCode Claude extension
|
||||
local config_json_path="$CLAUDE_CONFIG_DIR/config.json"
|
||||
cat <<EOF > "$config_json_path"
|
||||
cat <<EOF >"$config_json_path"
|
||||
{
|
||||
"primaryApiKey": "xcodecli"
|
||||
}
|
||||
EOF
|
||||
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
|
||||
@@ -187,10 +253,10 @@ main() {
|
||||
print_info "Claude Code Configuration Script for XCodeCLI"
|
||||
echo "======================================================="
|
||||
echo
|
||||
|
||||
|
||||
# Check dependencies
|
||||
check_jq
|
||||
|
||||
|
||||
# Parse command line arguments
|
||||
local api_key=""
|
||||
local test_only=false
|
||||
@@ -203,20 +269,20 @@ main() {
|
||||
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
-k|--key)
|
||||
api_key="$2"
|
||||
shift 2
|
||||
;;
|
||||
-t|--test)
|
||||
test_only=true
|
||||
shift
|
||||
;;
|
||||
-s|--show)
|
||||
show_settings=true
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
cat <<EOF
|
||||
-k | --key)
|
||||
api_key="$2"
|
||||
shift 2
|
||||
;;
|
||||
-t | --test)
|
||||
test_only=true
|
||||
shift
|
||||
;;
|
||||
-s | --show)
|
||||
show_settings=true
|
||||
shift
|
||||
;;
|
||||
-h | --help)
|
||||
cat <<EOF
|
||||
Usage: $0 [OPTIONS]
|
||||
|
||||
This script automatically tests multiple API endpoints and selects the working one:
|
||||
@@ -240,22 +306,22 @@ Interactive mode (no arguments):
|
||||
Environment Variables:
|
||||
API_KEY API key for authentication
|
||||
EOF
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
print_info "Use --help for usage information"
|
||||
exit 1
|
||||
;;
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
print_info "Use --help for usage information"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
|
||||
# Show settings and exit if requested
|
||||
if [ "$show_settings" = true ]; then
|
||||
display_settings
|
||||
exit 0
|
||||
fi
|
||||
|
||||
|
||||
# Interactive mode if no API key provided
|
||||
if [ -z "$api_key" ]; then
|
||||
print_info "Interactive setup mode"
|
||||
@@ -271,19 +337,19 @@ EOF
|
||||
fi
|
||||
done
|
||||
fi
|
||||
|
||||
|
||||
# Validate inputs
|
||||
if [ -z "$api_key" ]; then
|
||||
print_error "API key is required"
|
||||
print_info "Use --help for usage information"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
|
||||
# Validate API key
|
||||
if ! validate_api_key "$api_key"; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
|
||||
print_info "API Key: ${api_key:0:8}...${api_key: -4}"
|
||||
echo
|
||||
|
||||
@@ -309,19 +375,19 @@ EOF
|
||||
else
|
||||
print_info "Selected working base URL: $base_url"
|
||||
fi
|
||||
|
||||
|
||||
# Exit if test only
|
||||
if [ "$test_only" = true ]; then
|
||||
print_success "API test completed successfully"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
|
||||
# Create settings directory
|
||||
create_settings_dir
|
||||
|
||||
|
||||
# Backup existing settings
|
||||
backup_settings
|
||||
|
||||
|
||||
# Create new settings
|
||||
if create_settings "$base_url" "$api_key"; then
|
||||
echo
|
||||
@@ -332,7 +398,7 @@ EOF
|
||||
print_info " claude --version"
|
||||
print_info ""
|
||||
print_info "Configuration file location: $CLAUDE_SETTINGS_FILE"
|
||||
|
||||
|
||||
if [ -f "$CLAUDE_SETTINGS_FILE" ]; then
|
||||
echo
|
||||
print_info "Current settings:"
|
||||
|
||||
Reference in New Issue
Block a user