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

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";
      nreb = "sudo nixos-rebuild switch --flake ~/.config/nix#$(hostname)";
      
    } // lib.optionalAttrs pkgs.stdenv.isDarwin {
      # macOS-specific app aliases
      preview = "open -a Preview";
      slide = "open -a SlidePilot";
      pixel = "open -a 'Pixelmator Pro'";
      inkscape = "open -a Inkscape";
    };
    
    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 target
        target=$(echo "" | fzf --bind "change:reload:fd --hidden --follow --exclude .git {q} ~ 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 target
        target=$(echo "" | fzf --bind "change:reload:fd --hidden --follow --exclude .git {q} ~ 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
      }
      
      # Function to search and open all macOS applications (macOS only)
      ${lib.optionalString pkgs.stdenv.isDarwin ''
        function app() {
          local app_path
          local file_to_open="$1"
          
          app_path=$( (find -L /Applications -name "*.app" -maxdepth 2 2>/dev/null; \
                       find -L ~/Applications -name "*.app" -maxdepth 3 2>/dev/null; \
                       find /System/Applications -name "*.app" -maxdepth 2 2>/dev/null; \
                       find /System/Applications/Utilities -name "*.app" -maxdepth 1 2>/dev/null) | 
            sort | uniq | 
            fzf --header="Select app to open''${file_to_open:+ file: $file_to_open}" \
                --preview 'basename {} .app' \
                --preview-window=up:1 \
                --height=40%)
          
          if [[ -n "$app_path" ]]; then
            if [[ -n "$file_to_open" ]]; then
              open -a "$app_path" "$file_to_open"
            else
              open "$app_path"
            fi
          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;
}