setup.sh

#!/bin/bash
set -e

# Store the initial directory to ensure we can always return to it
INITIAL_DIR=$(pwd)

# Function to ensure we return to the initial directory
cleanup() {
    cd "$INITIAL_DIR"
}

# Set up trap to ensure we always return to the initial directory
trap cleanup EXIT

# Install poetry if not already installed
echo "Checking for poetry installation..."
if ! command -v poetry &> /dev/null; then
    echo "Installing poetry..."
    pip install poetry
else
    echo "Poetry already installed."
fi

# Install project dependencies with poetry
echo "Installing project dependencies..."
poetry install

# Enable widgets
echo "Enabling widgets..."
poetry run enable-widgets

# Apply the patch to fix the config package version
echo "Checking translation-server patch status..."
cd translation-server

# Check if patch has already been applied
if git apply --check ../translation-server-config-fix.patch &> /dev/null; then
    echo "Applying patch to translation-server..."
    git apply ../translation-server-config-fix.patch
else
    echo "Patch already applied or not applicable."
fi

# Install translation-server dependencies if node_modules doesn't exist or package.json is newer
if [ ! -d "node_modules" ] || [ "package.json" -nt "node_modules" ]; then
    echo "Installing translation-server dependencies..."
    npm install
else
    echo "Translation-server dependencies already installed."
fi

# Return to the project root directory
cd "$INITIAL_DIR"

echo "Setup complete! The project dependencies and translation-server are now installed."

# Check if .env file exists and contains required variables with values
if [ ! -f .env ] || \
   ! grep -q "OXYLABS_USERNAME=.*[^[:space:]]" .env || \
   ! grep -q "OXYLABS_PASSWORD=.*[^[:space:]]" .env || \
   ! grep -q "OPENAI_API_KEY=.*[^[:space:]]" .env; then
    echo "Don't forget to set up your environment variables in a .env file:"
    echo "OXYLABS_USERNAME="
    echo "OXYLABS_PASSWORD="
    echo "OPENAI_API_KEY="
    echo ""
fi

echo "To start the Zotero translation server, run:"
echo "cd translation-server && npm start"