#!/bin/bash

export LC_ALL=C

HWSETUP_DIR=/etc/conf.d/hwsetup
HWINFO="/usr/sbin/hwinfo"

UNSUPPORTED_GFXCARD_DRIVERS=""
UNSUPPORTED_NETCARD_DRIVERS=""
if [ -f ${HWSETUP_DIR}/unsupported-gfxcard-drivers ]
then
	UNSUPPORTED_GFXCARD_DRIVERS="$(< ${HWSETUP_DIR}/unsupported-gfxcard-drivers)"
fi
if [ -f ${HWSETUP_DIR}/unsupported-netcard-drivers ]
then
	UNSUPPORTED_NETCARD_DRIVERS="$(< ${HWSETUP_DIR}/unsupported-netcard-drivers)"
fi

clearconfig()
{
	local file="$1"
	: > ${HWSETUP_DIR}/${file}
}

addconfig()
{
	local file="$1"
	local data="$2"

	echo "${data}" >> ${HWSETUP_DIR}/${file}
}

# check_unsupported_netcard_drivers UNSUPPORTED_ARRAY DRIVER
check_unsupported_drivers()
{
	local unsupported="$1"
	local driver="$2"
	local i

	# filter unsupported devices
	for i in ${unsupported}
	do
		[[ ${i} = ${driver} ]] && return 1
	done
	return 0
}

get_gfxcard_sysfsid_from_specific_hwid()
{
	local SEARCH_ID="$1"
	local hwid
	local sysfsid
	local counter

	declare -i counter=0

	echo "${GFXCARD_INFO}" | while read line
	do
		# get hwinfo device id, its always the first line
		if [[ ${counter} = 0 ]]
		then
			# hwids begins all with ^ID:, sub informations have an empty space
			hwid=$(echo "${line}" | grep '^[0-9].*' | sed 's:\(^[0-9]*\)\:\ .*:\1:')
		fi

		# search for device id
		if [[ ${SEARCH_ID} = ${hwid} ]]
		then
			case "${line}" in
				*'SysFS ID:'*)
					# subst the id
					sysfsid="${line#*:}"
					# trim spaces
					sysfsid="${sysfsid// }"
					echo "${sysfsid}"
					;;
			esac
		fi

		# next hardware item
		if [[ -z ${line} ]]
		then
			counter=0
		else
			((counter++))
		fi
	done
}

get_primary_display_hwid()
{
	local hwid
	hwid="$(echo "${GFXCARD_INFO}" | grep 'Primary display adapter:' | sed 's:.*\:\ \#\([0-9]*\):\1:')"
	echo "${hwid}"
}


# disable probeonly, load all modules by default
probeonly=0

# check getops
for i in $*
do
	case $1 in
		# ignoring -p,-v,-a,-s
		-p|-v|-a|-s) shift;;
		# only support dry-run
		-n) shift; probeonly=1 ;;
	esac
	shift
done

echo -n "Autoconfiguring devices...  " 1>&2

GFXCARD_INFO="$(${HWINFO} --gfxcard)"
# check for a primary device and rerun hwinfo for this device only
primary_display_hwid="$(get_primary_display_hwid)"
if [[ -n ${primary_display_hwid} ]]
then
	GFXCARD_INFO="$(${HWINFO} --gfxcard --only $(get_gfxcard_sysfsid_from_specific_hwid ${primary_display_hwid}))"
fi
NETCARD_INFO="$(${HWINFO} --netcard)"
MOUSE_INFO="$(${HWINFO} --mouse)"

# eval arrays with all modules and descriptions
#
# graphic
eval $(echo GFXCARD_MODULE=\($(echo "${GFXCARD_INFO}" | grep 'XFree86.*Module:' | sed 's:.*\:\ \(.*\)$:\"\1\":')\))
eval $(echo GFXCARD_DESC=\($(echo "${GFXCARD_INFO}" | grep 'Model:' | sed 's:.*\:\ \"\(.*\)\"$:\"\1\":')\))
# the xserver is always xorg, no array needed
GFXCARD_XSERVER="Xorg"
#
# network
eval $(echo NETCARD_MODULE=\($(echo "${NETCARD_INFO}" | grep 'Driver Modules:' | sed 's:.*\:\ \"\(.*\)\"$:\"\1\":')\))
eval $(echo NETCARD_DESC=\($(echo "${NETCARD_INFO}" | grep 'Model:' | sed 's:.*\:\ \"\(.*\)\"$:\"\1\":')\))
# fallback (only needed for systems without netlink)
eval $(echo NETCARD_MODULE_FB=\($(echo "${NETCARD_INFO}" | grep 'Driver Activation Cmd:' | sed 's:.*\:\ \"modprobe\ \(.*\)\"$:\"\1\":')\))
#
# mouse
eval $(echo MOUSE_MODULE=\($(echo "${MOUSE_INFO}" | grep 'Driver Modules:' | sed 's:.*\:\ \"\(.*\)\"$:\"\1\":')\))
eval $(echo MOUSE_DESC=\($(echo "${MOUSE_INFO}" | grep 'Model:' | sed 's:.*\:\ \"\(.*\)\"$:\"\1\":')\))
eval $(echo MOUSE_DEVICE=\($(echo "${MOUSE_INFO}" | grep 'Device File:' | sed 's:.*\:\ \(.*\)[$\ ].*:\"\1\":')\))
eval $(echo MOUSE_GPM_PROTO=\($(echo "${MOUSE_INFO}" | grep 'GPM.*Protocol:' | sed 's:.*\:\ \(.*\)$:\"\1\":')\))
eval $(echo MOUSE_X11_PROTO=\($(echo "${MOUSE_INFO}" | grep 'XFree86.*Protocol:' | sed 's:.*\:\ \(.*\)$:\"\1\":')\))

