trax/scripts/setup_dev.sh

103 lines
3.0 KiB
Bash
Executable File

#!/bin/bash
set -e
echo "🚀 Setting up Trax development environment..."
# Color codes
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color
# Check Python version - prefer python3.11
if command -v python3.11 &> /dev/null; then
PYTHON_CMD="python3.11"
python_version=$($PYTHON_CMD --version | cut -d' ' -f2)
echo -e "${GREEN}✅ Python $python_version${NC}"
elif command -v python3 &> /dev/null; then
PYTHON_CMD="python3"
python_version=$($PYTHON_CMD --version | cut -d' ' -f2 | cut -d'.' -f1,2)
required_version="3.11"
if [ "$(printf '%s\n' "$required_version" "$python_version" | sort -V | head -n1)" != "$required_version" ]; then
echo -e "${RED}❌ Python 3.11+ required (found $python_version)${NC}"
echo -e "${YELLOW}💡 Try: brew install python@3.11${NC}"
exit 1
fi
echo -e "${GREEN}✅ Python $python_version${NC}"
else
echo -e "${RED}❌ Python 3 not found${NC}"
exit 1
fi
# Install uv if needed
if ! command -v uv &> /dev/null; then
echo -e "${YELLOW}📦 Installing uv...${NC}"
curl -LsSf https://astral.sh/uv/install.sh | sh
export PATH="$HOME/.cargo/bin:$PATH"
fi
echo -e "${GREEN}✅ uv installed${NC}"
# Setup virtual environment
echo -e "${YELLOW}🔧 Creating virtual environment...${NC}"
uv venv
source .venv/bin/activate
# Install dependencies
echo -e "${YELLOW}📚 Installing dependencies...${NC}"
uv pip install -e ".[dev]"
# Setup pre-commit hooks
echo -e "${YELLOW}🪝 Setting up pre-commit hooks...${NC}"
cat > .git/hooks/pre-commit << 'EOF'
#!/bin/bash
source .venv/bin/activate
echo "Running pre-commit checks..."
uv run black --check src/ tests/
uv run ruff check src/ tests/
uv run mypy src/
EOF
chmod +x .git/hooks/pre-commit
# Create directories
echo -e "${YELLOW}📁 Creating project directories...${NC}"
mkdir -p data/{media,exports,cache}
mkdir -p tests/{unit,integration,fixtures/audio,fixtures/transcripts}
mkdir -p src/agents/rules
mkdir -p docs/{reports,team,architecture}
# Check PostgreSQL
if command -v psql &> /dev/null; then
echo -e "${GREEN}✅ PostgreSQL installed${NC}"
else
echo -e "${YELLOW}⚠️ PostgreSQL not found - please install${NC}"
fi
# Check FFmpeg
if command -v ffmpeg &> /dev/null; then
echo -e "${GREEN}✅ FFmpeg installed${NC}"
else
echo -e "${YELLOW}⚠️ FFmpeg not found - please install${NC}"
fi
# Setup test data
echo -e "${YELLOW}🎵 Setting up test fixtures...${NC}"
cat > tests/fixtures/README.md << 'EOF'
# Test Fixtures
Place test audio files here:
- sample_5s.wav (5-second test)
- sample_30s.mp3 (30-second test)
- sample_2m.mp4 (2-minute test)
These should be real audio files for testing.
EOF
echo -e "${GREEN}✅ Development environment ready!${NC}"
echo ""
echo "📝 Next steps:"
echo " 1. source .venv/bin/activate"
echo " 2. Set up PostgreSQL database"
echo " 3. Add test audio files to tests/fixtures/audio/"
echo " 4. uv run pytest # Run tests"
echo " 5. uv run python src/cli/main.py --help # Run CLI"