#!/usr/bin/python3

from __future__ import absolute_import, print_function, unicode_literals

'''
Heart Rate Monitor test script
'''

from optparse import OptionParser, make_option
import sys
import dbus
import dbus.service
import dbus.mainloop.glib
try:
  from gi.repository import GObject
except ImportError:
  import gobject as GObject
import bluezutils

BUS_NAME = 'org.bluez'
HEARTRATE_MANAGER_INTERFACE = 'org.bluez.HeartRateManager1'
HEARTRATE_WATCHER_INTERFACE = 'org.bluez.HeartRateWatcher1'
HEARTRATE_INTERFACE = 'org.bluez.HeartRate1'

class Watcher(dbus.service.Object):
	@dbus.service.method(HEARTRATE_WATCHER_INTERFACE,
					in_signature="oa{sv}", out_signature="")
	def MeasurementReceived(self, device, measure):
		print("Measurement received from %s" % device)
		print("Value: ", measure["Value"])

		if "Energy" in measure:
			print("Energy: ", measure["Energy"])

		if "Contact" in measure:
			print("Contact: ", measure["Contact"])

		if "Interval" in measure:
			for i in measure["Interval"]:
				print("Interval: ", i)

if __name__ == "__main__":
	dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)

	bus = dbus.SystemBus()

	option_list = [
		make_option("-i", "--adapter", action="store",
			type="string", dest="adapter"),
		make_option("-b", "--device", action="store",
			type="string", dest="address"),
		]

	parser = OptionParser(option_list=option_list)

	(options, args) = parser.parse_args()

	if not options.address:
		print("Usage: %s [-i <adapter>] -b <bdaddr> [cmd]" % (sys.argv[0]))
		print("Possible commands:")
		print("\tReset")
		sys.exit(1)

	managed_objects = bluezutils.get_managed_objects()
	adapter = bluezutils.find_adapter_in_objects(managed_objects,
								options.adapter)
	adapter_path = adapter.object_path

	heartrateManager = dbus.Interface(bus.get_object(BUS_NAME,
				adapter_path), HEARTRATE_MANAGER_INTERFACE)

	path = "/test/watcher"
	heartrateManager.RegisterWatcher(path)

	device = bluezutils.find_device_in_objects(managed_objects,
								options.address,
								options.adapter)
	device_path = device.object_path

	heartrate = dbus.Interface(bus.get_object(BUS_NAME, device_path),
							HEARTRATE_INTERFACE)

	watcher = Watcher(bus, path)

	dev_prop = dbus.Interface(bus.get_object(BUS_NAME, device_path),
					"org.freedesktop.DBus.Properties")

	properties = dev_prop.GetAll(HEARTRATE_INTERFACE)

	if "Location" in properties:
		print("Sensor location: %s" % properties["Location"])
	else:
		print("Sensor location is not supported")

	if len(args) > 0:
		if args[0] == "Reset":
			reset_sup = properties["ResetSupported"]
			if reset_sup:
				heartrate.Reset()
			else:
				print("Reset not supported")
				sys.exit(1)
		else:
			print("unknown command")
			sys.exit(1)

	mainloop = GObject.MainLoop()
	mainloop.run()
