#!/bin/bash
# OpenClaw Framework Setup Script
# Version: 1.0.0
# Purpose: Initialize OpenClaw agent orchestration environment

set -e

echo "=== OpenClaw Framework Setup ==="

# Configuration
OPENCLAW_DIR="${HOME}/projects/openclaw"
PYTHON_VERSION="3.11"

# Check Python version
echo "Checking Python version..."
python3 --version | grep -q "$PYTHON_VERSION" || echo "Warning: Python $PYTHON_VERSION recommended"

# Create project directory
echo "Creating OpenClaw directory..."
mkdir -p "$OPENCLAW_DIR"
cd "$OPENCLAW_DIR"

# Initialize Python virtual environment
echo "Setting up virtual environment..."
python3 -m venv venv
source venv/bin/activate

# Install OpenClaw dependencies
echo "Installing OpenClaw dependencies..."
pip install --upgrade pip
pip install openclaw-agent
pip install pydantic fastapi uvicorn
pip install langchain langchain-community

# Configure environment variables
echo "Creating .env file..."
cat > .env << 'EOF'
# OpenClaw Configuration
OPENCLAW_API_KEY=your-api-key-here
OPENCLAW_ORCHESTRATOR_URL=http://localhost:8000

# Model Configuration
OPENAI_API_KEY=your-openai-key-here
OLLAMA_BASE_URL=http://localhost:11434
QWEN35_ENDPOINT=192.168.68.8:8080

# Agent Registry Settings
AGENT_REGISTRY_DIR=./agents
MAX_CONCURRENT_AGENTS=5
WORKFLOW_TIMEOUT=3600
EOF

# Create agents directory structure
echo "Creating agent directory structure..."
mkdir -p "$AGENT_REGISTRY_DIR"/researcher
mkdir -p "$AGENT_REGISTRY_DIR"/analyst
mkdir -p "$AGENT_REGISTRY_DIR"/writer

# Initialize git repository
echo "Initializing git repository..."
git init
git add .
git commit -m "Initial OpenClaw framework setup"

# Display configuration
echo "=== Setup Complete ==="
echo "OpenClaw installed at: $OPENCLAW_DIR"
echo "Run: source venv/bin/activate"
echo "Start orchestrator: python orchestrator.py"
echo ""
echo "Next steps:"
echo "1. Set your API keys in .env file"
echo "2. Create custom agent definitions in $AGENT_REGISTRY_DIR/"
echo "3. Test with: python test_agents.py"