#! /bin/sh

. "${PM_FUNCTIONS}"

test_pulse_system() {
    getent passwd pulse | awk -F: '{print $3}'
}

get_pulse_users() {
    PULSE_SYSTEM_USER=$(test_pulse_system)
    if [ -z "${PULSE_SYSTEM_USER}" ]; then
        ps -C pulseaudio -o uid= | tr -d ' '
    else
        ps -C pulseaudio -o uid= | tr -d ' ' | sed s/${PULSE_SYSTEM_USER}//
    fi
}

# $1 = username
save_pulse_default() {
    su "${1}" -c -- "pacmd dump" | grep ^set-default | \
        while read field value; do
            case ${field} in
                (set-default-sink)
                    savestate pulse:${1}:default-sink ${value}
                    ;;
                (set-default-source)
                    savestate pulse:${1}:default-source ${value}
                    ;;
            esac
        done
}

# $1 = sink|source
# $2 = username
save_pulse_state() {
    su "${2}" -c -- "pacmd list-${1}s" | \
        sed -n "s/^[[:space:]*]*//; /^\(name\|mute\)/p" | \
        (name="";
            while read field value; do
                if [ ${field%:} = "name" ]; then
                    name="$(echo ${value} | sed -n 's/^<//p' | sed -n 's/>$//p')"
                else
                    savestate pulse:"${2}":${1}:${name} ${value}
                fi
            done)
}

# $1 = username
save_pulse_volume() {
    su "${1}" -c -- "pacmd dump | grep '^set-.*-volume'" | \
        while read field device value; do
            if [ ${field} = 'set-sink-volume' ]; then
                savestate pulse:${1}:sink-volume:${device} ${value}
            else # set-source-volume
                savestate pulse:${1}:source-volume:${device} ${value}
            fi
        done
}

# $1 = sink|source
# $2 = username
save_pulse_connection() {
    case "${1}" in
        (sink)
            cmd='list-sink-inputs'
            name='sink-input'
            ;;
        (source)
            cmd='list-source-outputs'
            name='source-output'
            ;;
    esac
    su "${2}" -c -- "pacmd ${cmd}" | \
        sed -n "s/^[[:space:]*]*//; /^\(index\|sink\|source\)/p" | \
        (index=""; device="";
            while read field value; do
                if [ ${field%:} = "index" ]; then
                    index="${value}"
                else
                    device="$(echo ${value} | sed -n "s/.*<\(.*\)>/\1/p")"
                    savestate pulse:"${2}":${name}:${index} ${device}
                fi
            done)
}

# $1 = username
restore_pulse_default() {
    if state_exists pulse:${1}:default-sink; then
        su "${1}" -c -- "pacmd \
            set-default-sink \
            $(restorestate pulse:${1}:default-sink)" > /dev/null 2>&1
    fi
    if state_exists pulse:${1}:default-source; then
        su "${1}" -c -- "pacmd \
            set-default-source \
            $(restorestate pulse:${1}:default-source)" > /dev/null 2>&1
    fi
}

# $1 = username
restore_pulse_volume() {
    su "${1}" -c -- "pacmd dump | grep '^set-.*-volume'" | \
        while read field device value; do
            if [ ${field} = 'set-sink-volume' ]; then
                if state_exists pulse:${1}:sink-volume:${device}; then
                    su "${1}" -c -- "pacmd set-sink-volume ${device} \
                        $(restorestate pulse:${1}:sink-volume:${device})" > /dev/null 2>&1
                fi
            else # set-source-volume
                if state_exists pulse:${1}:source-volume:${device}; then
                    su "${1}" -c -- "pacmd set-source-volume ${device} \
                        $(restorestate pulse:${1}:source-volume:${device})" > /dev/null 2>&1
                fi
            fi
        done
}

# $1 = sink|source
# $2 = username
restore_pulse_state() {
    su "${2}" -c -- "pacmd list-${1}s" | \
        sed -n "s/^[[:space:]*]*name: //p" | sed -n 's/^<//p' | sed -n 's/>$//p' | \
        while read name; do
            if state_exists pulse:"${2}":${1}:${name}; then
                su "${2}" -c -- "pacmd \
                    set-${1}-mute \
                    ${name} \
                    $(restorestate pulse:"${2}":${1}:${name})"
            fi
        done
}

# $1 = sink|source
# $2 = username
restore_pulse_connection() {
    case "${1}" in
        (sink)
            cmd='list-sink-inputs'
            name='sink-input'
            ;;
        (source)
            cmd='list-source-outputs'
            name='source-output'
            ;;
    esac
    su "${2}" -c -- "pacmd ${cmd}" | \
        sed -n "s/^[[:space:]*]*index: //p" | \
        while read index; do
            if state_exists pulse:"${2}":${name}:${index}; then
                su "${2}" -c -- "pacmd \
                    move-${name} \
                    ${index} \
                    $(restorestate pulse:"${2}":${name}:${index})"
            fi
        done
}

# $1 = sink|source
# $2 = username
mute_pulse() {
    su "${2}" -c -- "pacmd list-${1}s" | \
        sed -n "s/^[[:space:]*]*//; /^\(name\|mute\)/p" | \
        (name="";
            while read field value; do
                if [ ${field%:} = "name" ]; then
                    name=${value}
                    su "${2}" -c -- "pacmd \
                        set-${1}-mute ${name} yes"
                fi
            done)
}

suspend_pulse() {
    for i in $(get_pulse_users); do
        THIS_USER="$(getent passwd ${i} | cut -f1 -d:)"
        save_pulse_default "${THIS_USER}"
        save_pulse_volume "${THIS_USER}"
        save_pulse_state sink "${THIS_USER}"
        save_pulse_state source "${THIS_USER}"
        save_pulse_connection sink "${THIS_USER}"
        save_pulse_connection source "${THIS_USER}"
        su "${THIS_USER}" -c -- 'pacmd suspend true' > /dev/null 2>&1
    done
    for i in $(get_pulse_users); do
        THIS_USER="$(getent passwd ${i} | cut -f1 -d:)"
        if su "${THIS_USER}" -c -- 'ck-list-sessions | grep "active = TRUE"' > /dev/null 2>&1; then
            mute_pulse sink "${THIS_USER}"
            mute_pulse source "${THIS_USER}"
            break
        fi
    done
}

resume_pulse() {
    for i in $(get_pulse_users); do
        THIS_USER="$(getent passwd ${i} | cut -f1 -d:)"
        restore_pulse_default "${THIS_USER}"
        restore_pulse_volume "${THIS_USER}"
        restore_pulse_state sink "${THIS_USER}"
        restore_pulse_state source "${THIS_USER}"
        restore_pulse_connection sink "${THIS_USER}"
        restore_pulse_connection source "${THIS_USER}"
        su "${THIS_USER}" -c -- 'pacmd suspend false' > /dev/null 2>&1
    done
}

case $1 in
    hibernate|suspend)
        suspend_pulse
        ;;
    thaw|resume)
        resume_pulse
        ;;
    *) exit $NA
        ;;
esac
