BangleApps/apps/bthrm/boot.js

(function() {
  var settings = Object.assign(
    require('Storage').readJSON("bthrm.default.json", true) || {},
    require('Storage').readJSON("bthrm.json", true) || {}
  );
  
  var log = function(text, param){
    if (settings.debuglog){
      var logline = new Date().toISOString() + " - " + text;
      if (param){
        logline += " " + JSON.stringify(param);
      }
      print(logline);
    }
  };
  
  log("Settings: ", settings);
  
  if (settings.enabled){

    function clearCache(){
      return require('Storage').erase("bthrm.cache.json");
    }

    function getCache(){
      return require('Storage').readJSON("bthrm.cache.json", true) || {};
    }
    
    function addNotificationHandler(characteristic){
      log("Setting notification handler: " + supportedCharacteristics[characteristic.uuid].handler);
      characteristic.on('characteristicvaluechanged', supportedCharacteristics[characteristic.uuid].handler);
    }
    
    function writeCache(cache){
      var oldCache = getCache();
      if (oldCache != cache) {
        log("Writing cache");
        require('Storage').writeJSON("bthrm.cache.json", cache)
      } else {
        log("No changes, don't write cache");
      }
      
    }

    function characteristicsToCache(characteristics){
      log("Cache characteristics");
      var cache = getCache();
      if (!cache.characteristics) cache.characteristics = {};
      for (var c of characteristics){
        //"handle_value":16,"handle_decl":15
        log("Saving handle " + c.handle_value + " for characteristic: ", c);
        cache.characteristics[c.uuid] = {
          "handle": c.handle_value,
          "uuid": c.uuid,
          "notify": c.properties.notify,
          "read": c.properties.read
        };
      }
      writeCache(cache);
    }

    function characteristicsFromCache(){
      log("Read cached characteristics");
      var cache = getCache();
      if (!cache.characteristics) return [];
      var restored = [];
      for (var c in cache.characteristics){
        var cached = cache.characteristics[c];
        var r = new BluetoothRemoteGATTCharacteristic();
        log("Restoring characteristic ", cached);
        r.handle_value = cached.handle;
        r.uuid = cached.uuid;
        r.properties = {};
        r.properties.notify = cached.notify;
        r.properties.read = cached.read;
        addNotificationHandler(r);
        log("Restored characteristic: ", r);
        restored.push(r);
      }
      return restored;
    }

    log("Start");

    var lastReceivedData={
    };

    var serviceFilters = [{
      services: [ "180d" ]
    }];

    supportedServices = [
      "0x180d", "0x180f"
    ];

    var supportedCharacteristics = {
      "0x2a37": {
        //Heart rate measurement
        handler: function (event){          
          var dv = event.target.value;
          var flags = dv.getUint8(0);
          
          var bpm = (flags & 1) ? (dv.getUint16(1) / 100 /* ? */ ) : dv.getUint8(1); // 8 or 16 bit
          
          var sensorContact;
          
          if (flags & 2){
            sensorContact = (flags & 4) ? true : false;
          }
          
          var idx = 2 + (flags&1);
          
          var energyExpended;
          if (flags & 8){
            energyExpended = dv.getUint16(idx,1);
            idx += 2;
          }
          var interval;
          if (flags & 16) {
            interval = [];
            maxIntervalBytes = (dv.byteLength - idx);
            log("Found " + (maxIntervalBytes / 2) + " rr data fields");
            for(var i = 0 ; i < maxIntervalBytes / 2; i++){
              interval[i] = dv.getUint16(idx,1); // in milliseconds
              idx += 2
            }
          }

          var location;
          if (lastReceivedData && lastReceivedData["0x180d"] && lastReceivedData["0x180d"]["0x2a38"]){
            location = lastReceivedData["0x180d"]["0x2a38"];
          }

          var battery;
          if (lastReceivedData && lastReceivedData["0x180f"] && lastReceivedData["0x180f"]["0x2a19"]){
            battery = lastReceivedData["0x180f"]["0x2a19"];
          }

          if (settings.replace){
            var newEvent = {
              bpm: bpm,
              confidence: (sensorContact || sensorContact === undefined)? 100 : 0,
              src: "bthrm"
            };
          
            log("Emitting HRM: ", newEvent);
            Bangle.emit("HRM", newEvent);
          }

          var newEvent = {
            bpm: bpm
          };
          
          if (location) newEvent.location = location;
          if (interval) newEvent.rr = interval;
          if (energyExpended) newEvent.energy = energyExpended;
          if (battery) newEvent.battery = battery;
          if (sensorContact) newEvent.contact = sensorContact;
          
          log("Emitting BTHRM: ", newEvent);
          Bangle.emit("BTHRM", newEvent);
        }
      },
      "0x2a38": {
        //Body sensor location
        handler: function(data){
          if (!lastReceivedData["0x180d"]) lastReceivedData["0x180d"] = {};
          if (!lastReceivedData["0x180d"]["0x2a38"]) lastReceivedData["0x180d"]["0x2a38"] = data.target.value;
        }
      },
      "0x2a19": {
        //Battery
        handler: function (event){
          if (!lastReceivedData["0x180f"]) lastReceivedData["0x180f"] = {};
          if (!lastReceivedData["0x180f"]["0x2a19"]) lastReceivedData["0x180f"]["0x2a19"] = event.target.value.getUint8(0);
        }
      }

    };

    var device;
    var gatt;
    var characteristics = [];
    var blockInit = false;
    var currentRetryTimeout;
    var initialRetryTime = 40;
    var maxRetryTime = 60000;
    var retryTime = initialRetryTime;

    var connectSettings = {
      minInterval: 7.5,
      maxInterval: 1500
    };

    function waitingPromise(timeout) {
      return new Promise(function(resolve){
        log("Start waiting for " + timeout);
        setTimeout(()=>{
          log("Done waiting for " + timeout);
          resolve();
        }, timeout);
      });
    }

    if (settings.enabled){
      Bangle.isBTHRMOn = function(){
        return (Bangle._PWR && Bangle._PWR.BTHRM && Bangle._PWR.BTHRM.length > 0);
      };

      Bangle.isBTHRMConnected = function(){
        return gatt && gatt.connected;
      };
    }


    if (settings.replace){
      var origIsHRMOn = Bangle.isHRMOn;

      Bangle.isHRMOn = function() {
        if (settings.enabled && !settings.replace){
            return origIsHRMOn();
        } else if (settings.enabled && settings.replace){
            return Bangle.isBTHRMOn();
        }
        return origIsHRMOn() || Bangle.isBTHRMOn();
      };
    }

    function clearRetryTimeout(){
      if (currentRetryTimeout){
        log("Clearing timeout " + currentRetryTimeout);
        clearTimeout(currentRetryTimeout);
        currentRetryTimeout = undefined;
      }
    }

    function retry(){
      log("Retry");

      if (!currentRetryTimeout){

        var clampedTime = retryTime < 100 ? 100 : retryTime;
        
        log("Set timeout for retry as " + clampedTime);      
        clearRetryTimeout();
        currentRetryTimeout = setTimeout(() => {
          log("Retrying");
          currentRetryTimeout = undefined;
          initBt();
        }, clampedTime);

        retryTime = Math.pow(retryTime, 1.1);
        if (retryTime > maxRetryTime){
          retryTime = maxRetryTime;
        }
      } else {
        log("Already in retry...");
      }
    }

    var buzzing = false;
    function onDisconnect(reason) {
      log("Disconnect: " + reason);
      log("GATT: ", gatt);
      log("Characteristics: ", characteristics);
      retryTime = initialRetryTime;
      clearRetryTimeout();
      switchInternalHrm();
      blockInit = false;
      if (settings.warnDisconnect && !buzzing){
        buzzing = true;
        Bangle.buzz(500,0.3).then(()=>waitingPromise(4500)).then(()=>{buzzing = false;});
      }
      if (Bangle.isBTHRMOn()){
        retry();
      }
    }

    function createCharacteristicPromise(newCharacteristic){
      log("Create characteristic promise: ", newCharacteristic);
      var result = Promise.resolve();
      if (newCharacteristic.properties.notify){
        result = result.then(()=>{
          log("Starting notifications for: ", newCharacteristic);
          var startPromise = newCharacteristic.startNotifications().then(()=>log("Notifications started for ", newCharacteristic));
          if (settings.gracePeriodNotification > 0){
            log("Add " + settings.gracePeriodNotification + "ms grace period after starting notifications");
            startPromise = startPromise.then(()=>{
              log("Wait after connect");
              waitingPromise(settings.gracePeriodNotification)
            });
          }
          return startPromise;
        });
      } else if (newCharacteristic.read){
        result = result.then(()=>{
          readData(newCharacteristic);
          log("Reading data for " + newCharacteristic);
          return newCharacteristic.read().then((data)=>{
            supportedCharacteristics[newCharacteristic.uuid].handler(data);
          });
        });
      }
      return result.then(()=>log("Handled characteristic: ", newCharacteristic));
    }
    
    function attachCharacteristicPromise(promise, characteristic){
      return promise.then(()=>{
        log("Handling characteristic:", characteristic);
        return createCharacteristicPromise(characteristic);
      });
    }
    
    function createCharacteristicsPromise(newCharacteristics){
      log("Create characteristics promise: ", newCharacteristics);
      var result = Promise.resolve();
      for (var c of newCharacteristics){
        if (!supportedCharacteristics[c.uuid]) continue;
        log("Supporting characteristic: ", c);
        characteristics.push(c);
        if (c.properties.notify){
          addNotificationHandler(c);
        }
        
        result = attachCharacteristicPromise(result, c);
      }
      return result.then(()=>log("Handled characteristics"));
    }
    
    function createServicePromise(service){
      log("Create service promise: ", service);
      var result = Promise.resolve();
      result = result.then(()=>{
        log("Handling service: " + service.uuid);
        return service.getCharacteristics().then((c)=>createCharacteristicsPromise(c));
      });
      return result.then(()=>log("Handled service" + service.uuid));
    }
    
    function attachServicePromise(promise, service){
      return promise.then(()=>createServicePromise(service));
    }
    
    var reUseCounter = 0;

    function initBt() {
      log("initBt with blockInit: " + blockInit);
      if (blockInit){
        retry();
        return;
      }

      blockInit = true;

      if (reUseCounter > 10){
        log("Reuse counter to high");
        gatt=undefined;
        reUseCounter = 0;
      }
      
      var promise;
      
      if (!device){
        promise = NRF.requestDevice({ filters: serviceFilters });
        
        if (settings.gracePeriodRequest){
          log("Add " + settings.gracePeriodRequest + "ms grace period after request");
        }
          
        promise = promise.then((d)=>{
          log("Got device: ", d);
          d.on('gattserverdisconnected', onDisconnect);
          device = d;
        });
        
        promise = promise.then(()=>{
          log("Wait after request");
          return waitingPromise(settings.gracePeriodRequest);
        });
        
      } else {
        promise = Promise.resolve();
        log("Reuse device: ", device);
      }
      
      promise = promise.then(()=>{
        if (gatt){
          log("Reuse GATT: ", gatt);
        } else {
          log("GATT is new: ", gatt);
          characteristics = [];
          var cachedName = getCache().name;
          if (device.name != cachedName){
            log("Device name changed from " + cachedName + " to " + device.name + ", clearing cache");
            clearCache();
          }
          var newCache = getCache();
          newCache.name = device.name;
          writeCache(newCache);
          gatt = device.gatt;
        }
        
        return Promise.resolve(gatt);
      });
      
      promise = promise.then((gatt)=>{
        if (!gatt.connected){
          var connectPromise = gatt.connect(connectSettings);
          if (settings.gracePeriodConnect > 0){
            log("Add " + settings.gracePeriodConnect + "ms grace period after connecting");
            connectPromise = connectPromise.then(()=>{
              log("Wait after connect");
              return waitingPromise(settings.gracePeriodConnect);
            });
          }
          return connectPromise;
        } else {
          return Promise.resolve();
        }
      });
      
      promise = promise.then(()=>{
        if (!characteristics || characteristics.length == 0){
          characteristics = characteristicsFromCache();
        }
      });

      promise = promise.then(()=>{
        var characteristicsPromise = Promise.resolve();
        if (characteristics.length == 0){
          characteristicsPromise = characteristicsPromise.then(()=>{
            log("Getting services");
            return gatt.getPrimaryServices();
          });

          characteristicsPromise = characteristicsPromise.then((services)=>{
            log("Got services:", services);
            var result = Promise.resolve();
            for (var service of services){
              if (!(supportedServices.includes(service.uuid))) continue;
              log("Supporting service: ", service.uuid);
              result = attachServicePromise(result, service);
            }
            if (settings.gracePeriodService > 0) {
              log("Add " + settings.gracePeriodService + "ms grace period after services");
              result = result.then(()=>{
                log("Wait after services");
                return waitingPromise(settings.gracePeriodService)
              });
            }
            return result;
          });
          
        } else {
          for (var characteristic of characteristics){
            characteristicsPromise = attachCharacteristicPromise(characteristicsPromise, characteristic, true);
          }
        }
        
        return characteristicsPromise;
      });
      
      promise = promise.then(()=>{
        log("Connection established, waiting for notifications");
        reUseCounter = 0;
        characteristicsToCache(characteristics);
        clearRetryTimeout();
      }).catch((e) => {
        characteristics = [];
        log("Error:", e);
        onDisconnect(e);
      });
    }

    Bangle.setBTHRMPower = function(isOn, app) {
      // Do app power handling
      if (!app) app="?";
      if (Bangle._PWR===undefined) Bangle._PWR={};
      if (Bangle._PWR.BTHRM===undefined) Bangle._PWR.BTHRM=[];
      if (isOn && !Bangle._PWR.BTHRM.includes(app)) Bangle._PWR.BTHRM.push(app);
      if (!isOn && Bangle._PWR.BTHRM.includes(app)) Bangle._PWR.BTHRM = Bangle._PWR.BTHRM.filter(a=>a!=app);
      isOn = Bangle._PWR.BTHRM.length;
      // so now we know if we're really on
      if (isOn) {
        if (!Bangle.isBTHRMConnected()) initBt();
      } else { // not on
        log("Power off for " + app);
        if (gatt) {
          if (gatt.connected){
            log("Disconnect with gatt: ", gatt);
            gatt.disconnect().then(()=>{
              log("Successful disconnect");
            }).catch((e)=>{
              log("Error during disconnect", e);
            });
          }
        }
      }
    };
    
    var origSetHRMPower = Bangle.setHRMPower;

    if (settings.startWithHrm){

      Bangle.setHRMPower = function(isOn, app) {
        log("setHRMPower for " + app + ": " + (isOn?"on":"off"));
        if (settings.enabled){
          Bangle.setBTHRMPower(isOn, app);
        }
        if ((settings.enabled && !settings.replace) || !settings.enabled){
          origSetHRMPower(isOn, app);
        }
      };
    }
    
    
    var fallbackInterval;
    
    function switchInternalHrm(){
      if (settings.allowFallback && !fallbackInterval){
        log("Fallback to HRM enabled");
        origSetHRMPower(1, "bthrm_fallback");
        fallbackInterval = setInterval(()=>{
          if (Bangle.isBTHRMConnected()){
            origSetHRMPower(0, "bthrm_fallback");
            clearInterval(fallbackInterval);
            fallbackInterval = undefined;
            log("Fallback to HRM disabled");
          }
        }, settings.fallbackTimeout);
      }
    }

    if (settings.replace){
      log("Replace HRM event");
      if (Bangle._PWR && Bangle._PWR.HRM){
        for (var i = 0; i < Bangle._PWR.HRM.length; i++){
          var app = Bangle._PWR.HRM[i];
          log("Moving app " + app);
          origSetHRMPower(0, app);
          Bangle.setBTHRMPower(1, app);
          if (Bangle._PWR.HRM===undefined) break;
        }
      }
      switchInternalHrm();
    }
    
    E.on("kill", ()=>{
      if (gatt && gatt.connected){
        log("Got killed, trying to disconnect");
        var promise = gatt.disconnect().then(()=>log("Disconnected on kill")).catch((e)=>log("Error during disconnnect on kill", e));
      }
    });
  }
})();