add: better rofi theme selector

1 files changed, 139 insertions, 60 deletions

diff --git a/config/hypr/scripts/RofiThemeSelector.sh b/config/hypr/scripts/RofiThemeSelector.sh
index 6fd8a6f8..8423b2fe 100755
--- a/config/hypr/scripts/RofiThemeSelector.sh
+++ b/config/hypr/scripts/RofiThemeSelector.sh
@@ -1,75 +1,154 @@
 #!/bin/bash
-# /* ---- 💫 https://github.com/JaKooLit 💫 ---- */  ##
-# Script for adding a selected theme to the Rofi config
+# /* ---- 💫 https://github.com/JaKooLit 💫 ---- */  #
+# Rofi Themes - Script to preview and apply themes by live-reloading the config.
 
-IFS=$'\n\t'
+# --- Configuration ---
+ROFI_THEMES_DIR_CONFIG="$HOME/.config/rofi/themes"
+ROFI_THEMES_DIR_LOCAL="$HOME/.local/share/rofi/themes"
+ROFI_CONFIG_FILE="$HOME/.config/rofi/config.rasi"
+ROFI_THEME_FOR_THIS_SCRIPT="$HOME/.config/rofi/config-rofi-theme.rasi" # A separate rofi theme for the picker itself
+IDIR="$HOME/.config/swaync/images"                                     # For notifications
 
-# Define directories and variables
-rofi_theme_dir="$HOME/.config/rofi/themes"
-rofi_config_file="$HOME/.config/rofi/config.rasi"
-SED=$(which sed)
-iDIR="$HOME/.config/swaync/images"
-rofi_theme="$HOME/.config/rofi/config-rofi-theme.rasi"
+# --- Helper Functions ---
 
