#!/usr/bin/env fish

# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - 

function print_in_color -a $color $text
  printf '%b' \
    "$(tput setaf $color 2> /dev/null)" \
    $text \
    "$(tput sgr0 2> /dev/null)"
end

function print_in_red -a $text
  print_in_color $text 1
end

function print_in_yellow -a $text
  print_in_color $text 3
end

function print_in_green -a $text
  print_in_color $text 2
end

function print_in_purple -a $text
  print_in_color $text 5
end

function print_title -a $text
  print_in_purple "\n • $text\n\n"
end

function print_success -a $text
  print_in_green "   [✔] $text\n"
end

function print_warning -a $text
  print_in_yellow "   [!] $text\n"
end

function print_error -a $text $text2
  print_in_red "   [✖] $text $text2\n"
end

function print_question -a $text
  print_in_yellow "   [?] $text\n"
end

function print_result -a $exit_code $text
  if [ "$exit_code" -eq 0 ]
    print_success $text
  else
    print_error $text
  end

  return $exit_code
end

function print_error_stream
  while read -r line
    print_error "↳ ERROR: $line"
  end
end

# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - 

function show_spinner -a $pid $cmds $msg
  set -l frames '/-\|'

  set -l number_of_frames (string length $FRAMES)

  set -l i 0
  set -l frame_text ""

  printf "\n\n\n"
  tput cuu 3
  tput sc

  while kill -0 "$pid" &> /dev/null
    set i math $i + 1
    set -l num math % $number_of_frames
    set -l frame (string sub -s $num -l 1)
    set frame_text " [$frame] $msg"

    printf '%s' $frame_text

    sleep 0.2

    tput rc
  end
end

# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - 

function set_trap -a $sig $func
  trap -p "$sig" | grep "$func" &> /dev/null \
    || trap "$func" "$sig"
end

function kill_all_subproccesses
  set -l i ""

  for i in (jobs -p)
    kill "$i"
    wait "$i" &> /dev/null
  end
end

# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - 

function execute -a $cmds
  set -q $argv[2] and set -l msg $argv[2] or set -l msg "$cmds"

  set -l tmp_file "$(mktmp /tmp/XXXXX)"

  set -l exit_code 0

  set_trap 'EXIT' 'kill_all_subproccesses'

  eval "$cmds" &> /dev/null 2> $tmp_file &
  set cmds_pid $last_pid

  show_spinner $cmds_pid $cmds $msg

  wait $cmds_pid &> /dev/null
  set exit_code $status

  print_result $exit_code $msg

  if [ $exit_code -ne 0 ]
    print_error_stream < $tmp_file
  end

  rm -rf "$tmp_file"

  return $exit_code
end