#!/bin/sh
# works on any init system
# requirements: rofi -dmenu

kbquery()
{
  kblayout=$(swaymsg --raw -t get_inputs | jq -r '.[] | select(.type == "keyboard") | .xkb_active_layout_name' | sort -u) || exit 1
  kblayout=$(echo "$kblayout" | sed 's/(/\\(/g; s/)/\\)/g') || exit 1
  kblayout=$(rg -oP "^\s*\K\w+(?=\s+$kblayout)" /usr/share/X11/xkb/rules/base.lst) || exit 1
  echo $kblayout
}

kb=$(kbquery)

case $BLOCK_BUTTON in
  1) kb_choice="$(awk '/! layout/{flag=1; next} /! variant/{flag=0} flag {print $2, "- " $1}' /usr/share/X11/xkb/rules/base.lst | rofi -dmenu -l 15)"
    [ -z "$kb_choice" ] && exit 0
    kb="$(echo "$kb_choice" | awk '{print $3}')"
    swaymsg input "type:keyboard" xkb_layout "$kb" >/dev/null 2>&1
    pkill -RTMIN+30 "${STATUSBAR:-waybar}";;
  3) notify-send "⌨  Keyboard/language module" "$(printf "%s" "\- Current layout: $(kbquery)")
- Left click to change keyboard.";;
  8) setsid -f "$TERMINAL" -e "$EDITOR" "$0" >/dev/null 2>&1 ;;
esac

echo "$kb"