#!/bin/bash

set -e -o pipefail 

DEFAULT_INFILE="http://geolite.maxmind.com/download/geoip/database/GeoIPCountryCSV.zip"
DEFAULT_CACHEFILE="/etc/haproxy/GeoIPCountryCSV.zip"

function usage()
{
	cat >&2 << EOF

Usage: $0 [-l] [-i <input file>] [<country code>...]

Options :
  -h              display this help and exit
  -l              applies a lowercase filter to the output
  -c <cachefile>  when using an HTTP URL, defines where the file will be cached
                    on disk. This prevents downloading a file that was not
                    modified
                    Default: $DEFAULT_CACHEFILE
  -i <infile>     the input file containing the GeOIP database.
                    it can be a local file or an HTTP URL. If the file name
                    ends with ".zip", it will be automatically unzipped
                    Default: $DEFAULT_INFILE

  <country code>  if specified, only those countries will be sent to the
                    output. If not specified, all of them will


Example :
$ $0 -l -i $DEFAULT_INFILE FR UK IT US
EOF
	exit 1
}
# Initialize the parameters from the command line
filters=""
infile=$DEFAULT_INFILE
cachefile=$DEFAULT_CACHEFILE
lowercase_opts=0
while getopts "c:i:hl" opt;
do
	case $opt in
		c) cachefile=$OPTARG;;
		i) infile=$OPTARG;;
		l) lowercase_opts=1;;
		h | \?) usage;;
	esac
done
shift $(($OPTIND - 1)) 


# Fetch the file if it's a HTTP URL, then use the cache file as input
if [[ "$infile" =~ ^http://.* ]];
then
	curl -fs -z $cachefile -o $cachefile $infile || { echo "Unable to fetch the GeoIP data file"; exit 1; }
	infile=$cachefile
fi

if [ ! -e $infile ];
then
	echo "Le fichier source de géolocalisation $infile n'existe pas". >&2
	exit 1
fi

# Should we decompress it on the fly ?
if [[ "$infile" =~ ^.*\.zip ]];
then
	reader="funzip"
else
	reader="cat"
fi

# Initialize the country filter
countries=""
for country in $*;
do
	if [ "$countries" = "" ];
	then
		countries="$country"
	else
		countries="$countries|$country"
	fi
done

if [ "$countries" != "" ];
then
	filters="$filters|grep -Ei '($countries)'"
fi

# Initialize the lowercase filter
if [ $lowercase_opts -eq 1 ];
then
	filters="$filters|tr A-Z a-z"
fi

# Now we can launch the conversion process
eval "$reader $infile | cut -d, -f1,2,5 | iprange | sed 's/\"//g' $filters" || { echo "GeoIP data conversion failure"; exit 1; }