#!/usr/bin/python
# -*- coding: UTF-8 -*-
# -*- Mode: Python -*-
#
# Copyright (C) 2006 Bertera Pietro <pietro@bertera.it>
# Response time monitoring with perfdata modification by Ivan Savcic <isavcic@gmail.com> and Milos Buncic, 2012.
# From: https://github.com/isavcic/check_imap_login

# This file may be distributed and/or modified under the terms of
# the GNU General Public License version 2 as published by
# the Free Software Foundation.
# This file is distributed without any warranty; without even the implied
# warranty of merchantability or fitness for a particular purpose.

import sys, os, imaplib, getopt
from time import time

def usage():
	print sys.argv[0] + " -u <user> -p <password> -H <host> [-s] -w <warning threshold (sec)> -c <critical threshold (sec)>\n -s is for using IMAPS"

def main():
	try:
		opts, args = getopt.getopt(sys.argv[1:], "u:p:sH:w:c:")
	except getopt.GetoptError:
		usage()
		return 3
    
	user = host = password = use_ssl = warning = critical = None

	for o, a in opts:
		if o == "-u":
			user = a
		elif o == "-p":
			password = a
		elif o == "-s":
			use_ssl = True
		elif o == "-H":
			host = a
		elif o == "-w":
			warning = float(a)
		elif o == "-c":
			critical = float(a)

	if user == None or password == None or host == None or warning == None or critical == None:
		usage()
		return 1
    
	if use_ssl:
		M = imaplib.IMAP4_SSL(host=host)
	else:
		M = imaplib.IMAP4(host)
    
	timestamp = time()
	
	try:	
		M.login(user, password)
	except Exception, e:
		print "CRITICAL IMAP Login Failed: %s" % e
		return 2

	M.logout()

	timestamp = time() - timestamp

	if timestamp < warning:
		status = "OK"
		exitcode = 0
	if timestamp >= warning:
		status = "WARNING"
		exitcode = 1
	if timestamp >= critical:
		status = "CRITICAL"
		exitcode = 2

	print '%s IMAP Login | response_time=%.3fs;%.3f;%.3f' % (status, timestamp, warning, critical)

	return exitcode

if __name__ == "__main__":
	sys.exit(main())