{ config, pkgs, ... }:

{
  # Install fastfetch
  environment.systemPackages = with pkgs; [
    fastfetch
  ];

  # Create the fastfetch config file with unicode icons
  environment.etc."fastfetch/config.jsonc".text = ''
    {
      "$schema": "https://github.com/fastfetch-cli/fastfetch/raw/dev/doc/json_schema.json",
      "logo": {
        "source": "auto",
        "padding": {
          "top": 1,
          "left": 2
        }
      },
      "display": {
        "separator": " ",
        "color": {
          "keys": "blue",
          "title": "cyan"
        }
      },
      "modules": [
        {
          "type": "title",
          "format": "{user-name-colored}@{host-name-colored}"
        },
        {
          "type": "separator",
          "string": "─────────────────────────────────"
        },
        {
          "type": "os",
          "key": " ",
          "keyColor": "blue"
        },
        {
          "type": "host",
          "key": "󰌢 ",
          "keyColor": "blue"
        },
        {
          "type": "kernel",
          "key": " ",
          "keyColor": "blue"
        },
        {
          "type": "uptime",
          "key": "󰔟 ",
          "keyColor": "blue"
        },
        {
          "type": "packages",
          "key": "󰏖 ",
          "keyColor": "blue"
        },
        {
          "type": "shell",
          "key": " ",
          "keyColor": "blue"
        },
        {
          "type": "terminal",
          "key": " ",
          "keyColor": "blue"
        },
        {
          "type": "de",
          "key": " ",
          "keyColor": "blue"
        },
        {
          "type": "wm",
          "key": " ",
          "keyColor": "blue"
        },
        {
          "type": "wmtheme",
          "key": "󰉼 ",
          "keyColor": "blue"
        },
        {
          "type": "icons",
          "key": "󰀻 ",
          "keyColor": "blue"
        },
        {
          "type": "cpu",
          "key": "󰻠 ",
          "keyColor": "green"
        },
        {
          "type": "gpu",
          "key": "󰍛 ",
          "keyColor": "green"
        },
        {
          "type": "memory",
          "key": "󰑭 ",
          "keyColor": "yellow"
        },
        {
          "type": "disk",
          "key": "󰋊 ",
          "keyColor": "yellow"
        },
        {
          "type": "separator",
          "string": "─────────────────────────────────"
        },
        {
          "type": "colors",
          "paddingLeft": 2,
          "symbol": "circle"
        }
      ]
    }
  '';

  # Set up bash with fastfetch and a nice prompt
  programs.bash.interactiveShellInit = ''
    # Run fastfetch on terminal start (but not in non-interactive shells)
    if [[ $- == *i* ]]; then
      ${pkgs.fastfetch}/bin/fastfetch --config /etc/fastfetch/config.jsonc
    fi

    # Color definitions
    RESET="\[\033[0m\]"
    BOLD="\[\033[1m\]"
    
    # Colors
    CYAN="\[\033[0;36m\]"
    BLUE="\[\033[0;34m\]"
    PURPLE="\[\033[0;35m\]"
    GREEN="\[\033[0;32m\]"
    YELLOW="\[\033[0;33m\]"
    RED="\[\033[0;31m\]"
    
    # Bold colors
    BCYAN="\[\033[1;36m\]"
    BBLUE="\[\033[1;34m\]"
    BPURPLE="\[\033[1;35m\]"
    BGREEN="\[\033[1;32m\]"
    
    # Function to get git branch
    parse_git_branch() {
      git branch 2> /dev/null | sed -e '/^[^*]/d' -e 's/* \(.*\)/ (\1)/'
    }
    
    # Stylish prompt with icons
    # Shows: ╭─ user@host  ~/current/directory (git-branch)
    #        ╰─❯ 
    PS1="${RESET}${BCYAN}╭─${RESET} ${BPURPLE}\u${RESET}${CYAN}@${RESET}${BBLUE}\h${RESET} ${BGREEN} \w${RESET}${YELLOW}\$(parse_git_branch)${RESET}\n${BCYAN}╰─${BPURPLE}❯${RESET} "
  '';
}