#!/bin/sh
# ============================================================================
#
# This file is part of the 'Darkstat' package
# 
# This program is free software; you can redistribute it and/or modify it 
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2, or (at your option) any
# later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# Author: Mijzelf <Mijzelf@live.com>
#
# ============================================================================
PKG_NAME="Darkstat"
THIS_SCRIPT="Darkstat"
PKG_FIRMWARE=4

############################################################
## Check if we are in upgradig mode. From the point of view of the package
## an upgrade is a uninstall followed by a install. So if you want to preserve
## something over an upgrade, you have to detect it somehow
##
## Function sets PKG_UPGRADE > 0 when upgrading, 
## and PKG_UPGRADE=0 when not
############################################################
IsUpgrading()
{
	PKG_UPGRADING=0
	local dir=
	for dir in "/usr/local/zy_pkgs/" "/i-data/.system/zy-pkgs/"
	do
		local logfile=${dir}tmp/zypkg.log
		[ ! -f ${logfile} ] && continue

		local lastupdate=` grep "Upgrade Mode" ${logfile} | tail -n 1 | grep "\[${PKG_NAME}\]" | awk -v OFS=' ' '{print $1, $2}' `

	        [ "${lastupdate}" = "" ] && return 1

		local since1970=` TZ=GMT date -d "${lastupdate}" +%s `
		local now=`date +%s `
		local ago=""
		let ago=now-since1970
		[ $ago -eq 0 ] && ago=1
		[ 300 -gt $ago ] && PKG_UPGRADING=$ago && return 0

		return 1
	done

	# We shouldn't be here
	return 1
}

#############################################################
## Dummy logcode
#############################################################
Log() 
{
	return
}

######################################
## Find the IP address of the box
FrameworkGetIPAddress()
{
	# Detect interfaces
	local interfaces=` ifconfig | grep -e "^egiga" -e "^bond" -e "^eth" | cut -d ' ' -f 1 `

	# And find the IP (v4) address
	for interface in $interfaces
	do
		local address=` ifconfig ${interface} | grep "inet addr" | tr -s ' ' `
		if [ "${address}" != "" ] ; then
			echo ${address} | cut -d ' ' -f 2 | cut -d ':' -f 2
			return
		fi
	done
	hostname # What else?
}

#####################################
## Echo the URL of the package
FrameworkGetLink()
{
	GetLink
}

######################################
## Status get and set
FrameworkStatus()
{
	local pkg_config_file="${PKG_ROOT}/config/${PKG_NAME}/config"
	if [ "get" = "$1" ] ; then
		if [ ! -f ${pkg_config_file} ] ; then
			FrameworkStatus set Disabled
		fi
		cat ${pkg_config_file}
		return
	elif [ "set" = "$1" ] ; then
		mkdir -p ` dirname ${pkg_config_file} `
		
		echo $2 >${pkg_config_file}
	fi
}

################################################
## Run or stop the package
FrameworkRun()
{
	if [ $1 -eq 1 ] ; then
		# Startup
		STATUS=` FrameworkStatus get `
		[ "$STATUS" != "Enabled" ] && return	# We're not enabled

		Startup
		return
	fi

	Shutdown
	
}

FrameworkVersion()
{
	local controlfile=${PKG_ROOT}/zypkg_conf/info/${PKG_NAME}.control
	local version=$( grep "^Version:" ${controlfile} | awk -F"zypkg" '{print $2}' )
	exit $version
}

FrameworkParseParams()
{
	case $1 in
		getlink)
			FrameworkGetLink
			;;
		status)
			FrameworkStatus get
			;;
		enable)
			FrameworkStatus set Enabled 
			FrameworkRun 1
			;;
		disable)
			FrameworkRun 0
			FrameworkStatus set Disabled 
			;;
		startup)
			FrameworkRun 1
			;;
		shutdown)
			FrameworkRun 0
			;;
		version)
			FrameworkVersion
			;;
	esac
}

Log

PKG_ROOT="/usr/local/zy-pkgs"

PKG_USER=nobody
PKG_PORT=667

GetLink()
{
    local ADDR=` FrameworkGetIPAddress `
    echo  "http://${ADDR}:${PKG_PORT}/"
}

Startup()
{
    local iface=$( route | grep ^default | awk '{print $8}' )
    mkdir -p ${PKG_ROOT}/var
    chmod a+w ${PKG_ROOT}/var

    ${PKG_ROOT}/sbin/darkstat -i ${iface} -p ${PKG_PORT} --user ${PKG_USER} --chroot ${PKG_ROOT}/var
}

Shutdown()
{
    killall darkstat
}

#usage: darkstat [ -i interface ]
#                [ -f filter ]
#                [ -r capfile ]
#                [ -p port ]
#                [ -b bindaddr ]
#                [ -l network/netmask ]
#                [ --base path ]
#                [ --local-only ]
#                [ --snaplen bytes ]
#                [ --pppoe ]
#                [ --syslog ]
#                [ --verbose ]
#                [ --no-daemon ]
#                [ --no-promisc ]
#                [ --no-dns ]
#                [ --no-macs ]
#                [ --no-lastseen ]
#                [ --chroot dir ]
#                [ --user username ]
#                [ --daylog filename ]
#                [ --import filename ]
#                [ --export filename ]
#                [ --pidfile filename ]
#                [ --hosts-max count ]
#                [ --hosts-keep count ]
#                [ --ports-max count ]
#                [ --ports-keep count ]
#                [ --highest-port port ]
#                [ --wait secs ]
#                [ --hexdump ]
#                [ --version ]
#                [ --help ]

FrameworkParseParams "$@"

exit 0
