# Copyright 1999-2006 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Header: /var/cvsroot/gentoo-x86/net-misc/htbinit/htbinit-0.8.5-r1.ebuild,v 1.1 2006/10/13 11:01:22 bass Exp $

inherit eutils linux-mod

DESCRIPTION="Sets up Hierachical Token Bucket based traffic control (QoS) with iproute2"
HOMEPAGE="http://www.sourceforge.net/projects/htbinit"
SRC_URI="mirror://sourceforge/htbinit/htb.init-v${PV}"

LICENSE="GPL-2"
SLOT="0"
KEYWORDS="~x86 ~ppc ~sparc alpha ~amd64"
IUSE="ipv6 esfq"

DEPEND="sys-apps/iproute2"

S=${WORKDIR}

pkg_setup() {
        linux-mod_pkg_setup

	ebegin "Checking for SCH_HTB support"
        linux_chkconfig_present NET_SCH_HTB
    	eend $?
    	if [[ $? -ne 0 ]] ; then
        	eerror "This version needs sch_htb support!"
	        die "sch_htb support not detected!"
    	fi
	ebegin "Checking for SCH_SFQ support"
        linux_chkconfig_present NET_SCH_SFQ
    	eend $?
    	if [[ $? -ne 0 ]] ; then
        	eerror "This version needs sch_sfq support!"
	        die "sch_sfqsupport not detected!"
    	fi
	ebegin "Checking for CLS_FW support"
        linux_chkconfig_present NET_CLS_FW
    	eend $?
    	if [[ $? -ne 0 ]] ; then
        	eerror "This version needs cls_fw support!"
	        die "cls_fw support not detected!"
    	fi
	ebegin "Checking for CLS_U32 support"
        linux_chkconfig_present NET_CLS_U32
    	eend $?
    	if [[ $? -ne 0 ]] ; then
        	eerror "This version needs sch_u32 support!"
	        die "sch_u32 support not detected!"
    	fi
	ebegin "Checking for CLS_ROUTE support"
        linux_chkconfig_present NET_CLS_ROUTE
    	eend $?
    	if [[ $? -ne 0 ]] ; then
        	eerror "This version needs cls_route support!"
	        die "cls_route support not detected!"
    	fi

	if use esfq; then
		ebegin "Checking for NET_SCH_ESFQ support"
    	        linux_chkconfig_present NET_SCH_ESFQ
    		eend $?

    		if [[ $? -ne 0 ]] ; then
            		eerror "This version needs sch_esfq support!"
            		eerror "See    http://fatooh.org/esfq-2.6/"
	                die "sch_esfq support not detected!"
    		fi
	fi
}

src_unpack() {
	cp ${DISTDIR}/htb.init-v${PV} ${S}/htb.init
}

src_compile() {
	sed -i 's|/etc/sysconfig/htb|/etc/htb|g' ${S}/htb.init
	epatch ${FILESDIR}/htb.init-v0.8.5_tos.patch
	use ipv6 && epatch ${FILESDIR}/htb_0.8.5_ipv6.diff
	use esfq && epatch ${FILESDIR}/htb_0.8.5_esfq.diff
	epatch ${FILESDIR}/prio_rule.patch
	epatch ${FILESDIR}/timecheck_fix.patch
	epatch ${FILESDIR}/htb.init_find_fix.patch
}

src_install() {
	dosbin htb.init

	newinitd ${FILESDIR}/htbinit.rc htbinit

	keepdir /etc/htb
}

pkg_postinst() {
	einfo 'Run "rc-update add htbinit default" to run htb.init at startup.'
	einfo 'Please, read carefully the htb.init documentation.'
	einfo 'new directory to store configuration /etc/htb'
}