mirror of
https://github.com/QwenLM/qwen-code.git
synced 2026-04-28 11:41:04 +00:00
557 lines
16 KiB
Bash
Executable file
557 lines
16 KiB
Bash
Executable file
#!/bin/bash
|
||
|
||
# Qwen Code Installation Script
|
||
# This script installs Node.js (via NVM) and Qwen Code CLI
|
||
# Supports Linux and macOS
|
||
#
|
||
# Usage: install-qwen-with-source.sh --source [github|npm|internal|local-build]
|
||
# install-qwen-with-source.sh -s [github|npm|internal|local-build]
|
||
|
||
# Re-execute with bash if running with sh or other shells
|
||
# This block must use POSIX-compliant syntax ([ not [[) since it runs before we know bash is available
|
||
if [ -z "${BASH_VERSION}" ] && [ -z "${__QWEN_INSTALL_REEXEC:-}" ]; then
|
||
# Check if we're in a git hook environment
|
||
case "${0}" in
|
||
*.git/hooks/*) export __QWEN_IN_GIT_HOOK=1 ;;
|
||
esac
|
||
if [ -n "${GIT_DIR:-}" ]; then
|
||
export __QWEN_IN_GIT_HOOK=1
|
||
fi
|
||
|
||
# Try to find bash
|
||
if command -v bash >/dev/null 2>&1; then
|
||
export __QWEN_INSTALL_REEXEC=1
|
||
# Re-exec with bash, preserving all arguments
|
||
exec bash -- "${0}" "$@"
|
||
else
|
||
echo "Error: This script requires bash. Please install bash first."
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# Enable strict mode (bash-specific options)
|
||
# pipefail requires bash 3+; check before setting
|
||
if [ -n "${BASH_VERSION:-}" ]; then
|
||
# shellcheck disable=SC3040
|
||
set -eo pipefail
|
||
else
|
||
set -e
|
||
fi
|
||
|
||
# ============================================
|
||
# Color definitions
|
||
# ============================================
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[0;34m'
|
||
NC='\033[0m' # No Color
|
||
|
||
# ============================================
|
||
# Log functions
|
||
# ============================================
|
||
log_info() {
|
||
echo -e "${BLUE}ℹ️ $1${NC}"
|
||
}
|
||
|
||
log_success() {
|
||
echo -e "${GREEN}✅ $1${NC}"
|
||
}
|
||
|
||
log_warning() {
|
||
echo -e "${YELLOW}⚠️ $1${NC}"
|
||
}
|
||
|
||
log_error() {
|
||
echo -e "${RED}❌ $1${NC}"
|
||
}
|
||
|
||
# ============================================
|
||
# Utility functions
|
||
# ============================================
|
||
command_exists() {
|
||
command -v "$1" >/dev/null 2>&1
|
||
}
|
||
|
||
get_shell_profile() {
|
||
local current_shell
|
||
current_shell=$(basename "${SHELL}")
|
||
case "${current_shell}" in
|
||
bash)
|
||
echo "${HOME}/.bashrc"
|
||
;;
|
||
zsh)
|
||
echo "${HOME}/.zshrc"
|
||
;;
|
||
fish)
|
||
echo "${HOME}/.config/fish/config.fish"
|
||
;;
|
||
*)
|
||
echo "${HOME}/.profile"
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# ============================================
|
||
# Parse command line arguments
|
||
# ============================================
|
||
SOURCE="unknown"
|
||
while [[ $# -gt 0 ]]; do
|
||
case $1 in
|
||
-s|--source)
|
||
if [[ -z "$2" ]] || [[ "$2" == -* ]]; then
|
||
log_error "--source requires a value"
|
||
exit 1
|
||
fi
|
||
SOURCE="$2"
|
||
shift 2
|
||
;;
|
||
-h|--help)
|
||
echo "Usage: $0 [OPTIONS]"
|
||
echo ""
|
||
echo "Options:"
|
||
echo " -s, --source SOURCE Specify the installation source (e.g., github, npm, internal)"
|
||
echo " -h, --help Show this help message"
|
||
echo ""
|
||
exit 0
|
||
;;
|
||
*)
|
||
log_error "Unknown option: $1"
|
||
exit 1
|
||
;;
|
||
esac
|
||
done
|
||
|
||
# ============================================
|
||
# Print header
|
||
# ============================================
|
||
echo "=========================================="
|
||
echo " Qwen Code Installation Script"
|
||
echo "=========================================="
|
||
echo ""
|
||
log_info "System: $(uname -s) $(uname -r)" || true
|
||
log_info "Shell: $(basename "${SHELL}")"
|
||
echo ""
|
||
|
||
# ============================================
|
||
# Ensure download tool is available
|
||
# ============================================
|
||
ensure_download_tool() {
|
||
if command_exists curl; then
|
||
DOWNLOAD_CMD="curl"
|
||
DOWNLOAD_ARGS="-fsSL"
|
||
return 0
|
||
fi
|
||
|
||
if command_exists wget; then
|
||
DOWNLOAD_CMD="wget"
|
||
DOWNLOAD_ARGS="-qO -"
|
||
return 0
|
||
fi
|
||
|
||
log_error "Neither curl nor wget found"
|
||
log_info "Please install curl or wget manually:"
|
||
echo " - macOS: brew install curl"
|
||
echo " - Ubuntu/Debian: sudo apt-get install curl"
|
||
echo " - CentOS/RHEL: sudo yum install curl"
|
||
exit 1
|
||
}
|
||
|
||
# ============================================
|
||
# Clean npm configuration conflicts
|
||
# ============================================
|
||
clean_npmrc_conflict() {
|
||
local npmrc="${HOME}/.npmrc"
|
||
if [[ -f "${npmrc}" ]]; then
|
||
log_info "Cleaning npmrc conflicts..."
|
||
grep -Ev '^(prefix|globalconfig) *= *' "${npmrc}" > "${npmrc}.tmp" || true
|
||
mv -f "${npmrc}.tmp" "${npmrc}" || true
|
||
fi
|
||
}
|
||
|
||
# ============================================
|
||
# Install NVM
|
||
# ============================================
|
||
install_nvm() {
|
||
local NVM_DIR="${NVM_DIR:-${HOME}/.nvm}"
|
||
local NVM_VERSION="${NVM_VERSION:-v0.40.3}"
|
||
|
||
if [[ -s "${NVM_DIR}/nvm.sh" ]]; then
|
||
log_info "NVM is already installed at ${NVM_DIR}"
|
||
return 0
|
||
fi
|
||
|
||
log_info "Installing NVM ${NVM_VERSION}..."
|
||
|
||
# Download and install NVM from Aliyun OSS
|
||
# Use temporary file instead of pipe to avoid potential subshell issues
|
||
local NVM_INSTALL_TEMP
|
||
NVM_INSTALL_TEMP=$(mktemp)
|
||
if "${DOWNLOAD_CMD}" "${DOWNLOAD_ARGS}" "https://qwen-code-assets.oss-cn-hangzhou.aliyuncs.com/installation/install_nvm.sh" > "${NVM_INSTALL_TEMP}"; then
|
||
# Run the script in current shell environment
|
||
# shellcheck source=/dev/null
|
||
. "${NVM_INSTALL_TEMP}"
|
||
rm -f "${NVM_INSTALL_TEMP}"
|
||
log_success "NVM installed successfully"
|
||
else
|
||
rm -f "${NVM_INSTALL_TEMP}"
|
||
log_error "Failed to install NVM"
|
||
log_info "Please install NVM manually: https://github.com/nvm-sh/nvm#install--update-script"
|
||
exit 1
|
||
fi
|
||
|
||
# Configure shell profile
|
||
local PROFILE_FILE
|
||
PROFILE_FILE=$(get_shell_profile)
|
||
|
||
# Check if profile file is writable
|
||
if [[ -f "${PROFILE_FILE}" ]] && [[ ! -w "${PROFILE_FILE}" ]]; then
|
||
log_warning "Cannot write to ${PROFILE_FILE} (permission denied)"
|
||
log_info "Skipping shell profile configuration"
|
||
log_info "You may need to manually add NVM configuration to your shell profile"
|
||
elif ! grep -q 'NVM_DIR' "${PROFILE_FILE}" 2>/dev/null; then
|
||
# shellcheck disable=SC2016
|
||
# The following echo statements intentionally use single quotes to write literal strings
|
||
{
|
||
echo ""
|
||
echo "# NVM configuration (added by Qwen Code installer)"
|
||
echo "export NVM_DIR=\"\$HOME/.nvm\""
|
||
echo '[ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"'
|
||
echo '[ -s "$NVM_DIR/bash_completion" ] && \. "$NVM_DIR/bash_completion"'
|
||
} >> "${PROFILE_FILE}" 2>/dev/null || {
|
||
log_warning "Failed to write to ${PROFILE_FILE}"
|
||
log_info "Skipping shell profile configuration"
|
||
return 0
|
||
}
|
||
log_info "Added NVM config to ${PROFILE_FILE}"
|
||
fi
|
||
|
||
# Load NVM for current session
|
||
export NVM_DIR="${NVM_DIR}"
|
||
# shellcheck source=/dev/null
|
||
[[ -s "${NVM_DIR}/nvm.sh" ]] && \. "${NVM_DIR}/nvm.sh"
|
||
|
||
log_success "NVM configured successfully"
|
||
return 0
|
||
}
|
||
|
||
# ============================================
|
||
# Install Node.js via NVM
|
||
# ============================================
|
||
install_nodejs_with_nvm() {
|
||
local NODE_VERSION="${NODE_VERSION:-20}"
|
||
local NVM_DIR="${NVM_DIR:-${HOME}/.nvm}"
|
||
|
||
# Ensure NVM is loaded
|
||
export NVM_DIR="${NVM_DIR}"
|
||
# shellcheck source=/dev/null
|
||
[[ -s "${NVM_DIR}/nvm.sh" ]] && \. "${NVM_DIR}/nvm.sh"
|
||
|
||
if ! command_exists nvm; then
|
||
log_error "NVM not loaded properly"
|
||
return 1
|
||
fi
|
||
|
||
# Set Node.js mirror source for faster downloads in China
|
||
export NVM_NODEJS_ORG_MIRROR="https://npmmirror.com/mirrors/node"
|
||
|
||
# Install Node.js
|
||
log_info "Installing Node.js v${NODE_VERSION}..."
|
||
if nvm install "${NODE_VERSION}"; then
|
||
nvm alias default "${NODE_VERSION}" || true
|
||
nvm use default || true
|
||
log_success "Node.js v${NODE_VERSION} installed successfully"
|
||
|
||
# Verify installation
|
||
log_info "Node.js version: $(node -v)" || true
|
||
log_info "npm version: $(npm -v)" || true
|
||
|
||
return 0
|
||
else
|
||
log_error "Failed to install Node.js"
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
# ============================================
|
||
# Check Node.js version
|
||
# ============================================
|
||
check_node_version() {
|
||
if ! command_exists node; then
|
||
return 1
|
||
fi
|
||
|
||
local current_version
|
||
current_version=$(node -v | sed 's/v//')
|
||
local major_version
|
||
major_version=$(echo "${current_version}" | cut -d. -f1)
|
||
|
||
if [[ "${major_version}" -ge 20 ]]; then
|
||
log_success "Node.js v${current_version} is already installed (>= 20)"
|
||
return 0
|
||
else
|
||
log_warning "Node.js v${current_version} is installed but version < 20"
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
# ============================================
|
||
# Install Node.js
|
||
# ============================================
|
||
install_nodejs() {
|
||
local platform
|
||
platform=$(uname -s)
|
||
|
||
case "${platform}" in
|
||
Linux|Darwin)
|
||
log_info "Installing Node.js on ${platform}..."
|
||
|
||
# Install NVM
|
||
if ! install_nvm; then
|
||
log_error "Failed to install NVM"
|
||
return 1
|
||
fi
|
||
|
||
# Load NVM
|
||
export NVM_DIR="${HOME}/.nvm"
|
||
# shellcheck source=/dev/null
|
||
[[ -s "${NVM_DIR}/nvm.sh" ]] && \. "${NVM_DIR}/nvm.sh"
|
||
|
||
# Install Node.js
|
||
if ! install_nodejs_with_nvm; then
|
||
log_error "Failed to install Node.js"
|
||
return 1
|
||
fi
|
||
;;
|
||
MINGW*|CYGWIN*|MSYS*)
|
||
log_error "Windows platform detected. Please use Windows installer or WSL."
|
||
log_info "Visit: https://nodejs.org/en/download/"
|
||
exit 1
|
||
;;
|
||
*)
|
||
log_error "Unsupported platform: ${platform}"
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# ============================================
|
||
# Check and install Node.js
|
||
# ============================================
|
||
check_and_install_nodejs() {
|
||
if check_node_version; then
|
||
log_info "Using existing Node.js installation"
|
||
clean_npmrc_conflict
|
||
else
|
||
log_warning "Installing or upgrading Node.js..."
|
||
install_nodejs
|
||
fi
|
||
}
|
||
|
||
# ============================================
|
||
# Fix npm permissions (without using sudo)
|
||
# ============================================
|
||
fix_npm_permissions() {
|
||
log_info "Checking npm permissions..."
|
||
|
||
local NPM_GLOBAL_DIR
|
||
NPM_GLOBAL_DIR=$(npm config get prefix 2>/dev/null) || true
|
||
if [[ -z "${NPM_GLOBAL_DIR}" ]] || [[ "${NPM_GLOBAL_DIR}" == *"error"* ]]; then
|
||
NPM_GLOBAL_DIR="${HOME}/.npm-global"
|
||
npm config set prefix "${NPM_GLOBAL_DIR}"
|
||
log_info "Set npm prefix to user directory: ${NPM_GLOBAL_DIR}"
|
||
return 0
|
||
fi
|
||
|
||
# SAFETY CHECK: Never modify system directories
|
||
# This prevents catastrophic failures like breaking sudo setuid binaries
|
||
case "${NPM_GLOBAL_DIR}" in
|
||
/|/usr|/usr/local|/bin|/sbin|/lib|/lib64|/opt|/snap|/var|/etc)
|
||
log_warning "npm prefix is a system directory (${NPM_GLOBAL_DIR})."
|
||
log_info "Using user directory instead to avoid breaking system binaries."
|
||
NPM_GLOBAL_DIR="${HOME}/.npm-global"
|
||
npm config set prefix "${NPM_GLOBAL_DIR}"
|
||
log_success "npm prefix set to: ${NPM_GLOBAL_DIR}"
|
||
return 0
|
||
;;
|
||
*)
|
||
# Safe to proceed with non-system directory
|
||
;;
|
||
esac
|
||
|
||
# Check if npm global directory is writable
|
||
if [[ -w "${NPM_GLOBAL_DIR}" ]]; then
|
||
log_info "npm global directory is writable"
|
||
return 0
|
||
fi
|
||
|
||
# If not writable, use user directory
|
||
log_warning "npm global directory is not writable: ${NPM_GLOBAL_DIR}"
|
||
log_info "Setting npm prefix to user directory..."
|
||
|
||
NPM_GLOBAL_DIR="${HOME}/.npm-global"
|
||
mkdir -p "${NPM_GLOBAL_DIR}"
|
||
npm config set prefix "${NPM_GLOBAL_DIR}"
|
||
|
||
log_success "npm prefix set to: ${NPM_GLOBAL_DIR}"
|
||
|
||
# Add to PATH in shell profile
|
||
local PROFILE_FILE
|
||
PROFILE_FILE=$(get_shell_profile)
|
||
if ! grep -q '.npm-global/bin' "${PROFILE_FILE}" 2>/dev/null; then
|
||
{
|
||
echo ""
|
||
echo "# NPM global bin (added by Qwen Code installer)"
|
||
echo "export PATH=\"\$HOME/.npm-global/bin:\$PATH\""
|
||
} >> "${PROFILE_FILE}"
|
||
log_info "Added npm global bin to PATH in ${PROFILE_FILE}"
|
||
fi
|
||
|
||
return 0
|
||
}
|
||
|
||
# ============================================
|
||
# Install Qwen Code
|
||
# ============================================
|
||
install_qwen_code() {
|
||
# Ensure NVM node is in PATH
|
||
export NVM_DIR="${HOME}/.nvm"
|
||
# shellcheck source=/dev/null
|
||
[[ -s "${NVM_DIR}/nvm.sh" ]] && \. "${NVM_DIR}/nvm.sh" 2>/dev/null || true
|
||
|
||
# Add npm global bin to PATH
|
||
local NPM_GLOBAL_BIN
|
||
NPM_GLOBAL_BIN=$(npm bin -g 2>/dev/null) || true
|
||
if [[ -n "${NPM_GLOBAL_BIN}" ]]; then
|
||
export PATH="${NPM_GLOBAL_BIN}:${PATH}"
|
||
fi
|
||
|
||
if command_exists qwen; then
|
||
local QWEN_VERSION
|
||
QWEN_VERSION=$(qwen --version 2>/dev/null) || echo "unknown"
|
||
log_success "Qwen Code is already installed: ${QWEN_VERSION}"
|
||
log_info "Upgrading to the latest version..."
|
||
fi
|
||
|
||
# Clean npmrc conflicts
|
||
clean_npmrc_conflict
|
||
|
||
# Fix npm permissions if needed
|
||
fix_npm_permissions
|
||
|
||
# Configure npm registry for faster downloads in China
|
||
npm config set registry https://registry.npmmirror.com
|
||
log_info "npm registry set to npmmirror"
|
||
|
||
# Install Qwen Code
|
||
log_info "Installing Qwen Code..."
|
||
if npm install -g @qwen-code/qwen-code@latest; then
|
||
log_success "Qwen Code installed successfully!"
|
||
|
||
# Verify installation
|
||
if command_exists qwen; then
|
||
local qwen_version
|
||
qwen_version=$(qwen --version 2>/dev/null) || qwen_version="unknown"
|
||
log_info "Qwen Code version: ${qwen_version}"
|
||
fi
|
||
else
|
||
log_error "Failed to install Qwen Code!"
|
||
log_info "Please check your internet connection and try again"
|
||
exit 1
|
||
fi
|
||
|
||
# Create source.json if source parameter was provided
|
||
if [[ "${SOURCE}" != "unknown" ]]; then
|
||
create_source_json
|
||
fi
|
||
}
|
||
|
||
# ============================================
|
||
# Create source.json
|
||
# ============================================
|
||
create_source_json() {
|
||
local QWEN_DIR="${HOME}/.qwen"
|
||
|
||
mkdir -p "${QWEN_DIR}"
|
||
|
||
# Escape special characters in SOURCE for JSON
|
||
local ESCAPED_SOURCE
|
||
ESCAPED_SOURCE=$(printf '%s' "${SOURCE}" | sed 's/\\/\\\\/g; s/"/\\"/g')
|
||
|
||
cat > "${QWEN_DIR}/source.json" <<EOF
|
||
{
|
||
"source": "${ESCAPED_SOURCE}"
|
||
}
|
||
EOF
|
||
|
||
log_success "Installation source saved to ~/.qwen/source.json"
|
||
}
|
||
|
||
# ============================================
|
||
# Main function
|
||
# ============================================
|
||
main() {
|
||
# Validate HOME variable
|
||
if [[ -z "${HOME}" ]]; then
|
||
log_warning "HOME environment variable is not set"
|
||
local MAIN_UID
|
||
MAIN_UID=$(id -u) || true
|
||
if [[ "${MAIN_UID}" -eq 0 ]]; then
|
||
export HOME="/root"
|
||
else
|
||
local CURRENT_USER
|
||
CURRENT_USER=$(whoami) || true
|
||
local user_home
|
||
user_home=$(eval echo "~${CURRENT_USER}") || true
|
||
export HOME="${user_home}"
|
||
fi
|
||
log_info "Using HOME=${HOME}"
|
||
fi
|
||
|
||
# Ensure download tool is available
|
||
ensure_download_tool
|
||
|
||
# Check and install Node.js
|
||
check_and_install_nodejs
|
||
echo ""
|
||
|
||
# Install Qwen Code
|
||
install_qwen_code
|
||
echo ""
|
||
|
||
# ============================================
|
||
# Final instructions
|
||
# ============================================
|
||
echo "=========================================="
|
||
echo "✅ Installation completed!"
|
||
echo "=========================================="
|
||
echo ""
|
||
|
||
# Ensure NVM and npm global bin are in PATH
|
||
export NVM_DIR="${HOME}/.nvm"
|
||
# shellcheck source=/dev/null
|
||
[[ -s "${NVM_DIR}/nvm.sh" ]] && \. "${NVM_DIR}/nvm.sh" 2>/dev/null || true
|
||
local NPM_GLOBAL_BIN
|
||
NPM_GLOBAL_BIN=$(npm bin -g 2>/dev/null) || true
|
||
if [[ -n "${NPM_GLOBAL_BIN}" ]]; then
|
||
export PATH="${NPM_GLOBAL_BIN}:${PATH}"
|
||
fi
|
||
|
||
# Check if qwen is immediately available
|
||
if command_exists qwen; then
|
||
log_success "Qwen Code is ready to use!"
|
||
echo ""
|
||
echo "You can now run: qwen"
|
||
else
|
||
log_warning "To start using Qwen Code, please run:"
|
||
echo ""
|
||
local PROFILE_FILE
|
||
PROFILE_FILE=$(get_shell_profile)
|
||
echo " source ${PROFILE_FILE}"
|
||
echo ""
|
||
echo "Or simply restart your terminal, then run: qwen"
|
||
fi
|
||
}
|
||
|
||
# Run main function
|
||
main "$@"
|