shell.nix

# Shell configuration for zsh (frequently used)

{ config, lib, pkgs, ... }:

let
  # Set all shell aliases programatically
  shellAliases = {
    # Aliases for commonly used tools
    grep = "grep --color=auto";
    ll = "ls -lh";
    tf = "terraform";
    hms = "home-manager switch";

    # Reload zsh
    szsh = "source ~/.zshrc";

    # Reload home manager and zsh
    reload = "NIXPKGS_ALLOW_UNFREE=1 home-manager switch && source ~/.zshrc";

    # Nix garbage collection
    garbage = "nix-collect-garbage -d";

    # Bundle Rails C
    brc = "bundle exec rails c";
    # Bundle Rails S
    brs = "bundle exec rails s";
    # Database MigrAte
    dma = "bundle exec rake db:migrate";
    # Database (M) Rollback
    dmr = "bundle exec rake db:rollback";
    # Visual Studio Code
    vsc = "code .";
    # HooGLe server
    hgl = "hoogle server --local --port 8080 &";
  };
in
{
  # Fancy filesystem navigator
  programs.broot = {
    enable = true;
    enableZshIntegration = true;
  };

  # zsh settings
  programs.zsh = {
    inherit shellAliases;
    enable = true;
    autosuggestion.enable = true;
    enableCompletion = true;
    history.extended = true;

    initExtraFirst = ''
      export ZSH=${pkgs.oh-my-zsh}/share/oh-my-zsh/
    '';

    # Called whenever zsh is initialized
    initExtra = ''
      export TERM="xterm-256color"
      bindkey -e

      # Nix setup (environment variables, etc.)
      # https://discourse.nixos.org/t/how-to-restore-nix-and-home-manager-after-macos-upgrade/25474
      if [ -e '/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh' ]; then
        . '/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh'
      fi

      # Load environment variables from a file; this approach allows me to not
      # commit secrets like API keys to Git
      if [ -e ~/.env ]; then
        . ~/.env
      fi

      # Start up Starship shell
      eval "$(starship init zsh)"

      # direnv setup
      eval "$(direnv hook zsh)"

      # direnv hook
      eval "$(direnv hook zsh)"

      function setCabalProjectLocalToBuild() {
        echo "optimization: False" > cabal.project.local
        echo "program-options" >> cabal.project.local
        echo "  ghc-options: -Wall" >> cabal.project.local
      }

      function setCabalProjectLocalToDebug() {
        echo "optimization: False" > cabal.project.local
        echo "program-options" >> cabal.project.local
        echo "  ghc-options: -Wwarn -Wunused-top-binds -Werror=unused-top-binds" >> cabal.project.local
      }

      function setCabalProjectLocalToCoverage() {
        echo "package *" >> cabal.project.local
        echo "  coverage: True" >> cabal.project.local
        echo "  library-coverage: True" >> cabal.project.local
        echo "package order-processing" >> cabal.project.local
        echo "  coverage: False" >> cabal.project.local
        echo "  library-coverage: False" >> cabal.project.local
      }

      # Build and test a Haskell project
      function hbt() {
        setCabalProjectLocalToBuild

        TOOL_NAME=$1
        clear && cabal --builddir=./dist-build build $TOOL_NAME && cabal --builddir=./dist-build test $TOOL_NAME
      }

      # Build, test, and install a Haskell tool
      function hbti() {
        setCabalProjectLocalToBuild

        TOOL_NAME=$1
        clear && cabal --builddir=./dist-build build $TOOL_NAME && cabal --builddir=./dist-build test $TOOL_NAME && cabal --builddir=./dist-build install $TOOL_NAME --overwrite-policy=always
      }

      # Debug a Haskell project with ghcid
      function hdbg() {
        setCabalProjectLocalToDebug

        TOOL_NAME=$1
        ghcid -c "cabal --builddir=./dist-debug repl $TOOL_NAME"
      }

      # Run the Haskell REPL
      function hrepl() {
        setCabalProjectLocalToDebug

        TOOL_NAME=$1
        cabal --builddir=./dist-debug repl $TOOL_NAME
      }

      # Run tests with coverage
      function hbtc() {
        setCabalProjectLocalToBuild
        setCabalProjectLocalToCoverage

        TOOL_NAME=$1
        cabal --builddir=./dist-coverage build $TOOL_NAME && cabal --builddir=./dist-coverage test $TOOL_NAME
      }

      # Do cabal run
      function hbr() {
        setCabalProjectLocalToBuild

        TOOL_NAME=$1
        cabal --builddir=./dist-build run $TOOL_NAME -- ''${@:2}
      }

      PATH=$PATH:~/.local/bin
    '';

    oh-my-zsh = {
      enable = true;
      plugins = [
        "git"
        "bundler"
        "gem"
        "powder"
        "rake"
        "themes"
        "history"
        "z"
        "brew"
      ];
      theme = "muse";
    };
  };
}