#!/bin/bash

GEOIP_URL="http://geolite.maxmind.com/download/geoip/database/"

GEOLITE_COUNTRY_PATH="GeoLiteCountry/"
GEOLITE_COUNTRY_FILE="GeoIP.dat"

GEOLITE_COUNTRY_IPV6_PATH=""
GEOLITE_COUNTRY_IPV6_FILE="GeoIPv6.dat"

GEOLITE_CITY_PATH=""
GEOLITE_CITY_FILE="GeoLiteCity.dat"
GEOLITE_CITY_LINK="GeoIPCity.dat"

GEOLITE_ASNUM_PATH="asnum/"
GEOLITE_ASNUM_FILE="GeoIPASNum.dat"

FAILED=0

DEST=/usr/share/GeoIP

for url in \
    "$GEOIP_URL$GEOLITE_COUNTRY_PATH$GEOLITE_COUNTRY_FILE.gz" \
    "$GEOIP_URL$GEOLITE_COUNTRY_IPV6_PATH$GEOLITE_COUNTRY_IPV6_FILE.gz" \
    "$GEOIP_URL$GEOLITE_CITY_PATH$GEOLITE_CITY_FILE.gz" \
    "$GEOIP_URL$GEOLITE_ASNUM_PATH$GEOLITE_ASNUM_FILE.gz"
do
    FILE=$(basename $url)

    [ -f $DEST/$FILE ] && rm -f $DEST/$FILE
    /usr/bin/wget -q -t3 -T15 -P $DEST/ "$url" && \
        /bin/gunzip -f $DEST/$FILE
    if [ "$?" != "0" ]
    then
        FAILED=1
    fi
done

[ -h $DEST/$GEOLITE_CITY_LINK ] || ln -s $GEOLITE_CITY_FILE $DEST/$GEOLITE_CITY_LINK

if [ "$FAILED" == "0" ]; then
	for chrdir in /chroot/*; do
		if [ -d $chrdir/$DEST ]; then
			cp -d $DEST/* $chrdir/$DEST/
		fi
	done
fi

exit $FAILED