#!/bin/bash
#
# imaps_check_ - munin plugin for IMAPs benchmark check
# Copyright (C) 2008 Marek Mahut <mmahut@fedoraproject.org>, special thanks to Lubomir Rintel <lkundrak@v3.sk>
#
# Usage:
#       ln -s /usr/share/munin/plugins/imaps_check_ /etc/munin/plugins/imaps_check_mailserver.example.com:993
#
#       imaps_login - IMAPS login 
#       imaps_password - IMAPS password
#       imaps_mailbox - Mailbox to examine
#       imaps_mails - Messages to download in format: from_mailid:to_mailid [flag|fast|all|full|...]
#
#       More info in RFC 2060 (INTERNET MESSAGE ACCESS PROTOCOL - VERSION 4rev1). This script logins to the imaps server,
#	examine mailbox folder and list first 100 mails before exiting.
#
# 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
# of the License, 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.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

host=`basename $0 | sed 's/^imaps_check_//g'`
imaps_login=${imaps_login:-user@example.com}
imaps_password=${imaps_password:-SecretPassw0rd}
imaps_mailbox=${imaps_mailbox:-INBOX}
imaps_mails=${imaps_mails:-1:100 full}

if [ "$1" == "config" ]; then

        echo "graph_title IMAPS login and fetch response time"
        echo "graph_vlabel response in sec"
        echo "graph_period minute"
        echo "graph_args --base 1000 --lower-limit 0"
        echo "imaps_check.label $host"

elif [ "$1" == "autoconf" ]; then

        if [ -x /usr/bin/time ] && [ -x /usr/bin/expect ] && [ -x /usr/bin/openssl ]; then
                echo "yes"
                exit 1
        else
                echo "no (/usr/bin/time or /usr/bin/expect or /usr/bin/openssl missing)"
                exit 0
        fi

else

        response=`echo "spawn openssl s_client -connect $host
        expect -re \"OK.*\n\"
        send \". login $imaps_login $imaps_password\n\" 
        expect -re \"OK.*\n\"
        send \". examine $imaps_mailbox\n\"
        expect -re \"completed.*\n\"
        send \". fetch $imaps_mails\n\"
        expect -re \"completed.*\n\"
        send \"QUIT\"
        " | /usr/bin/time -f "%e" time expect 2>&1 | tail -n 1`
        echo "imaps_check.value $response"

fi