#!/usr/bin/env bash

# This file is part of The RetroPie Project
#
# The RetroPie Project is the legal property of its developers, whose names are
# too numerous to list here. Please refer to the COPYRIGHT.md file distributed with this source.
#
# See the LICENSE.md file at the top-level directory of this distribution and
# at https://raw.githubusercontent.com/RetroPie/RetroPie-Setup/master/LICENSE.md
#

rp_module_id="joy2key"
rp_module_desc="Provides joystick to keyboard conversion for navigation of RetroPie dialog menus"
rp_module_section="core"

function _update_hook_joy2key() {
    # make sure joy2key is always updated when updating retropie-setup
    rp_isInstalled "$md_id" && rp_callModule "$md_id"
}

function depends_joy2key() {
    local depends=(python3-urwid python3-uinput)
    # 'python3-sdl2' might not be available
    # it's packaged in Debian starting with version 11 (Bullseye)
    local p_ver
    p_ver="$(apt-cache madison python3-sdl2 | cut -d" " -f3 | head -n1)"
    if [[ -n "$p_ver" ]]; then
        depends+=(python3-sdl2)
    fi

    getDepends "${depends[@]}"
}

function install_bin_joy2key() {
    local file
    for file in "joy2key.py" "joy2key_sdl.py" "osk.py"; do
        cp "$md_data/$file" "$md_inst/"
        chmod +x "$md_inst/$file"
        python3 -m compileall "$md_inst/$file"
    done

    local wrapper="$md_inst/joy2key"
    cat >"$wrapper" <<_EOF_
#!/bin/bash
mode="\$1"
[[ -z "\$mode" ]] && mode="start"
shift

# allow overriding joystick device via __joy2key_dev env (by default will use /dev/input/jsX which will scan all)
device="/dev/input/jsX"
[[ -n "\$__joy2key_dev" ]] && device="\$__joy2key_dev"

params=("\$@")
if [[ "\${#params[@]}" -eq 0 ]]; then
    # Default button-to-keyboard mappings:
    # * cursor keys for axis/dpad
    # * enter, space, esc and tab for buttons 'a', 'b', 'x' and 'y'
    # * page up/page down for buttons 5,6 (shoulder buttons)
    params=(kcub1 kcuf1 kcuu1 kcud1 0x0a 0x20 0x1b 0x09 kpp knp)
fi

script="joy2key_sdl.py"
grep --basic-regexp --quiet --no-messages '^legacy_joy2key[[:space:]]*=[[:space:]]*"\?1"\?' $configdir/all/runcommand.cfg && script="joy2key.py"

case "\$mode" in
    start)
        if pgrep -f "\$script" &>/dev/null; then
            "\$0" stop
        fi
        "$md_inst/\$script" "\$device" "\${params[@]}" || exit 1
        ;;
    stop)
        if pid=\$(pgrep -f "\$script"); then
            /sbin/start-stop-daemon --stop --oknodo --pid \$pid --retry 1
        fi
        ;;
esac
exit 0
_EOF_
    chmod +x "$wrapper"
    if ! grep -q "uinput" /etc/modules; then
        addLineToFile "uinput" "/etc/modules"
    fi
    # add an udev rule to give 'input' group write access to `/dev/uinput`
    echo 'KERNEL=="uinput", MODE="0660", GROUP="input"' > /etc/udev/rules.d/80-rpi-uinput.rules
    udevadm control --reload

    modprobe uinput

    # make sure the install user is part of 'input' group
    usermod -a -G input "$__user"

    joy2keyStart
}

function remove_joy2key() {
    joy2keyStop
}