ble_nus.js

'use strict';

const bleNusServiceUUID  = '6e400001-b5a3-f393-e0a9-e50e24dcca9e';
const bleNusCharRXUUID   = '6e400002-b5a3-f393-e0a9-e50e24dcca9e';
const bleNusCharTXUUID   = '6e400003-b5a3-f393-e0a9-e50e24dcca9e';
const MTU = 20;


var bleDevice = {};
var nusService = {};
var rxCharacteristic = {};
var txCharacteristic = {};
var connected = {};
var bleBusy = false;

for (const cube of ["cube1", "cube2"]) {
    bleDevice[cube] = null;
    nusService[cube] = null;
    rxCharacteristic[cube] = null;
    txCharacteristic[cube] = null;
    connected[cube] = false;
}

function connectionToggle(cube) {
    if (connected[cube]) {
        disconnect(cube);
    } else {
        connect(cube);
    }
}

// Sets button to either Connect or Disconnect
function changeConnectionState(state, cube) {
    connected[cube] = state;
    if (state) {
        setButtonShadow(cube, cube + "_clientConnectButton", "Disconnect")
        openFullscreen(document.body);
    } else {
        setButtonShadow("", cube + "_clientConnectButton", "Connect")
        document.exitFullscreen();
    }
}

function connect(cube) {
    if (!checkBluetooth()) {
        terminal_writeln('WebBluetooth API is not available on your browser.\r\n' +
                         'Please make sure the Web Bluetooth flag is enabled.');
        return;
    }
    terminal_writeln('Requesting Bluetooth Device...');
    navigator.bluetooth.requestDevice({
        filters: [{services: [bleNusServiceUUID]}]
    })
    .then(device => {
        bleDevice[cube] = device; 
        terminal_writeln('Found ' + device.name);
        terminal_writeln('Connecting to GATT Server...');
        const onDisconnectedCube = function () {
            onDisconnected(cube);
        };
        bleDevice[cube].addEventListener('gattserverdisconnected', onDisconnectedCube);
        return device.gatt.connect();
    })
    .then(server => {
        terminal_writeln('Locate NUS service');
        return server.getPrimaryService(bleNusServiceUUID);
    }).then(service => {
        nusService[cube] = service;
        terminal_writeln('Found NUS service: ' + service.uuid);
    })
    .then(() => {
        terminal_writeln('Locate RX characteristic');
        return nusService[cube].getCharacteristic(bleNusCharRXUUID);
    })
    .then(characteristic => {
        rxCharacteristic[cube] = characteristic;
        terminal_writeln('Found RX characteristic');
    })
    .then(() => {
        terminal_writeln('Locate TX characteristic');
        return nusService[cube].getCharacteristic(bleNusCharTXUUID);
    })
    .then(characteristic => {
        txCharacteristic[cube] = characteristic;
        terminal_writeln('Found TX characteristic');
    })
    .then(() => {
        terminal_writeln('Enable notifications');
        return txCharacteristic[cube].startNotifications();
    })
    .then(() => {
        terminal_writeln('Notifications started');
        txCharacteristic[cube].addEventListener('characteristicvaluechanged',
                                          handleNotifications);
        changeConnectionState(true, cube);
        terminal_writeln('\r\n' + bleDevice[cube].name + ' Connected.');
    })
    .catch(error => {
        terminal_writeln('' + error);
        if(bleDevice[cube] && bleDevice[cube].gatt.connected)
        {
            bleDevice[cube].gatt.disconnect();
        }
    });
}

function disconnect(cube) {
    if (!bleDevice[cube]) {
        terminal_writeln('No Bluetooth Device connected...');
        return;
    }
    terminal_writeln('Disconnecting from Bluetooth Device...');
    if (bleDevice[cube].gatt.connected) {
        bleDevice[cube].gatt.disconnect();
        changeConnectionState(false, cube);
        terminal_writeln('Bluetooth Device connected: ' + bleDevice[cube].gatt.connected);
    } else {
        terminal_writeln('> Bluetooth Device is already disconnected');
    }
}

function onDisconnected(cube) {
    changeConnectionState(false, cube);
    terminal_writeln('\r\n' + bleDevice[cube].name + ' Disconnected.');
}

function handleNotifications(event) {
    terminal_write('notification');
    let value = event.target.value;
    // Convert raw data bytes to character values and use these to 
    // construct a string.
    let str = "";
    for (let i = 0; i < value.byteLength; i++) {
        str += String.fromCharCode(value.getUint8(i));
    }
    terminal_writeln(str);
}

function nusSendString(s, cube, log=true) {
    if (bleDevice[cube] && bleDevice[cube].gatt.connected) {
        if (log) terminal_writeln("send: " + s);
        let val_arr = new Uint8Array(s.length)
        for (let i = 0; i < s.length; i++) {
            let val = s[i].charCodeAt(0);
            val_arr[i] = val;
        }
        sendNextChunk(val_arr, cube);
    } else {
        terminal_writeln('Not connected to a device yet.');
    }
}

function sendNextChunk(a, cube) {
    let chunk = a.slice(0, MTU);
    rxCharacteristic[cube].writeValue(chunk)
        .then(function() {
            if (a.length > MTU) {
                bleBusy = true;
                sendNextChunk(a.slice(MTU), cube);
            } else {
                bleBusy = false;
            }
        })
        .catch(function(error) {
            if (error == "NetworkError: GATT operation already in progress.") {
                console.log(error + ' Retrying...');
                delay(50).then(() => sendNextChunk(a, cube));
            }
        });
}