#!/bin/sh -e

. ../common-script.sh
. ../common-service-script.sh

# setleds can be used in all distros
# This method works by calling a script using systemd service

# Create a script to toggle numlock

create_file() {
  printf "%b\n" "Creating script..."
  "$ESCALATION_TOOL" tee "/usr/local/bin/numlock" >/dev/null <<'EOF'
#!/bin/bash

for tty in /dev/tty{1..6}
do
    /usr/bin/setleds -D +num < "$tty"; 
done
EOF

  "$ESCALATION_TOOL" chmod +x /usr/local/bin/numlock
}

# Create a systemd service to run the script on boot
create_service() {
  printf "%b\n" "Creating service..."
  "$ESCALATION_TOOL" tee "/etc/systemd/system/numlock.service" >/dev/null <<'EOF'
[Unit]
Description=numlock
        
[Service]
ExecStart=/usr/local/bin/numlock
StandardInput=tty
RemainAfterExit=yes

[Install]
WantedBy=multi-user.target
EOF
}

numlockSetup() {
  # Check if the script and service files exists
  if [ "$PACKAGER" = "apk" ]; then
    printf "%b\n" "${RED}Unsupported package manager.${RC}"
    exit 1
  fi

  if [ ! -f "/usr/local/bin/numlock" ]; then
    create_file
  fi

  if [ ! -f "/etc/systemd/system/numlock.service" ]; then
    create_service
  fi

  printf "%b" "Do you want to enable Numlock on boot? (y/N): "
  read -r confirm
  if [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]; then
    enableService numlock
    printf "%b\n" "Numlock will be enabled on boot"
  else
    disableService numlock
    printf "%b\n" "Numlock will not be enabled on boot"
  fi
}

checkEnv
checkEscalationTool
numlockSetup