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 --impure --extra-experimental-features nix-command && 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;
  };

  programs.starship = {
    enable = true;
    enableZshIntegration = true;
    settings = {
      time = {
        disabled = false;
        format = "[$time]($style) ";
        time_format = "%T"; # 24-hour format HH:MM:SS
        style = "bold yellow";
      };
      right_format = "$time";
    };
  };

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

    # Called whenever zsh is initialized
    initContent = lib.mkBefore ''
      export ZSH=${pkgs.oh-my-zsh}/share/oh-my-zsh/
      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

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

      # Build and test a Haskell project
      function hbt() {
        TOOL_NAME=$1
        clear && cabal build $TOOL_NAME && cabal test $TOOL_NAME
      }

      # Build, test, and install a Haskell tool
      function hbti() {
        TOOL_NAME=$1
        clear && cabal build $TOOL_NAME && cabal test $TOOL_NAME && cabal install $TOOL_NAME --overwrite-policy=always
      }

      # Debug a Haskell project with ghcid
      function hdbg() {
        TOOL_NAME=$1
        ghcid -c "cabal repl --enable-multi-repl --ghc-options=-Wwarn --builddir=./dist-debug $TOOL_NAME"
      }

      # Run the Haskell REPL
      function hrepl() {
        TOOL_NAME=$1
        cabal repl --enable-multi-repl --ghc-options=-Wwarn --builddir=./dist-debug $TOOL_NAME
      }

      # Do cabal run
      function hbr() {
        TOOL_NAME=$1
        cabal run $TOOL_NAME -- ''${@:2}
      }

      # Create git worktree with Haskell project files
      function gt() {
        # Check if we're in a git repository root
        if [[ ! -d .git ]]; then
          echo "Error: Not in a git repository root directory"
          return 1
        fi

        # Require branch name argument
        if [[ -z "$1" ]]; then
          echo "Usage: gt <branch-name>"
          echo "Creates a git worktree and copies Haskell build artifacts if present"
          return 1
        fi

        local branch_name="$1"
        local worktree_path=".worktrees/$branch_name"

        # Create the git worktree
        echo "Creating git worktree for branch '$branch_name'..."
        git worktree add -b "$branch_name" "$worktree_path"

        if [[ $? -ne 0 ]]; then
          echo "Failed to create git worktree"
          return 1
        fi

        echo "Worktree created at: $worktree_path"

        # Copy cabal.project.local if it exists
        if [[ -f cabal.project.local ]]; then
          echo "Copying cabal.project.local..."
          cp cabal.project.local "$worktree_path/"
        fi

        # Copy dist directory if it exists
        if [[ -d dist-newstyle ]]; then
          echo "Copying dist-newstyle..."
          cp -r dist-newstyle "$worktree_path/"
        fi

        # Copy vendor directory if it exists
        if [[ -d vendor ]]; then
          echo "Copying vendor..."
          cp -r vendor "$worktree_path/"
        fi

        # Copy pre-commit config symlink if it exists
        if [[ -L .pre-commit-config.yaml ]]; then
          echo "Copying .pre-commit-config.yaml symlink..."
          local target=$(readlink .pre-commit-config.yaml)
          ln -sf "$target" "$worktree_path/.pre-commit-config.yaml"
        fi

        echo "Git worktree setup complete!"
      }

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

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