group

#!/bin/sh
#
# group

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

    cat >&2 << EOF
Usage:
    $ $base -a | add \$group \$wid : Add \$wid to \$group.
    $ $base -A | scr \$group      : Add all windows on screen to \$group.
    $ $base -d | del \$wid        : Delete \$wid from all groups.
    $ $base -t | tog \$group      : Toggle \$group visibility.
    $ $base -s | smart \$group    : Toggle \$group visibility smartly.
    $ $base -l | list            : List all windows in groups.
    $ $base -h | help            : Show this help.
EOF

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

list_groups() {
    for wid in $(lsw -a); do
        wg="$(atomx GROUP "$wid")"
        [ -n "$wg" ] && printf 'GROUP_%s %s %s\n' \
                        "$wg" "$wid" "$(atomx WM_CLASS "$wid")"
    done
}

sort_groups() {
    list_groups | sort
}

show_group() {
    # focus window in group that is at the top
    PFW="$(list_groups | grep "GROUP_$1" | tail -n 1 | cut -d\  -f 2)"

    for wid in $(lsw -u); do
        wg=$(atomx GROUP "$wid")
        [ -z "$wg" ] && continue
        [ "$wg" -eq "$1" ] && {
            mapw -m "$wid"
            chwso -r "$wid"
        }
    done

    focus -w "$PFW"

    # set group to active
    atomx GROUP_"$1"=on "$ROOT" > /dev/null
}

hide_group() {
    for wid in $(lsw); do
        wg=$(atomx GROUP "$wid")
        [ -z "$wg" ] && continue
        [ "$wg" -eq "$1" ] && {
            mapw -u "$wid"
            chwb -s "$BW" -c "$INACTIVE" "$wid"
        }
    done

    # set group to inactive
    atomx GROUP_"$1"=off "$ROOT" > /dev/null
}

toggle_group() {
    case $(atomx GROUP_"$1" "$ROOT") in
        on)  hide_group "$1" ;;
        off) show_group "$1" ;;
        *)   atomx GROUP_"$1"=on "$ROOT" > /dev/null ;;
    esac
}

del_wid() {
    grp="$(atomx GROUP "$1") $(lsw -r)"

    list_groups | grep -q "$grp" || {
        atomx -d GROUP_"$grp" "$(lsw -r)"
    }

    atomx -d GROUP "$1"
}

smart_toggle_group() {
    case $(atomx GROUP_"$1" "$ROOT") in
        on)
            windows="$(list_groups | grep "GROUP_$1" | cut -d\  -f 2)"

            for wid in $windows; do
                if [ "$wid" = "$PFW" ]; then
                    # if [ "$PFW" != "$(lsw | head -n 1)" ]; then
                        hide_group "$1"
                        return
                    # fi
                fi
            done

            focus -w "$(printf '%s\n' "$windows" | tail -n 1)"
            ;;
        off)
            show_group "$1"
            ;;
        *)
            smart_toggle_group "$1"
            ;;
    esac
}

add_group() {
    atomx GROUP="$1" "$2" > /dev/null

    # automatically hide window if group is hidden
    case "$(atomx GROUP_"$1" "$ROOT")" in
        off)
            mapw -u "$2"
            chwb -s "$BW" -c "$INACTIVE" "$2"
            ;;
        *)
            # set group to active
            atomx GROUP_"$1"=on "$ROOT" > /dev/null
            ;;
    esac
}

add_screen() {
    for wid in $(listwindows); do
        add_group "$1" "$wid"
    done

    # set group to active
    atomx GROUP_"$1"=on "$ROOT" > /dev/null
}

main() {
    . fwmrc
    wmenv
    wmcolours

    # store group information inside root window properties
    ROOT="$(lsw -r)"

    case "$1" in
        -a|--add|add)     add_group "$2" "$3"     ;;
        -A|--scr|scr)     add_screen "$2"         ;;
        -d|--del|del)     del_wid "$2"            ;;
        -u|--hide|hide)   hide_group "$2"         ;;
        -t|--tog|tog)     toggle_group "$2"       ;;
        -s|--smart|smart) smart_toggle_group "$2" ;;
        -l|--list|list)   sort_groups             ;;
        -h|--help|help)   usage 0                 ;;
        *)                usage 1                 ;;
    esac
}

main "$@"