# get the number of devices for each class
GFXCARD_COUNT=$(echo "${GFXCARD_INFO}" | grep -c "[0-9a-zA-Z]\: .*\:\ .*")
NETCARD_COUNT=$(echo "${NETCARD_INFO}" | grep -c "[0-9a-zA-Z]\: .*\:\ .*")
MOUSE_COUNT=$(echo "${MOUSE_INFO}" | grep -c "[0-9a-zA-Z]\: .*\:\ .*")


# clear all config files
clearconfig xserver
clearconfig netcard
clearconfig knoppix
clearconfig mouse

for ((i=0; i<GFXCARD_COUNT; i++))
do
	# fallback to vesa
	if [[ -z ${GFXCARD_MODULE[${i}]} ]]
	then
		GFXCARD_MODULE[${i}]="vesa"
	fi

	# exclude unsupported drivers
	check_unsupported_drivers "${UNSUPPORTED_GFXCARD_DRIVERS}" "${NETCARD_MODULE[${i}]}" || continue

	addconfig xserver "XSERVER=\"${GFXCARD_XSERVER}\"" 
	addconfig xserver "XMODULE=\"${GFXCARD_MODULE[${i}]}\""
	addconfig xserver "XDESC=\"${GFXCARD_DESC[${i}]}\""

	# add them to knoppix too
	addconfig knoppix "XSERVER=\"${GFXCARD_XSERVER}\"" 
	addconfig knoppix "XMODULE=\"${GFXCARD_MODULE[${i}]}\""
	addconfig knoppix "XDESC=\"${GFXCARD_DESC[${i}]}\""

	# load some needed modules for xorg-servers without udev or hal support, which the driver would normally autoload
	case "${GFXCARD_MODULE[${i}]}" in
		intel|i810) modprobe -q intel-agp ;;
		nv) modprobe -q nvidia-agp ;;
		sis) modprobe -q sis-agp ;;
		ati|radeon|r128|mach64) modprobe -q ati-agp ;;
		via|openchrome|unichrome) modprobe -q via-agp ;;
	esac
done

for ((i=0; i<NETCARD_COUNT; i++))
do
	# use fallback
	if [[ -z ${NETCARD_MODULE[${i}]} ]]
	then
		NETCARD_MODULE[${i}]="${NETCARD_MODULE_FB[${i}]}"
	fi

	# exclude this driver if its still empty, the provided kernel does not support this hardware
	if [[ -z ${NETCARD_MODULE[${i}]} ]]
	then
		continue
	fi

	# exclude unsupported drivers
	check_unsupported_drivers "${UNSUPPORTED_NETCARD_DRIVERS}" "${NETCARD_MODULE[${i}]}" || continue

	addconfig netcard "FULLNAME=\"${NETCARD_DESC[${i}]}\"" 
	addconfig netcard "DRIVER=\"${NETCARD_MODULE[${i}]}\""

	# add them to knoppix too
	addconfig knoppix "NETCARD_FULLNAME=\"${NETCARD_DESC[${i}]}\""
	addconfig knoppix "NETCARD_DRIVER=\"${NETCARD_MODULE[${i}]}\""

	# load the netcard modules
	if [[ ${probeonly} = 0 ]]
	then
		modprobe -q "${NETCARD_MODULE[${i}]}"
	fi
done

for ((i=0; i<MOUSE_COUNT; i++))
do
	# fix mouseproto to be hwsetup compatible (it just guesses the proto :/ )
	case "${MOUSE_GPM_PROTO[${i}]}" in
		exps2) MOUSE_GPM_PROTO[${i}]="ps2" ;;
		"") MOUSE_GPM_PROTO[${i}]="ps2" ;;
	esac
	case "${MOUSE_X11_PROTO[${i}]}" in
		explorerps/2) MOUSE_X11_PROTO[${i}]="IMPS/2" ;;
		ps/2) MOUSE_X11_PROTO[${i}]="PS/2" ;;
		"") MOUSE_X11_PROTO[${i}]="PS/2" ;;
	esac

	addconfig mouse "MOUSETYPE=\"${MOUSE_GPM_PROTO[${i}]}\""
	addconfig mouse "XMOUSETYPE=\"${MOUSE_X11_PROTO[${i}]}\""
	addconfig mouse "FULLNAME=\"${MOUSE_DESC[${i}]}\""
	addconfig mouse "DEVICE=\"${MOUSE_DEVICE[${i}]}\""
	addconfig mouse "DRIVER=\"${MOUSE_MODULE[${i}]}\""

	# add them to knoppix too
	addconfig knoppix "MOUSE_MOUSETYPE=\"${MOUSE_GPM_PROTO[${i}]}\""
	addconfig knoppix "MOUSE_XMOUSETYPE=\"${MOUSE_X11_PROTO[${i}]}\""
	addconfig knoppix "MOUSE_FULLNAME=\"${MOUSE_DESC[${i}]}\""
	addconfig knoppix "MOUSE_DEVICE=\"${MOUSE_DEVICE[${i}]}\""
	addconfig knoppix "MOUSE_DRIVER=\"${MOUSE_MODULE[${i}]}\""
done

echo "Done." 1>&2