#!/bin/sh

cachefile="$HOME/.cache/layout"

# Handle manual override first
if [ -n "$1" ]; then
    echo "$1" >"$cachefile"
else
    # Ensure the cache file exists, or initialize it
    [ -f "$cachefile" ] || echo "us" >"$cachefile"
    layout=$(cat "$cachefile")

    case "$layout" in
    ca)
        echo us >"$cachefile"
        ;;
    us)
        echo ca >"$cachefile"
        ;;
    *)
        echo us >"$cachefile"
        ;;
    esac
fi

# Apply keyboard layout
riverctl keyboard-layout -model pc104 "$(cat "$cachefile")"

# Restart the correct block if waybar is running
waybar_pid=$(pidof waybar)
if [ -n "$waybar_pid" ]; then
    kill -38 "$waybar_pid"
fi