#!/bin/bash
# ------------------------------------------------------
# Setup
# ------------------------------------------------------
echo -e "${GREEN}"
cat <<"EOF"
 _  __          _                         _ 
| |/ /___ _   _| |__   ___   __ _ _ __ __| |
| ' // _ \ | | | '_ \ / _ \ / _` | '__/ _` |
| . \  __/ |_| | |_) | (_) | (_| | | | (_| |
|_|\_\___|\__, |_.__/ \___/ \__,_|_|  \__,_|
          |___/                             

EOF
echo -e "${NONE}"

# Default layout and variants
keyboard_layout="us"
keyboard_variant=""

_setupKeyboardLayout() {
    if gum confirm "Current keyboard layout: $keyboard_layout! Do you want to change it?" ;then
        keyboard_layout=$(localectl list-x11-keymap-layouts | gum filter --placeholder "Select keyboard layout...")
        echo ""
        echo "Keyboard layout changed to $keyboard_layout"
    elif [ $? -eq 130 ]; then
        exit 130
    fi
    _setupKeyboardVariant
}

_setupKeyboardVariant() {
    if gum confirm "Current keyboard variant: $keyboard_variant! Do you want to change it?" ;then
        keyboard_variant=$(localectl list-x11-keymap-variants | gum filter --placeholder "Select keyboard variant...")
        echo ""
        echo "Keyboard layout changed to $keyboard_variant"
    elif [ $? -eq 130 ]; then
        exit 130
    fi
    echo ""
    _confirmKeyboard
}

_confirmKeyboard() {
    echo "Current selected keyboard setup:"
    echo "Keyboard layout: $keyboard_layout"
    echo "Keyboard variant: $keyboard_variant"
    if gum confirm "Do you want proceed with this keyboard setup?" --affirmative "Proceed" --negative "Change" ;then
        return 0
    elif [ $? -eq 130 ]; then
        exit 130
    else
        _setupKeyboardLayout
    fi
}

if [ "$restored" == "1" ]; then
    echo "You have already restored your settings into the new installation."
else
    _confirmKeyboard
    
    cp .install/templates/keyboard.conf ~/dotfiles-versions/$version/hypr/conf/keyboard.conf
    cp .install/templates/keyboard.py ~/dotfiles-versions/$version/qtile/conf/keyboard.py

    SEARCH="KEYBOARD_LAYOUT"
    REPLACE="$keyboard_layout"
    sed -i "s/$SEARCH/$REPLACE/g" ~/dotfiles-versions/$version/hypr/conf/keyboard.conf

    SEARCH="\"KEYBOARD_LAYOUT\""
    REPLACE="\"$keyboard_layout\""
    sed -i "s/$SEARCH/$REPLACE/g" ~/dotfiles-versions/$version/qtile/conf/keyboard.py

    SEARCH="KEYBOARD_VARIANT"
    REPLACE="$keyboard_variant"
    sed -i "s/$SEARCH/$REPLACE/g" ~/dotfiles-versions/$version/hypr/conf/keyboard.conf

    echo ""
    echo "Keyboard setup updated successfully."
fi