|
| 1 | +#!/usr/bin/env bash |
| 2 | + |
| 3 | +show_help() { |
| 4 | + echo "Usage: $0 [--what/-w all|r|python|julia] [--force/-f] [--help/-h]" |
| 5 | + echo " --what/-w: Specify what to initialise (default: all)." |
| 6 | + echo " all: Initialise R (renv), Python (uv), and Julia (project)." |
| 7 | + echo " r: Initialise R (renv)." |
| 8 | + echo " python: Initialise Python (uv)." |
| 9 | + echo " julia: Initialise Julia (project)." |
| 10 | + echo " --force/-f: Force initialisation regardless of existing files." |
| 11 | + echo " --help/-h: Show this help message." |
| 12 | +} |
| 13 | + |
| 14 | +initialise_r() { |
| 15 | + if [ "$FORCE" = true ] || [ ! -f "renv.lock" ]; then |
| 16 | + if grep -q 'source("renv/activate.R")' .Rprofile; then |
| 17 | + sed -i '' '/source("renv\/activate.R")/d' .Rprofile |
| 18 | + fi |
| 19 | + Rscript -e 'renv::init(bare = FALSE)' |
| 20 | + Rscript -e 'renv::install(c("rmarkdown", "prompt", "languageserver", "lintr", "styler", "cli"))' |
| 21 | + Rscript -e 'renv::snapshot(type = "all")' |
| 22 | + fi |
| 23 | +} |
| 24 | + |
| 25 | +initialise_python() { |
| 26 | + if [ "$FORCE" = true ] || [ ! -f "requirements.txt" ]; then |
| 27 | + python3 -m venv .venv |
| 28 | + source .venv/bin/activate |
| 29 | + python3 -m pip install jupyter papermill |
| 30 | + python3 -m pip freeze > requirements.txt |
| 31 | + fi |
| 32 | +} |
| 33 | + |
| 34 | +initialise_uv() { |
| 35 | + if [ "$FORCE" = true ] || [ ! -f "uv.lock" ]; then |
| 36 | + uv init --no-package --vcs none --bare --no-readme --author-from none |
| 37 | + uv venv |
| 38 | + source .venv/bin/activate |
| 39 | + uv add jupyter papermill |
| 40 | + uv sync |
| 41 | + fi |
| 42 | +} |
| 43 | + |
| 44 | +initialise_julia() { |
| 45 | + if [ "$FORCE" = true ] || [ ! -f "Project.toml" ]; then |
| 46 | + julia -e 'using Pkg; Pkg.activate("."); Pkg.instantiate()' |
| 47 | + julia --project=. -e 'using Pkg; Pkg.add("IJulia")' |
| 48 | + fi |
| 49 | +} |
| 50 | + |
| 51 | +WHAT="all" |
| 52 | +FORCE=false |
| 53 | + |
| 54 | +while [[ "$#" -gt 0 ]]; do |
| 55 | + case $1 in |
| 56 | + --what|-w) |
| 57 | + WHAT="$2" |
| 58 | + shift |
| 59 | + ;; |
| 60 | + --force|-f) |
| 61 | + FORCE=true |
| 62 | + ;; |
| 63 | + --help|-h) |
| 64 | + show_help |
| 65 | + exit 0 |
| 66 | + ;; |
| 67 | + *) |
| 68 | + echo "Unknown parameter passed: $1" |
| 69 | + show_help |
| 70 | + exit 1 |
| 71 | + ;; |
| 72 | + esac |
| 73 | + shift |
| 74 | +done |
| 75 | + |
| 76 | +case $WHAT in |
| 77 | + all) |
| 78 | + initialise_r |
| 79 | + initialise_uv |
| 80 | + initialise_julia |
| 81 | + ;; |
| 82 | + r) |
| 83 | + initialise_r |
| 84 | + ;; |
| 85 | + python) |
| 86 | + initialise_uv |
| 87 | + ;; |
| 88 | + julia) |
| 89 | + initialise_julia |
| 90 | + ;; |
| 91 | + *) |
| 92 | + echo "Unknown option for --what: $WHAT" |
| 93 | + show_help |
| 94 | + exit 1 |
| 95 | + ;; |
| 96 | +esac |
0 commit comments