setup.sh

#!/bin/bash

# Check for flags
TEST_MODE=false
SKIP_INSTALL=false
for arg in "$@"; do
  if [ "$arg" == "--test" ]; then
    TEST_MODE=true
    echo "Running in test mode - will skip analysis step"
  elif [ "$arg" == "--skip-install" ]; then
    SKIP_INSTALL=true
    echo "Skipping installation steps"
  fi
done

if [ "$SKIP_INSTALL" = false ]; then
  echo "Installing dependencies for langgraph..."
  echo "1. make venv and install.."
  python -m venv venv || { echo "Failed to create virtual environment"; exit 1; }
  source venv/bin/activate || { echo "Failed to activate virtual environment"; exit 1; }
  pip install langgraph-cli || { echo "Failed to install langgraph-cli"; exit 1; }
  pip install -e ./app || { echo "Failed to install app package"; exit 1; }
else
  echo "Skipping installation steps - activating existing venv"
  source venv/bin/activate || { echo "Failed to activate virtual environment"; exit 1; }
fi

echo -e "\033[31m2. make callgraph...\033[0m"
echo -e "\033[31mplease input root directory of project\033[0m"
read root_dir
echo -e "\033[31mplease input src directory of project\033[0m"
read src_dir

echo -e "\033[31m3. Prepare analyze data and callgraph...\033[0m"

# Only run AnalyzeSolidity if not in test mode
if [ "$TEST_MODE" = false ]; then
  output_analyze_solidity="app/final_analyze_results.json"
  if [ ! -f "$output_analyze_solidity" ]; then
    echo "Analyzing Solidity..."
    python app/src/utils/AnalyzeSolidity.py $src_dir $output_analyze_solidity
    if [ $? -ne 0 ]; then
      echo "AnalyzeSolidity.py failed"
      exit 1
    fi
  else
    echo "Analyze Solidity file $output_analyze_solidity already exists. Skipping analysis."
  fi
else
  echo "Skipping AnalyzeSolidity.py in test mode"
fi

echo -e "\033[31m4. prepare Redis server..\033[0m"

make clean
make redis
if [ $? -ne 0 ]; then
  echo "Failed to start Redis server"
  exit 1
fi

echo -e "\033[31m5. run langgraph...\033[0m"
cd app || { echo "Failed to change directory to app"; exit 1; }
langgraph dev --allow-blocking