#!/usr/bin/env python
# -*- coding: ISO-8859-1 -*-

##################################
#  @program        smon
#  @description    simulation monitor
#  @copyright      Copyright “(c)2009 Centre National de la Recherche Scientifique CNRS. 
#                             All Rights Reserved”
#  @svn_file       $Id: watcher 2545 2013-02-01 09:58:10Z jripsl $
#  @version        $Rev: 2545 $
#  @lastrevision   $Date: 2013-02-01 10:58:10 +0100 (Fri, 01 Feb 2013) $
#  @license        CeCILL (http://dods.ipsl.jussieu.fr/jripsl/smon/LICENSE)
##################################

#from smon import dao
import pika
import base64
import json
import sys
import traceback
import smtplib
from email.mime.text import MIMEText

class Watcher():
	message_code_action_mapping = {"0000":["log","mail"],"1000":["log"],"2000":["log"],"3000":["log"],"9000":["log"],"9999":["log","mail"]}

	@classmethod
	def get_fake_progress_messages(cls):
		pass

	@classmethod
	def start(cls):
		#dao.insert_progress_messages(cls.get_fake_progress_messages())
		pass

	@classmethod
	def stop(cls):
		pass

	@classmethod
	def add(cls,message):
		pass

	@classmethod
	def mail_example(cls):
		me="jripsl@ipsl.jussieu.fr"
		you="jripsl@ipsl.jussieu.fr"
		body="Alarm"
		object="Supervisor"

		cls.mail(me,you,object,body)

	@classmethod
	def mail(cls):
		cls.mail_example()

	@classmethod
	def send_mail(cls,me,you,object,body):
		msg = MIMEText(body)
		msg['Subject'] = object
		msg['From'] = me
		msg['To'] = you

		# Send the message via our own SMTP server, but don't include the # envelope header.
		s = smtplib.SMTP('localhost')
		s.sendmail(me,[you], msg.as_string())
		s.quit()

	@classmethod
	def log(cls,message):

		with open("/home/jripsl/supervisor/log/supervisor.log", "a") as log_file:
			log_file.write("%s %s\n"%(message["code"],message["jobid"]))

	@classmethod
	def execActions(cls,message):

		message_code=message["code"]

		for action in cls.message_code_action_mapping[message_code]:
			proc_name=action

			try:
				getattr(cls, proc_name)(message)
			except Exception,e:
				traceback.print_exc()

				raise Exception("WATCH-ERR002","procedure error (%s,%s)"%(proc_name,str(e)))

def main():

	"""
	# parse args
	parser = argparse.ArgumentParser(prog='watcher')
	parser.add_argument('-v', dest='verbose',required=False,action='store_true')
	args = parser.parse_args()

	# check
	if not os.path.exists(SMON.smon_home):
		sys.exit(1)

	SMON.init_singleton()
	"""

	connection = pika.BlockingConnection(pika.ConnectionParameters( host='localhost'))
	channel = connection.channel()

	#channel.queue_declare(queue='myqueue')

	print ' [*] Waiting for messages. To exit press CTRL+C'

	def callback(ch, method, properties, raw_msg):
		#
		#return

		base64_decoded_msg=base64.b64decode(raw_msg)

		# debug
		#print " [x] Received %s" % raw_msg
		#print " [x] Received %s (uudecoded)" % base64_decoded_msg 

		try:
			message=json.loads(base64_decoded_msg)
		except Exception,e:
			print base64_decoded_msg
		
		# message code based action
		Watcher.execActions(message)

	channel.basic_consume(callback, queue='myqueue', no_ack=True)

	channel.start_consuming()


	"""
	SMON.free_singleton()
	"""

if __name__ == '__main__':
	try:
		main()

		sys.exit(0)

	except Exception, e:

		traceback.print_exc()

		sys.exit(1)