soln.sh

#!/bin/bash

function soln() {
  if [ -z $1 ]; then
      echo "Usage: soln <name>"
      return
  fi

  solution_dir="$DMOJ_SOLUTIONS/$1"
  template_dir="$DMOJ_SOLUTIONS/__template__"

  if [ -d "$solution_dir" ]; then
    echo "Switched to $1"
    cd "$solution_dir"
  else
    echo -n "No solution for $1. Create directory? (y or n)"
    while true; do
        read -n 1 -s yn
        case $yn in
            y ) echo; break;;
            n ) echo; return;;
            * ) continue;;
        esac
    done

    cp -r "$template_dir" "$solution_dir"
    echo "Created $1"
    cd "$solution_dir"
  fi
}