import_alt and export_alt directories are now created and logged

[python-TSD.git] / bin / tsd-projectdir-setup

#!/bin/sh
#
# Fix permissions on a freshly created project directory.  Assume the
# path is /tsd/pXX, and all project members are member of the file
# group pXX-member.  This setup assume all project members have umask
# 00X (as in 002 or 005).

set -e

if [ ! "$1" ] ; then
   cat <<EOF
Usage: $0 <path-to-project-dir> [<path-to-project-dir> ...]
EOF
   exit 1
fi

infomsg() {
    if [ -t 0 ] ; then
	echo "info:" "$@"
    fi
}
logmsg() {
    if [ -t 0 ] ; then
	echo "info:" "$@"
    else
	logger -t tsd-projectdir-setup info: "$@"
    fi
}
errormsg() {
    if [ -t 0 ] ; then
	echo "error:" "$@"
    else
	logger -t tsd-projectdir-setup error: "$@"
    fi
}

groupexist() {
    if getent group $1 > /dev/null ; then
	true
    else
	false
    fi
}

setup_projectdir() {
    pdir="$1"
    pnum=$(echo $pdir|cut -d/ -f3)
    group=$pnum-member-group
    agroup=$pnum-admin-group
    egroup=$pnum-export-group
    igroup=$pnum-import-group
    vgroup=$pnum-vidyo-group
    cd "$pdir"
    mkdir -p "home"
    mkdir -p "fx/export"
    mkdir -p "fx/import"
    mkdir -p "data/durable"
    mkdir -p "data/no-backup"
    chown root:$group . home fx data
    chmod 750 . home fx data
    chmod 2770 fx/export fx/import data/durable data/no-backup
    if ! groupexist $group || ! groupexist $egroup || ! groupexist $igroup ; then
	errormsg "unable to find needed groups, can not to set up $pdir"
	return 1
    fi
    infomsg "updating file permissions for $pdir"
    chown root:$group data/durable data/no-backup
    chown root:$egroup fx/export
    chown root:$igroup fx/import

    if getent netgroup $pnum-hpc-user-group > /dev/null 2>&1 ; then
	chmod 2770 data/colossus
	chown root:$group data/colossus
    elif [ -d data/colossus ] ; then
	errormsg "project $pnum have data/colossus but lack $pnum-hpc-user-group"
    fi

    if getent netgroup $pnum-fx-alt-group > /dev/null 2>&1 ; then
        if ! [ -d "fx/export_alt" -a -d "fx/import_alt" ] ; then
            mkdir -p "fx/export_alt"
            mkdir -p "fx/import_alt"
            logmsg "created import_alt and export_alt directories for project $pnum"
        fi

	chown root:$egroup fx/export_alt
	chmod 2770         fx/export_alt
	chown root:$igroup fx/import_alt
	chmod 2770         fx/import_alt
    fi

    if getent group $vgroup > /dev/null 2>&1 ; then
        if [ ! -d "${pdir}/fx/import_alt/vidyo" ]; then
            mkdir -pm 2770     fx/import_alt/vidyo
            chown root:$vgroup fx/import_alt/vidyo
            logmsg "created vidyo directory for project $pnum"
        fi
    fi

    if true ; then
	for u in $(getent group $group | cut -d: -f4- | tr , " "); do
	    home=home/$u
	    if [ ! -d $home ] ; then
		infomsg "trying to create home directory for $u"
		if groupexist $u-group \
		    && mkdir $home.new \
		    && chown $u $home.new \
		    && chmod 700 $home.new \
		    && chgrp -f $u-group $home.new \
		    ; then
		    mv $home.new $home
		    pwd="$(pwd)"
		    logmsg "created home for $u in $pwd/$home"
		else
		    errormsg "creating home directory for user $u failed"
		    if [ -d $home.new ] ; then
			rmdir $home.new
		    fi
		fi
	    fi
	done
    fi
    return 0
}

setup_project_sharedir() {
    pdir="$1"
    pnum=$(echo $pdir|cut -d/ -f3|sed 's/^\(p[0-9]*\).*/\1/')
    group=$pnum-member-group
    cd $pdir
    mkdir -p data/durable
    mkdir -p data/no-backup

    chown root:$group . data
    chmod 755 . data
    chmod 2775 data/durable data/no-backup
    if ! groupexist $group ; then
	errormsg "unable to find needed group $group, can not to set up $pdir as sharedir"
	return 1
    fi
    infomsg "updating project share file permissions for $pdir"
    chown root:$group data/durable data/no-backup
}

# Split disk volumes can not have files on the top level.  Move flag
# files to subdirectory data/durable/projectinfo/ because of this.
flagpresent() {
    pdir="$1"
    flag="$2"
    if [ -e "$pdir/$flag" ]\
	|| [ -e "$pdir/data/durable/projectinfo/$flag" ] ; then
	return 0
    fi
    return 1
}

while [ "$1" ] ; do
    pdir="$1"
    shift
    if [ -d "$pdir" ] ; then
	if [ -d "$pdir/data/durable" ] ; then
	    if flagpresent "$pdir" "migration-override" ; then
		errormsg "$pdir/migration-override exist, ignoring project disk"
	    elif flagpresent "$pdir" "project-share" ; then
		setup_project_sharedir "$pdir" || true
	    else
		setup_projectdir "$pdir" || true
	    fi
	else
	    errormsg "missing data/durable in $pdir, skipping partition"
	fi
    else
	errormsg "unable to find directory $pdir"
    fi
done