#!/bin/bash

# Constants
gfetch="gfetch -d churn contributors --no-color-palette --no-title"
eza="eza --header --icons --all --color=always --group-directories-first --hyperlink"
projects_dir="$HOME/dev"
projects_command="find $projects_dir -mindepth 2 -maxdepth 2 -type d"
fzf_command="fzf --height=100%"

yaml_config() {
  local yaml_file="$1"
  local dest_dir="$2"
  local dest_yaml_file="$dest_dir/.tmuxp.yaml"

  cp "$yaml_file" "$dest_yaml_file"
  sed -i "1i start_directory: $dest_dir\nsession_name: $tab_title" "$dest_yaml_file"
}

tmuxp_load() {
  if [[ -n "$TMUX" ]]; then
    tmuxp load -d "$selected_dir"
  else
    tmuxp load "$selected_dir"

  fi
  tmux switch-client -t "$tab_title"
}

handle_tmuxp() {
  if [[ -f "$selected_dir/.tmuxp.yaml" ]]; then
    tmuxp_load
  else
    if [[ -d "$XDG_CONFIG_HOME/tmuxp" ]]; then
      local yaml_file=$(find "$XDG_CONFIG_HOME/tmuxp" -name "*.yaml" -print0 |
        xargs -0 -n 1 basename | sed 's/\.yaml$//' |
        fzf --preview "bat --style=numbers --color=always $XDG_CONFIG_HOME/tmuxp/{}.yaml" \
          --prompt="Select a session file: ")
      if [[ -n "$yaml_file" ]]; then
        yaml_config "$XDG_CONFIG_HOME/tmuxp/$yaml_file.yaml" "$selected_dir"
        tmuxp_load
      fi
    fi
  fi
}

handle_tmux() {
  if [[ -n "$TMUX" ]]; then
    tmux new-session -ds "$tab_title" -c "$selected_dir"
    tmux switch-client -t "$tab_title"
  else
    tmux new-session -s "$tab_title" -c "$selected_dir"
  fi
}

handle_kitty() {
  kitty @ set-tab-title "$tab_title"
  cd "$selected_dir"
  $gfetch
  $eza
}

selected_dir=$($projects_command | $fzf_command --prompt='Select a directory: ' \
  --header='CTRL-P: Projects / CTRL-Z: Zoxide' \
  --bind "ctrl-p:reload($projects_command)" \
  --bind "ctrl-z:reload(zoxide query -l)" \
  --preview "$gfetch {} 2>&1 | grep -v 'Error: Could not find a git repository in' || true && $eza {} || $eza {}")

tab_title=$(basename "$selected_dir")

if [[ -n "$selected_dir" ]]; then
  if command -v tmux &>/dev/null; then
    if command -v tmuxp &>/dev/null; then
      handle_tmuxp
    else
      handle_tmux
    fi
  elif [[ "$TERM" == "xterm-kitty" ]]; then
    handle_kitty
  fi
fi