#!/bin/bash

# Check if the base directory is provided as an argument

if [ "$#" -ne 1 ]; then

    echo "Usage: $0 <base_notebook_directory>"

    exit 1

fi

# Base directory for notebooks

base_dir=$1

# Define notebook directories or patterns

declare -a notebooks=(

    "$base_dir/examples/plotting/*.ipynb"

    "$base_dir/examples/problems/*.ipynb"

    "$base_dir/examples/solvers/*.ipynb"

)

# Initialize an array to hold valid notebook paths

declare -a valid_notebooks

# Gather all valid notebook files from the patterns

echo "Gathering notebooks..."

for pattern in "${notebooks[@]}"; do

    for nb in $pattern; do

        if [[ -f "$nb" ]]; then  # Check if the file exists

            valid_notebooks+=("$nb")  # Add to the list of valid notebooks

        fi

    done

done

# Check if we have any notebooks to run

if [ ${#valid_notebooks[@]} -eq 0 ]; then

    echo "No notebooks found to run."

    exit 1

fi

# Echo the notebooks that will be run for clarity

echo "Preparing to run the following notebooks:"

for nb in "${valid_notebooks[@]}"; do

    echo "$nb"

done

# Initialize a flag to track the success of all commands

all_success=true

# Execute all valid notebooks

for nb in "${valid_notebooks[@]}"; do

    echo "Running $nb"

    jupytext -k moscot --execute "$nb" || {

        echo "Failed to run $nb"

        all_success=false

    }

done

# Check if any executions failed

if [ "$all_success" = false ]; then

    echo "One or more notebooks failed to execute."

    exit 1

fi

echo "All notebooks executed successfully."