#!/usr/bin/env python3
import sys
from datetime import datetime
import bluetooth._bluetooth as bluez

from bluetooth_utils import (toggle_device, enable_le_scan,
                            parse_le_advertising_events,
                            disable_le_scan, raw_packet_to_str)

# Use 0 for hci0
dev_id = 0
toggle_device(dev_id, True)

try:
   sock = bluez.hci_open_dev(dev_id)
except:
   print("Cannot open bluetooth device %i" % dev_id)
   raise

# Set filter to "True" to see only one packet per device
enable_le_scan(sock, filter_duplicates=False)

try:
   def le_advertise_packet_handler(mac, adv_type, data, rssi):
       data_str = raw_packet_to_str(data)
       # Check for ATC preamble
       if data_str[6:10] == '1a18':
           temp = int(data_str[22:26], 16) / 10
           hum = int(data_str[26:28], 16)
           batt = int(data_str[28:30], 16)
           print("%s - Device: %s Temp: %sc Humidity: %s%% Batt: %s%%" % \
                (datetime.now().strftime("%Y-%m-%d %H:%M:%S"), mac, temp, hum, batt))

   # Called on new LE packet
   parse_le_advertising_events(sock,
                               handler=le_advertise_packet_handler,
                               debug=False)
# Scan until Ctrl-C
except KeyboardInterrupt:
   disable_le_scan(sock)

 ' ESP32 BLE logger
 ' TassyJim April 2021
 OPTION EXPLICIT
 OPTION DEFAULT INTEGER
 ' Global variables
 CONST qt$=CHR$(34)
 CONST crlf$=CHR$(13)+CHR$(10)
 '
 dim string comPort$ = "com1:921600,1024" '"com1:115200,1024"
 dim float temperature, humidity, batpc, batV
 dim inTxt$, ink$
 dim integer quit, CMM2, x, Verbose = 0, showDebug
 
 OPEN comPort$ AS #1
 PAUSE 2000 'wait for ESP32 to wake up afer power on
 CLOSE #1 'clear the buffer
 OPEN comPort$ AS #1
 
 IF instr(MM.DEVICE$,"Colour Maximite 2") THEN
   CMM2 = 1
 ELSE
   CMM2 = 0
 ENDIF
 
 ESPcommand("AT") ' test ESP for responce
 IF OK()THEN
   PRINT "ESP alive!"
 ELSE
   PAUSE 200
   ESPcommand("AT")
   IF OK()THEN
     PRINT "ESP alive!"
   ELSE
     PRINT inTxt$
   ENDIF
 ENDIF
 
 pause 500
 getinput
 ESPcommand("AT+CWMODE=0")
 if not OK() then print "Error 1"
 ESPcommand("AT+BLEINIT=1")
 if not OK() then print "Error 2"
 ESPcommand("AT+BLESCANPARAM?")
 if not OK() then print "Error 3"
 ESPcommand("AT+BLESCAN=1,0")
 
 DO
   ink$ = INKEY$
   SELECT CASE ink$
     CASE CHR$(27),"Q","q"
       quit = 1
     CASE "D","d"
       showDebug = 1 - showDebug
       IF showDebug THEN PRINT "Debug on" ELSE PRINT "Debug off"
     CASE "V","v"
       verbose = 1 - verbose
       
   END SELECT
   '
   x = waitFor(30000,250,crlf$)
   if x = 1 then ' we have a crlf pair
     decode inTxt$
   endif
 LOOP UNTIL quit = 1
 
 PRINT "Shutting down..."
 ESPcommand "AT+BLESCAN=0"
 if not OK() then print "Error shutting down"
END
 '
 
 'send ESP related command
SUB ESPcommand(txt$)
 getInput
 PRINT #1,txt$+crlf$;
 IF showDebug THEN PRINT txt$ 'DEBUG
END SUB
 
