#!/usr/bin/env bash
# ==================================================
#  KoolDots (2026)
#  Project URL: https://github.com/LinuxBeginnings
#  License: GNU GPLv3
#  SPDX-License-Identifier: GPL-3.0-or-later
# ==================================================
# Ghostty theme selector

config_file="${XDG_CONFIG_HOME:-$HOME/.config}/ghostty/config"
iDIR="${XDG_CONFIG_HOME:-$HOME/.config}/swaync/images"
rofi_theme_primary="${XDG_CONFIG_HOME:-$HOME/.config}/rofi/config-ghostty-theme.rasi"
rofi_theme_fallback="${XDG_CONFIG_HOME:-$HOME/.config}/rofi/config-edit.rasi"
wallust_include_path="${XDG_CONFIG_HOME:-$HOME/.config}/ghostty/wallust.conf"
wallust_option_label="Set by wallpaper"
default_option_label="Default - no color"
wallust_refresh_script="${XDG_CONFIG_HOME:-$HOME/.config}/hypr/scripts/WallustSwww.sh"

notify_user() {
  local icon="$1"
  local title="$2"
  local body="$3"
  if [[ -n "$icon" && -f "$icon" ]]; then
    notify-send -u low -i "$icon" "$title" "$body"
  else
    notify-send -u low "$title" "$body"
  fi
}

refresh_wallpaper_theme() {
  if [[ -x "$wallust_refresh_script" ]]; then
    "$wallust_refresh_script" >/dev/null 2>&1 || true
  fi
  pkill -SIGUSR2 ghostty >/dev/null 2>&1 || true
}

if [[ ! -f "$config_file" ]]; then
  notify_user "$iDIR/error.png" "Ghostty Theme" "Config not found: $config_file"
  exit 1
fi

rofi_config_args=()
if [[ -f "$rofi_theme_primary" ]]; then
  rofi_config_args=(-config "$rofi_theme_primary")
elif [[ -f "$rofi_theme_fallback" ]]; then
  rofi_config_args=(-config "$rofi_theme_fallback")
fi

current_theme=$(
  awk -F'=' '/^[[:space:]]*theme[[:space:]]*=/ {
    val=$2
    sub(/^[[:space:]]+/, "", val)
    sub(/[[:space:]]+$/, "", val)
    gsub(/^"|"$/, "", val)
    print val
    exit
  }' "$config_file"
)

wallust_enabled=$(
  awk '/^[[:space:]]*config-file[[:space:]]*=/ && $0 !~ /^[[:space:]]*#/ && /wallust\.conf/ {print "1"; exit}' "$config_file"
)
[[ "$wallust_enabled" != "1" ]] && wallust_enabled="0"

mapfile -t available_theme_names < <(
  awk -F'=' '/^[[:space:]]*#[[:space:]]*theme[[:space:]]*=/ {
    val=$2
    sub(/^[[:space:]]+/, "", val)
    sub(/[[:space:]]+$/, "", val)
    gsub(/^"|"$/, "", val)
    print val
  }' "$config_file"
)

