nixos/modules/bash/default.nix

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

with lib;
let cfg = config.modules.bash;

in {
  options.modules.bash = { enable = mkEnableOption "bash"; };
  config = mkIf cfg.enable {
    programs.bash = {
      enable = true;
      enableCompletion = true;

      initExtra = ''
        eval "$(direnv hook bash)"

        is_ssh_session() {
          if [ -n "$SSH_CLIENT" ] || [ -n "$SSH_TTY" ]; then
            return 0
          else
            return 1
          fi
        }

        function set_ps1_prompt() {
          local git_branch=""
          local flake_icon=""
          local cur_dir=""

          if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
            git_branch="$(git symbolic-ref --short HEAD 2>/dev/null)"

            if [ $? -ne 0 ]; then
              git_branch="$(git rev-parse --short HEAD 2>/dev/null)"
            fi

            git_branch=" \[\033[01;31m\]$git_branch󰘬:\[\033[00m\]"

            if [ -f "$(git rev-parse --show-toplevel)/flake.nix" ]; then
                # If it exists, set the flake icon and color it blue
                flake_icon="\[\033[01;34m\] \[\033[00m\]"
            fi

            git_root="$(basename "$(git rev-parse --show-toplevel)")"

            cur_dir=$(realpath --relative-to=$(git rev-parse --show-toplevel) .)
            if [ "$cur_dir" == "." ]; then
              cur_dir="\[\033[01;34m\] $git_root\[\033[00m\]"
            else
              cur_dir="\[\033[01;34m\] $git_root/$cur_dir\[\033[00m\]"
            fi
          else
            cur_dir="\[\033[01;34m\]\w\[\033[00m\]"
          fi

          if [ -n "${IN_NIX_SHELL:+x}" ]; then
            PS1="$cur_dir\n$flake_icon\[\033[01;32m\]nixShell>$git_branch\[\033[00m\]"
          else
            if ! is_ssh_session; then
              PS1="\n$cur_dir\n$flake_icon\[\033[01;32m\]>$git_branch\[\033[00m\]"
            else
              PS1="\n\[\033[01;34m\]\w\[\033[00m\]\n\[\033[01;32m\]\u@\h:\[\033[00m\] "
            fi
          fi
          unset flake_icon
        }
        PROMPT_COMMAND="set_ps1_prompt; $PROMPT_COMMAND"
        '';

        bashrcExtra = ''
          function cdg() {
            if [[ $1 == "--help" ]]; then
              echo "A simple utility for navigating to the root of a git repo"
              return 0
            fi

            # Check for invalid command
            if [[ -n "$1" ]]; then
              echo "Invalid command: $1. Try 'cdg --help'."
              return 1
            fi

            local root_dir
            root_dir=$(git rev-parse --show-toplevel 2>/dev/null)
            local git_status=$?

            if [ $git_status -ne 0 ]; then
              echo "Error: Not a git repo."
              return 1
            fi

            cd "$root_dir"
          }

          function ldv() {
          if [[ $1 == "help" ]] || [[ $1 == "-h" ]] || [[ $1 == "--help" ]]; then
            cat << EOF
          ldv
          A simple utility for setting up development environments effortlessly.
          Commands:
            ldv               Start a preconfigured nix shell.
            init              Create a new dev template in the current working directory.
            help              Show available commands and options.

          Contributions welcome: https://github.com/itme-brain/ldv
          EOF

          elif [[ $1 == "init" ]] || [[ $1 == "-i" ]] || [[ $1 == "--init" ]]; then
            if [[ -e ./shell.nix ]] || [[ -e ./.envrc ]]; then
              cat << EOF
          Existing environment found.
          Initialization cancelled.
          EOF
              return
            fi
            
            cat << EOF
          Initializing a new environment...
          Select an environment:
          1. Web
          2. Elixir
          3. Haskell
          EOF

            read -p "Enter the number of your choice: " choice

            case $choice in
              1)
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/shells/web.nix -O shell.nix
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/utils/flake -O flake.nix
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/utils/envrc -O .envrc
                direnv allow
                ;;
              2)
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/shells/elixir.nix -O shell.nix
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/utils/flake -O flake.nix
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/utils/envrc -O .envrc
                direnv allow
                ;;
              3)
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/shells/haskell.nix -O shell.nix
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/utils/flake -O flake.nix
                wget -q https://raw.githubusercontent.com/itme-brain/ldv/main/utils/envrc -O .envrc
                direnv allow
                ;;
              *)
                echo "Invalid choice."
                ;;
            esac
          elif [[ -z $1 ]]; then
            cat << EOF
          Select an environment:
          1. Web
          2. Elixir
          3. Haskell
          EOF

            read -p "Enter the number of your choice: " choice

            case $choice in
              1)
                (nix develop github:itme-brain/ldv#web)
                ;;
              2)
                (nix develop github:itme-brain/ldv#elixir)
                ;;
              3)
                (nix develop github:itme-brain/ldv#haskell)
                ;;
              # Add more cases here...
              *)
                echo "Invalid choice."
                ;;
            esac
          else
            echo "Error: Invalid command. Try 'ldv --help'"
          fi
          }
        '';

      shellAliases = {
        ls = "lsd";
        hmup = "home-manager switch --flake '$HOME/Documents/projects/nixos#bryan'";
        nixup = "sudo nixos-rebuild switch --flake '$HOME/Documents/projects/nixos#socrates'";
      };
    };
  };
}