sub getInput
 IF LOC(#1) > 0 THEN
   do
     inTxt$=INPUT$(255,#1)
     if verbose then print inTxt$
   loop until LOC(#1)=0
 ENDIF
 
END sub
 
 ' wait timeout% mS or 250 bytes for "OK"
FUNCTION OK(timeout%)
 IF timeout% = 0 THEN timeout% = 500
 IF waitFor(timeout%,250,"OK") = 1 THEN OK = 1
END FUNCTION
 
 ' wait for text string(s). tested in supplied order and stops after first blank text
 ' returns index to found txt or 0 if timeout
 ' if maxlen is non zero, function will return 255 if max len is reached
 ' and global variable inTxt$ will contain the received text.
FUNCTION waitFor(timeout%,maxlen%,a1$,a2$,a3$,a4$,a5$,a6$) AS INTEGER
 LOCAL INTEGER n, c, endTime = TIMER + timeout%
 LOCAL a$
 DO
   IF LOC(#1) THEN
     a$ = a$ + INPUT$(1,#1)
     INC c
     IF c >= maxlen% THEN n = 255 : EXIT DO
     IF RIGHT$(a$,LEN(a1$)) = a1$ THEN n = 1 : EXIT DO
     IF a2$="" THEN CONTINUE DO
     IF RIGHT$(a$,LEN(a2$)) = a2$ THEN n = 2 : EXIT DO
     IF a3$="" THEN CONTINUE DO
     IF RIGHT$(a$,LEN(a3$)) = a3$ THEN n = 3 : EXIT DO
     IF a4$="" THEN CONTINUE DO
     IF RIGHT$(a$,LEN(a4$)) = a4$ THEN n = 4 : EXIT DO
     IF a5$="" THEN CONTINUE DO
     IF RIGHT$(a$,LEN(a5$)) = a5$ THEN n = 5 : EXIT DO
     IF a6$="" THEN CONTINUE DO
     IF RIGHT$(a$,LEN(a6$)) = a6$ THEN n = 6 : EXIT DO
   ENDIF
 LOOP UNTIL TIMER > endtime OR c > 254
 IF maxlen% THEN inTxt$ = a$
 waitFor = n
END FUNCTION
 
sub decode txt$
 local integer startTxt
 local adv_data$, fld$
 startTxt = instr(txt$,"BLESCAN")
 if startTxt then
   txt$ = mid$(txt$,starttxt+8)
   adv_data$ = field$(txt$,3,",")
   if verbose then print adv_data$
   fld$ = mid$(adv_data$,5,4)
   if fld$ = "1a18" then ' we have the correct data
     fld$ = mid$(adv_data$,9,12)
     print "MAC address: ";fld$
     fld$ = mid$(adv_data$,21,4) ': print fld$
     temperature = signed16(val("&h"+fld$))
     temperature = temperature/10
     fld$ = mid$(adv_data$,25,2)': print fld$
     humidity =  val("&h"+fld$)
     fld$ = mid$(adv_data$,27,2)': print fld$
     batpc =  val("&h"+fld$)
     fld$ = mid$(adv_data$,29,4)': print fld$
     batV =  val("&h"+fld$)/1000
     
     print Time$;"  ";temperature;"`C ";Humidity;"% ";batpc;"% ";batV;"V "
     print
   endif
 endif
end sub
 
function signed16(x)
 if x > 32767 then
   signed16 = x - 65536
 else
   signed16 = x
 endif
end function

 ' ESP32 BLE logger
 ' TassyJim April 2021
 OPTION EXPLICIT
 OPTION DEFAULT INTEGER
 ' Global variables
 CONST qt$=CHR$(34)
 CONST crlf$=CHR$(13)+CHR$(10)
 '
 DIM STRING comPort$ = "com1:921600,2048" '"com1:115200,2048"
 DIM FLOAT temperature, humidity, batpc, batV, counter
 DIM inTxt$, ink$, j$, MACadr$
 DIM INTEGER quit, Verbose = 1
 
 OPEN comPort$ AS #1
 PAUSE 2000 'wait for ESP32 to wake up afer power on
 CLOSE #1 'clear the buffer
 OPEN comPort$ AS #1

 ESPcommand("AT") ' test ESP for responce
 IF OK()THEN
   PRINT "ESP alive!"
 ELSE
   PAUSE 200
   ESPcommand("AT")
   IF OK()THEN
     PRINT "ESP alive!!"
   ELSE
     PRINT inTxt$
   ENDIF
 ENDIF
 
 DO ' flush the comms buffer
   j$ = INPUT$(255,#1)
   IF verbose THEN PRINT j$
 LOOP UNTIL j$ = ""
 
 ESPcommand("AT+CWMODE=0")
 IF NOT OK() THEN PRINT "Error 1"
 ESPcommand("AT+BLEINIT=1")
 IF NOT OK() THEN PRINT "Error 2"
 ESPcommand("AT+BLESCANPARAM?")
 IF NOT OK() THEN PRINT "Error 3"
 IF verbose THEN PRINT inTxt$
 ESPcommand("AT+BLESCAN=1,0")
 IF verbose THEN PRINT "Scanning..."
 
 DO
   ink$ = INKEY$
   SELECT CASE ink$
     CASE CHR$(27),"Q","q"
       quit = 1
     CASE "V","v"
       verbose = 1 - verbose
       
   END SELECT
   IF LOC(#1) THEN getInput
   '
 LOOP UNTIL quit = 1
 
 PRINT "Shutting down..."
 ESPcommand "AT+BLESCAN=0"
 IF NOT OK() THEN PRINT "Error shutting down"
 CLOSE #1
END
 '
 
 'send ESP related command
SUB ESPcommand(txt$)
 PRINT #1,txt$+crlf$;
 IF verbose THEN PRINT txt$
END SUB
 
SUB getInput
 LOCAL INTEGER x
   x = waitfor(500,crlf$)
   IF verbose THEN PRINT inTxt$;
   IF x = 1 THEN
     decode inTxt$
   ENDIF
END SUB
 
 ' wait timeout% mS or 250 bytes for "OK"
FUNCTION OK(timeout%)
 IF timeout% = 0 THEN timeout% = 500
 IF waitFor(timeout%,"OK"+crlf$) = 1 THEN
 OK = 1
 IF verbose THEN PRINT "OK"
 ENDIF
END FUNCTION
 
 ' wait for text string.
 ' returns 1 for end match, 255 if 250 characters reached  or 0 if timeout
 ' global variable inTxt$ will contain the received text.
FUNCTION waitFor(timeout%,a1$) AS INTEGER
 LOCAL INTEGER n, c, endTime = TIMER + timeout%
 LOCAL a$
 DO
   IF LOC(#1) THEN
     a$ = a$ + INPUT$(1,#1)
     'if verbose then print right$(a$,1);
     c = c + 1
     IF c >= 250 THEN n = 255 : EXIT DO
     IF RIGHT$(a$,LEN(a1$)) = a1$ THEN n = 1 : EXIT DO
   ENDIF
 LOOP UNTIL TIMER > endtime
 IF n THEN inTxt$ = a$
 waitFor = n
END FUNCTION
 
SUB decode txt$
 LOCAL INTEGER startTxt
 LOCAL adv_data$, fld$
 startTxt = INSTR(txt$,"BLESCAN")
 IF startTxt THEN
   txt$ = MID$(txt$,starttxt+8)
   adv_data$ = FIELD$(txt$,3,",")
   'if verbose then print adv_data$
   fld$ = MID$(adv_data$,5,4)
   IF fld$ = "1a18" THEN ' we have the correct data
     MACadr$ = MID$(adv_data$,9,12)
     fld$ = MID$(adv_data$,21,4) ': print fld$
     temperature = signed16(VAL("&h"+fld$))
     temperature = temperature/10
     fld$ = MID$(adv_data$,25,2)': print fld$
     humidity =  VAL("&h"+fld$)
     fld$ = MID$(adv_data$,27,2)': print fld$
     batpc =  VAL("&h"+fld$)
     fld$ = MID$(adv_data$,29,4)': print fld$
     batV =  VAL("&h"+fld$)/1000
     fld$ = MID$(adv_data$,33,2)': print fld$
     counter =  VAL("&h"+fld$)
     PRINT TIME$;"  ";MACadr$;"  ";temperature;"`C ";Humidity;"%rh ";batpc;"% ";batV;"V ";counter
     PRINT
   ENDIF
 ENDIF
END SUB
 
FUNCTION signed16(x%) AS INTEGER
 IF x% > 32767 THEN
   signed16 = x% - 65536
 ELSE
   signed16 = x%
 ENDIF
END FUNCTION