-# Function to display menu options
-menu() {
-    options=()
-    while IFS= read -r file; do
-        options+=("$file")
-    done < <(find -L "$rofi_theme_dir" -maxdepth 1 -type f -exec basename {} \; | sort -V)
-
-    printf '%s\n' "${options[@]}"
+# Function to send a notification
+notify_user() {
+  notify-send -u low -i "$1" "$2" "$3"
 }
 
-# Function to add or update theme in the config.rasi
-add_theme_to_config() {
-    local theme_name="$1"
-    local theme_path="$rofi_theme_dir/$theme_name"
-        
-    # if config in $HOME to write as $HOME 
-    if [[ "$theme_path" == $HOME/* ]]; then
-        theme_path_with_tilde="~${theme_path#$HOME}"
-    else
-        theme_path_with_tilde="$theme_path"
-    fi
+# Function to apply the selected rofi theme to the main config file
+apply_rofi_theme_to_config() {
+  local theme_name_to_apply="$1"
 
-    # If no @theme is in the file, add it
-    if ! grep -q '^\s*@theme' "$rofi_config_file"; then
-        echo -e "\n\n@theme \"$theme_path_with_tilde\"" >> "$rofi_config_file"
-        echo "Added @theme \"$theme_path_with_tilde\" to $rofi_config_file"
-    else
-        $SED -i "s/^\(\s*@theme.*\)/\/\/\1/" "$rofi_config_file"
-        echo -e "@theme \"$theme_path_with_tilde\"" >> "$rofi_config_file"
-        echo "Updated @theme line to $theme_path_with_tilde"
-    fi
+  # Find the full path of the theme file
+  local theme_path
+  if [[ -f "$ROFI_THEMES_DIR_CONFIG/$theme_name_to_apply" ]]; then
+    theme_path="$ROFI_THEMES_DIR_CONFIG/$theme_name_to_apply"
+  elif [[ -f "$ROFI_THEMES_DIR_LOCAL/$theme_name_to_apply" ]]; then
+    theme_path="$ROFI_THEMES_DIR_LOCAL/$theme_name_to_apply"
+  else
+    notify_user "$IDIR/error.png" "Error" "Theme file not found: $theme_name_to_apply"
+    return 1
+  fi
 
-    # Ensure no more than max # of lines with //@theme lines
-    max_line="9"
-    total_lines=$(grep -c '^\s*//@theme' "$rofi_config_file")
+  # Use ~ for the home directory in the config path
+  local theme_path_with_tilde="~${theme_path#$HOME}"
 
-    if [ "$total_lines" -gt "$max_line" ]; then
-        excess=$((total_lines - max_line))
-        # Remove the oldest or the very top //@theme lines
-        for i in $(seq 1 "$excess"); do
-            $SED -i '0,/^\s*\/\/@theme/ { /^\s*\/\/@theme/ {d; q; }}' "$rofi_config_file"
-        done
-        echo "Removed excess //@theme lines"
-    fi
-}
+  # Create a temporary file to safely edit the config
+  local temp_rofi_config_file
+  temp_rofi_config_file=$(mktemp)
+  cp "$ROFI_CONFIG_FILE" "$temp_rofi_config_file"
 
-# Main function
-main() {
-    choice=$(menu | rofi rofi -dmenu -i -config $rofi_theme)
+  # Comment out any existing @theme entry
+  sed -i -E 's/^(\s*@theme)/\\/\\/\1/' "$temp_rofi_config_file"
 
-    if [[ -z "$choice" ]]; then
-        exit 0
-    fi
-    add_theme_to_config "$choice"
-    notify-send -i "$iDIR/ja.png" -u low 'Rofi Theme applied:' "$choice"
+  # Add the new @theme entry at the end of the file
+  echo "@theme \"$theme_path_with_tilde\"" >>"$temp_rofi_config_file"
+
+  # Overwrite the original config file
+  cp "$temp_rofi_config_file" "$ROFI_CONFIG_FILE"
+  rm "$temp_rofi_config_file"
+
+  # Prune old commented-out theme lines to prevent clutter
+  local max_lines=10
+  local total_lines=$(grep -c '^//\s*@theme' "$ROFI_CONFIG_FILE")
+  if [ "$total_lines" -gt "$max_lines" ]; then
+    local excess=$((total_lines - max_lines))
+    for ((i = 1; i <= excess; i++)); do
+      sed -i '0,/^\s*\/\/@theme/s///' "$ROFI_CONFIG_FILE"
+    done
+  fi
+
+  return 0
 }
 
-if pgrep -x "rofi" >/dev/null; then
-    pkill rofi
+# --- Main Script Execution ---
+
+# Check for required directories and files
+if [ ! -d "$ROFI_THEMES_DIR_CONFIG" ] && [ ! -d "$ROFI_THEMES_DIR_LOCAL" ]; then
+  notify_user "$IDIR/error.png" "E-R-R-O-R" "No Rofi themes directory found."
+  exit 1
+fi
+
+if [ ! -f "$ROFI_CONFIG_FILE" ]; then
+  notify_user "$IDIR/error.png" "E-R-R-O-R" "Rofi config file not found: $ROFI_CONFIG_FILE"
+  exit 1
+fi
+
+# Backup the original config content
+original_rofi_config_content_backup=$(cat "$ROFI_CONFIG_FILE")
+
+# Generate a sorted list of available theme file names
+mapfile -t available_theme_names < <((
+  find "$ROFI_THEMES_DIR_CONFIG" -maxdepth 1 -name "*.rasi" -type f -printf "%f\n" 2>/dev/null
+  find "$ROFI_THEMES_DIR_LOCAL" -maxdepth 1 -name "*.rasi" -type f -printf "%f\n" 2>/dev/null
+) | sort -u)
+
+if [ ${#available_theme_names[@]} -eq 0 ]; then
+  notify_user "$IDIR/error.png" "No Rofi Themes" "No .rasi files found in theme directories."
+  exit 1
 fi
 
-main
+# Find the currently active theme to set as the initial selection
+current_selection_index=0
+current_active_theme_path=$(grep -oP '^\s*@theme\s*"\K[^"]+' "$ROFI_CONFIG_FILE" | tail -n 1)
+if [ -n "$current_active_theme_path" ]; then
+  current_active_theme_name=$(basename "$current_active_theme_path")
+  for i in "${!available_theme_names[@]}"; do
+    if [[ "${available_theme_names[$i]}" == "$current_active_theme_name" ]]; then
+      current_selection_index=$i
+      break
+    fi
+  done
+fi
+
+# Main preview loop
+while true; do
+  theme_to_preview_now="${available_theme_names[$current_selection_index]}"
+
+  # Apply the theme for preview
+  if ! apply_rofi_theme_to_config "$theme_to_preview_now"; then
+    echo "$original_rofi_config_content_backup" >"$ROFI_CONFIG_FILE"
+    notify_user "$IDIR/error.png" "Preview Error" "Failed to apply $theme_to_preview_now. Reverted."
+    exit 1
+  fi
+
+  # Prepare theme list for Rofi
+  rofi_input_list=""
+  for theme_name_in_list in "${available_theme_names[@]}"; do
+    rofi_input_list+="$(basename "$theme_name_in_list" .rasi)\n"
+  done
+  rofi_input_list_trimmed="${rofi_input_list%\\n}"
+
+  # Launch Rofi and get user's choice
+  chosen_index_from_rofi=$(echo -e "$rofi_input_list_trimmed" |
+    rofi -dmenu -i \
+      -format 'i' \
+      -p "Rofi Theme" \
+      -mesg "Preview: $(basename "$theme_to_preview_now" .rasi) | Enter: Preview | Ctrl+S: Apply & Exit | Esc: Cancel" \
+      -config "$ROFI_THEME_FOR_THIS_SCRIPT" \
+      -selected-row "$current_selection_index" \
+      -kb-custom-1 "Control+s")
+
+  rofi_exit_code=$?
+
+  # Handle Rofi's exit code
+  if [ $rofi_exit_code -eq 0 ]; then # Enter
+    if [[ "$chosen_index_from_rofi" =~ ^[0-9]+$ ]] && [ "$chosen_index_from_rofi" -lt "${#available_theme_names[@]}" ]; then
+      current_selection_index="$chosen_index_from_rofi"
+    fi
+  elif [ $rofi_exit_code -eq 1 ]; then # Escape
+    notify_user "$IDIR/note.png" "Rofi Theme" "Selection cancelled. Reverting to original theme."
+    echo "$original_rofi_config_content_backup" >"$ROFI_CONFIG_FILE"
+    break
+  elif [ $rofi_exit_code -eq 10 ]; then # Custom bind 1 (Ctrl+S)
+    notify_user "$IDIR/ja.png" "Rofi Theme Applied" "$(basename "$theme_to_preview_now" .rasi)"
+    break
+  else # Error or unexpected exit code
+    notify_user "$IDIR/error.png" "Rofi Error" "Unexpected Rofi exit ($rofi_exit_code). Reverting."
+    echo "$original_rofi_config_content_backup" >"$ROFI_CONFIG_FILE"
+    break
+  fi
+done
+
+exit 0