#!/usr/bin/env python2

import zmq, msgpack, time, yaml, psutil, fnmatch

ctx = zmq.Context()

sock = ctx.socket(zmq.PUSH)
sock.connect("ipc:///tmp/cstatsd")

with open("config/processes.yaml", "r") as cfile:
	config = yaml.safe_load(cfile)

interval = config["interval"]

old_status = {}

while True:
	all_procs = psutil.get_process_list()
	
	for service_name, patterns in config["processes"].iteritems():
		matching = []
		for proc in all_procs:  # Can't use filter() because of exceptions...
			try:
				if len(proc.cmdline) > 0 and fnmatch.fnmatch(proc.cmdline[0], patterns["name"]):
					failed = False
					try:
						for arg, pattern in patterns["args"].iteritems():
							try:
								if len(proc.cmdline) < (arg + 1) or not fnmatch.fnmatch(proc.cmdline[arg], pattern):
									failed = True
							except KeyError, e:
								pass
					except KeyError, e:
						pass
					if failed == False:
						matching.append(proc)
			except psutil._error.NoSuchProcess, e:
				pass
			
		if len(matching) > 0:
			up = True
		else:
			up = False
			
		try:
			if up == old_status[service_name]:
				send_notice = False
			else:
				send_notice = True
				initial = False
		except KeyError, e:
			send_notice = True
			initial = True
			
		old_status[service_name] = up
			
		if send_notice:
			if up:
				msg_type = "up"
			else:
				msg_type = "down"
				
			sock.send(msgpack.packb({
				"service": "process",
				"msg_type": msg_type,
				"unit": service_name,
				"initial": initial
			}))
			
			
	time.sleep(interval)