#!/bin/sh

db()
{
	/usr/bin/mysql $@ -ugld -pgld gld
}

run()
{
  _REGEXP="$1"
  _COMMENT="$2"
  grep -E "$_REGEXP" /var/log/mail.log /var/log/mail.log.0 |sed 's/.*\[\([^]]*\)\].*/insert into whitelist values ("\1", "'"$_COMMENT"'");/'|sort -u | db --force 2>&1 | grep -v "Duplicate entry"
}

count()
{
	echo "select count(*) from whitelist" | db --skip-column-names 
}

before=`count`

run '\.google\.com\[' "Gmail"
run '\.tfbnw\.net\[' "Facebook"
run '\.pixmania\.com\[' "Pixmania"

after=`count`

echo "Added `expr $after - $before` hosts to whitelist."