#!/usr/bin/env python3
import subprocess
import time
import eq3crypto
import os
import config
import sys

pstart = time.time()
writeHandle = 0x411
readHandle  = 0x421

def tsLog(m = ""):
    global pstart
    print("{:.3f}".format(time.time() - pstart) + " " + m)

def getBuffer(p):
    val = p.stdout.peek()
    val = p.stdout.read(len(val))
    #print(val)
    return val

def expect(p, s, t=0):
    b = ""
    while True:
        b = b + getBuffer(p).decode()
        if s in b:
            return b

def getNotif(p):
    expect(p, "Notification")


def sendCmd(p, s):
    print("<< " + s)
    p.stdin.write(s.encode())
    p.stdin.write(b"\n")
    p.stdin.flush()

def getNotifVal(s):
    r = s.split("value: ")[-1].split("\n")[0]
    print(">> " + r)
    return r

actionStr = os.getenv("HOME")
if "door" in actionStr:
   print("Door user")
   actionStr = sys.argv[-1]
action = 0
tsLog("Start")
if "open" in actionStr:
    action = 2
if "unlock" in actionStr:
    action = 1
try:
    p = subprocess.Popen(["gatttool", "-I"], stdout=subprocess.PIPE, stdin=subprocess.PIPE)
    expect(p, "[LE]")
    tsLog("INIT DONE")
    sendCmd(p, "connect " + config.device_mac)
    p.stdin.flush()
    expect(p, "Connection successful")
    tsLog("CONNECT DONE")
    sendCmd(p, "char-write-req 0x0411 8002014DC7e11BAE13FECB0000000000")
    notifNonce = bytes.fromhex(getNotifVal(expect(p, "Notification")))
    if notifNonce[0:2] == b'\x80\x03':
        tsLog("Nonce received")
        remoteNonce = notifNonce[3:(3+8)]
        tp = time.strftime("%y %m %d %H %M %S").split()
        msg = eq3crypto.getMsg(0x82, bytes([int(tp[0]),int(tp[1]),int(tp[2]),int(tp[3]),int(tp[4]),int(tp[5]),0,0]), remoteNonce, bytes.fromhex(config.user_key))
        sendCmd(p, "char-write-req 0x0411 " + msg.hex())
        #getNotifVal(expect(p, "Notification"))
        #tsLog("MOVING")
        getNotifVal(expect(p, "Notification"))
        tsLog("DONE")
finally:
    print("FINALLY")
    try:
        p.communicate(input=b"exit\n",timeout=2)
        print("NORMAL EXIT")
    except:
        print("KILL")
        p.kill()