wmenu

#!/bin/sh
#
# wmenu

usage() {
    base="$(basename "$0")"

    cat >&2 << EOF
Usage:
    $ $base -b | bar <-au>    : Show as a bar on primary screen.
    $ $base -m | menu <-au>   : Show as a window.
    $ $base -c | center <-au> : Show as a window centered above focused window.
    $ $base -h | help         : Show this help.
EOF

    [ $# -eq 0 ] || exit "$1"
}

width() {
    ps=$(txtw -f "$FONT" "$PROMPT")
    char="$(output | wc -L)"
    W=$(echo "($ps + $char * 4.9 + 8) / 1" | bc)
}

output() {
    for wid in $($LSWCMD); do
        printf '%s\n' "$wid $(atomx WM_CLASS "$wid") | $(wname "$wid")"
    done | grep -v "$PFW"
}

selection() {
    output | sort -rk 2 | \
        dmenu -name "$(basename "$0")" -f -l "$nWindows" -i -fn "$FONT" -p "$PROMPT" \
        -nf "#$NF" -sf "#$SF" -nb "#$NB" -sb "#$SB" -bc "#$BC" \
        -s "$(mattr d "$PFM")" -x "$X" -y "$Y" -w "$W" -h "$FH" -bw "$BW" | \
        cut -d\  -f 1
}

groupoutput() {
    for wid in $(printf '%s\n' "$grplist" | grep "GROUP_$grp" | cut -d' ' -f 2); do
        printf '%s\n' "$wid $(atomx WM_CLASS "$wid") | $(wname "$wid")"
    done
}

groupselection() {
    groupoutput | sort -rk 2 | \
        dmenu -name "$(basename "$0")" -f -l "$nWindows" -i -fn "$FONT" -p "$PROMPT" \
        -nf "#$NF" -sf "#$SF" -nb "#$NB" -sb "#$SB" -bc "#$BC" \
        -s "$(mattr d "$PFM")" -x "$X" -y "$Y" -w "$W" -h "$FH" -bw "$BW" | \
        cut -d\  -f 1
}

menu() {
    width

    # menu geometry
    X0="$(mattr x "$PFM")"
    Y0="$(mattr y "$PFM")"

    mouse="$(wmp)"
    X=$((${mouse% *} - X0 - 20))
    Y=$((${mouse#* } - Y0 - 20))
}

center() {
    width
    H=$((nWindows * FH))

    # finding x y cords for centering
    if [ "$PFW" = "0x00000000" ]; then
        X=$(($(mattr w "$PFM")/2 - W/2 - 10))
        Y=$(($(mattr h "$PFM")/2 - H/2 - 10))
    else
        X0="$(mattr x "$PFM")"
        Y0="$(mattr y "$PFM")"
        X=$(($(wattr x "$PFW") + $(wattr w "$PFW")/2 - W/2 - X0))
        Y=$(($(wattr y "$PFW") + $(wattr h "$PFW")/2 - H/2 - Y0))

        # move mouse to optimal selection position
        wmp -a $(($(wattr x "$PFW") + $(wattr w "$PFW") / 2 - W / 2 + 20)) \
               $(($(wattr y "$PFW") + $(wattr h "$PFW") / 2 - H / 2 + 20))
    fi
}

bar() {
    X=$((LGAP))
    width

    if [ "$BAR" = "top" ]; then
        H=$((TGAP/2 + BW*2))
        Y=$((H - TGAP/4))
    else
        H=$((BGAP/2 + BW*2))
        Y=$(($(mattr h "$PRI") - FH * (nWindows + 1) - BGAP))
    fi

    # position mouse
    wmp -a "$(($(mattr x "$PRI") + 40))" "$(($(mattr y "$PRI") + Y + 20))"
}

groupbar() {
    grp="$1"
    grplist="$(group -l)"

    case "$(printf '%s\n' "$grplist" | grep -v "$PFW" | grep -c "GROUP_$grp")" in
        0)
            exit 1
            ;;
        *)
            LSWCMD="printf '%s\n' $grplist | grep GROUP_$grp | cut -d' ' -f 2"

            output

            exit 0
            ;;
    esac
}

main() {
    . fwmrc
    wmenv
    wmgaps
    dcolours

    PFM="$(pfm)"
    FONT=$(awk '/font/ {print $3}' < ~/.Xresources | cut -d',' -f 1)
    FH=15
    PROMPT=" Window >"

    case "$2" in
        -a|all) LSWCMD="lsw -a" ;;
        -u|hid) LSWCMD="lsw -u" ;;
        *)
            LSWCMD="lsw"
            ;;
    esac

    nWindows=$($LSWCMD | grep -cv "$PFW")
    [ "$nWindows" -eq 0 ] && exit 1

    case "$1" in
        -b|bar)    bar           ;;
        -m|menu)   menu          ;;
        -c|center) center        ;;
        -g|group)  groupbar "$2" ;;
        -h|help)   usage 0       ;;
        *)         usage 1       ;;
    esac

    # unfocus from current window to add contrast
    focus -u

    # grab window id from user input
    wid="$(selection)"

    # if dmenu returns no output refocus previous window
    # else focus selected window
    if [ -n "$wid" ]; then
        focus -w "$wid"
    else
        focus -w "$PFW"
    fi
}

main "$@"