rob
/
CmdForge
1
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases Wiki Activity
CmdForge/install.sh

414 lines
12 KiB
Bash
Executable File
Raw Blame History

#!/bin/bash
#
# CmdForge Installer
#
# Usage:
# ./install.sh # Interactive installation
# ./install.sh --yes # Accept all defaults (non-interactive)
# ./install.sh --no-venv # Install to system Python (not recommended)
# ./install.sh --no-examples # Skip example tools
#
set -e
# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color
BOLD='\033[1m'
# Configuration
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
VENV_DIR="$HOME/.cmdforge-venv"
LOCAL_BIN="$HOME/.local/bin"
MIN_PYTHON_VERSION="3.10"
# Parse arguments
USE_VENV=true
INSTALL_EXAMPLES=true
AUTO_YES=false
while [[ $# -gt 0 ]]; do
case $1 in
--yes|-y)
AUTO_YES=true
shift
;;
--no-venv)
USE_VENV=false
shift
;;
--no-examples)
INSTALL_EXAMPLES=false
shift
;;
--help|-h)
echo "CmdForge Installer"
echo ""
echo "Usage: ./install.sh [OPTIONS]"
echo ""
echo "Options:"
echo " --yes, -y Accept all defaults (non-interactive)"
echo " --no-venv Install to system Python (not recommended)"
echo " --no-examples Skip installing example tools"
echo " --help, -h Show this help message"
exit 0
;;
*)
echo "Unknown option: $1"
echo "Run './install.sh --help' for usage"
exit 1
;;
esac
done
# Helper functions
print_header() {
echo ""
echo -e "${CYAN}${BOLD}$1${NC}"
echo -e "${CYAN}$(printf '=%.0s' {1..60})${NC}"
}
print_step() {
echo -e "${BLUE}[*]${NC} $1"
}
print_success() {
echo -e "${GREEN}[+]${NC} $1"
}
print_warning() {
echo -e "${YELLOW}[!]${NC} $1"
}
print_error() {
echo -e "${RED}[X]${NC} $1"
}
ask_yes_no() {
if $AUTO_YES; then
return 0
fi
local prompt="$1"
local default="${2:-y}"
if [[ "$default" == "y" ]]; then
prompt="$prompt [Y/n] "
else
prompt="$prompt [y/N] "
fi
read -r -p "$prompt" response
response=${response:-$default}
[[ "$response" =~ ^[Yy]$ ]]
}
version_gte() {
# Check if version $1 >= version $2
printf '%s\n%s\n' "$2" "$1" | sort -V -C
}
# Start installation
clear 2>/dev/null || true
echo ""
echo -e "${CYAN}${BOLD}"
echo " ____ _ _____ "
echo " / ___|_ __ ___ __| | ___|__ _ __ __ _ ___ "
echo " | | | '_ \` _ \\ / _\` | |_ / _ \\| '__/ _\` |/ _ \\"
echo " | |___| | | | | | (_| | _| (_) | | | (_| | __/"
echo " \\____|_| |_| |_|\\__,_|_| \\___/|_| \\__, |\\___|"
echo " |___/ "
echo -e "${NC}"
echo " AI-powered CLI command builder"
echo ""
# ==============================================================================
# Step 1: Check prerequisites
# ==============================================================================
print_header "Step 1: Checking Prerequisites"
# Check Python
print_step "Checking Python version..."
PYTHON_CMD=""
for cmd in python3 python; do
if command -v "$cmd" &> /dev/null; then
version=$("$cmd" --version 2>&1 | awk '{print $2}')
if version_gte "$version" "$MIN_PYTHON_VERSION"; then
PYTHON_CMD="$cmd"
break
fi
fi
done
if [[ -z "$PYTHON_CMD" ]]; then
print_error "Python $MIN_PYTHON_VERSION or higher is required"
echo ""
echo "Please install Python first:"
echo " Ubuntu/Debian: sudo apt install python3 python3-pip python3-venv"
echo " macOS: brew install python3"
echo " Windows: https://www.python.org/downloads/"
exit 1
fi
PYTHON_VERSION=$("$PYTHON_CMD" --version 2>&1 | awk '{print $2}')
print_success "Python $PYTHON_VERSION found ($PYTHON_CMD)"
# Check pip
print_step "Checking pip..."
if ! "$PYTHON_CMD" -m pip --version &> /dev/null; then
print_error "pip is not installed"
echo ""
echo "Please install pip:"
echo " Ubuntu/Debian: sudo apt install python3-pip"
echo " macOS: python3 -m ensurepip"
exit 1
fi
print_success "pip is available"
# Check venv module (if using venv)
if $USE_VENV; then
print_step "Checking venv module..."
if ! "$PYTHON_CMD" -m venv --help &> /dev/null; then
print_warning "venv module not available"
echo ""
if ask_yes_no "Install python3-venv? (recommended)" "y"; then
if command -v apt-get &> /dev/null; then
sudo apt-get update && sudo apt-get install -y python3-venv
else
print_error "Cannot auto-install venv. Please install manually."
echo " Ubuntu/Debian: sudo apt install python3-venv"
exit 1
fi
else
print_warning "Continuing without venv (installing to system Python)"
USE_VENV=false
fi
else
print_success "venv module is available"
fi
fi
# Check we're in the CmdForge directory
print_step "Checking CmdForge source..."
if [[ ! -f "$SCRIPT_DIR/pyproject.toml" ]]; then
print_error "Cannot find CmdForge source files"
echo "Please run this script from the CmdForge directory"
exit 1
fi
print_success "CmdForge source found at $SCRIPT_DIR"
# ==============================================================================
# Step 2: Installation options
# ==============================================================================
print_header "Step 2: Installation Options"
if ! $AUTO_YES; then
echo ""
echo "Installation options:"
echo ""
if $USE_VENV; then
echo -e " ${GREEN}1.${NC} Use virtual environment at $VENV_DIR ${GREEN}(recommended)${NC}"
else
echo -e " ${YELLOW}1.${NC} Install to system Python ${YELLOW}(--no-venv specified)${NC}"
fi
if $INSTALL_EXAMPLES; then
echo -e " ${GREEN}2.${NC} Install 28 example tools"
else
echo -e " ${YELLOW}2.${NC} Skip example tools ${YELLOW}(--no-examples specified)${NC}"
fi
echo ""
if ! ask_yes_no "Proceed with installation?" "y"; then
echo "Installation cancelled."
exit 0
fi
fi
# ==============================================================================
# Step 3: Create virtual environment (if using)
# ==============================================================================
if $USE_VENV; then
print_header "Step 3: Setting Up Virtual Environment"
if [[ -d "$VENV_DIR" ]]; then
print_warning "Virtual environment already exists at $VENV_DIR"
if ask_yes_no "Remove and recreate?" "n"; then
rm -rf "$VENV_DIR"
fi
fi
if [[ ! -d "$VENV_DIR" ]]; then
print_step "Creating virtual environment..."
"$PYTHON_CMD" -m venv "$VENV_DIR"
print_success "Virtual environment created at $VENV_DIR"
fi
print_step "Activating virtual environment..."
source "$VENV_DIR/bin/activate"
print_success "Virtual environment activated"
# Upgrade pip in venv
print_step "Upgrading pip..."
pip install --upgrade pip --quiet
print_success "pip upgraded"
else
print_header "Step 3: Using System Python"
print_warning "Installing to system Python (use --no-venv was specified)"
fi
# ==============================================================================
# Step 4: Install CmdForge
# ==============================================================================
print_header "Step 4: Installing CmdForge"
print_step "Installing CmdForge and dependencies..."
cd "$SCRIPT_DIR"
# Install with all optional dependencies
pip install -e ".[all]" --quiet
print_success "CmdForge installed successfully"
# Verify installation
print_step "Verifying installation..."
if command -v cmdforge &> /dev/null; then
VERSION=$(cmdforge --version 2>&1 | head -1)
print_success "cmdforge command available: $VERSION"
else
print_error "cmdforge command not found in PATH"
exit 1
fi
# ==============================================================================
# Step 5: Set up PATH
# ==============================================================================
print_header "Step 5: Setting Up PATH"
# Create local bin directory
mkdir -p "$LOCAL_BIN"
# Determine shell config file
SHELL_CONFIG=""
if [[ -f "$HOME/.bashrc" ]]; then
SHELL_CONFIG="$HOME/.bashrc"
elif [[ -f "$HOME/.zshrc" ]]; then
SHELL_CONFIG="$HOME/.zshrc"
elif [[ -f "$HOME/.bash_profile" ]]; then
SHELL_CONFIG="$HOME/.bash_profile"
fi
# Check if PATH already includes local bin
PATH_LINE='export PATH="$HOME/.local/bin:$PATH"'
PATH_CONFIGURED=false
if [[ -n "$SHELL_CONFIG" ]]; then
if grep -q '.local/bin' "$SHELL_CONFIG" 2>/dev/null; then
print_success "PATH already configured in $SHELL_CONFIG"
PATH_CONFIGURED=true
fi
fi
# Add venv activation if using venv
if $USE_VENV; then
VENV_LINE="source $VENV_DIR/bin/activate"
if [[ -n "$SHELL_CONFIG" ]] && ! grep -q "cmdforge-venv" "$SHELL_CONFIG" 2>/dev/null; then
print_step "Adding virtual environment activation to $SHELL_CONFIG..."
echo "" >> "$SHELL_CONFIG"
echo "# CmdForge virtual environment" >> "$SHELL_CONFIG"
echo "$VENV_LINE" >> "$SHELL_CONFIG"
print_success "Added venv activation to $SHELL_CONFIG"
fi
fi
# Add PATH if not already there
if ! $PATH_CONFIGURED && [[ -n "$SHELL_CONFIG" ]]; then
print_step "Adding ~/.local/bin to PATH in $SHELL_CONFIG..."
echo "" >> "$SHELL_CONFIG"
echo "# CmdForge tool wrappers" >> "$SHELL_CONFIG"
echo "$PATH_LINE" >> "$SHELL_CONFIG"
print_success "Added PATH configuration to $SHELL_CONFIG"
fi
# Export for current session
export PATH="$LOCAL_BIN:$PATH"
# ==============================================================================
# Step 6: Create wrapper scripts
# ==============================================================================
print_header "Step 6: Creating Tool Wrappers"
print_step "Running cmdforge refresh..."
cmdforge refresh
print_success "Wrapper scripts created in $LOCAL_BIN"
# ==============================================================================
# Step 7: Install example tools (optional)
# ==============================================================================
if $INSTALL_EXAMPLES; then
print_header "Step 7: Installing Example Tools"
print_step "Installing 28 example tools..."
"$PYTHON_CMD" "$SCRIPT_DIR/examples/install.py"
print_step "Refreshing wrappers for new tools..."
cmdforge refresh
print_success "Example tools installed"
else
print_header "Step 7: Skipping Example Tools"
print_warning "Example tools not installed (use 'python examples/install.py' later)"
fi
# ==============================================================================
# Complete!
# ==============================================================================
print_header "Installation Complete!"
echo ""
echo -e "${GREEN}${BOLD}CmdForge has been installed successfully!${NC}"
echo ""
# Show installed tools count
TOOL_COUNT=$(cmdforge list 2>/dev/null | grep -c "^" || echo "0")
echo -e " ${CYAN}Tools installed:${NC} $TOOL_COUNT"
echo -e " ${CYAN}Config location:${NC} ~/.cmdforge/"
echo -e " ${CYAN}Tool wrappers:${NC} ~/.local/bin/"
if $USE_VENV; then
echo -e " ${CYAN}Virtual env:${NC} $VENV_DIR"
fi
echo ""
echo -e "${BOLD}Next Steps:${NC}"
echo ""
echo " 1. Reload your shell (or run: source $SHELL_CONFIG)"
echo ""
echo " 2. Set up an AI provider:"
echo -e " ${CYAN}cmdforge providers install${NC}"
echo ""
echo " 3. Launch the UI:"
echo -e " ${CYAN}cmdforge ui${NC}"
echo ""
echo " 4. Or try a tool directly:"
echo -e " ${CYAN}echo 'Hello world' | eli5${NC}"
echo ""
if $USE_VENV; then
echo -e "${YELLOW}Note:${NC} The virtual environment will be activated automatically"
echo " in new shell sessions. For this session, run:"
echo -e " ${CYAN}source $VENV_DIR/bin/activate${NC}"
echo ""
fi
echo -e "${CYAN}Documentation:${NC} https://pages.brrd.tech/rob/cmdforge/"
echo -e "${CYAN}Source code:${NC} https://gitea.brrd.tech/rob/CmdForge"
echo ""
Reference in New Issue View Git Blame Copy Permalink