#!/bin/bash

[[ -z "$@" ]] && /usr/bin/su && exit 0

user="$1"
if [ "$user" != "root" ] && [[ ! "$(ls /home)" =~ "$user" ]]; then
    cmd="$user"
    user="root"
else
    shift
    [ "$1" = "-c" ] && shift
    cmd="$1"
fi

[ "$user" = "root" ] && home="/root" || home="/home/$user"
if [ -n "$cmd" ]; then
    su $user -c ". $home/.profile; $cmd"
else
    su - $user
fi
