Simple CLI for AWS SageMaker notebook instances

smnb

#!/bin/bash

# TODO:
# - start, stop, status commands
# - jupyterlab option
# - pass settings (e.g. profile) to aws cli

PROGNAME=$(basename $0)
AWS="aws"
OPEN="xdg-open"

cmd_help() {
  echo "Usage: $PROGNAME <command> [args]\n"
  cat <<EOF
Subcommands:
  list
  open <instance>
EOF
}

cmd_list() {
  $AWS sagemaker list-notebook-instances --query 'NotebookInstances[*].[NotebookInstanceName]' --output text
}

cmd_open() {
  $OPEN $($AWS sagemaker create-presigned-notebook-instance-url --notebook-instance-name $1 --query AuthorizedUrl --output text)
}

subcmd=$1
case $subcmd in
  "" | "-h" | "--help")
    cmd_help
    ;;
  *)
    shift
    cmd_${subcmd} $@
    if [ $? = 127 ]; then
      echo "Unrecognized subcommand, run '$PROGNAME --help' to list subcommands"
      exit 1
    fi
    ;;
esac