Mount.md5key/fuse-workspace-union by Stef Bon

#!/bin/bash

#    This script is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.

#    This software is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.

#    You should have received a copy of the GNU General Public License.
#    If not, see http://www.gnu.org/licenses/.

#    credits:
#
#    Stef Bon. the Netherlands
#

manage_consolekit_data()
{
local laction="$1"


if [ -n "$CK_SESSION_SHORT" ]; then


	if [ "$laction" = "session_added" ]; then


	    if [ ! -d "/tmp/ConsoleKit/sessions/$CK_SESSION_SHORT" ]; then

		install --directory "/tmp/ConsoleKit/sessions/$CK_SESSION_SHORT"

	    fi
	
	    echo $userid > "/tmp/ConsoleKit/sessions/$CK_SESSION_SHORT"/unixuser
	    echo $CK_SESSION_SHORT > "/tmp/ConsoleKit/sessions/$CK_SESSION_SHORT"/session
	
	    if [ -n "$userid" ]; then
	
		if [ ! -d "/tmp/ConsoleKit/users/$userid" ]; then
	    
		    install --directory "/tmp/ConsoleKit/users/$userid"
		
		fi
	
		if [ -h "/tmp/ConsoleKit/users/$userid/$CK_SESSION_SHORT" ]; then
	
		    rm --force "/tmp/ConsoleKit/users/$userid/$CK_SESSION_SHORT"

		fi
	
		ln --symbolic --force ../../sessions/$CK_SESSION_SHORT "/tmp/ConsoleKit/users/$userid/$CK_SESSION_SHORT"

	    fi
	    
	elif [ "$laction" = "session_removed" ]; then


	    if [ -d "/tmp/ConsoleKit/sessions/$CK_SESSION_SHORT" ]; then

		rm --recursive --force "/tmp/ConsoleKit/sessions/$CK_SESSION_SHORT"

	    fi


	    if [ -n "$userid" ]; then

		if [ -h "/tmp/ConsoleKit/users/$userid/$CK_SESSION_SHORT" ]; then
	
		    rm --force "/tmp/ConsoleKit/users/$userid/$CK_SESSION_SHORT"
		    
		    if [ $(ls -v "/tmp/ConsoleKit/users/$userid/" | wc --words) -eq 0 ]; then
		    
			rm --force "/tmp/ConsoleKit/users/$userid/"
			
		    fi

		fi

	    fi


	fi


fi
}


function count_sessions()
{
local luserid="$1"
local ncount=0

if [ -n "$luserid" ]; then

    if [ -d "/tmp/ConsoleKit/users/$luserid" ]; then
    
	ncount=$(ls -v "/tmp/ConsoleKit/users/$luserid" | wc --words)
	
    fi
    
fi

echo $ncount
}



# defaults

logpriority="daemon.info"



if [ -f /etc/session-scripts.conf ]; then


    . /etc/session-scripts.conf


elif [ -f /etc/session.d/session-scripts.conf ]; then


    . /etc/session.d/session-scripts.conf


else


    echo "Configurationfile session-scripts.conf not found."

fi

if [ -z "$BASE_SESSION_SCRIPTS_DIR" ]; then

    echo "Parameter SESSION_FUNCTIONS_DIR not set. Taking default."
    BASE_SESSION_SCRIPTS_DIR=/etc/session.d

fi



if [ -z "$SESSION_FUNCTIONS_FILE" ]; then

    echo "Parameter SESSION_FUNCTIONS_DIR not set."
    SESSION_FUNCTIONS_FILE=${BASE_SESSION_SCRIPTS_DIR}/scripts/functions
    


elif [ -z "$SESSION_CK_ADDED_DIR" ]; then

    echo "Parameter SESSION_CK_ADDED_DIR not set. Taking default."
    SESSION_CK_ADDED_DIR=${BASE_SESSION_SCRIPTS_DIR}/CK-added


elif [ -z "$SESSION_CK_REMOVED_DIR" ]; then

    echo "Parameter SESSION_CK_REMOVED_DIR not set. Taking default."
    SESSION_CK_REMOVED_DIR=${BASE_SESSION_SCRIPTS_DIR}/CK-removed

fi



if [ -f "$SESSION_FUNCTIONS_FILE" ]; then

    . "$SESSION_FUNCTIONS_FILE"

fi


if [ -z "$(type -t do_log)" ]; then

	echo "Function do_log not defined."
	exit

fi


if [ -z "$(type -t find_command)" ]; then

	echo "Function find_command not defined."
	exit

fi


if [ -z "$(type -t manage_consolekit_data)" ]; then

	echo "Function manage_consolekit_data not defined."
	exit

fi


if [ -z "$SESSION_USE_ATD" ]; then

    SESSION_USE_ATD=yes

fi


CK_SESSION_SHORT=""



