#!/bin/sh -e

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

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_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() {
    if [ "$INIT_MANAGER" = "rc-service" ]; then
        printf "%b\n" "${RED}Unsupported init system.${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