if [[ ${#available_theme_names[@]} -eq 0 ]]; then
  notify_user "$iDIR/error.png" "Ghostty Theme" "No commented themes found in $config_file"
  exit 1
fi
menu_entries=("$wallust_option_label" "$default_option_label")
for t in "${available_theme_names[@]}"; do
  menu_entries+=("$t")
done
current_selection_index=0
if [[ "$wallust_enabled" == "1" ]]; then
  current_selection_index=0
elif [[ -z "$current_theme" ]]; then
  current_selection_index=1
else
  current_selection_index=1
  for i in "${!available_theme_names[@]}"; do
    if [[ "${available_theme_names[$i]}" == "$current_theme" ]]; then
      current_selection_index=$((i + 2))
      break
    fi
  done
fi

choice=$(
  printf "%s\n" "${menu_entries[@]}" |
    rofi -i -dmenu -p "Ghostty Theme" "${rofi_config_args[@]}" -mesg "Select a theme to apply" -selected-row "$current_selection_index"
)

[[ -z "$choice" ]] && exit 0


selected_theme="$choice"

if [[ "$selected_theme" == "$wallust_option_label" ]]; then
  if [[ "$wallust_enabled" == "1" ]]; then
    exit 0
  fi

  tmp_file=$(mktemp)
  awk -v wallust_include_path="$wallust_include_path" '
function trim(s) { sub(/^[[:space:]]+/, "", s); sub(/[[:space:]]+$/, "", s); return s }
{
  line=$0
  if ($0 ~ /^[[:space:]]*theme[[:space:]]*=/) {
    sub(/^[[:space:]]*theme[[:space:]]*=/, "#theme =", line)
    print line
    next
  }
  if ($0 ~ /^[[:space:]]*#?[[:space:]]*config-file[[:space:]]*=/ && $0 ~ /wallust\.conf/) {
    print "config-file = " wallust_include_path
    wallust_set=1
    next
  }
  print $0
}
END {
  if (!wallust_set) {
    print "config-file = " wallust_include_path
  }
}
' "$config_file" > "$tmp_file"
  mv "$tmp_file" "$config_file"
  refresh_wallpaper_theme
  notify_user "$iDIR/ja.png" "Ghostty Theme Applied" "$wallust_option_label"
  exit 0
fi
if [[ "$selected_theme" == "$default_option_label" ]]; then
  if [[ "$wallust_enabled" != "1" && -z "$current_theme" ]]; then
    exit 0
  fi

  tmp_file=$(mktemp)
  awk -v wallust_include_path="$wallust_include_path" '
{
  line=$0
  if ($0 ~ /^[[:space:]]*theme[[:space:]]*=/) {
    sub(/^[[:space:]]*theme[[:space:]]*=/, "#theme =", line)
    print line
    next
  }
  if ($0 ~ /^[[:space:]]*#?[[:space:]]*config-file[[:space:]]*=/ && $0 ~ /wallust\.conf/) {
    print "#config-file = " wallust_include_path
    wallust_seen=1
    next
  }
  print $0
}
END {
  if (!wallust_seen) {
    print "#config-file = " wallust_include_path
  }
}
' "$config_file" > "$tmp_file"
  mv "$tmp_file" "$config_file"

  pkill -SIGUSR2 ghostty >/dev/null 2>&1 || true
  notify_user "$iDIR/ja.png" "Ghostty Theme Applied" "$default_option_label"
  exit 0
fi

if [[ "$wallust_enabled" != "1" && -n "$current_theme" && "$selected_theme" == "$current_theme" ]]; then
  exit 0
fi

format_theme_value() {
  if [[ "$1" =~ [[:space:]] ]]; then
    printf "\"%s\"" "$1"
  else
    printf "%s" "$1"
  fi
}

selected_formatted=$(format_theme_value "$selected_theme")

tmp_file=$(mktemp)
awk -v selected="$selected_theme" -v selected_formatted="$selected_formatted" -v wallust_include_path="$wallust_include_path" '
function trim(s) { sub(/^[[:space:]]+/, "", s); sub(/[[:space:]]+$/, "", s); return s }
function strip_quotes(s) { gsub(/^"|"$/, "", s); return s }
{
  line=$0
  if ($0 ~ /^[[:space:]]*#?[[:space:]]*config-file[[:space:]]*=/ && $0 ~ /wallust\.conf/) {
    print "#config-file = " wallust_include_path
    wallust_seen=1
    next
  }
  if ($0 ~ /^[[:space:]]*theme[[:space:]]*=/) {
    sub(/^[[:space:]]*theme[[:space:]]*=/, "#theme =", line)
    print line
    next
  }
  if ($0 ~ /^[[:space:]]*#[[:space:]]*theme[[:space:]]*=/) {
    val=$0
    sub(/^[[:space:]]*#[[:space:]]*theme[[:space:]]*=[[:space:]]*/, "", val)
    val=trim(val)
    val=strip_quotes(val)
    if (val == selected) {
      print "theme = " selected_formatted
      next
    }
  }
  print $0
}
END {
  if (!wallust_seen) {
    print "#config-file = " wallust_include_path
  }
}' "$config_file" > "$tmp_file"

mv "$tmp_file" "$config_file"

pkill -SIGUSR2 ghostty >/dev/null 2>&1 || true

notify_user "$iDIR/ja.png" "Ghostty Theme Applied" "$selected_theme"

exit 0