if [ -n "$CK_SESSION_USER_UID" -a -n "$CK_SESSION_ID" ]; then


    # called through ConsoleKit

    do_log "Starting $0 through CK with parameters $@."


    # determine the session 


    CK_SESSION_SHORT=$(echo "$CK_SESSION_ID" | rev | sed 's@^/*@@' | awk -F '/' '{ print $1 }' | rev)

    

else

    # cannot determine how called: exit
    
    do_log "Called $0 without the required CK parameters. Quitting."

    exit

fi


# lookup user

userproperties=$(getent passwd $CK_SESSION_USER_UID | grep --max-count=1 ":$CK_SESSION_USER_UID:")
userid=$(echo $userproperties | cut --delimiter=":" --fields=1)

if [ -z "$userproperties" ]; then

    # something wrong : the basic properties for this user are not found!
    do_log "User $userid not found. Quitting."
    exit 1

else

    do_log "User $userid found."


fi


# determine TMPDIR

if [ -z "$TMPDIR" ]; then

    if [ ! -d /tmp ]; then

	exit 1
	
    else

	TMPDIR=/tmp/run-scripts-with-CK
	
    fi

fi


if [ ! -d $TMPDIR ]; then

    install --directory $TMPDIR

fi

set >> $TMPDIR/set.ck.log

if [ "$1" = "session_added" ]; then

    # maintain a simple tree containing information about sessions and users
    # which is easy accessible to shellscripts

    manage_consolekit_data "session_added"
    
    # only run scripts when this is the first session for this user
    
    if [ $(count_sessions $userid) -gt 1 ]; then
    
	# not the first session
    
	do_log "This is not the first session for this user."

	exit
    fi

    # a session is added

    SCRIPTS_DIRECTORY="$SESSION_CK_ADDED_DIR"
    LOCK_FILE="$TMPDIR/run-add-scripts-for-$userid"


elif [ "$1" = "session_removed" ]; then


    # maintain a simple tree containing information about sessions and users
    
    manage_consolekit_data "session_removed"

    # only run scripts when this is the last session for this user

    if [ $(count_sessions $userid) -gt 0 ]; then
    
	# there are still other sessions

	do_log "This is not the last session for this user."

	exit

    fi
    
    # a session is removed
    
    SCRIPTS_DIRECTORY="$SESSION_CK_REMOVED_DIR"
    LOCK_FILE="$TMPDIR/run-remove-scripts-for-$userid"

else

    do_log "Parameter $1 not handled here."
    exit
    
fi



# try to run scripts via the at daemon; if not possible run these directly

if [ "$SESSION_USE_ATD" = "yes" ]; then


    AT_COMMAND=$(find_command at)


    if [ -z "$AT_COMMAND" ]; then

	do_log "At command not found. Running scripts directy."
	SESSION_USE_ATD="no"

    else

	do_log "At command found. Check the daemon is running."

	if [ -z "$(ps --no-headers -C atd)" ]; then

	    do_log "The at daemon not running. Running scripts directly."

	    SESSION_USE_ATD="no"

	fi

    fi
    
fi



# create the lockfile

if [ -f "$LOCK_FILE" ]; then

    sleep 1
    
    if [ -f "$LOCK_FILE" ]; then
    
	do_log "Existing lockfile $LOCK_FILE found. Cannot continue."
	
	exit
	
    fi
    
else

    touch "$LOCK_FILE"
    
fi



if [ -d "$SCRIPTS_DIRECTORY" ]; then


    for at_script in $(ls -v "$SCRIPTS_DIRECTORY"); do


	if [ -x "$SCRIPTS_DIRECTORY/$at_script" ]; then


	    if [ "$SESSION_USE_ATD" = "yes" ]; then


		# run commands through at daemon
		# create a sh script file and feed it to the at daemon 

		AT_SH_FILE="$TMPDIR/$at_script"
	    
		if [ -f "$AT_SH_FILE" ]; then
	    
		    rm --force "$AT_SH_FILE"
		
		fi

		echo "sleep 1" >> "$AT_SH_FILE"

		echo "$SCRIPTS_DIRECTORY/$at_script $userid" >> "$AT_SH_FILE"
	    
		$AT_COMMAND -q "b" -f "$AT_SH_FILE" NOW >> /dev/null
		
		do_log "created script $AT_SH_FILE with $at_script... and running it with at $AT_COMMAND."


	    else
	    
	    
		# run commands directly
	    

		$SCRIPTS_DIRECTORY/$at_script $userid &


	    fi
	
	else
	
	    # script is not executable
	    
	    do_log "Script $at_script found in $SCRIPT_DIRECTORY not executable."
	
	fi
	
    done
    
else

    # this should not happen
    
    do_log "Cannot find directory $SCRIPTS_DIRECTORY where the scripts should be. Cannot continue."
    
    
fi

rm --force "$LOCK_FILE"
Linux

Intro

Fuse module

Mount.md5key

Sessionscripts

Screenshots

Wiki

Run scripts when session starts en when it ends.

Licence
