#!/usr/bin/env sh

# This script generates a menu from the given list of items. Choosing an item runs the command for that item.
# Usage: 
#   change_my_password="Change my password%passwd"
#   list_users="List all users%cut -d: -f1 /etc/passwd"
#   exit="kill -2 $$"
#   menu "User management options:" "$change_my_password" "$list_users" "$exit"
# Use up/down arrow keys to select a menu item, then press Enter. Or cancel with ESC.

. colors

if [ ${#} -lt 2 ]; then
	mud-menu
	exit 0
fi

description=$1
shift

# Parse args. Each arg is "name%command"
items=""
commands=""
while [ ${#} -gt 0 ]; do
    IFS='%' read -r item command <<EOF
$1
EOF
    items="${items}:${item}"
    commands="${commands}:${command}"
    shift
done

selected=0
displayed=0

# Calculate the position of the menu
menu_x=$((0))
menu_y=$(fathom-cursor -y)

# Calculate the number of rows that the menu will occupy
num_rows=$(echo "$items" | tr -cd ':' | wc -c)

# Calculate the maximum length of the item names
# Todo: abstract this out into a max-length function in another script. I gave up on trying to find an elegant way to pass a list of strings with spaces. Even separating them with newlines didn't work.
max_name_length=0

# Find the longest line
set -f IFS=':'; set -- $items; set +f
for line; do
    line_length=${#line}
    if [ $line_length -gt $max_name_length ]; then
        max_name_length=$line_length
    fi
done

# Make sure the menu items don't wrap, multiline menu items are not supported
window_width=$(fathom-terminal -w)
if [ "${max_name_length:-0}" -gt "${window_width:-80}" ]; then
	max_name_length=$window_width
fi

# Define the function to display the menu
display_menu() {
  # Get the dimensions of the terminal window (again each time in case window is resized)
  window_width=$(fathom-terminal -w)
  window_height=$(fathom-terminal -h)
  
  # Recalculate the position of the menu
  menu_y=$(fathom-cursor -y)
  
  # Move the cursor to the correct position
  if [ $displayed -eq 0 ]; then
    # Don't move the cursor or clear the screen the first time
    displayed=1
  else
    if [ $((menu_y + num_rows)) -ge $window_height ]; then
    	menu_y=$((window_height))
    fi
    move-cursor $menu_x $((menu_y - num_rows)) 
  fi
  
  # Print the items and commands
  i=0
	set -f IFS=':'; set -- $items; set +f
	for item; do
		set -f IFS=':'; set -- $commands; set +f
		for j in $(seq 1 $selected); do shift; done
		command=$1
		max_command_length=$(( $window_width - $max_name_length - 3 )) # -3 is for the > and the space between item and command
		truncated_command=$(echo "${command}" | cut -c -$max_command_length)
		if [ $i -eq $selected ]; then
		  printf "${CYAN}> %-${max_name_length}s${RESET} ${GREY}%s${RESET}\n" "${item}" "$truncated_command"
		else
		  printf "  %-${max_name_length}s %*s\n" "${item}" "${#truncated_command}"
		fi
		i=$((i + 1))
	done
}

# Define the function to handle key presses
handle_key() {
  key=$(await-keypress)
  case "$key" in
    up)
      selected=$((selected - 1))
      if [ $selected -lt 0 ]; then
        selected=$((num_rows - 1))
      fi
      ;;
    down)
      selected=$((selected + 1))
      if [ $selected -ge $num_rows ]; then
        selected=0
      fi
      ;;
    enter)
      cursor-blink on
      command="${commands[$selected]}"
      printf "\n"
      eval $command 
      exit 0
      ;;
    escape) 
      cursor-blink on
      echo ESC
      exit
      ;;
    esac
}

# Catch Ctrl-C and restore the cursor blink, so the user doesn't get stuck with an invisible cursor
handle_ctrl_c() {
  trap 'cursor-blink on; exit' INT
}

handle_ctrl_c

test_menu() {
  assert_equal "$(./menu "Please choose" "item1 item2 item3" "ls cat command3" -v)" "Please choose\n1) item1\n2) item2\n3) item3\n" "menu -v output differs"
  assert_equal "$(./menu "Please choose" "item1 item2 item3" "ls cat command3")" "1) item1\n2) item2\n3) item3\n" "menu output differs"
}

cursor-blink off

# Display the menu's description
echo $description

# Menu loop
while true; do
  display_menu
  handle_key $key
done