#!/usr/bin/env python3
# coding: utf-8
# vim: noet ts=2 sw=2 sts=2

import os
import collectd

from sem6000 import SEMSocket
import bluepy

instances = []

def init_func():
	pass

def config_func(cfg):
	global instances

	config = {}

	for node in cfg.children:
		key = node.key.lower()
		value = node.values[0]

		if key in ['address', 'socketname']:
			config[key] = value

	if 'address' not in config.keys():
		collectd.error('sem6000: address must be set')
		return

	if 'socketname' not in config.keys():
		config['socketname'] = config['address'].replace(':', '')

	instances.append( {'config': config, 'socket': None} )

def read_func():
	global instances

	for inst in instances:
		config = inst['config']

		try:
			if inst['socket'] == None:
				collectd.info("sem6000: Connecting to {}...".format(config['address']))

				inst['socket'] = SEMSocket(config['address'])
				collectd.info("sem6000: Connected.")

			inst['socket'].getStatus()
		except (SEMSocket.NotConnectedException, bluepy.btle.BTLEDisconnectError, BrokenPipeError) as e:
			collectd.warning("sem6000: Exception caught: {}".format(e))
			collectd.warning("sem6000: Restarting on next cycle...")

			if inst['socket'] != None:
				inst['socket'].disconnect()
				inst['socket'] = None

		socket = inst['socket']

		if socket != None and socket.voltage != 0:
			collectd.debug("Uploading values for {}".format(socket.mac_address))

			val = collectd.Values(plugin = 'sem6000-{}'.format(config['socketname']))

			val.type = 'voltage'
			val.type_instance = 'grid'
			val.values = [ socket.voltage ]
			val.dispatch()

			val.type = 'current'
			val.type_instance = 'load'
			val.values = [ socket.current ]
			val.dispatch()

			val.type = 'power'
			val.type_instance = 'real_power'
			val.values = [ socket.power ]
			val.dispatch()

			val.type = 'gauge'
			val.type_instance = 'power_factor'
			val.values = [ socket.power_factor ]
			val.dispatch()

			val.type = 'gauge'
			val.type_instance = 'load_on'
			val.values = [ socket.powered ]
			val.dispatch()

			val.type = 'frequency'
			val.type_instance = 'grid'
			val.values = [ socket.frequency ]
			val.dispatch()

def shutdown_func():
	global instances

	for inst in instances:
		if inst['socket'] != None:
			inst['socket'].disconnect()

	instances = []

collectd.register_config(config_func)
collectd.register_init(init_func)
collectd.register_read(read_func)
collectd.register_shutdown(shutdown_func)