dotfiles/modules/nixos/graphical/rofi/power.nix

{ config, pkgs, ... }:
let

  rofi = config.home-manager.users.${config.user}.programs.rofi.finalPackage;

in {

  # Adapted from:
  # https://gitlab.com/vahnrr/rofi-menus/-/blob/b1f0e8a676eda5552e27ef631b0d43e660b23b8e/scripts/rofi-power
  # A rofi powered menu to execute power related action.

  config.powerCommand = builtins.toString (pkgs.writeShellScript "powermenu" ''
    power_off=''
    reboot=''
    lock=''
    suspend='󰒲'
    log_out=''

    chosen=$(printf '%s;%s;%s;%s;%s\n' \
        "$power_off" \
        "$reboot" \
        "$lock" \
        "$suspend" \
        "$log_out" \
        | ${rofi}/bin/rofi \
            -theme-str '@import "power.rasi"' \
            -hover-select \
            -me-select-entry ''' \
            -me-accept-entry MousePrimary \
            -dmenu \
            -sep ';' \
            -selected-row 2)

    case "$chosen" in
        "$power_off")
            ${
              builtins.toString ./rofi-prompt.sh
            } 'Shutdown?' && doas shutdown now
            ;;

        "$reboot")
            ${builtins.toString ./rofi-prompt.sh} 'Reboot?' && doas reboot
            ;;

        "$lock")
            ${pkgs.betterlockscreen}/bin/betterlockscreen --lock --display 1 --blur 0.5 --span
            ;;

        "$suspend")
            systemctl suspend
            ;;

        "$log_out")
            ${builtins.toString ./rofi-prompt.sh} 'Logout?' && i3-msg exit
            ;;

        *) exit 1 ;;
    esac
  '');

}