#!/bin/sh

set -euo pipefail

# Validate environment variables
[ -z "$SITE_DOMAIN" ]       && { echo "SITE_DOMAIN env variable required";       exit 1; }
[ -z "$SITE_ALIASES" ]      && { echo "SITE_DOMAIN env variable required";       exit 1; }
[ -z "$GEOIP_LICENSE_KEY" ] && { echo "GEOIP_LICENSE_KEY env variable required"; exit 1; }

# Create the config template
cat > /etc/awstats/awstats.${SITE_DOMAIN}.conf <<EOF
LogFile="/input/${SITE_DOMAIN}.log"
DirData="/output"
LogFormat = "%virtualname %host - %other %time1 %methodurl %code %bytesd %refererquot %uaquot"
SiteDomain="${SITE_DOMAIN}"
HostAliases="${SITE_DOMAIN} ${SITE_ALIASES}"
#ShowScreenSizeStats=1
DefaultFile="index.html default.html"
AllowFullYearView=3

LoadPlugin="ipv6"
# Should be enabled for build only
LoadPlugin="geoip2 /geoip/GeoLite2-Country.mmdb"
LoadPlugin="geoip2_city /geoip/GeoLite2-City.mmdb"

#ExtraSectionName1="Redirected Hit"
#ExtraSectionCodeFilter1="302"
#ExtraSectionCondition1="URL,\/offsite"
#ExtraSectionFirstColumnTitle1="Url"
#ExtraSectionFirstColumnValues1="QUERY_STRING,url=([^&]+)"
#ExtraSectionStatTypes1=HL
#MaxNbOfExtra1=500
#MinHitExtra1=1
#ExtraSectionAddSumRow1=1
EOF

# Download and setup GeoIP Databases
curl -s "https://download.maxmind.com/app/geoip_download?edition_id=GeoLite2-ASN&suffix=tar.gz&license_key=${GEOIP_LICENSE_KEY}" | tar -xz -C /tmp
curl -s "https://download.maxmind.com/app/geoip_download?edition_id=GeoLite2-City&suffix=tar.gz&license_key=${GEOIP_LICENSE_KEY}" | tar -xz -C /tmp
curl -s "https://download.maxmind.com/app/geoip_download?edition_id=GeoLite2-Country&suffix=tar.gz&license_key=${GEOIP_LICENSE_KEY}" | tar -xz -C /tmp

find /tmp -name '*.mmdb' -exec cp '{}' /geoip/ \;
rm -rf /tmp/GeoLite2*

/opt/awstats/wwwroot/cgi-bin/awstats.pl -config=${SITE_DOMAIN} -update -dir=/output