#!/usr/bin/env bash

install-jupyter-lab() {
    activate-venv

    echo 'Ensure Jupyter Lab requirements'
    local req_file='requirements.txt'
    touch "$req_file"

    grep -q '^jupyterlab$' "$req_file" || echo 'jupyterlab' >> "$req_file"
    grep -q '^papermill$' "$req_file" || echo 'papermill' >> "$req_file"
    grep -q '^pandas$' "$req_file" || echo 'pandas' >> "$req_file"
    grep -q '^openpyxl$' "$req_file" || echo 'openpyxl' >> "$req_file"
    grep -q '^jupyter-ai$' "$req_file" || echo 'jupyter-ai' >> "$req_file"

    echo 'Install project dependencies'
    uv pip install -r "$req_file"

    if [[ -f ".gitignore" ]]; then
        if ! grep -q '^\.ipynb_checkpoints$' ".gitignore"; then
            echo '' >> ".gitignore"
            echo '# Jupyter' >> ".gitignore"
            echo '.ipynb_checkpoints' >> ".gitignore"
            echo 'Added .ipynb_checkpoints to .gitignore'
        else
            echo '.ipynb_checkpoints already in .gitignore'
        fi
    else
        echo 'Warning: no .gitignore found; skipping .ipynb_checkpoints entry'
    fi
}

ijl() {
    install-jupyter-lab "$@"
}
