#!/bin/bash

ffoo import inigrep
ffoo import pretty
ffoo import sync

__zleep_user_part() {
    saturnin iam undocking
    slock &
}

__zleep_try_umount() {
    local mp=$1
    mountpoint -q "$mp" || return 0
    umount        "$mp" || return 1
    mount | grep  "$mp" && return 1
    return 0
}

__zleep_try_umount_all() {
    local mp
    for mp in $(inigrep -p iam.mounting.point);
    do
        __zleep_try_umount "$mp" || return 1
    done
    return 0
}

__zleep_sudo_part() {
    wait_until -d 5 -t 30 __zleep_try_umount_all || die "unable to unmount drives"
    pm-suspend
}

__zleep_sudo_part_started() {
    # i.e. password prompt went ok
    test -e "$ZLEEP_ON_ROOT"
    rv=$?
    debug -v rv
    return $rv
}

__zleep_user_part_done() {
    ! test -e "$ZLEEP_ON_USER"
    rv=$?
    debug -v rv
    return $rv
}

__zleep_min_path() {
    echo -n "$(sudo env | grep ^PATH)"
    echo -n :
    echo -n "$(dirname "$(which ffoom)")"
    echo -n :
    echo -n "$(dirname "$(which saturnin)")"
}

zleep() {
    #
    # Put the host on sleep
    #
    # We need to
    #
    #  1. authenticate
    #  2. start user's part in the background -- mainly slock
    #  3. start root's part -- put system on sleep (unmount, suspend)
    #
    # This is solved by forking with sudo, and waiting for user part to indicate
    # it has finished.  That way, we can ensure that suspend happens *after* the
    # screen has been locked, and at the same time we leave authentication part
    # on sudo.
    #
    if test -z "$ZLEEP_ON_USER";    ## we are parent (running as normal user)
    then
        ZLEEP_ON_USER=$(mktemp)
        ZLEEP_ON_ROOT=$(mktemp -u)
        SUDO_PATH=$(__zleep_min_path)   # needed while bins are still under local
        debug -v ZLEEP_ON_USER ZLEEP_ON_ROOT SUDO_PATH
        sudo env \
            ZLEEP_ON_ROOT="$ZLEEP_ON_ROOT" \
            ZLEEP_ON_USER="$ZLEEP_ON_USER" \
            PATH="$SUDO_PATH" \
            saturnin iam zleeping &
        wait_until -t 60 __zleep_sudo_part_started    # i.e. password prompt done
        __zleep_user_part
        rm -f "$ZLEEP_ON_USER"
    else                            ## we are child (running as root)
        touch "$ZLEEP_ON_ROOT"
        test "$(id -u)" -eq 0               || die "must be root"
        wait_until __zleep_user_part_done   || die "timeout waiting for parent"
        __zleep_sudo_part
        rm -f "$ZLEEP_ON_ROOT"
        return $?
    fi
}