#!/bin/bash

if [[ $1 != 'dkms' ]]; then
    echo '## Making package ##'
    make

    echo '## Installing package ##'
    make install
else
    echo '## Installing package with DKMS ##'
    make dkms
fi

quirk='0x0c45:0x7603:0x0007'
modquirk="options usbhid quirks=$quirk"
grubquirk="usbhid.quirks=$quirk"

if (lsmod | grep 'usbhid'); then
    echo '## usbhid is module ##'

    # Making sure the quirk does not get added multiple times
    if ! (cat /etc/modprobe.d/usbhid.conf | grep "$modquirk"); then
        echo '## Writing to /etc/modprobe.d/usbhid.conf ##'
        echo $modquirk >> /etc/modprobe.d/usbhid.conf
        if [[ $1 != 'dkms' ]]; then sudo echo 'perixxkbd' >> /etc/modules; fi
    else
       echo 'NOTICE - modprobe config files have already been updated'
    fi

    echo '## Starting module ##'
    modprobe perixxkbd

    # Note: this line may fail if you have other drivers loaded that depend
    # on usbhid. For example, your mouse driver. In that case you would have
    # to remove those drivers first, then load them again.

    echo '## Attempting to reload usbhid module ##'
    rmmod usbhid && modprobe usbhid quirks=$quirk
    echo '## Reload done. Please reboot if the keyboard is not working yet! ##'
else
    echo '## usbhid is compiled into kernel ##'

    # Making sure the quirk does not get added multiple times
    if ! (cat /etc/default/grub.d/perixxkbd.conf | grep "$grubquirk"); then
        echo '## Writing to /etc/default/grub.d/perixxkbd.conf ##'
        echo $grubquirk >> /etc/default/grub.d/perixxkbd.conf
        $distro = $(lsb_release -si)
        if ($distro | grep 'Ubuntu'); then
            update-grub
        fi
        if [ -f "/etc/arch-release" ]; then
            update-grub
        fi
        if [ -f "/etc/fedora-release" ]; then
            grub2-mkconfig -o /boot/grub2/grub.cfg
        fi
    else
       echo 'NOTICE - grub config file has already been updated'
    fi

    echo '## You must reboot to load the module ##'
fi