setup.sh

#!/bin/bash
set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo -e "${GREEN}======================================${NC}"
echo -e "${GREEN}Kernel Knowledge RAG System Setup${NC}"
echo -e "${GREEN}======================================${NC}"
echo

# Get the directory where the script is located
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
VENV_DIR="$SCRIPT_DIR/venv"

# Check Python version
echo -e "${YELLOW}Checking Python version...${NC}"
if ! command -v python3 &> /dev/null; then
    echo -e "${RED}Error: Python 3 is not installed${NC}"
    exit 1
fi

PYTHON_VERSION=$(python3 --version | cut -d' ' -f2)
echo -e "${GREEN}Found Python $PYTHON_VERSION${NC}"
echo

# Create virtual environment
echo -e "${YELLOW}Creating virtual environment...${NC}"
if [ -d "$VENV_DIR" ]; then
    echo -e "${YELLOW}Virtual environment already exists. Removing...${NC}"
    rm -rf "$VENV_DIR"
fi

python3 -m venv "$VENV_DIR"
echo -e "${GREEN}Virtual environment created${NC}"
echo

# Activate virtual environment
echo -e "${YELLOW}Activating virtual environment...${NC}"
source "$VENV_DIR/bin/activate"
echo -e "${GREEN}Virtual environment activated${NC}"
echo

# Upgrade pip
echo -e "${YELLOW}Upgrading pip...${NC}"
pip install --upgrade pip
echo

# Install dependencies
echo -e "${YELLOW}Installing dependencies...${NC}"
pip install -r "$SCRIPT_DIR/requirements.txt"
echo -e "${GREEN}Dependencies installed${NC}"
echo

# Download embedding model
echo -e "${YELLOW}Downloading embedding model (this may take a few minutes)...${NC}"
python3 << EOF
from sentence_transformers import SentenceTransformer
print("Loading model...")
model = SentenceTransformer('all-MiniLM-L6-v2')
print("Model downloaded and cached successfully!")
EOF
echo -e "${GREEN}Embedding model ready${NC}"
echo

# Create directories if they don't exist
echo -e "${YELLOW}Ensuring directory structure...${NC}"
mkdir -p "$SCRIPT_DIR/journals"
mkdir -p "$SCRIPT_DIR/vector-db"
mkdir -p "$SCRIPT_DIR/mcp-server"
echo -e "${GREEN}Directories ready${NC}"
echo

# Make scripts executable
echo -e "${YELLOW}Setting executable permissions...${NC}"
chmod +x "$SCRIPT_DIR/mcp-server/indexer_service.py"
chmod +x "$SCRIPT_DIR/mcp-server/server.py"
chmod +x "$SCRIPT_DIR/start-indexer.sh"
chmod +x "$SCRIPT_DIR/stop-indexer.sh"
chmod +x "$SCRIPT_DIR/increase-inotify-limit.sh"
chmod +x "$SCRIPT_DIR/test-mcp-server.sh"
echo -e "${GREEN}Permissions set${NC}"
echo

echo -e "${GREEN}======================================${NC}"
echo -e "${GREEN}Setup Complete!${NC}"
echo -e "${GREEN}======================================${NC}"
echo
echo -e "${YELLOW}Next steps:${NC}"
echo
echo "1. Start the indexer service:"
echo -e "   ${GREEN}./start-indexer.sh${NC}"
echo -e "   ${YELLOW}Or with custom base directory:${NC}"
echo -e "   ${GREEN}./start-indexer.sh --base-dir /path/to/knowledge-base${NC}"
echo
echo "2. Add your first journal entry:"
echo -e "   ${GREEN}cd journals/${NC}"
echo -e "   ${GREEN}vim my-first-note.md${NC}"
echo
echo "3. Configure GitHub Copilot to use the MCP server:"
echo -e "   ${GREEN}./configure-vscode.sh${NC}"
echo
echo "4. Test the MCP server:"
echo -e "   ${GREEN}./test-mcp-server.sh${NC}"
echo
echo "5. Stop the indexer service:"
echo -e "   ${GREEN}./stop-indexer.sh${NC}"
echo
echo -e "${YELLOW}Note:${NC} If the indexer fails with 'inotify watch limit reached', run:"
echo -e "   ${GREEN}./increase-inotify-limit.sh${NC}"
echo
echo -e "${YELLOW}Documentation is available in:${NC}"
echo -e "   ${GREEN}README.md${NC}"
echo