#!/bin/sh
############################################################################
# Copyright:
#      (C) 2008 - 2010  Alexander Shaduri <ashaduri 'at' gmail.com>
# License: See LICENSE_zlib.txt file
############################################################################

# Run gsmartcontrol with root, asking for root password first.
# export GSMARTCONTROL_SU to override a su command (e.g. "kdesu -c").

EXEC_BIN="/usr/local/bin/gsmartcontrol";
prog_name="gsmartcontrol"

DESKTOP="$1";

# This works for --help too
if [ "$DESKTOP" = "" ]; then
	DESKTOP="auto";  # default

elif [ "$DESKTOP" != "auto" ] && [ "$DESKTOP" != "kde" ] && \
		[ "$DESKTOP" != "gnome" ] && [ "$DESKTOP" != "other" ]; then
	echo "Usage: $0 [<auto|kde|gnome|other> [<${prog_name}_options>] ]";
	exit 1;
fi


# Auto-detect current desktop if auto was specified.
if [ "$DESKTOP" = "auto" ]; then
	# KDE_SESSION_UID is present on kde3 and kde4.
	# Note that it may be empty (but still set)
	if [ "${KDE_SESSION_UID+set}" = "set" ]; then
		DESKTOP="kde";
	# same with gnome
	elif [ "${GNOME_DESKTOP_SESSION_ID+set}" = "set" ]; then
		DESKTOP="gnome";
	else
		DESKTOP="other";
	fi
fi

# echo $DESKTOP;


# They're basically the same, only the order is different.
# sux requires xterm to ask for the password.
# xdg-su is basically like this script, except worse :)
# su-to-root is a debian/ubuntu official method (although gksu is available).
gnome_sus="gksu-polkit";
kde_sus="kdesu";
other_sus="$gnome_sus";


candidates="";
found_su=""

if [ "$DESKTOP" = "gnome" ]; then
	candidates="$gnome_sus";
elif [ "$DESKTOP" = "kde" ]; then
	candidates="$kde_sus";
elif [ "$DESKTOP" = "other" ]; then
	candidates="$other_sus";
fi

if [ "$GSMARTCONTROL_SU" = "" ]; then
	for subin in $candidates; do
		which $subin > /dev/null 2>&1
		if [ $? -eq 0 ]; then
			found_su="$subin";
			break;
		fi
	done

	if [ "$found_su" = "" ]; then
		xmessage "Error launching ${prog_name}: No suitable su mechanism found.
Try installing kdesu or gksu-polkit first.";
		exit 1;
	fi
fi


# gnomesu and gksu (but not kdesu, not sure about others) fail to adopt
# root's PATH. Since the user's PATH may not contain /usr/sbin (with smartctl)
# on some distributions (e.g. mandriva), add it manually. We also add
# /usr/local/sbin, since that's the default location of custom-compiled smartctl.
# Add these directories _before_ existing PATH, so that the user is not
# tricked into running it from some other path (we're running as root with
# the user's env after all).
# Note that beesu won't show a GUI login box if /usr/sbin is before /usr/bin,
# so add it first as well.
EXTRA_PATHS="/usr/bin:/usr/sbin:/usr/local/sbin";
export PATH="$EXTRA_PATHS:$PATH"


# echo $found_su;

# Examples:
# gnomesu -c 'gsmartcontrol --no-scan'
# kdesu -c 'gsmartcontrol --no-scan'
# beesu -P 'gsmartcontrol --no-scan'
# su-to-root -X -c 'gsmartcontrol --no-scan'
# xterm -e sux -c 'gsmartcontrol --no-scan'  # sux asks for password in a terminal

full_cmd="";

if [ "$GSMARTCONTROL_SU" != "" ]; then
	full_cmd="$GSMARTCONTROL_SU '$EXEC_BIN $@'";

elif [ "$found_su" = "sux" ]; then
	full_cmd="xterm -e sux -c '$EXEC_BIN $@'";

elif [ "$found_su" = "gksu" ]; then
	full_cmd="$found_su '$EXEC_BIN $@'";

elif [ "$found_su" = "beesu" ]; then
	full_cmd="$found_su -P '$EXEC_BIN $@'";

elif [ "$found_su" = "su-to-root" ]; then
	full_cmd="$found_su -X -c '$EXEC_BIN $@'";

elif [ "$found_su" = "$gnome_sus" ]; then
	full_cmd="$found_su $EXEC_BIN $@";

else  # kdesu, xdg-su
	full_cmd="$found_su -c '$EXEC_BIN $@'";
fi


#echo $full_cmd
eval $full_cmd




