from py9b.link.base import LinkOpenException, LinkTimeoutException
from py9b.link.tcp import TCPLink
from py9b.link.ble import BLELink
from py9b.transport.base import BaseTransport as BT
from py9b.transport.packet import BasePacket as PKT
from py9b.transport.xiaomi import XiaomiTransport

READ_CHUNK_SIZE = 0x40

#link = SerialLink()
#link = TCPLink()
link = BLELink()

with link:
	print "Scanning..."
	ports = link.scan()
	print ports

	tran = XiaomiTransport(link)

	#link.open(("192.168.1.45", 6000))
	link.open(ports[0][1])
	print "Connected"

	req = PKT(src=BT.HOST, dst=BT.ESC, cmd=0x01, reg=0, data=chr(READ_CHUNK_SIZE))

	hfo = open("EscRegs.bin", "wb")
	for i in xrange(0, 0x200, READ_CHUNK_SIZE):
		print ".",
		req.reg = i>>1
		for retry in xrange(3):
			tran.send(req)
			try:
				rsp = tran.recv()
			except LinkTimeoutException:
				continue
			break
		hfo.write(rsp.data)

	hfo.close()
	link.close()