#!/bin/bash

set -euo pipefail

# Pipx to install pre-commit. Otherwise, pre-commit is broken when installed
# with /usr/bin/pip3 (alinux), but we don't want to use ~/anaconda/bin/pip3
# either to minimize polluting its site packages.
~/anaconda3/bin/pip3 install --no-cache-dir pipx

# Relocate pipx packages to ~/SageMaker to survive reboot
export PIPX_HOME=~/SageMaker/.initsmnb.d/pipx
export PIPX_BIN_DIR=~/SageMaker/.initsmnb.d/bin
cat << EOF | sudo tee /etc/profile.d/initsmnb-pipx.sh

# Add pipx binaries to PATH. In addition, add also ~/.local/bin so that its
# commands are usable by Jupyter kernels (notable example: docker-compose for
# SageMaker local mode).
export PATH=\$PATH:$PIPX_BIN_DIR:/home/ec2-user/.local/bin
export PIPX_HOME=$PIPX_HOME
export PIPX_BIN_DIR=$PIPX_BIN_DIR
EOF

# Stop 'pipx install' from warning about path
export PATH=$PATH:$PIPX_BIN_DIR

declare -a PKG=(
    pre-commit
    ranger-fm
    cookiecutter
    jupytext
    s4cmd
    nvitop
    gpustat
    awslogs
    ruff
    #black
    #nbqa
    #isort
    #pyupgrade
)

for i in "${PKG[@]}"; do
    ~/anaconda3/bin/pipx install $i
done

~/anaconda3/bin/pipx upgrade-all

# ranger defaults to relative line number
mkdir -p ~/.config/ranger/
echo set line_numbers relative >> ~/.config/ranger/rc.conf