nix/modules/zsh.nix

{ config, pkgs, ... }:

let
  projectsConfig = import ../config/projects.nix { homeDirectory = config.home.homeDirectory; };
  projectLauncher = "${config.home.homeDirectory}/.config/nix/scripts/project-launcher.sh";
in
{
  programs.zsh = {
    enable = true;
    defaultKeymap = "viins";
    enableVteIntegration = true;
    enableCompletion = true;
    autosuggestion.enable = true;
    syntaxHighlighting.enable = true;
    
    sessionVariables = {
      COLORTERM = "truecolor";
      EDITOR = "nvim";
      TERM = "xterm-256color";
    };
    
    shellAliases = {
      ll = "ls -alF";
      zi = "z -i";  # Interactive selection with fzf

      # Nix helpers
      hm = "home-manager";
      hms = "home-manager switch --flake ~/.config/nix#$(whoami)@$(hostname)";
      hms-offline = "home-manager switch --flake ~/.config/nix#$(whoami)@$(hostname) --option substitute false";
      fs = "oss && hms";

      # Steam Deck screenshot and game recording sync
      deck-ss = "rsync -avh --progress 'deck:.local/share/Steam/userdata/*/760/remote/*/screenshots/' ~/Pictures/Steam/ && rsync -avh --progress 'deck:.local/share/Steam/userdata/*/gamerecordings/' ~/Pictures/Steam/recordings/";
    };
    
    initContent = ''
      # Load Powerlevel10k theme
      if [[ -f ${pkgs.zsh-powerlevel10k}/share/zsh-powerlevel10k/powerlevel10k.zsh-theme ]]; then
        source ${pkgs.zsh-powerlevel10k}/share/zsh-powerlevel10k/powerlevel10k.zsh-theme
      fi

      # Load Powerlevel10k configuration (managed by Nix)
      source ~/.p10k.zsh

      # Vim mode configuration
      # Reduce delay when switching modes (10ms instead of 400ms)
      export KEYTIMEOUT=1
      
      # Cursor shape changes for vim modes
      function zle-keymap-select {
        case $KEYMAP in
          vicmd)      echo -ne '\e[1 q';;  # block cursor for normal mode
          viins|main) echo -ne '\e[5 q';;  # line cursor for insert mode
        esac
      }
      zle -N zle-keymap-select
      
      # Ensure we start with line cursor in insert mode
      function zle-line-init {
        echo -ne '\e[5 q'
      }
      zle -N zle-line-init
      
      # Fix cursor after each command
      function preexec {
        echo -ne '\e[5 q'
      }
      
      # Additional vim-like bindings
      bindkey -M vicmd 'k' history-search-backward
      bindkey -M vicmd 'j' history-search-forward
      bindkey -M vicmd '/' history-incremental-search-backward
      bindkey -M vicmd '?' history-incremental-search-forward
      
      # Better word movement in insert mode
      bindkey '^[[1;5C' forward-word      # Ctrl+Right
      bindkey '^[[1;5D' backward-word     # Ctrl+Left
      
      # Fix backspace in vim insert mode
      bindkey '^?' backward-delete-char   # Backspace
      bindkey '^H' backward-delete-char   # Ctrl+H (alternative backspace)
      
      # Prevent Shift+A from triggering autocomplete in vim insert mode
      # Try multiple potential key sequences for Shift+A across different terminals
      bindkey -M viins 'A' self-insert
      bindkey -M viins '^[[1;2A' self-insert
      bindkey -M viins '^[[65;2u' self-insert
      
      # Disable expand-or-complete on potential problematic keys in vim insert mode
      bindkey -M viins '^I' expand-or-complete   # Keep tab completion but be explicit
      
      # Configure autosuggestions to not interfere with vim mode
      ZSH_AUTOSUGGEST_CLEAR_WIDGETS+=(vi-add-eol)
      ZSH_AUTOSUGGEST_CLEAR_WIDGETS+=(vi-add-next)
      ZSH_AUTOSUGGEST_ACCEPT_WIDGETS=(end-of-line vi-end-of-line vi-add-eol)
      
      # Zoxide configuration - replace cd with z for smart directory jumping
      eval "$(zoxide init zsh --cmd cd)"
      
      # Function to cd to directory containing a file selected with fzf
      function cdf() {
        local search_dir="''${1:-~}"
        local target
        target=$(echo "" | fzf --bind "change:reload:fd --hidden --follow --exclude .git {q} ''$search_dir 2>/dev/null || true" --header="Type to search, Enter to cd" --preview '([[ -d {} ]] && ls -la {}) || ([[ -f {} ]] && head -20 {})' --height 40% --ansi)
        if [[ -n "$target" ]]; then
          [[ -d "$target" ]] && cd "$target" || cd "$(dirname "$target")"
        fi
      }
      
      # Function to print path of file/directory selected with fzf
      function pwdf() {
        local search_dir="''${1:-~}"
        local target
        target=$(echo "" | fzf --bind "change:reload:fd --hidden --follow --exclude .git {q} ''$search_dir 2>/dev/null || true" --header="Type to search, Enter to print path" --preview '([[ -d {} ]] && ls -la {}) || ([[ -f {} ]] && head -20 {})' --height 40% --ansi)
        if [[ -n "$target" ]]; then
          echo "$target"
        fi
      }
      
      
      # Interactive project launcher with fzf
      function proj() {
        local project_json="$HOME/.config/nix/config/projects.json"
        local launcher="${projectLauncher}"
        
        # If arguments provided, pass directly to launcher
        if [[ $# -gt 0 ]]; then
          "$launcher" "$@"
          return
        fi
        
        # Interactive mode with fzf
        local project=$(jq -r '.projects | keys[]' "$project_json" 2>/dev/null | \
          fzf --header="Select project (ESC to cancel)" \
              --preview="echo '==== {} ====' && \
                         jq -r '.projects.\"{}\" | \"Description: \" + .description' \"$project_json\" && \
                         echo && \
                         echo 'Windows:' && \
                         jq -r '.projects.\"{}\" | .windows[] | \"  - \" + .name + \": \" + .path' \"$project_json\" && \
                         echo && \
                         if tmux has-session -t {} 2>/dev/null; then \
                           echo 'Status: \033[32mRunning\033[0m' && \
                           echo && \
                           echo 'Tmux windows:' && \
                           tmux list-windows -t {} -F \"  #{window_index}: #{window_name}\" 2>/dev/null; \
                         else \
                           echo 'Status: \033[33mNot running\033[0m'; \
                         fi" \
              --preview-window=right:60%:wrap \
              --height=80% \
              --ansi)
        
        [[ -n "$project" ]] && "$launcher" "$project"
      }
    '';
  };
  
  # Essential packages for enhanced zsh experience
  home.packages = with pkgs; [
    zsh-powerlevel10k
    fzf
    fd
    ripgrep
    bat
    jq
  ];
  
  programs.fzf = {
    enable = true;
    enableZshIntegration = true;
  };
  
  programs.zoxide = {
    enable = true;
    enableZshIntegration = true;
  };
  
  # Manage Powerlevel10k configuration
  home.file.".p10k.zsh".source = ../config/p10k.zsh;

  # Generate projects.json for shell scripts
  home.file.".config/nix/config/projects.json".text = builtins.toJSON projectsConfig;

  # Ensure Steam screenshots and recordings directories exist
  home.file."Pictures/Steam/.gitkeep".text = "";
  home.file."Pictures/Steam/recordings/.gitkeep".text = "";
}